Fix issue of ignoring multiple tool calls

This issue is described here: #431 (comment)
Ironclad · Jul 20, 2024 · 93370d8 · 93370d8
1 parent 0b0f5fd
commit 93370d8
Showing 1 changed file with 19 additions and 31 deletions.
diff --git a/packages/core/src/model/nodes/ChatNode.ts b/packages/core/src/model/nodes/ChatNode.ts
@@ -910,37 +910,25 @@ export class ChatNodeImpl extends NodeImpl<ChatNode> {
             }
 
             if (functionCalls.length > 0) {
-              if (isMultiResponse) {
-                output['function-call' as PortId] = {
-                  type: 'object[]',
-                  value: functionCalls.map((functionCalls) => ({
-                    name: functionCalls[0]?.name,
-                    arguments: functionCalls[0]?.lastParsedArguments,
-                    id: functionCalls[0]?.id,
-                  })),
-                };
-              } else {
-                if (this.data.parallelFunctionCalling) {
-                  console.dir({ functionCalls });
-                  output['function-calls' as PortId] = {
-                    type: 'object[]',
-                    value: functionCalls[0]!.map((functionCall) => ({
-                      name: functionCall.name,
-                      arguments: functionCall.lastParsedArguments,
-                      id: functionCall.id,
-                    })),
-                  };
-                } else {
-                  output['function-call' as PortId] = {
-                    type: 'object',
-                    value: {
-                      name: functionCalls[0]![0]?.name,
-                      arguments: functionCalls[0]![0]?.lastParsedArguments,
-                      id: functionCalls[0]![0]?.id,
-                    } as Record<string, unknown>,
-                  };
-                }
-              }
+							if (isMultiResponse) {
+								output['function-call' as PortId] = {
+									type: 'object[]',
+									value: functionCalls.flat().map((functionCall) => ({
+										name: functionCall.name,
+										arguments: functionCall.lastParsedArguments,
+										id: functionCall.id,
+									})),
+								}
+							} else {
+								output['function-call' as PortId] = {
+									type: 'object[]',
+									value: functionCalls[0]!.map((functionCall) => ({
+										name: functionCall.name,
+										arguments: functionCall.lastParsedArguments,
+										id: functionCall.id,
+									})),
+								}
+							}
             }
 
             context.onPartialOutputs?.(output);