langchain[patch]: Take all required fields into account for OpenAPI c…

…hain (#6700) Co-authored-by: Brace Sproul <braceasproul@gmail.com>
langchain-ai · Sep 5, 2024 · f0a6094 · f0a6094
1 parent da42b13
commit f0a6094
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/langchain/src/chains/openai_functions/openapi.ts b/langchain/src/chains/openai_functions/openapi.ts
@@ -175,7 +175,11 @@ export function convertOpenAPISchemaToJSONSchema(
           openAPIProperty,
           spec
         );
-        if (openAPIProperty.required && jsonSchema.required !== undefined) {
+        if (
+          (openAPIProperty.required ||
+            schema.required?.includes(propertyName)) &&
+          jsonSchema.required !== undefined
+        ) {
           jsonSchema.required.push(propertyName);
         }
         return jsonSchema;

diff --git a/langchain/src/chains/openai_functions/tests/openapi.test.ts b/langchain/src/chains/openai_functions/tests/openapi.test.ts
@@ -46,6 +46,19 @@ test("Test convert OpenAPI params to JSON Schema", async () => {
                 },
               },
             },
+            {
+              name: "objectParamWithRequiredFields",
+              in: "query",
+              schema: {
+                type: "object",
+                required: ["fooRequired"],
+                properties: {
+                  fooRequired: {
+                    type: "string",
+                  },
+                },
+              },
+            },
             {
               name: "stringArrayParam",
               in: "query",
@@ -195,6 +208,12 @@ test("Test convert OpenAPI params to JSON Schema", async () => {
   expectType("string", typedObjectParamSchema.properties.foo);
   expectType("number", typedObjectParamSchema.properties.bar);
 
+  const objectParamWithRequiredFieldSchema = convertOpenAPISchemaToJSONSchema(
+    getParamSchema(createWidget, "objectParamWithRequiredFields"),
+    spec
+  ) as JsonSchema7ObjectType;
+  expect(objectParamWithRequiredFieldSchema.required).toContain("fooRequired");
+
   const stringArrayParamSchema = convertOpenAPISchemaToJSONSchema(
     getParamSchema(createWidget, "stringArrayParam"),
     spec