fix: anthropic tool nested objects (#568)

lunary-ai · Sep 20, 2024 · c035de5 · c035de5
1 parent 2ee2eb0
commit c035de5
Showing 1 changed file with 22 additions and 7 deletions.
diff --git a/packages/frontend/components/prompts/Provider.tsx b/packages/frontend/components/prompts/Provider.tsx
@@ -23,6 +23,26 @@ import Link from "next/link";
 import { IconInfoCircle, IconTools } from "@tabler/icons-react";
 
 function convertOpenAIToolsToAnthropic(openAITools) {
+  function convertProperties(openAIProperties) {
+    const anthropicProperties = {};
+
+    for (const [key, value] of Object.entries(openAIProperties)) {
+      const anthropicProperty = {
+        type: value.type,
+        description: value.description,
+      };
+
+      if (value.type === "object" && value.properties) {
+        anthropicProperty.properties = convertProperties(value.properties);
+        anthropicProperty.required = value.required || [];
+      }
+
+      anthropicProperties[key] = anthropicProperty;
+    }
+
+    return anthropicProperties;
+  }
+
   return openAITools.map((openAITool) => {
     const openAIFunction = openAITool.function;
 
@@ -40,14 +60,9 @@ function convertOpenAIToolsToAnthropic(openAITools) {
       },
     };
 
-    for (const [key, value] of Object.entries(
+    anthropicTool.input_schema.properties = convertProperties(
       openAIFunction.parameters.properties,
-    )) {
-      anthropicTool.input_schema.properties[key] = {
-        type: value.type,
-        description: value.description,
-      };
-    }
+    );
 
     return anthropicTool;
   });