@illuma-ai/agents 1.0.90 → 1.0.93

package/src/tools/__tests__/ToolNode.recovery.test.js

@@ -0,0 +1,200 @@
+import { z } from 'zod';
+import { tool } from '@langchain/core/tools';
+import { AIMessage } from '@langchain/core/messages';
+import { describe, it, expect } from '@jest/globals';
+import { ToolNode } from '../ToolNode';
+import { StreamingToolCallBuffer } from '../StreamingToolCallBuffer';
+/**
+ * Creates a mock content_tool that captures the args it receives.
+ * Used to verify that truncated args are recovered from the streaming buffer.
+ */
+function createMockContentTool(capturedArgs) {
+    return tool(async (input) => {
+        capturedArgs.push({ ...input });
+        return `Wrote ${input.filename ?? 'file'}`;
+    }, {
+        name: 'content_tool',
+        description: 'Read, write, and manage files',
+        schema: z.object({
+            action: z.string(),
+            filename: z.string().optional(),
+            content: z.string().optional(),
+        }),
+    });
+}
+function createAIMessageWithContentTool(callId, args) {
+    return new AIMessage({
+        content: '',
+        tool_calls: [
+            {
+                id: callId,
+                name: 'content_tool',
+                args,
+            },
+        ],
+    });
+}
+describe('ToolNode truncated arg recovery', () => {
+    it('recovers missing content field from streaming buffer', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        // Simulate streaming: buffer accumulated the full raw args
+        buffer.append('tc_1', '{"action":"write","filename":"app.tsx","content":"function App() {\\n  return <div>Hello</div>;\\n}"}');
+        buffer.setToolName('tc_1', 'content_tool');
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        // Simulate truncated args: parsePartialJson lost the content field
+        const aiMsg = createAIMessageWithContentTool('tc_1', {
+            action: 'write',
+            filename: 'app.tsx',
+            // content is MISSING — truncated by max_tokens
+        });
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].action).toBe('write');
+        expect(capturedArgs[0].filename).toBe('app.tsx');
+        expect(capturedArgs[0].content).toBe('function App() {\n  return <div>Hello</div>;\n}');
+    });
+    it('recovers completely empty args from streaming buffer', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        // Buffer has full args, but parsePartialJson returned empty object
+        buffer.append('tc_1', '{"action":"write","filename":"test.js","content":"const x = 1;"}');
+        buffer.setToolName('tc_1', 'content_tool');
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        // Empty args — JSON parsing aborted completely
+        const aiMsg = createAIMessageWithContentTool('tc_1', {});
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].action).toBe('write');
+        expect(capturedArgs[0].filename).toBe('test.js');
+        expect(capturedArgs[0].content).toBe('const x = 1;');
+    });
+    it('does not modify args when buffer has no entry', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        // No buffer entry for tc_1
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        const aiMsg = createAIMessageWithContentTool('tc_1', {
+            action: 'read',
+            filename: 'test.js',
+        });
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].action).toBe('read');
+        expect(capturedArgs[0].filename).toBe('test.js');
+        expect(capturedArgs[0].content).toBeUndefined();
+    });
+    it('does not overwrite existing parsed values', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        // Buffer has different values than parsed
+        buffer.append('tc_1', '{"action":"write","filename":"old.tsx","content":"old content"}');
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        // Parsed args have action and filename already
+        const aiMsg = createAIMessageWithContentTool('tc_1', {
+            action: 'write',
+            filename: 'new.tsx',
+            // content is missing — should be recovered
+        });
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].action).toBe('write');
+        expect(capturedArgs[0].filename).toBe('new.tsx'); // Keeps parsed value, not buffer value
+        expect(capturedArgs[0].content).toBe('old content'); // Recovered from buffer
+    });
+    it('clears buffer after recovery', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        buffer.append('tc_1', '{"action":"write","content":"test"}');
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        const aiMsg = createAIMessageWithContentTool('tc_1', { action: 'write' });
+        await toolNode.invoke({ messages: [aiMsg] });
+        // Buffer should be cleared after processing
+        expect(buffer.has('tc_1')).toBe(false);
+    });
+    it('works with tools other than content_tool', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        buffer.append('tc_1', '{"query":"SELECT * FROM users","database":"prod"}');
+        buffer.setToolName('tc_1', 'sql_query');
+        const sqlTool = tool(async (input) => {
+            capturedArgs.push({ ...input });
+            return 'Results: 42 rows';
+        }, {
+            name: 'sql_query',
+            description: 'Execute SQL query',
+            schema: z.object({
+                query: z.string(),
+                database: z.string().optional(),
+            }),
+        });
+        const toolNode = new ToolNode({
+            tools: [sqlTool],
+            streamingToolCallBuffer: buffer,
+        });
+        // Truncated: only has query, missing database
+        const aiMsg = new AIMessage({
+            content: '',
+            tool_calls: [
+                { id: 'tc_1', name: 'sql_query', args: { query: 'SELECT * FROM users' } },
+            ],
+        });
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].query).toBe('SELECT * FROM users');
+        expect(capturedArgs[0].database).toBe('prod');
+    });
+    it('recovers truncated content with newlines and special chars', async () => {
+        const capturedArgs = [];
+        const buffer = new StreamingToolCallBuffer();
+        const largeContent = 'import React from \\"react\\";\\n\\n' +
+            'export default function Dashboard() {\\n' +
+            '  const [count, setCount] = React.useState(0);\\n' +
+            '  return (\\n' +
+            '    <div className=\\"dashboard\\">\\n' +
+            '      <h1>Dashboard</h1>\\n' +
+            '      <button onClick={() => setCount(c => c + 1)}>\\n' +
+            '        Count: {count}\\n' +
+            '      </button>\\n' +
+            '    </div>\\n' +
+            '  );\\n' +
+            '}';
+        buffer.append('tc_1', `{"action":"write","filename":"Dashboard.tsx","content":"${largeContent}"}`);
+        const contentTool = createMockContentTool(capturedArgs);
+        const toolNode = new ToolNode({
+            tools: [contentTool],
+            streamingToolCallBuffer: buffer,
+        });
+        // content_tool with missing content
+        const aiMsg = createAIMessageWithContentTool('tc_1', {
+            action: 'write',
+            filename: 'Dashboard.tsx',
+        });
+        await toolNode.invoke({ messages: [aiMsg] });
+        expect(capturedArgs).toHaveLength(1);
+        expect(capturedArgs[0].content).toContain('import React from "react"');
+        expect(capturedArgs[0].content).toContain('export default function Dashboard()');
+        expect(capturedArgs[0].content).toContain('Count: {count}');
+    });
+});
+//# sourceMappingURL=ToolNode.recovery.test.js.map

package/src/tools/__tests__/ToolNode.recovery.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolNode.recovery.test.js","sourceRoot":"","sources":["ToolNode.recovery.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAErD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAErE;;;GAGG;AACH,SAAS,qBAAqB,CAC5B,YAAuC;IAEvC,OAAO,IAAI,CACT,KAAK,EAAE,KAAK,EAAE,EAAE;QACd,YAAY,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;QAChC,OAAO,SAAU,KAAiC,CAAC,QAAQ,IAAI,MAAM,EAAE,CAAC;IAC1E,CAAC,EACD;QACE,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,+BAA+B;QAC5C,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;YACf,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YAC/B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAC/B,CAAC;KACH,CACoC,CAAC;AAC1C,CAAC;AAED,SAAS,8BAA8B,CACrC,MAAc,EACd,IAA6B;IAE7B,OAAO,IAAI,SAAS,CAAC;QACnB,OAAO,EAAE,EAAE;QACX,UAAU,EAAE;YACV;gBACE,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,cAAc;gBACpB,IAAI;aACL;SACF;KACF,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,2DAA2D;QAC3D,MAAM,CAAC,MAAM,CACX,MAAM,EACN,uGAAuG,CACxG,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAE3C,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,mEAAmE;QACnE,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE;YACnD,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,SAAS;YACnB,+CAA+C;SAChD,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAClC,iDAAiD,CAClD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,kEAAkE,CAAC,CAAC;QAC1F,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAE3C,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,+CAA+C;QAC/C,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEzD,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAC7C,2BAA2B;QAE3B,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE;YACnD,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,SAAS;SACpB,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,0CAA0C;QAC1C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,iEAAiE,CAAC,CAAC;QAEzF,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,+CAA+C;QAC/C,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE;YACnD,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,SAAS;YACnB,2CAA2C;SAC5C,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,uCAAuC;QACzF,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,wBAAwB;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,qCAAqC,CAAC,CAAC;QAE7D,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1E,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,4CAA4C;QAC5C,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,mDAAmD,CAAC,CAAC;QAC3E,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,OAAO,GAAG,IAAI,CAClB,KAAK,EAAE,KAAK,EAAE,EAAE;YACd,YAAY,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YAChC,OAAO,kBAAkB,CAAC;QAC5B,CAAC,EACD;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,mBAAmB;YAChC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;gBACf,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;gBACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;aAChC,CAAC;SACH,CACoC,CAAC;QAExC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,OAAO,CAAC;YAChB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,8CAA8C;QAC9C,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC;YAC1B,OAAO,EAAE,EAAE;YACX,UAAU,EAAE;gBACV,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,EAAE;aAC1E;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,YAAY,GAA8B,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE7C,MAAM,YAAY,GAChB,sCAAsC;YACtC,0CAA0C;YAC1C,mDAAmD;YACnD,eAAe;YACf,wCAAwC;YACxC,6BAA6B;YAC7B,wDAAwD;YACxD,2BAA2B;YAC3B,oBAAoB;YACpB,eAAe;YACf,SAAS;YACT,GAAG,CAAC;QAEN,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,2DAA2D,YAAY,IAAI,CAAC,CAAC;QAEnG,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,uBAAuB,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,KAAK,GAAG,8BAA8B,CAAC,MAAM,EAAE;YACnD,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,eAAe;SAC1B,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QACvE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACjF,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/tools/__tests__/ToolNode.recovery.test.ts

@@ -0,0 +1,276 @@
+import { z } from 'zod';
+import { tool } from '@langchain/core/tools';
+import { AIMessage } from '@langchain/core/messages';
+import { describe, it, expect } from '@jest/globals';
+import type { StructuredToolInterface } from '@langchain/core/tools';
+import { ToolNode } from '../ToolNode';
+import { StreamingToolCallBuffer } from '../StreamingToolCallBuffer';
+
+/**
+ * Creates a mock content_tool that captures the args it receives.
+ * Used to verify that truncated args are recovered from the streaming buffer.
+ */
+function createMockContentTool(
+  capturedArgs: Record<string, unknown>[]
+): StructuredToolInterface {
+  return tool(
+    async (input) => {
+      capturedArgs.push({ ...input });
+      return `Wrote ${(input as Record<string, unknown>).filename ?? 'file'}`;
+    },
+    {
+      name: 'content_tool',
+      description: 'Read, write, and manage files',
+      schema: z.object({
+        action: z.string(),
+        filename: z.string().optional(),
+        content: z.string().optional(),
+      }),
+    }
+  ) as unknown as StructuredToolInterface;
+}
+
+function createAIMessageWithContentTool(
+  callId: string,
+  args: Record<string, unknown>
+): AIMessage {
+  return new AIMessage({
+    content: '',
+    tool_calls: [
+      {
+        id: callId,
+        name: 'content_tool',
+        args,
+      },
+    ],
+  });
+}
+
+describe('ToolNode truncated arg recovery', () => {
+  it('recovers missing content field from streaming buffer', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    // Simulate streaming: buffer accumulated the full raw args
+    buffer.append(
+      'tc_1',
+      '{"action":"write","filename":"app.tsx","content":"function App() {\\n  return <div>Hello</div>;\\n}"}'
+    );
+    buffer.setToolName('tc_1', 'content_tool');
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    // Simulate truncated args: parsePartialJson lost the content field
+    const aiMsg = createAIMessageWithContentTool('tc_1', {
+      action: 'write',
+      filename: 'app.tsx',
+      // content is MISSING — truncated by max_tokens
+    });
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].action).toBe('write');
+    expect(capturedArgs[0].filename).toBe('app.tsx');
+    expect(capturedArgs[0].content).toBe(
+      'function App() {\n  return <div>Hello</div>;\n}'
+    );
+  });
+
+  it('recovers completely empty args from streaming buffer', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    // Buffer has full args, but parsePartialJson returned empty object
+    buffer.append(
+      'tc_1',
+      '{"action":"write","filename":"test.js","content":"const x = 1;"}'
+    );
+    buffer.setToolName('tc_1', 'content_tool');
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    // Empty args — JSON parsing aborted completely
+    const aiMsg = createAIMessageWithContentTool('tc_1', {});
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].action).toBe('write');
+    expect(capturedArgs[0].filename).toBe('test.js');
+    expect(capturedArgs[0].content).toBe('const x = 1;');
+  });
+
+  it('does not modify args when buffer has no entry', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+    // No buffer entry for tc_1
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    const aiMsg = createAIMessageWithContentTool('tc_1', {
+      action: 'read',
+      filename: 'test.js',
+    });
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].action).toBe('read');
+    expect(capturedArgs[0].filename).toBe('test.js');
+    expect(capturedArgs[0].content).toBeUndefined();
+  });
+
+  it('does not overwrite existing parsed values', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    // Buffer has different values than parsed
+    buffer.append(
+      'tc_1',
+      '{"action":"write","filename":"old.tsx","content":"old content"}'
+    );
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    // Parsed args have action and filename already
+    const aiMsg = createAIMessageWithContentTool('tc_1', {
+      action: 'write',
+      filename: 'new.tsx',
+      // content is missing — should be recovered
+    });
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].action).toBe('write');
+    expect(capturedArgs[0].filename).toBe('new.tsx'); // Keeps parsed value, not buffer value
+    expect(capturedArgs[0].content).toBe('old content'); // Recovered from buffer
+  });
+
+  it('clears buffer after recovery', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    buffer.append('tc_1', '{"action":"write","content":"test"}');
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    const aiMsg = createAIMessageWithContentTool('tc_1', { action: 'write' });
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    // Buffer should be cleared after processing
+    expect(buffer.has('tc_1')).toBe(false);
+  });
+
+  it('works with tools other than content_tool', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    buffer.append('tc_1', '{"query":"SELECT * FROM users","database":"prod"}');
+    buffer.setToolName('tc_1', 'sql_query');
+
+    const sqlTool = tool(
+      async (input) => {
+        capturedArgs.push({ ...input });
+        return 'Results: 42 rows';
+      },
+      {
+        name: 'sql_query',
+        description: 'Execute SQL query',
+        schema: z.object({
+          query: z.string(),
+          database: z.string().optional(),
+        }),
+      }
+    ) as unknown as StructuredToolInterface;
+
+    const toolNode = new ToolNode({
+      tools: [sqlTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    // Truncated: only has query, missing database
+    const aiMsg = new AIMessage({
+      content: '',
+      tool_calls: [
+        {
+          id: 'tc_1',
+          name: 'sql_query',
+          args: { query: 'SELECT * FROM users' },
+        },
+      ],
+    });
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].query).toBe('SELECT * FROM users');
+    expect(capturedArgs[0].database).toBe('prod');
+  });
+
+  it('recovers truncated content with newlines and special chars', async () => {
+    const capturedArgs: Record<string, unknown>[] = [];
+    const buffer = new StreamingToolCallBuffer();
+
+    const largeContent =
+      'import React from \\"react\\";\\n\\n' +
+      'export default function Dashboard() {\\n' +
+      '  const [count, setCount] = React.useState(0);\\n' +
+      '  return (\\n' +
+      '    <div className=\\"dashboard\\">\\n' +
+      '      <h1>Dashboard</h1>\\n' +
+      '      <button onClick={() => setCount(c => c + 1)}>\\n' +
+      '        Count: {count}\\n' +
+      '      </button>\\n' +
+      '    </div>\\n' +
+      '  );\\n' +
+      '}';
+
+    buffer.append(
+      'tc_1',
+      `{"action":"write","filename":"Dashboard.tsx","content":"${largeContent}"}`
+    );
+
+    const contentTool = createMockContentTool(capturedArgs);
+    const toolNode = new ToolNode({
+      tools: [contentTool],
+      streamingToolCallBuffer: buffer,
+    });
+
+    // content_tool with missing content
+    const aiMsg = createAIMessageWithContentTool('tc_1', {
+      action: 'write',
+      filename: 'Dashboard.tsx',
+    });
+
+    await toolNode.invoke({ messages: [aiMsg] });
+
+    expect(capturedArgs).toHaveLength(1);
+    expect(capturedArgs[0].content).toContain('import React from "react"');
+    expect(capturedArgs[0].content).toContain(
+      'export default function Dashboard()'
+    );
+    expect(capturedArgs[0].content).toContain('Count: {count}');
+  });
+});