update schema

mike-gee · Oct 30, 2023 · b65c683 · b65c683
1 parent 927bc83
commit b65c683
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/webtranspose/openai.py b/src/webtranspose/openai.py
@@ -127,7 +127,16 @@ def transform_schema(self, schema: dict) -> dict:
         properties = {}
         for key, value in schema.items():
             if isinstance(value, dict):
-                if "type" in value:
+                if "type" in value and value["type"] == "array":
+                    properties[key] = {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": self.transform_schema(value["items"]),
+                        },
+                        "required": list(value["items"].keys()),
+                    }
+                elif "type" in value:
                     properties[key] = value
                 else:
                     properties[key] = self.transform_schema(value)
@@ -145,4 +154,5 @@ def transform_schema(self, schema: dict) -> dict:
                     "type": value,
                     "description": key,
                 }
+
         return properties