@illuma-ai/agents 1.0.89 → 1.0.93

package/src/scripts/test-tool-before-handoff-role-order.ts

@@ -0,0 +1,276 @@
+import { config } from 'dotenv';
+config();
+
+import { z } from 'zod';
+import { tool } from '@langchain/core/tools';
+import { HumanMessage, BaseMessage } from '@langchain/core/messages';
+import type { RunnableConfig } from '@langchain/core/runnables';
+import type * as t from '@/types';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { GraphEvents, Providers } from '@/common';
+import { Run } from '@/run';
+
+const conversationHistory: BaseMessage[] = [];
+
+/**
+ * Test: Tool call followed by handoff (role order validation)
+ *
+ * Reproduces the bug from issue #54:
+ * When a router agent runs a non-handoff tool (e.g. list_upload_sessions)
+ * and then hands off to another agent in the same turn, the receiving agent
+ * gets a message sequence of `... tool → user` which many chat APIs reject
+ * with: "400 Unexpected role 'user' after role 'tool'"
+ *
+ * The fix ensures handoff instructions are injected into the last ToolMessage
+ * (instead of appending a new HumanMessage) when the filtered messages end
+ * with a ToolMessage.
+ */
+async function testToolBeforeHandoffRoleOrder(): Promise<void> {
+  console.log('='.repeat(60));
+  console.log('Test: Tool Call Before Handoff (Role Order Validation)');
+  console.log('='.repeat(60));
+  console.log('\nThis test verifies that:');
+  console.log('1. Router calls a regular tool AND then hands off');
+  console.log('2. The receiving agent does NOT get tool → user role sequence');
+  console.log('3. No 400 API error occurs after the handoff\n');
+
+  const { contentParts, aggregateContent } = createContentAggregator();
+
+  let currentAgent = '';
+  let toolCallCount = 0;
+  let handoffOccurred = false;
+
+  const customHandlers = {
+    [GraphEvents.TOOL_END]: new ToolEndHandler(async () => {
+      toolCallCount++;
+      console.log(`\n  Tool completed (total: ${toolCallCount})`);
+    }),
+    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+    [GraphEvents.ON_RUN_STEP]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP,
+        data: t.StreamEventData
+      ): void => {
+        const runStep = data as t.RunStep;
+        if (runStep.agentId) {
+          currentAgent = runStep.agentId;
+          console.log(`\n[Agent: ${currentAgent}] Processing...`);
+        }
+        aggregateContent({ event, data: runStep });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
+    },
+    [GraphEvents.ON_MESSAGE_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.StreamEventData
+      ): void => {
+        aggregateContent({ event, data: data as t.MessageDeltaEvent });
+      },
+    },
+    [GraphEvents.TOOL_START]: {
+      handle: (
+        _event: string,
+        data: t.StreamEventData,
+        _metadata?: Record<string, unknown>
+      ): void => {
+        const toolData = data as { name?: string };
+        if (toolData?.name?.includes('transfer_to_')) {
+          handoffOccurred = true;
+          const specialist = toolData.name.replace('lc_transfer_to_', '');
+          console.log(`\n  Handoff initiated to: ${specialist}`);
+        }
+      },
+    },
+  };
+
+  /**
+   * Create a simple tool for the router agent.
+   * This simulates the list_upload_sessions scenario from issue #54:
+   * the router calls a regular tool and THEN hands off in the same turn.
+   */
+  const listSessions = tool(
+    async () => {
+      return JSON.stringify({
+        sessions: [
+          { id: 'sess_1', name: 'Q4 Report', status: 'ready' },
+          { id: 'sess_2', name: 'Budget Analysis', status: 'pending' },
+        ],
+      });
+    },
+    {
+      name: 'list_upload_sessions',
+      description: 'List available upload sessions for data analysis',
+      schema: z.object({}),
+    }
+  );
+
+  const agents: t.AgentInputs[] = [
+    {
+      agentId: 'router',
+      provider: Providers.OPENAI,
+      clientOptions: {
+        modelName: 'gpt-4.1-mini',
+        apiKey: process.env.OPENAI_API_KEY,
+      },
+      tools: [listSessions],
+      instructions: `You are a Router agent with access to upload sessions and a data analysis specialist.
+
+Your workflow for data-related requests:
+1. FIRST: Call list_upload_sessions to check available data
+2. THEN: Transfer to the data_analyst with your findings
+
+CRITICAL: You MUST call list_upload_sessions first, then immediately transfer to data_analyst.
+Do NOT write a long response. Just call the tool and hand off.`,
+      maxContextTokens: 8000,
+    },
+    {
+      agentId: 'data_analyst',
+      provider: Providers.OPENAI,
+      clientOptions: {
+        modelName: 'gpt-4.1-mini',
+        apiKey: process.env.OPENAI_API_KEY,
+      },
+      instructions: `You are a Data Analyst specialist. When you receive a request:
+1. Review any data or context provided
+2. Provide a concise analysis or recommendation
+3. Keep your response brief and focused`,
+      maxContextTokens: 8000,
+    },
+  ];
+
+  const edges: t.GraphEdge[] = [
+    {
+      from: 'router',
+      to: 'data_analyst',
+      description: 'Transfer to data analyst after checking sessions',
+      edgeType: 'handoff',
+      prompt:
+        'Provide specific instructions for the data analyst about what to analyze',
+      promptKey: 'instructions',
+    },
+  ];
+
+  const runConfig: t.RunConfig = {
+    runId: `tool-before-handoff-role-order-${Date.now()}`,
+    graphConfig: {
+      type: 'multi-agent',
+      agents,
+      edges,
+    },
+    customHandlers,
+    returnContent: true,
+    skipCleanup: true,
+  };
+
+  const run = await Run.create(runConfig);
+
+  const streamConfig: Partial<RunnableConfig> & {
+    version: 'v1' | 'v2';
+    streamMode: string;
+  } = {
+    configurable: {
+      thread_id: 'tool-before-handoff-role-order-1',
+    },
+    streamMode: 'values',
+    version: 'v2' as const,
+  };
+
+  try {
+    const query =
+      'I want to visualize my CSV data. Can you check what upload sessions are available and have the analyst help me?';
+
+    console.log('\n' + '-'.repeat(60));
+    console.log(`USER QUERY: "${query}"`);
+    console.log('-'.repeat(60));
+    console.log('\nExpected behavior:');
+    console.log('1. Router calls list_upload_sessions tool');
+    console.log('2. Router hands off to data_analyst');
+    console.log('3. data_analyst responds WITHOUT 400 error\n');
+
+    conversationHistory.push(new HumanMessage(query));
+    const inputs = { messages: conversationHistory };
+
+    await run.processStream(inputs, streamConfig);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+      conversationHistory.push(...finalMessages);
+    }
+
+    /** Results */
+    console.log(`\n${'='.repeat(60)}`);
+    console.log('TEST RESULTS:');
+    console.log('='.repeat(60));
+    console.log(`Tool calls made: ${toolCallCount}`);
+    console.log(`Handoff occurred: ${handoffOccurred ? 'Yes' : 'No'}`);
+    console.log(
+      `Test status: ${toolCallCount > 0 && handoffOccurred ? 'PASSED' : 'FAILED'}`
+    );
+
+    if (toolCallCount === 0) {
+      console.log('\nNote: Router did not call any tools before handoff.');
+      console.log(
+        'The bug only occurs when a non-handoff tool is called in the same turn as the handoff.'
+      );
+      console.log('Try running again - the model may need stronger prompting.');
+    }
+
+    console.log('='.repeat(60));
+
+    /** Show conversation history */
+    console.log('\nConversation History:');
+    console.log('-'.repeat(60));
+    conversationHistory.forEach((msg, idx) => {
+      const role = msg.getType();
+      const content =
+        typeof msg.content === 'string'
+          ? msg.content.substring(0, 150) +
+            (msg.content.length > 150 ? '...' : '')
+          : '[complex content]';
+      console.log(`  [${idx}] ${role}: ${content}`);
+    });
+  } catch (error) {
+    const errorMsg = error instanceof Error ? error.message : String(error);
+    console.error('\nTest FAILED with error:', errorMsg);
+
+    if (errorMsg.includes('Unexpected role') || errorMsg.includes('400')) {
+      console.error('\n>>> This is the exact bug from issue #54! <<<<');
+      console.error(
+        '>>> The tool→user role sequence caused a 400 API error. <<<'
+      );
+    }
+
+    console.log('\nConversation history at failure:');
+    console.dir(conversationHistory, { depth: null });
+  }
+}
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+  console.log('\nConversation history at failure:');
+  console.dir(conversationHistory, { depth: null });
+  process.exit(1);
+});
+
+process.on('uncaughtException', (err) => {
+  console.error('Uncaught Exception:', err);
+});
+
+testToolBeforeHandoffRoleOrder().catch((err) => {
+  console.error('Test failed:', err);
+  console.log('\nConversation history at failure:');
+  console.dir(conversationHistory, { depth: null });
+  process.exit(1);
+});

package/src/scripts/test-tools-before-handoff.js

@@ -0,0 +1,187 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { Run } from '@/run';
+import { Providers, GraphEvents } from '@/common';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+const conversationHistory = [];
+/**
+ * Test edge case: Agent performs 2 web searches before handing off
+ *
+ * This tests how the system behaves when an agent with handoff capabilities
+ * uses tools before transferring control to another agent.
+ */
+async function testToolsBeforeHandoff() {
+    console.log('Testing Tools Before Handoff Edge Case...\n');
+    // Set up content aggregator
+    const { contentParts, aggregateContent } = createContentAggregator();
+    // Track tool calls and handoffs
+    let toolCallCount = 0;
+    let handoffOccurred = false;
+    // Create custom handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(async () => {
+            console.log('\n✅ Tool completed');
+        }),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                const runStepData = data;
+                if (runStepData?.name) {
+                    console.log(`\n[${runStepData.name}] Processing...`);
+                }
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                // console.log('====== ON_MESSAGE_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                const toolData = data;
+                console.log(`\n🔧 Tool started:`);
+                console.dir({ toolData, metadata }, { depth: null });
+                if (toolData?.output?.name?.includes('transfer_to_')) {
+                    handoffOccurred = true;
+                    const specialist = toolData.name.replace('transfer_to_', '');
+                    console.log(`\n🔀 Handoff initiated to: ${specialist}`);
+                }
+            },
+        },
+    };
+    // Create the graph with research agent and report writer
+    function createGraphWithToolsAndHandoff() {
+        const agents = [
+            {
+                agentId: 'research_coordinator',
+                provider: Providers.OPENAI,
+                clientOptions: {
+                    modelName: 'gpt-4.1-mini',
+                    apiKey: process.env.OPENAI_API_KEY,
+                },
+                tools: [],
+                instructions: `You are a Research Coordinator with access to a report writer specialist.
+        
+        Your workflow MUST follow these steps IN ORDER:
+        1. FIRST: Write an initial response acknowledging the request
+           - Explain what you understand about the topic
+           - Provide any general knowledge you have
+        2. FINALLY: Transfer to the report writer
+           - Provide the report writer with a summary of the information
+        
+        CRITICAL: You MUST write your initial response before transferring to the report writer.`,
+                maxContextTokens: 8000,
+            },
+            {
+                agentId: 'report_writer',
+                provider: Providers.OPENAI,
+                clientOptions: {
+                    modelName: 'gpt-5-mini',
+                    apiKey: process.env.OPENAI_API_KEY,
+                },
+                instructions: `You are a Report Writer specialist. Your role is to:
+        1. Receive research findings from the Research Coordinator
+        2. Create a well-structured, comprehensive report
+        3. Include all key findings from the research
+        4. Format the report with clear sections and bullet points
+        5. Add a brief executive summary at the beginning
+        
+        Focus on clarity, completeness, and professional presentation.`,
+                maxContextTokens: 8000,
+            },
+        ];
+        // Create edge from research coordinator to report writer
+        const edges = [
+            {
+                from: 'research_coordinator',
+                to: 'report_writer',
+                description: 'Transfer to report writer after completing research',
+                edgeType: 'handoff',
+            },
+        ];
+        return {
+            runId: `tools-before-handoff-${Date.now()}`,
+            graphConfig: {
+                type: 'multi-agent',
+                agents,
+                edges,
+            },
+            customHandlers,
+            returnContent: true,
+            skipCleanup: true,
+        };
+    }
+    try {
+        // Single test query that requires handoff to report writer
+        const query = `Tell me about quantum computing developments, 
+    including major breakthroughs and commercial applications. 
+    I need a comprehensive report.`;
+        console.log('='.repeat(60));
+        console.log(`USER QUERY: "${query}"`);
+        console.log('='.repeat(60));
+        // Create the graph
+        const runConfig = createGraphWithToolsAndHandoff();
+        const run = await Run.create(runConfig);
+        console.log('\nExpected behavior:');
+        console.log('1. Research Coordinator writes initial response');
+        console.log('2. Research Coordinator hands off to Report Writer');
+        console.log('3. Report Writer creates final report\n');
+        // Process with streaming
+        conversationHistory.push(new HumanMessage(query));
+        const inputs = {
+            messages: conversationHistory,
+        };
+        const config = {
+            configurable: {
+                thread_id: 'tools-handoff-test-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        const finalContentParts = await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+        }
+        // Show results summary
+        console.log(`\n${'─'.repeat(60)}`);
+        console.log('EDGE CASE TEST RESULTS:');
+        console.log('─'.repeat(60));
+        console.log(`Tool calls before handoff: ${toolCallCount}`);
+        console.log(`Expected tool calls: 0 (no web search available)`);
+        console.log(`Handoff occurred: ${handoffOccurred ? 'Yes ✅' : 'No ❌'}`);
+        console.log(`Test status: ${handoffOccurred ? 'PASSED ✅' : 'FAILED ❌'}`);
+        console.log('─'.repeat(60));
+        // Display conversation history
+        console.log('\nConversation History:');
+        console.log('─'.repeat(60));
+        conversationHistory.forEach((msg, idx) => {
+            const role = msg.constructor.name.replace('Message', '');
+            console.log(`\n[${idx}] ${role}:`);
+            if (typeof msg.content === 'string') {
+                console.log(msg.content.substring(0, 200) +
+                    (msg.content.length > 200 ? '...' : ''));
+            }
+        });
+    }
+    catch (error) {
+        console.error('Error in tools-before-handoff test:', error);
+    }
+}
+// Run the test
+testToolsBeforeHandoff();
+//# sourceMappingURL=test-tools-before-handoff.js.map

package/src/scripts/test-tools-before-handoff.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-tools-before-handoff.js","sourceRoot":"","sources":["test-tools-before-handoff.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG3D,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;;;GAKG;AACH,KAAK,UAAU,sBAAsB;IACnC,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAE3D,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,gCAAgC;IAChC,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,eAAe,GAAG,KAAK,CAAC;IAE5B,yBAAyB;IACzB,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,CAAC,KAAK,IAAI,EAAE;YACpD,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACpC,CAAC,CAAC;QACF,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,MAAM,WAAW,GAAG,IAAW,CAAC;gBAChC,IAAI,WAAW,EAAE,IAAI,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,MAAM,WAAW,CAAC,IAAI,iBAAiB,CAAC,CAAC;gBACvD,CAAC;gBACD,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;gBACR,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,iDAAiD;gBACjD,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,MAAM,QAAQ,GAAG,IAAW,CAAC;gBAC7B,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAErD,IAAI,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;oBACrD,eAAe,GAAG,IAAI,CAAC;oBACvB,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;oBAC7D,OAAO,CAAC,GAAG,CAAC,8BAA8B,UAAU,EAAE,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;SACF;KACF,CAAC;IAEF,yDAAyD;IACzD,SAAS,8BAA8B;QACrC,MAAM,MAAM,GAAoB;YAC9B;gBACE,OAAO,EAAE,sBAAsB;gBAC/B,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,aAAa,EAAE;oBACb,SAAS,EAAE,cAAc;oBACzB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;iBACnC;gBACD,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE;;;;;;;;;iGAS2E;gBACzF,gBAAgB,EAAE,IAAI;aACvB;YACD;gBACE,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,aAAa,EAAE;oBACb,SAAS,EAAE,YAAY;oBACvB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;iBACnC;gBACD,YAAY,EAAE;;;;;;;uEAOiD;gBAC/D,gBAAgB,EAAE,IAAI;aACvB;SACF,CAAC;QAEF,yDAAyD;QACzD,MAAM,KAAK,GAAkB;YAC3B;gBACE,IAAI,EAAE,sBAAsB;gBAC5B,EAAE,EAAE,eAAe;gBACnB,WAAW,EAAE,qDAAqD;gBAClE,QAAQ,EAAE,SAAS;aACpB;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,wBAAwB,IAAI,CAAC,GAAG,EAAE,EAAE;YAC3C,WAAW,EAAE;gBACX,IAAI,EAAE,aAAa;gBACnB,MAAM;gBACN,KAAK;aACN;YACD,cAAc;YACd,aAAa,EAAE,IAAI;YACnB,WAAW,EAAE,IAAI;SAClB,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,2DAA2D;QAC3D,MAAM,KAAK,GAAG;;mCAEiB,CAAC;QAEhC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,GAAG,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,mBAAmB;QACnB,MAAM,SAAS,GAAG,8BAA8B,EAAE,CAAC;QACnD,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;QAClE,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;QAEvD,yBAAyB;QACzB,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,sBAAsB;aAClC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;QAE3C,IAAI,aAAa,EAAE,CAAC;YAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC7C,CAAC;QAED,uBAAuB;QACvB,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,8BAA8B,aAAa,EAAE,CAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;QAChE,OAAO,CAAC,GAAG,CAAC,qBAAqB,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CAAC,gBAAgB,eAAe,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC;QACzE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,+BAA+B;QAC/B,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,mBAAmB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACvC,MAAM,IAAI,GAAG,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACzD,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,KAAK,IAAI,GAAG,CAAC,CAAC;YACnC,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO,CAAC,GAAG,CACT,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;oBAC3B,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAC1C,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;IAC9D,CAAC;AACH,CAAC;AAED,eAAe;AACf,sBAAsB,EAAE,CAAC"}

package/src/scripts/test-tools-before-handoff.ts

@@ -28,14 +28,9 @@ async function testToolsBeforeHandoff() {
 
   // Create custom handlers
   const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(
-      undefined,
-      undefined,
-      (name?: string) => {
-        console.log(`\n✅ Tool completed: ${name}`);
-        return true;
-      }
-    ),
+    [GraphEvents.TOOL_END]: new ToolEndHandler(async () => {
+      console.log('\n✅ Tool completed');
+    }),
     [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
     [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
     [GraphEvents.ON_RUN_STEP]: {
@@ -151,6 +146,7 @@ async function testToolsBeforeHandoff() {
       },
       customHandlers,
       returnContent: true,
+      skipCleanup: true,
     };
   }