core[patch]: Make tool call chunk partial JSON parsing more strict (#…

…6639)
langchain-ai · Aug 26, 2024 · a64cf77 · a64cf77
1 parent 1a95392
commit a64cf77
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/langchain-core/src/messages/ai.ts b/langchain-core/src/messages/ai.ts
@@ -188,8 +188,12 @@ export class AIMessageChunk extends BaseMessageChunk {
       for (const toolCallChunk of fields.tool_call_chunks) {
         let parsedArgs = {};
         try {
-          parsedArgs = parsePartialJson(toolCallChunk.args ?? "{}") ?? {};
-          if (typeof parsedArgs !== "object" || Array.isArray(parsedArgs)) {
+          parsedArgs = parsePartialJson(toolCallChunk.args || "{}");
+          if (
+            parsedArgs === null ||
+            typeof parsedArgs !== "object" ||
+            Array.isArray(parsedArgs)
+          ) {
             throw new Error("Malformed tool call chunk args.");
           }
           toolCalls.push({