@illuma-ai/agents 1.0.90 → 1.0.93

package/src/scripts/test-parallel-handoffs.js

@@ -0,0 +1,229 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler } from '@/events';
+import { Providers, GraphEvents } from '@/common';
+import { sleep } from '@/utils/run';
+import { Run } from '@/run';
+const conversationHistory = [];
+/**
+ * Test parallel handoffs - where an LLM calls multiple transfer tools simultaneously
+ *
+ * Graph structure:
+ * coordinator -> [researcher, writer] (via parallel handoff tools)
+ *
+ * The coordinator agent has two transfer tools:
+ * - transfer_to_researcher
+ * - transfer_to_writer
+ *
+ * When given a task that needs both, it should call both tools in parallel.
+ */
+async function testParallelHandoffs() {
+    console.log('Testing Parallel Handoffs (LLM calling multiple transfers)...\n');
+    const { contentParts, aggregateContent } = createContentAggregator();
+    const agents = [
+        {
+            agentId: 'coordinator',
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4o-mini',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a COORDINATOR agent. Your job is to delegate tasks to specialized agents.
+
+You have access to two transfer tools:
+- transfer_to_researcher: For research and fact-finding tasks
+- transfer_to_writer: For content creation and writing tasks
+
+IMPORTANT: When a task requires BOTH research AND writing, you MUST call BOTH transfer tools SIMULTANEOUSLY in the same response. Do not call them sequentially.
+
+For example, if asked to "research and write about X", call both transfers at once to enable parallel work.
+
+When delegating, provide clear instructions to each agent about what they should do.`,
+        },
+        {
+            agentId: 'researcher',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are a RESEARCHER. When you receive a task:
+1. Provide concise research findings (100-150 words)
+2. Start your response with "📚 RESEARCH FINDINGS:"`,
+        },
+        {
+            agentId: 'writer',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are a WRITER. When you receive a task:
+1. Provide creative content (100-150 words)
+2. Start your response with "✍️ WRITTEN CONTENT:"`,
+        },
+    ];
+    /**
+     * Create handoff edges from coordinator to both researcher and writer.
+     * These are separate edges so the LLM sees both transfer tools.
+     */
+    const edges = [
+        {
+            from: 'coordinator',
+            to: 'researcher',
+            edgeType: 'handoff',
+            description: 'Transfer to researcher for research and fact-finding tasks',
+            prompt: 'Research task instructions',
+        },
+        {
+            from: 'coordinator',
+            to: 'writer',
+            edgeType: 'handoff',
+            description: 'Transfer to writer for content creation and writing tasks',
+            prompt: 'Writing task instructions',
+        },
+    ];
+    /** Track which agents are active and their timing */
+    const activeAgents = new Set();
+    const agentTimings = {};
+    const startTime = Date.now();
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: {
+            handle: (_event, _data, metadata) => {
+                const nodeName = metadata?.langgraph_node;
+                if (nodeName) {
+                    const elapsed = Date.now() - startTime;
+                    agentTimings[nodeName] = agentTimings[nodeName] || {};
+                    agentTimings[nodeName].end = elapsed;
+                    activeAgents.delete(nodeName);
+                    console.log(`\n⏱️  [${nodeName}] COMPLETED at ${elapsed}ms`);
+                }
+            },
+        },
+        [GraphEvents.CHAT_MODEL_START]: {
+            handle: (_event, _data, metadata) => {
+                const nodeName = metadata?.langgraph_node;
+                if (nodeName) {
+                    const elapsed = Date.now() - startTime;
+                    /** Store first start time for parallel overlap calculation */
+                    if (!agentTimings[nodeName]?.start) {
+                        agentTimings[nodeName] = agentTimings[nodeName] || {};
+                        agentTimings[nodeName].start = elapsed;
+                    }
+                    activeAgents.add(nodeName);
+                    console.log(`\n⏱️  [${nodeName}] STARTED at ${elapsed}ms`);
+                    console.log(`   Active agents: ${Array.from(activeAgents).join(', ')}`);
+                }
+            },
+        },
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+    };
+    const runConfig = {
+        runId: `parallel-handoffs-${Date.now()}`,
+        graphConfig: {
+            type: 'multi-agent',
+            agents,
+            edges,
+        },
+        customHandlers,
+        returnContent: true,
+        skipCleanup: true,
+    };
+    try {
+        const run = await Run.create(runConfig);
+        /** Prompt designed to trigger parallel handoffs without confusing language */
+        const userMessage = `Help me with two topics:
+1. The history of the internet
+2. A short poem about technology
+
+I need information on both topics.`;
+        conversationHistory.push(new HumanMessage(userMessage));
+        console.log('User message:', userMessage);
+        console.log('\nInvoking multi-agent graph with parallel handoff request...\n');
+        const config = {
+            configurable: {
+                thread_id: 'parallel-handoffs-test-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        const inputs = {
+            messages: conversationHistory,
+        };
+        await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+        }
+        /** Analyze parallel execution */
+        console.log('\n\n========== TIMING SUMMARY ==========');
+        console.log('Available timing keys:', Object.keys(agentTimings));
+        for (const [agent, timing] of Object.entries(agentTimings)) {
+            const duration = timing.end && timing.start ? timing.end - timing.start : 'N/A';
+            console.log(`${agent}: started=${timing.start}ms, ended=${timing.end}ms, duration=${duration}ms`);
+        }
+        /** Check if researcher and writer ran in parallel (handle key variations) */
+        const researcherKey = Object.keys(agentTimings).find((k) => k.includes('researcher'));
+        const writerKey = Object.keys(agentTimings).find((k) => k.includes('writer'));
+        const researcherTiming = researcherKey
+            ? agentTimings[researcherKey]
+            : undefined;
+        const writerTiming = writerKey ? agentTimings[writerKey] : undefined;
+        if (researcherTiming && writerTiming) {
+            const bothStarted = researcherTiming.start && writerTiming.start;
+            const bothEnded = researcherTiming.end && writerTiming.end;
+            if (bothStarted && bothEnded) {
+                const overlap = Math.min(researcherTiming.end, writerTiming.end) -
+                    Math.max(researcherTiming.start, writerTiming.start);
+                if (overlap > 0) {
+                    console.log(`\n✅ PARALLEL HANDOFFS SUCCESSFUL: ${overlap}ms overlap between researcher and writer`);
+                }
+                else {
+                    console.log(`\n⚠️  SEQUENTIAL EXECUTION: researcher and writer did not overlap`);
+                    console.log(`   This may indicate the LLM called transfers sequentially, not in parallel`);
+                }
+            }
+        }
+        else {
+            console.log('\n⚠️  Not all agents were invoked. Check if handoffs occurred.');
+            console.log('   researcher timing:', researcherTiming);
+            console.log('   writer timing:', writerTiming);
+        }
+        console.log('====================================\n');
+        console.log('Final content parts:', contentParts.length, 'parts');
+        console.dir(contentParts, { depth: null });
+        await sleep(3000);
+    }
+    catch (error) {
+        console.error('Error in parallel handoffs test:', error);
+        throw error;
+    }
+}
+testParallelHandoffs();
+//# sourceMappingURL=test-parallel-handoffs.js.map

package/src/scripts/test-parallel-handoffs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-parallel-handoffs.js","sourceRoot":"","sources":["test-parallel-handoffs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;;;;;;;;;GAWG;AACH,KAAK,UAAU,oBAAoB;IACjC,OAAO,CAAC,GAAG,CACT,iEAAiE,CAClE,CAAC;IAEF,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,MAAM,MAAM,GAAoB;QAC9B;YACE,OAAO,EAAE,aAAa;YACtB,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,aAAa;gBACxB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;;;;qFAUiE;SAChF;QACD;YACE,OAAO,EAAE,YAAY;YACrB,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EAAE;;oDAEgC;SAC/C;QACD;YACE,OAAO,EAAE,QAAQ;YACjB,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EAAE;;kDAE8B;SAC7C;KACF,CAAC;IAEF;;;OAGG;IACH,MAAM,KAAK,GAAkB;QAC3B;YACE,IAAI,EAAE,aAAa;YACnB,EAAE,EAAE,YAAY;YAChB,QAAQ,EAAE,SAAS;YACnB,WAAW,EAAE,4DAA4D;YACzE,MAAM,EAAE,4BAA4B;SACrC;QACD;YACE,IAAI,EAAE,aAAa;YACnB,EAAE,EAAE,QAAQ;YACZ,QAAQ,EAAE,SAAS;YACnB,WAAW,EAAE,2DAA2D;YACxE,MAAM,EAAE,2BAA2B;SACpC;KACF,CAAC;IAEF,qDAAqD;IACrD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IACvC,MAAM,YAAY,GAAqD,EAAE,CAAC;IAC1E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE;YAC5B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,MAAM,QAAQ,GAAG,QAAQ,EAAE,cAAwB,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;oBACvC,YAAY,CAAC,QAAQ,CAAC,GAAG,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACtD,YAAY,CAAC,QAAQ,CAAC,CAAC,GAAG,GAAG,OAAO,CAAC;oBACrC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,UAAU,QAAQ,kBAAkB,OAAO,IAAI,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,MAAM,QAAQ,GAAG,QAAQ,EAAE,cAAwB,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;oBACvC,8DAA8D;oBAC9D,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC;wBACnC,YAAY,CAAC,QAAQ,CAAC,GAAG,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACtD,YAAY,CAAC,QAAQ,CAAC,CAAC,KAAK,GAAG,OAAO,CAAC;oBACzC,CAAC;oBACD,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAC3B,OAAO,CAAC,GAAG,CAAC,UAAU,QAAQ,gBAAgB,OAAO,IAAI,CAAC,CAAC;oBAC3D,OAAO,CAAC,GAAG,CACT,qBAAqB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC3D,CAAC;gBACJ,CAAC;YACH,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,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,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,qBAAqB,IAAI,CAAC,GAAG,EAAE,EAAE;QACxC,WAAW,EAAE;YACX,IAAI,EAAE,aAAa;YACnB,MAAM;YACN,KAAK;SACN;QACD,cAAc;QACd,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;KAClB,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,8EAA8E;QAC9E,MAAM,WAAW,GAAG;;;;mCAIW,CAAC;QAEhC,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;QAExD,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAC1C,OAAO,CAAC,GAAG,CACT,iEAAiE,CAClE,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,0BAA0B;aACtC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,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,iCAAiC;QACjC,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3D,MAAM,QAAQ,GACZ,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;YACjE,OAAO,CAAC,GAAG,CACT,GAAG,KAAK,aAAa,MAAM,CAAC,KAAK,aAAa,MAAM,CAAC,GAAG,gBAAgB,QAAQ,IAAI,CACrF,CAAC;QACJ,CAAC;QAED,6EAA6E;QAC7E,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACzD,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CACzB,CAAC;QACF,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CACrB,CAAC;QACF,MAAM,gBAAgB,GAAG,aAAa;YACpC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC;YAC7B,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,YAAY,GAAG,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAErE,IAAI,gBAAgB,IAAI,YAAY,EAAE,CAAC;YACrC,MAAM,WAAW,GAAG,gBAAgB,CAAC,KAAK,IAAI,YAAY,CAAC,KAAK,CAAC;YACjE,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC;YAE3D,IAAI,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC7B,MAAM,OAAO,GACX,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAI,EAAE,YAAY,CAAC,GAAI,CAAC;oBAClD,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,KAAM,EAAE,YAAY,CAAC,KAAM,CAAC,CAAC;gBAEzD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,OAAO,CAAC,GAAG,CACT,qCAAqC,OAAO,0CAA0C,CACvF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CACT,mEAAmE,CACpE,CAAC;oBACF,OAAO,CAAC,GAAG,CACT,6EAA6E,CAC9E,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CACT,gEAAgE,CACjE,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,gBAAgB,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;QAEtD,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,oBAAoB,EAAE,CAAC"}

package/src/scripts/test-parallel-handoffs.ts

@@ -187,6 +187,7 @@ When delegating, provide clear instructions to each agent about what they should
     },
     customHandlers,
     returnContent: true,
+    skipCleanup: true,
   };
 
   try {

package/src/scripts/test-thinking-handoff-bedrock.js

@@ -0,0 +1,132 @@
+#!/usr/bin/env bun
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { Run } from '@/run';
+import { ChatModelStreamHandler } from '@/stream';
+import { Providers, GraphEvents } from '@/common';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+const conversationHistory = [];
+/**
+ * Test edge case: switching from OpenAI supervisor (no thinking) to Bedrock specialist (with thinking enabled)
+ * This should not throw an error about missing thinking blocks
+ */
+async function testBedrockThinkingHandoff() {
+    console.log('Testing OpenAI → Bedrock (with thinking) handoff...\n');
+    // Create custom handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data) => {
+                const toolData = data;
+                if (toolData?.name) {
+                    console.log(`\n🔧 Tool called: ${toolData.name}`);
+                }
+            },
+        },
+    };
+    // Create the graph configuration
+    function createGraphConfig() {
+        console.log('Creating graph with OpenAI supervisor and Bedrock specialist with thinking enabled.\n');
+        const agents = [
+            {
+                agentId: 'supervisor',
+                provider: Providers.OPENAI,
+                clientOptions: {
+                    modelName: 'gpt-4o-mini',
+                    apiKey: process.env.OPENAI_API_KEY,
+                },
+                instructions: `You are a task supervisor. When the user asks about code review, use transfer_to_code_reviewer to hand off to the specialist.`,
+                maxContextTokens: 8000,
+            },
+            {
+                agentId: 'code_reviewer',
+                provider: Providers.BEDROCK,
+                clientOptions: {
+                    region: process.env.BEDROCK_AWS_REGION || 'us-east-1',
+                    model: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
+                    credentials: {
+                        accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID,
+                        secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY,
+                    },
+                    additionalModelRequestFields: {
+                        thinking: {
+                            type: 'enabled',
+                            budget_tokens: 2000,
+                        },
+                    },
+                },
+                instructions: `You are a code review specialist using Bedrock with extended thinking. Think carefully about the code quality, best practices, and potential issues. Provide thoughtful feedback.`,
+                maxContextTokens: 8000,
+            },
+        ];
+        const edges = [
+            {
+                from: 'supervisor',
+                to: ['code_reviewer'],
+                description: 'Transfer to code review specialist',
+                edgeType: 'handoff',
+            },
+        ];
+        return {
+            runId: `bedrock-thinking-handoff-test-${Date.now()}`,
+            graphConfig: {
+                type: 'multi-agent',
+                agents,
+                edges,
+            },
+            customHandlers,
+            returnContent: true,
+            skipCleanup: true,
+        };
+    }
+    try {
+        // Test query that should trigger a handoff
+        const query = 'Can you review this function and tell me if there are any issues?\n\nfunction add(a, b) { return a + b; }';
+        console.log(`${'='.repeat(60)}`);
+        console.log(`USER QUERY: "${query}"`);
+        console.log('='.repeat(60));
+        // Initialize conversation
+        conversationHistory.push(new HumanMessage(query));
+        // Create and run the graph
+        const runConfig = createGraphConfig();
+        const run = await Run.create(runConfig);
+        const config = {
+            configurable: {
+                thread_id: 'bedrock-thinking-handoff-test-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        console.log('\nProcessing request...\n');
+        // Process with streaming
+        const inputs = {
+            messages: conversationHistory,
+        };
+        await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+        }
+        // Success!
+        console.log(`\n${'='.repeat(60)}`);
+        console.log('✅ TEST PASSED');
+        console.log('='.repeat(60));
+        console.log('\nSuccessfully handed off from OpenAI (no thinking) to');
+        console.log('Bedrock with thinking enabled without errors!');
+        console.log('\nThe ensureThinkingBlockInMessages() function correctly');
+        console.log('handled the transition by converting tool sequences to');
+        console.log('HumanMessages before calling the Bedrock API.');
+    }
+    catch (error) {
+        console.error('\n❌ TEST FAILED');
+        console.error('='.repeat(60));
+        console.error('Error:', error);
+        process.exit(1);
+    }
+}
+// Run the test
+testBedrockThinkingHandoff();
+//# sourceMappingURL=test-thinking-handoff-bedrock.js.map

package/src/scripts/test-thinking-handoff-bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-thinking-handoff-bedrock.js","sourceRoot":"","sources":["test-thinking-handoff-bedrock.ts"],"names":[],"mappings":";AAEA,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,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG3D,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;GAGG;AACH,KAAK,UAAU,0BAA0B;IACvC,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;IAErE,yBAAyB;IACzB,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,MAAM,QAAQ,GAAG,IAAW,CAAC;gBAC7B,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;oBACnB,OAAO,CAAC,GAAG,CAAC,qBAAqB,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;SACF;KACF,CAAC;IAEF,iCAAiC;IACjC,SAAS,iBAAiB;QACxB,OAAO,CAAC,GAAG,CACT,uFAAuF,CACxF,CAAC;QAEF,MAAM,MAAM,GAAoB;YAC9B;gBACE,OAAO,EAAE,YAAY;gBACrB,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,aAAa,EAAE;oBACb,SAAS,EAAE,aAAa;oBACxB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;iBACnC;gBACD,YAAY,EAAE,+HAA+H;gBAC7I,gBAAgB,EAAE,IAAI;aACvB;YACD;gBACE,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE,SAAS,CAAC,OAAO;gBAC3B,aAAa,EAAE;oBACb,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,WAAW;oBACrD,KAAK,EAAE,8CAA8C;oBACrD,WAAW,EAAE;wBACX,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,yBAA0B;wBACnD,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,6BAA8B;qBAC5D;oBACD,4BAA4B,EAAE;wBAC5B,QAAQ,EAAE;4BACR,IAAI,EAAE,SAAS;4BACf,aAAa,EAAE,IAAI;yBACpB;qBACF;iBACF;gBACD,YAAY,EAAE,mLAAmL;gBACjM,gBAAgB,EAAE,IAAI;aACvB;SACF,CAAC;QAEF,MAAM,KAAK,GAAkB;YAC3B;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,CAAC,eAAe,CAAC;gBACrB,WAAW,EAAE,oCAAoC;gBACjD,QAAQ,EAAE,SAAS;aACpB;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,iCAAiC,IAAI,CAAC,GAAG,EAAE,EAAE;YACpD,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,2CAA2C;QAC3C,MAAM,KAAK,GACT,2GAA2G,CAAC;QAE9G,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,GAAG,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,0BAA0B;QAC1B,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;QAElD,2BAA2B;QAC3B,MAAM,SAAS,GAAG,iBAAiB,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,iCAAiC;aAC7C;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAEzC,yBAAyB;QACzB,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,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,WAAW;QACX,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC7B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;QACtE,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAC7D,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;QACtE,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC/D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9B,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC/B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,eAAe;AACf,0BAA0B,EAAE,CAAC"}

package/src/scripts/test-thinking-handoff-bedrock.ts

@@ -91,6 +91,7 @@ async function testBedrockThinkingHandoff() {
       },
       customHandlers,
       returnContent: true,
+      skipCleanup: true,
     };
   }
 

package/src/scripts/test-thinking-handoff.js

@@ -0,0 +1,132 @@
+#!/usr/bin/env bun
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { ChatModelStreamHandler } from '@/stream';
+import { Providers, GraphEvents } from '@/common';
+import { Run } from '@/run';
+const conversationHistory = [];
+/**
+ * Test edge case: switching from OpenAI supervisor (no thinking) to Anthropic specialist (with thinking enabled)
+ * This should not throw an error about missing thinking blocks
+ */
+async function testThinkingHandoff() {
+    console.log('Testing OpenAI → Anthropic (with thinking) handoff...\n');
+    // Create custom handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data) => {
+                const toolData = data;
+                if (toolData?.name) {
+                    console.log(`\n🔧 Tool called: ${toolData.name}`);
+                }
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (_event, data) => {
+                const runStep = data;
+                console.log(`\n📍 ON_RUN_STEP: agentId=${runStep.agentId}, groupId=${runStep.groupId}`);
+            },
+        },
+    };
+    // Create the graph configuration
+    function createGraphConfig() {
+        console.log('Creating graph with OpenAI supervisor and Anthropic specialist with thinking enabled.\n');
+        const agents = [
+            {
+                agentId: 'supervisor',
+                provider: Providers.OPENAI,
+                clientOptions: {
+                    modelName: 'gpt-4o-mini',
+                    apiKey: process.env.OPENAI_API_KEY,
+                },
+                instructions: `You are a task supervisor. When the user asks about code review, use transfer_to_code_reviewer to hand off to the specialist.`,
+                maxContextTokens: 8000,
+            },
+            {
+                agentId: 'code_reviewer',
+                provider: Providers.ANTHROPIC,
+                clientOptions: {
+                    modelName: 'claude-3-7-sonnet-20250219',
+                    apiKey: process.env.ANTHROPIC_API_KEY,
+                    thinking: {
+                        type: 'enabled',
+                        budget_tokens: 2000,
+                    },
+                },
+                instructions: `You are a code review specialist. Think carefully about the code quality, best practices, and potential issues. Provide thoughtful feedback.`,
+                maxContextTokens: 8000,
+            },
+        ];
+        const edges = [
+            {
+                from: 'supervisor',
+                to: ['code_reviewer'],
+                description: 'Transfer to code review specialist',
+                edgeType: 'handoff',
+            },
+        ];
+        return {
+            runId: `thinking-handoff-test-${Date.now()}`,
+            graphConfig: {
+                type: 'multi-agent',
+                agents,
+                edges,
+            },
+            customHandlers,
+            returnContent: true,
+            skipCleanup: true,
+        };
+    }
+    try {
+        // Test query that should trigger a handoff
+        const query = 'Can you review this function and tell me if there are any issues?\n\nfunction add(a, b) { return a + b; }';
+        console.log(`${'='.repeat(60)}`);
+        console.log(`USER QUERY: "${query}"`);
+        console.log('='.repeat(60));
+        // Initialize conversation
+        conversationHistory.push(new HumanMessage(query));
+        // Create and run the graph
+        const runConfig = createGraphConfig();
+        const run = await Run.create(runConfig);
+        const config = {
+            configurable: {
+                thread_id: 'thinking-handoff-test-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        console.log('\nProcessing request...\n');
+        // Process with streaming
+        const inputs = {
+            messages: conversationHistory,
+        };
+        await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+        }
+        // Success!
+        console.log(`\n${'='.repeat(60)}`);
+        console.log('✅ TEST PASSED');
+        console.log('='.repeat(60));
+        console.log('\nSuccessfully handed off from OpenAI (no thinking) to');
+        console.log('Anthropic with thinking enabled without errors!');
+        console.log('\nThe ensureThinkingBlockInMessages() function correctly');
+        console.log('added a placeholder thinking block to the last assistant');
+        console.log('message before calling the Anthropic API.');
+    }
+    catch (error) {
+        console.error('\n❌ TEST FAILED');
+        console.error('='.repeat(60));
+        console.error('Error:', error);
+        process.exit(1);
+    }
+}
+// Run the test
+testThinkingHandoff();
+//# sourceMappingURL=test-thinking-handoff.js.map

package/src/scripts/test-thinking-handoff.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-thinking-handoff.js","sourceRoot":"","sources":["test-thinking-handoff.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;GAGG;AACH,KAAK,UAAU,mBAAmB;IAChC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IAEvE,yBAAyB;IACzB,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,MAAM,QAAQ,GAAG,IAAW,CAAC;gBAC7B,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;oBACnB,OAAO,CAAC,GAAG,CAAC,qBAAqB,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,MAAM,OAAO,GAAG,IAAiB,CAAC;gBAClC,OAAO,CAAC,GAAG,CACT,6BAA6B,OAAO,CAAC,OAAO,aAAa,OAAO,CAAC,OAAO,EAAE,CAC3E,CAAC;YACJ,CAAC;SACF;KACF,CAAC;IAEF,iCAAiC;IACjC,SAAS,iBAAiB;QACxB,OAAO,CAAC,GAAG,CACT,yFAAyF,CAC1F,CAAC;QAEF,MAAM,MAAM,GAAoB;YAC9B;gBACE,OAAO,EAAE,YAAY;gBACrB,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,aAAa,EAAE;oBACb,SAAS,EAAE,aAAa;oBACxB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;iBACnC;gBACD,YAAY,EAAE,+HAA+H;gBAC7I,gBAAgB,EAAE,IAAI;aACvB;YACD;gBACE,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE,SAAS,CAAC,SAAS;gBAC7B,aAAa,EAAE;oBACb,SAAS,EAAE,4BAA4B;oBACvC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;oBACrC,QAAQ,EAAE;wBACR,IAAI,EAAE,SAAS;wBACf,aAAa,EAAE,IAAI;qBACpB;iBACF;gBACD,YAAY,EAAE,8IAA8I;gBAC5J,gBAAgB,EAAE,IAAI;aACvB;SACF,CAAC;QAEF,MAAM,KAAK,GAAkB;YAC3B;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,CAAC,eAAe,CAAC;gBACrB,WAAW,EAAE,oCAAoC;gBACjD,QAAQ,EAAE,SAAS;aACpB;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,yBAAyB,IAAI,CAAC,GAAG,EAAE,EAAE;YAC5C,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,2CAA2C;QAC3C,MAAM,KAAK,GACT,2GAA2G,CAAC;QAE9G,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,GAAG,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,0BAA0B;QAC1B,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;QAElD,2BAA2B;QAC3B,MAAM,SAAS,GAAG,iBAAiB,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,yBAAyB;aACrC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAEzC,yBAAyB;QACzB,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,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,WAAW;QACX,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC7B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;QACtE,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9B,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC/B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,eAAe;AACf,mBAAmB,EAAE,CAAC"}

package/src/scripts/test-thinking-handoff.ts

@@ -93,6 +93,7 @@ async function testThinkingHandoff() {
       },
       customHandlers,
       returnContent: true,
+      skipCleanup: true,
     };
   }