Refactor (openapi_parser): update getBodySchemaName function to retur…

…n the appropriate schema due to pydantic v2 upgrade
Agenta-AI · May 25, 2024 · 754ead3 · 754ead3
1 parent f573878
commit 754ead3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agenta-web/src/lib/helpers/openapi_parser.ts b/agenta-web/src/lib/helpers/openapi_parser.ts
@@ -4,7 +4,9 @@ import {GenericObject, Parameter} from "../Types"
 
 const getBodySchemaName = (schema: GenericObject): string => {
     return (
-        schema?.paths?.["/generate"]?.post?.requestBody?.content["application/json"]?.schema["$ref"]
+        schema?.paths?.["/generate"]?.post?.requestBody?.content?.["application/json"]?.schema?.[
+            "allOf"
+        ]?.[0]?.["$ref"]
             ?.split("/")
             ?.pop() || ""
     )
@@ -20,6 +22,7 @@ export const detectChatVariantFromOpenAISchema = (schema: GenericObject) => {
 export const openAISchemaToParameters = (schema: GenericObject): Parameter[] => {
     const parameters: Parameter[] = []
     const bodySchemaName = getBodySchemaName(schema)
+    console.log("Body Schema Name: ", bodySchemaName)
 
     // get the actual schema for the body parameters
     Object.entries(schema.components.schemas[bodySchemaName].properties || {}).forEach(