Merge pull request #1 from tcgumus/dev-fix-unsupported-json-error

fix: Unsupported JSON schema type {type_}
arcee-ai · Oct 2, 2024 · b2f09e4 · b2f09e4
2 parents d353ea4 + bdb2ce9
commit b2f09e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/open_webui/utils/schemas.py b/backend/open_webui/utils/schemas.py
@@ -104,5 +104,9 @@ def json_schema_to_pydantic_type(json_schema: dict[str, Any]) -> Any:
         return Optional[Any]  # Use Optional[Any] for nullable fields
     elif type_ == "literal":
         return Literal[literal_eval(json_schema.get("enum"))]
+    elif type_ == "optional":
+        inner_schema = json_schema.get("items", {"type": "string"})
+        inner_type = json_schema_to_pydantic_type(inner_schema)
+        return Optional[inner_type]
     else:
         raise ValueError(f"Unsupported JSON schema type: {type_}")