From 94e6a07e3613c7426c67672d81c6b75a5d26b1c3 Mon Sep 17 00:00:00 2001
From: quobix <dave@quobix.com>
Date: Thu, 13 Feb 2025 21:27:52 -0500
Subject: [PATCH] Address #175

---
 builder/tree.go | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/builder/tree.go b/builder/tree.go
index fdfac5d..755760c 100644
--- a/builder/tree.go
+++ b/builder/tree.go
@@ -303,12 +303,25 @@ func DigIntoTreeNode[T any](parent *model.TreeNode, field reflect.Value, label s
 	}
 }
 
+func transformLabel(in string) string {
+	switch in {
+	case "ONEOF":
+		return "OneOf"
+	case "ALLOF":
+		return "AllOf"
+	case "ANYOF":
+		return "AnyOf"
+	default:
+		return in
+	}
+}
+
 func DigIntoTreeNodeSlice[T any](parent *model.TreeNode, field reflect.Value, label string) {
 	if !field.IsZero() {
 		for k := 0; k < field.Len(); k++ {
 			f := field.Index(k)
 			e := &model.TreeNode{
-				TitleString: label,
+				TitleString: transformLabel(label),
 				Key:         uuid.New().String(),
 				IsLeaf:      false,
 				Selectable:  false,