Improve rendering of intermediate tool calls

langchain-ai · Oct 23, 2024 · 2c89549 · 2c89549
1 parent 73abeae
commit 2c89549
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/js/src/vercel.ts b/js/src/vercel.ts
@@ -415,11 +415,22 @@ export class AISDKExporter {
 
         const outputs = ((): KVMap | undefined => {
           let result: KVMap | undefined = undefined;
+
           if (span.attributes["ai.response.toolCalls"]) {
+            let content = tryJson(span.attributes["ai.response.toolCalls"]);
+
+            if (Array.isArray(content)) {
+              content = content.map((i) => ({
+                type: "tool-call",
+                ...i,
+                args: tryJson(i.args),
+              }));
+            }
+
             result = {
               llm_output: convertCoreToSmith({
                 role: "assistant",
-                content: tryJson(span.attributes["ai.response.toolCalls"]),
+                content,
               } satisfies CoreAssistantMessage),
             };
           } else if (span.attributes["ai.response.text"]) {