Added support for enum tag in schema generation

sashabaranov · Dec 6, 2024 · 7b426eb · 7b426eb
1 parent c203ca0
commit 7b426eb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/jsonschema/json.go b/jsonschema/json.go
@@ -131,6 +131,12 @@ func reflectSchemaObject(t reflect.Type) (*Definition, error) {
 			required = false
 		}
 
+		enumTag := field.Tag.Get("enum")
+		var enumValues []string
+		if enumTag != "" {
+			enumValues = strings.Split(enumTag, ",")
+		}
+
 		item, err := reflectSchema(field.Type)
 		if err != nil {
 			return nil, err
@@ -139,6 +145,11 @@ func reflectSchemaObject(t reflect.Type) (*Definition, error) {
 		if description != "" {
 			item.Description = description
 		}
+
+		if len(enumValues) > 0 {
+			item.Enum = enumValues
+		}
+
 		properties[jsonTag] = *item
 
 		if s := field.Tag.Get("required"); s != "" {

diff --git a/test.mp3 b/test.mp3
@@ -0,0 +1 @@
+hello