@illuma-ai/agents 1.0.89 → 1.0.93

package/src/scripts/memory.ts

@@ -17,19 +17,27 @@ const memoryKv: Record<string, string> = {};
 const setMemory = tool(
   async ({ key, value }) => {
     if (!/^[a-z_]+$/.test(key)) {
-      throw new Error('Key must only contain lowercase letters and underscores');
+      throw new Error(
+        'Key must only contain lowercase letters and underscores'
+      );
     }
-    
+
     memoryKv[key] = value;
-    
+
     return { ok: true };
   },
   {
     name: 'set_memory',
     description: 'Saves important data about the user into memory.',
     schema: z.object({
-      key: z.string().describe('The key of the memory value. Always use lowercase and underscores, no other characters.'),
-      value: z.string().describe('Value can be anything represented as a string')
+      key: z
+        .string()
+        .describe(
+          'The key of the memory value. Always use lowercase and underscores, no other characters.'
+        ),
+      value: z
+        .string()
+        .describe('Value can be anything represented as a string'),
     }),
   }
 );
@@ -48,10 +56,12 @@ async function testStandardStreaming(): Promise<void> {
         streaming: false,
       },
       tools: [setMemory],
-      instructions: 'You can use the `set_memory` tool to save important data about the user into memory. If there is nothing to note about the user specifically, respond with `nothing`.',
+      instructions:
+        'You can use the `set_memory` tool to save important data about the user into memory. If there is nothing to note about the user specifically, respond with `nothing`.',
       toolEnd: true,
     },
     returnContent: true,
+    skipCleanup: true,
   });
 
   const config = {

package/src/scripts/multi-agent-chain.js

@@ -0,0 +1,271 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage, getBufferString, } from '@langchain/core/messages';
+import { Run } from '@/run';
+import { Providers, GraphEvents } from '@/common';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler } from '@/events';
+/**
+ * Helper function to create sequential chain edges with buffer string prompts
+ *
+ * @param agentIds - Array of agent IDs in order of execution
+ * @returns Array of edges configured for sequential chain with buffer prompts
+ */
+function createSequentialChainEdges(agentIds) {
+    const edges = [];
+    for (let i = 0; i < agentIds.length - 1; i++) {
+        const fromAgent = agentIds[i];
+        const toAgent = agentIds[i + 1];
+        edges.push({
+            from: fromAgent,
+            to: toAgent,
+            edgeType: 'direct',
+            // Use a prompt function to create the buffer string from all previous results
+            prompt: (messages, startIndex) => {
+                // Get only the messages from this run (after startIndex)
+                const runMessages = messages.slice(startIndex);
+                // Create buffer string from run messages
+                const bufferString = getBufferString(runMessages);
+                // Format the prompt for the next agent
+                return `Based on the following conversation and analysis from previous agents, please provide your insights:\n\n${bufferString}\n\nPlease add your specific expertise and perspective to this discussion.`;
+            },
+            // Critical: exclude previous results so only the prompt is passed
+            excludeResults: true,
+            description: `Sequential chain from ${fromAgent} to ${toAgent}`,
+        });
+    }
+    return edges;
+}
+const conversationHistory = [];
+/**
+ * Example of sequential agent chain mimicking the old chain behavior
+ *
+ * Graph structure:
+ * START -> researcher -> analyst -> reviewer -> summarizer -> END
+ *
+ * Each agent receives a buffer string of all previous results
+ */
+async function testSequentialAgentChain() {
+    console.log('Testing Sequential Agent Chain (Old Chain Pattern)...\n');
+    // Set up content aggregator
+    const { contentParts, aggregateContent } = createContentAggregator();
+    // Define four agents with specific roles
+    const agents = [
+        {
+            agentId: 'researcher',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are a Research Agent specializing in gathering initial information.
+      Your role is to:
+      1. Identify key aspects of the user's query
+      2. List important factors to consider
+      3. Provide initial research findings
+      
+      Format your response with clear sections and bullet points.
+      Start with "RESEARCH FINDINGS:" and be thorough but concise.`,
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'analyst',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are an Analysis Agent that builds upon research findings.
+      Your role is to:
+      1. Analyze the research provided by the previous agent
+      2. Identify patterns, risks, and opportunities
+      3. Provide deeper analytical insights
+      
+      Start with "ANALYSIS:" and structure your response with clear categories.
+      Reference specific points from the research when relevant.`,
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'reviewer',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are a Critical Review Agent that evaluates the work done so far.
+      Your role is to:
+      1. Review the research and analysis from previous agents
+      2. Identify any gaps or areas that need more attention
+      3. Suggest improvements or additional considerations
+      
+      Start with "CRITICAL REVIEW:" and be constructive in your feedback.
+      Highlight both strengths and areas for improvement.`,
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'summarizer',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are a Summary Agent that creates the final comprehensive output.
+      Your role is to:
+      1. Synthesize all insights from the researcher, analyst, and reviewer
+      2. Create a cohesive, actionable summary
+      3. Provide clear recommendations or conclusions
+      
+      Start with "EXECUTIVE SUMMARY:" followed by key sections.
+      End with "KEY RECOMMENDATIONS:" or "CONCLUSIONS:" as appropriate.`,
+            maxContextTokens: 8000,
+        },
+    ];
+    // Create sequential chain edges using our helper function
+    const agentIds = agents.map((a) => a.agentId);
+    const edges = createSequentialChainEdges(agentIds);
+    // Track agent progression
+    let currentAgent = '';
+    const startTime = Date.now();
+    let messageCount = 0;
+    // Create custom handlers with extensive metadata logging
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: {
+            handle: (_event, _data, metadata) => {
+                console.log('\n====== CHAT_MODEL_END METADATA ======');
+                console.dir(metadata, { depth: null });
+                const elapsed = Date.now() - startTime;
+                console.log(`⏱️  COMPLETED at ${elapsed}ms`);
+            },
+        },
+        [GraphEvents.CHAT_MODEL_START]: {
+            handle: (_event, _data, metadata) => {
+                console.log('\n====== CHAT_MODEL_START METADATA ======');
+                console.dir(metadata, { depth: null });
+                const elapsed = Date.now() - startTime;
+                console.log(`⏱️  STARTED at ${elapsed}ms`);
+            },
+        },
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data, metadata) => {
+                const runStepData = data;
+                console.log('\n====== ON_RUN_STEP ======');
+                console.log('DATA:');
+                console.dir(data, { depth: null });
+                console.log('METADATA:');
+                console.dir(metadata, { depth: null });
+                if (runStepData?.name) {
+                    currentAgent = runStepData.name;
+                    console.log(`\n→ ${currentAgent} is processing...`);
+                }
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data, metadata) => {
+                const runStepData = data;
+                console.log('\n====== ON_RUN_STEP_COMPLETED ======');
+                console.log('DATA:');
+                console.dir(data, { depth: null });
+                console.log('METADATA:');
+                console.dir(metadata, { depth: null });
+                if (runStepData?.name) {
+                    console.log(`✓ ${runStepData.name} completed`);
+                }
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data, metadata) => {
+                console.log('\n====== ON_RUN_STEP_DELTA ======');
+                console.log('DATA:');
+                console.dir(data, { depth: null });
+                console.log('METADATA:');
+                console.dir(metadata, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data, metadata) => {
+                messageCount++;
+                // Only log first few message deltas to avoid spam
+                if (messageCount <= 3) {
+                    console.log('\n====== ON_MESSAGE_DELTA ======');
+                    console.log('DATA:');
+                    console.dir(data, { depth: null });
+                    console.log('METADATA:');
+                    console.dir(metadata, { depth: null });
+                }
+                aggregateContent({ event, data: data });
+            },
+        },
+    };
+    // Create multi-agent run configuration
+    const runConfig = {
+        runId: `sequential-chain-${Date.now()}`,
+        graphConfig: {
+            type: 'multi-agent',
+            agents,
+            edges,
+        },
+        customHandlers,
+        returnContent: true,
+        skipCleanup: true,
+    };
+    try {
+        // Create and execute the run
+        const run = await Run.create(runConfig);
+        // Test with a complex question that benefits from multiple perspectives
+        const userMessage = 'I want to launch a new mobile app for personal finance management. What should I consider?';
+        conversationHistory.push(new HumanMessage(userMessage));
+        console.log(`User: "${userMessage}"\n`);
+        console.log('Starting sequential agent chain...\n');
+        const config = {
+            configurable: {
+                thread_id: 'sequential-chain-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        // Process with streaming
+        const inputs = {
+            messages: conversationHistory,
+        };
+        const finalContentParts = await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+        }
+        console.log('\n\n=== Final Output ===');
+        console.log('Sequential chain completed successfully!');
+        console.log(`Total content parts: ${contentParts.length}`);
+        // Display the buffer accumulation
+        console.log('\n=== Agent Outputs (Buffer Accumulation) ===');
+        const aiMessages = conversationHistory.filter((msg) => msg._getType() === 'ai');
+        aiMessages.forEach((msg, index) => {
+            console.log(`\n--- Agent ${index + 1}: ${agentIds[index]} ---`);
+            console.log(msg.content);
+            // Show buffer preview for next agent (except last)
+            if (index < aiMessages.length - 1) {
+                const bufferSoFar = getBufferString(aiMessages.slice(0, index + 1));
+                console.log(`\n[Buffer passed to ${agentIds[index + 1]}]: ${bufferSoFar.slice(0, 150)}...`);
+            }
+        });
+        // Demonstrate that each agent built upon previous results
+        console.log('\n=== Chain Analysis ===');
+        console.log('1. Researcher provided initial findings');
+        console.log('2. Analyst built upon research with deeper insights');
+        console.log('3. Reviewer critiqued and identified gaps');
+        console.log('4. Summarizer synthesized everything into actionable output');
+    }
+    catch (error) {
+        console.error('Error in sequential agent chain test:', error);
+    }
+}
+// Run the test
+testSequentialAgentChain();
+//# sourceMappingURL=multi-agent-chain.js.map

package/src/scripts/multi-agent-chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-agent-chain.js","sourceRoot":"","sources":["multi-agent-chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EACL,YAAY,EAEZ,eAAe,GAChB,MAAM,0BAA0B,CAAC;AAClC,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,EAAmB,MAAM,UAAU,CAAC;AAG3D;;;;;GAKG;AACH,SAAS,0BAA0B,CAAC,QAAkB;IACpD,MAAM,KAAK,GAAkB,EAAE,CAAC;IAEhC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7C,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAEhC,KAAK,CAAC,IAAI,CAAC;YACT,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,OAAO;YACX,QAAQ,EAAE,QAAQ;YAClB,8EAA8E;YAC9E,MAAM,EAAE,CAAC,QAAuB,EAAE,UAAkB,EAAE,EAAE;gBACtD,yDAAyD;gBACzD,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBAE/C,yCAAyC;gBACzC,MAAM,YAAY,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;gBAElD,uCAAuC;gBACvC,OAAO,2GAA2G,YAAY,4EAA4E,CAAC;YAC7M,CAAC;YACD,kEAAkE;YAClE,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,yBAAyB,SAAS,OAAO,OAAO,EAAE;SAChE,CAAC,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;;;;;GAOG;AACH,KAAK,UAAU,wBAAwB;IACrC,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IAEvE,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,yCAAyC;IACzC,MAAM,MAAM,GAAoB;QAC9B;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;;;;;;;mEAO+C;YAC7D,gBAAgB,EAAE,IAAI;SACvB;QACD;YACE,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EAAE;;;;;;;iEAO6C;YAC3D,gBAAgB,EAAE,IAAI;SACvB;QACD;YACE,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EAAE;;;;;;;0DAOsC;YACpD,gBAAgB,EAAE,IAAI;SACvB;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;;;;;;;wEAOoD;YAClE,gBAAgB,EAAE,IAAI;SACvB;KACF,CAAC;IAEF,0DAA0D;IAC1D,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IAEnD,0BAA0B;IAC1B,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,yDAAyD;IACzD,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,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;gBACvD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACvC,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,IAAI,CAAC,CAAC;YAC/C,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;gBACzD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACvC,OAAO,CAAC,GAAG,CAAC,kBAAkB,OAAO,IAAI,CAAC,CAAC;YAC7C,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,MAAM,WAAW,GAAG,IAAW,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC3C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,IAAI,WAAW,EAAE,IAAI,EAAE,CAAC;oBACtB,YAAY,GAAG,WAAW,CAAC,IAAI,CAAC;oBAChC,OAAO,CAAC,GAAG,CAAC,OAAO,YAAY,mBAAmB,CAAC,CAAC;gBACtD,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,EACvB,QAAkC,EAC5B,EAAE;gBACR,MAAM,WAAW,GAAG,IAAW,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;gBACrD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,IAAI,WAAW,EAAE,IAAI,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,KAAK,WAAW,CAAC,IAAI,YAAY,CAAC,CAAC;gBACjD,CAAC;gBACD,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,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,EACvB,QAAkC,EAC5B,EAAE;gBACR,YAAY,EAAE,CAAC;gBACf,kDAAkD;gBAClD,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;oBAChD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;oBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;oBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzC,CAAC;gBACD,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;KACF,CAAC;IAEF,uCAAuC;IACvC,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,oBAAoB,IAAI,CAAC,GAAG,EAAE,EAAE;QACvC,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,6BAA6B;QAC7B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,wEAAwE;QACxE,MAAM,WAAW,GACf,4FAA4F,CAAC;QAC/F,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;QAExD,OAAO,CAAC,GAAG,CAAC,UAAU,WAAW,KAAK,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,oBAAoB;aAChC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,yBAAyB;QACzB,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,mBAAmB;SAC9B,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,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,CAAC,wBAAwB,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3D,kCAAkC;QAClC,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAE7D,MAAM,UAAU,GAAG,mBAAmB,CAAC,MAAM,CAC3C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,IAAI,CACjC,CAAC;QAEF,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAChC,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,GAAG,CAAC,KAAK,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAEzB,mDAAmD;YACnD,IAAI,KAAK,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,WAAW,GAAG,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;gBACpE,OAAO,CAAC,GAAG,CACT,uBAAuB,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAC/E,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,0DAA0D;QAC1D,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CAAC,6DAA6D,CAAC,CAAC;IAC7E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED,eAAe;AACf,wBAAwB,EAAE,CAAC"}

package/src/scripts/multi-agent-chain.ts

@@ -258,6 +258,7 @@ async function testSequentialAgentChain() {
     },
     customHandlers,
     returnContent: true,
+    skipCleanup: true,
   };
 
   try {

package/src/scripts/multi-agent-conditional.js

@@ -0,0 +1,185 @@
+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 = [];
+/**
+ * Example of conditional multi-agent system
+ *
+ * Graph structure:
+ * START -> classifier
+ * classifier -> technical_expert (if technical question)
+ * classifier -> business_expert (if business question)
+ * classifier -> general_assistant (otherwise)
+ * [all experts] -> END
+ */
+async function testConditionalMultiAgent() {
+    console.log('Testing Conditional Multi-Agent System...\n');
+    // Set up content aggregator
+    const { contentParts, aggregateContent } = createContentAggregator();
+    // Define specialized agents
+    const agents = [
+        {
+            agentId: 'classifier',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: 'You are a query classifier. Analyze user questions and determine if they are technical, business-related, or general.',
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'technical_expert',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: 'You are a technical expert. Provide detailed technical answers about programming, systems, and technology.',
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'business_expert',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: 'You are a business expert. Provide insights on business strategy, operations, and management.',
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'general_assistant',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: 'You are a helpful general assistant. Answer questions on a wide range of topics.',
+            maxContextTokens: 8000,
+        },
+    ];
+    // Define conditional edges
+    // These create handoff tools with conditional routing logic
+    const edges = [
+        {
+            from: 'classifier',
+            to: ['technical_expert', 'business_expert', 'general_assistant'],
+            description: 'Route to appropriate expert based on query type',
+            condition: (state) => {
+                // Simple keyword-based routing for demo
+                // In a real system, this would use the classifier's analysis
+                const lastMessage = state.messages[state.messages.length - 1];
+                const content = lastMessage.content?.toString().toLowerCase() || '';
+                if (content.includes('code') ||
+                    content.includes('programming') ||
+                    content.includes('technical')) {
+                    return 'technical_expert';
+                }
+                else if (content.includes('business') ||
+                    content.includes('strategy') ||
+                    content.includes('market')) {
+                    return 'business_expert';
+                }
+                else {
+                    return 'general_assistant';
+                }
+            },
+        },
+    ];
+    // Track selected expert
+    let selectedExpert = '';
+    // Create custom handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [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) => {
+                const runStepData = data;
+                if (runStepData?.name && runStepData.name !== 'classifier') {
+                    selectedExpert = runStepData.name;
+                    console.log(`Routing to: ${selectedExpert}`);
+                }
+                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 });
+            },
+        },
+    };
+    // Create multi-agent run configuration
+    const runConfig = {
+        runId: `conditional-multi-agent-${Date.now()}`,
+        graphConfig: {
+            type: 'multi-agent',
+            agents,
+            edges,
+        },
+        customHandlers,
+        returnContent: true,
+        skipCleanup: true,
+    };
+    try {
+        // Create and execute the run
+        const run = await Run.create(runConfig);
+        // Test with different types of questions
+        const testQuestions = [
+            'How do I implement a binary search tree in Python?',
+            'What are the key strategies for market expansion?',
+            'What is the capital of France?',
+        ];
+        const config = {
+            configurable: {
+                thread_id: 'conditional-conversation-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        for (const question of testQuestions) {
+            console.log(`\n--- Processing: "${question}" ---\n`);
+            // Reset for each question
+            selectedExpert = '';
+            conversationHistory.length = 0;
+            conversationHistory.push(new HumanMessage(question));
+            // Process with streaming
+            const inputs = {
+                messages: conversationHistory,
+            };
+            const finalContentParts = await run.processStream(inputs, config);
+            const finalMessages = run.getRunMessages();
+            if (finalMessages) {
+                conversationHistory.push(...finalMessages);
+            }
+            console.log(`\n\nExpert used: ${selectedExpert}`);
+            console.log('Content parts:', contentParts.length);
+            console.log('---');
+            console.dir(contentParts, { depth: null });
+        }
+    }
+    catch (error) {
+        console.error('Error in conditional multi-agent test:', error);
+    }
+}
+// Run the test
+testConditionalMultiAgent();
+//# sourceMappingURL=multi-agent-conditional.js.map

package/src/scripts/multi-agent-conditional.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-agent-conditional.js","sourceRoot":"","sources":["multi-agent-conditional.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;;;;;;;;;GASG;AACH,KAAK,UAAU,yBAAyB;IACtC,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAE3D,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,4BAA4B;IAC5B,MAAM,MAAM,GAAoB;QAC9B;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,EACV,uHAAuH;YACzH,gBAAgB,EAAE,IAAI;SACvB;QACD;YACE,OAAO,EAAE,kBAAkB;YAC3B,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EACV,4GAA4G;YAC9G,gBAAgB,EAAE,IAAI;SACvB;QACD;YACE,OAAO,EAAE,iBAAiB;YAC1B,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EACV,+FAA+F;YACjG,gBAAgB,EAAE,IAAI;SACvB;QACD;YACE,OAAO,EAAE,mBAAmB;YAC5B,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,aAAa,EAAE;gBACb,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EACV,kFAAkF;YACpF,gBAAgB,EAAE,IAAI;SACvB;KACF,CAAC;IAEF,2BAA2B;IAC3B,4DAA4D;IAC5D,MAAM,KAAK,GAAkB;QAC3B;YACE,IAAI,EAAE,YAAY;YAClB,EAAE,EAAE,CAAC,kBAAkB,EAAE,iBAAiB,EAAE,mBAAmB,CAAC;YAChE,WAAW,EAAE,iDAAiD;YAC9D,SAAS,EAAE,CAAC,KAAuB,EAAE,EAAE;gBACrC,wCAAwC;gBACxC,6DAA6D;gBAC7D,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC9D,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;gBAEpE,IACE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;oBACxB,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;oBAC/B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAC7B,CAAC;oBACD,OAAO,kBAAkB,CAAC;gBAC5B,CAAC;qBAAM,IACL,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;oBAC5B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;oBAC5B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAC1B,CAAC;oBACD,OAAO,iBAAiB,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBACN,OAAO,mBAAmB,CAAC;gBAC7B,CAAC;YACH,CAAC;SACF;KACF,CAAC;IAEF,wBAAwB;IACxB,IAAI,cAAc,GAAG,EAAE,CAAC;IAExB,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,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,MAAM,WAAW,GAAG,IAAW,CAAC;gBAChC,IAAI,WAAW,EAAE,IAAI,IAAI,WAAW,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC3D,cAAc,GAAG,WAAW,CAAC,IAAI,CAAC;oBAClC,OAAO,CAAC,GAAG,CAAC,eAAe,cAAc,EAAE,CAAC,CAAC;gBAC/C,CAAC;gBACD,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,uCAAuC;IACvC,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,2BAA2B,IAAI,CAAC,GAAG,EAAE,EAAE;QAC9C,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,6BAA6B;QAC7B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,yCAAyC;QACzC,MAAM,aAAa,GAAG;YACpB,oDAAoD;YACpD,mDAAmD;YACnD,gCAAgC;SACjC,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,4BAA4B;aACxC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,OAAO,CAAC,GAAG,CAAC,sBAAsB,QAAQ,SAAS,CAAC,CAAC;YAErD,0BAA0B;YAC1B,cAAc,GAAG,EAAE,CAAC;YACpB,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAAC;YAC/B,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;YAErD,yBAAyB;YACzB,MAAM,MAAM,GAAG;gBACb,QAAQ,EAAE,mBAAmB;aAC9B,CAAC;YAEF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YAE3C,IAAI,aAAa,EAAE,CAAC;gBAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,oBAAoB,cAAc,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;IACjE,CAAC;AACH,CAAC;AAED,eAAe;AACf,yBAAyB,EAAE,CAAC"}

package/src/scripts/multi-agent-conditional.ts

@@ -166,6 +166,7 @@ async function testConditionalMultiAgent() {
     },
     customHandlers,
     returnContent: true,
+    skipCleanup: true,
   };
 
   try {