@illuma-ai/agents 1.0.90 → 1.0.93

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

@@ -0,0 +1,184 @@
+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 simple sequential multi-agent system
+ *
+ * Graph structure:
+ * START -> agent_a -> agent_b -> agent_c -> END
+ *
+ * No conditions, no tools, just automatic sequential flow
+ */
+async function testSequentialMultiAgent() {
+    console.log('Testing Sequential Multi-Agent System (A → B → C)...\n');
+    // Set up content aggregator
+    const { contentParts, aggregateContent, stepMap } = createContentAggregator();
+    // Define three simple agents
+    const agents = [
+        {
+            agentId: 'agent_a',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are Agent A, the first agent in a sequential workflow.
+      Your job is to:
+      1. Receive the initial user request
+      2. Process it and add your perspective (keep it brief, 2-3 sentences)
+      3. Pass it along to Agent B
+      
+      Start your response with "AGENT A:" and end with "Passing to Agent B..."`,
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'agent_b',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are Agent B, the second agent in a sequential workflow.
+      Your job is to:
+      1. Receive the context from Agent A
+      2. Add your own analysis or perspective (keep it brief, 2-3 sentences)
+      3. Pass it along to Agent C
+      
+      Start your response with "AGENT B:" and end with "Passing to Agent C..."`,
+            maxContextTokens: 8000,
+        },
+        {
+            agentId: 'agent_c',
+            provider: Providers.ANTHROPIC,
+            clientOptions: {
+                modelName: 'claude-haiku-4-5',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            },
+            instructions: `You are Agent C, the final agent in a sequential workflow.
+      Your job is to:
+      1. Receive the context from Agents A and B
+      2. Provide a final summary or conclusion based on all previous inputs
+      3. Complete the workflow
+      
+      Start your response with "AGENT C:" and end with "Workflow complete."`,
+            maxContextTokens: 8000,
+        },
+    ];
+    // Define sequential edges using 'direct' type to avoid tool creation
+    // This creates automatic transitions without requiring agents to call tools
+    const edges = [
+        {
+            from: 'agent_a',
+            to: 'agent_b',
+            edgeType: 'direct', // This creates direct edges without tools
+            description: 'Automatic transition from A to B',
+        },
+        {
+            from: 'agent_b',
+            to: 'agent_c',
+            edgeType: 'direct', // This creates direct edges without tools
+            description: 'Automatic transition from B to C',
+        },
+    ];
+    // Track agent progression
+    let currentAgent = '';
+    // 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]: {
+            handle: (event, data) => {
+                const runStepData = data;
+                if (runStepData?.name) {
+                    currentAgent = runStepData.name;
+                    console.log(`\n→ ${currentAgent} is processing...`);
+                }
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                const runStepData = data;
+                if (runStepData?.name) {
+                    console.log(`✓ ${runStepData.name} completed`);
+                }
+                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: `sequential-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 a simple question
+        const userMessage = 'What are the key considerations for building a recommendation system?';
+        conversationHistory.push(new HumanMessage(userMessage));
+        console.log(`User: "${userMessage}"\n`);
+        console.log('Starting sequential workflow...\n');
+        const config = {
+            configurable: {
+                thread_id: 'sequential-conversation-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 flow completed successfully!');
+        console.log(`Total content parts: ${contentParts.length}`);
+        console.log('\n=== Content Parts (clean, no metadata) ===');
+        console.dir(contentParts, { depth: null });
+        console.log('\n=== Step Map (separate from content) ===');
+        console.dir(Object.fromEntries(stepMap), { depth: null });
+        // Display the sequential responses
+        const aiMessages = conversationHistory.filter((msg) => msg._getType() === 'ai');
+        aiMessages.forEach((msg, index) => {
+            console.log(`\n--- Response ${index + 1} ---`);
+            console.log(msg.content);
+        });
+    }
+    catch (error) {
+        console.error('Error in sequential multi-agent test:', error);
+    }
+}
+// Run the test
+testSequentialMultiAgent();
+//# sourceMappingURL=multi-agent-sequence.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"multi-agent-sequence.js","sourceRoot":"","sources":["multi-agent-sequence.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;;;;;;;GAOG;AACH,KAAK,UAAU,wBAAwB;IACrC,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;IAEtE,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,OAAO,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAE9E,6BAA6B;IAC7B,MAAM,MAAM,GAAoB;QAC9B;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;;;;;;+EAM2D;YACzE,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;;;;;;+EAM2D;YACzE,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;;;;;;4EAMwD;YACtE,gBAAgB,EAAE,IAAI;SACvB;KACF,CAAC;IAEF,qEAAqE;IACrE,4EAA4E;IAC5E,MAAM,KAAK,GAAkB;QAC3B;YACE,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,SAAS;YACb,QAAQ,EAAE,QAAQ,EAAE,0CAA0C;YAC9D,WAAW,EAAE,kCAAkC;SAChD;QACD;YACE,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,SAAS;YACb,QAAQ,EAAE,QAAQ,EAAE,0CAA0C;YAC9D,WAAW,EAAE,kCAAkC;SAChD;KACF,CAAC;IAEF,0BAA0B;IAC1B,IAAI,YAAY,GAAG,EAAE,CAAC;IAEtB,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,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,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,EACjB,EAAE;gBACR,MAAM,WAAW,GAAG,IAAW,CAAC;gBAChC,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,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,0BAA0B,IAAI,CAAC,GAAG,EAAE,EAAE;QAC7C,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,8BAA8B;QAC9B,MAAM,WAAW,GACf,uEAAuE,CAAC;QAC1E,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,mCAAmC,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,2BAA2B;aACvC;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,yCAAyC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,wBAAwB,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;QAC1D,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,mCAAmC;QACnC,MAAM,UAAU,GAAG,mBAAmB,CAAC,MAAM,CAC3C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,IAAI,CACjC,CAAC;QACF,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAChC,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,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-sequence.ts

@@ -22,8 +22,7 @@ async function testSequentialMultiAgent() {
   console.log('Testing Sequential Multi-Agent System (A → B → C)...\n');
 
   // Set up content aggregator
-  const { contentParts, aggregateContent, contentMetadataMap } =
-    createContentAggregator();
+  const { contentParts, aggregateContent, stepMap } = createContentAggregator();
 
   // Define three simple agents
   const agents: t.AgentInputs[] = [
@@ -158,6 +157,7 @@ async function testSequentialMultiAgent() {
     },
     customHandlers,
     returnContent: true,
+    skipCleanup: true,
   };
 
   try {
@@ -197,8 +197,8 @@ async function testSequentialMultiAgent() {
     console.log(`Total content parts: ${contentParts.length}`);
     console.log('\n=== Content Parts (clean, no metadata) ===');
     console.dir(contentParts, { depth: null });
-    console.log('\n=== Content Metadata Map (separate from content) ===');
-    console.dir(Object.fromEntries(contentMetadataMap), { depth: null });
+    console.log('\n=== Step Map (separate from content) ===');
+    console.dir(Object.fromEntries(stepMap), { depth: null });
 
     // Display the sequential responses
     const aiMessages = conversationHistory.filter(

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

@@ -0,0 +1,324 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { Providers, GraphEvents } from '@/common';
+import { sleep } from '@/utils/run';
+import { Run } from '@/run';
+const conversationHistory = [];
+/**
+ * Example of supervisor-based multi-agent system
+ *
+ * The supervisor has handoff tools for 5 different specialists.
+ * To demonstrate the concept while respecting LangGraph constraints,
+ * we show two approaches:
+ *
+ * 1. All 5 specialists exist but share the same adaptive configuration
+ * 2. Only 2 agents total by using a single adaptive specialist (requires workaround)
+ */
+async function testSupervisorMultiAgent() {
+    console.log('Testing Supervisor-Based Multi-Agent System...\n');
+    // NOTE: To truly have only 2 agents with 5 handoff tools, you would need:
+    // 1. Custom tool implementation (see multi-agent-supervisor-mock.ts)
+    // 2. Or modify MultiAgentGraph to support "virtual" agents
+    // 3. Or use a single conditional edge with role parameter
+    //
+    // This example shows the concept using 6 agents that share configuration
+    // Set up content aggregator
+    const { contentParts, aggregateContent } = createContentAggregator();
+    // Define configurations for all possible specialists
+    const specialistConfigs = {
+        data_analyst: {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a Data Analyst specialist. Your expertise includes:
+      - Statistical analysis and data visualization
+      - SQL queries and database optimization
+      - Python/R for data science
+      - Machine learning model evaluation
+      - A/B testing and experiment design
+      
+      Follow the supervisor's specific instructions carefully.`,
+            maxContextTokens: 8000,
+        },
+        security_expert: {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a Security Expert. Your expertise includes:
+      - Cybersecurity best practices
+      - Vulnerability assessment and penetration testing
+      - Security architecture and threat modeling
+      - Compliance (GDPR, HIPAA, SOC2, etc.)
+      - Incident response and forensics
+      
+      Follow the supervisor's specific instructions carefully.`,
+            maxContextTokens: 8000,
+        },
+        product_designer: {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a Product Designer. Your expertise includes:
+      - User experience (UX) design principles
+      - User interface (UI) design and prototyping
+      - Design systems and component libraries
+      - User research and usability testing
+      - Accessibility and inclusive design
+      
+      Follow the supervisor's specific instructions carefully.`,
+            maxContextTokens: 8000,
+        },
+        devops_engineer: {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a DevOps Engineer. Your expertise includes:
+      - CI/CD pipeline design and optimization
+      - Infrastructure as Code (Terraform, CloudFormation)
+      - Container orchestration (Kubernetes, Docker)
+      - Cloud platforms (AWS, GCP, Azure)
+      - Monitoring, logging, and observability
+      
+      Follow the supervisor's specific instructions carefully.`,
+            maxContextTokens: 8000,
+        },
+        legal_advisor: {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are a Legal Advisor specializing in technology. Your expertise includes:
+      - Software licensing and open source compliance
+      - Data privacy and protection laws
+      - Intellectual property and patents
+      - Contract review and negotiation
+      - Regulatory compliance for tech companies
+      
+      Follow the supervisor's specific instructions carefully.`,
+            maxContextTokens: 8000,
+        },
+    };
+    // Track which specialist role was selected
+    let selectedRole = '';
+    let roleInstructions = '';
+    // 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]: {
+            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.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                const toolData = data;
+                if (toolData?.name?.includes('transfer_to_')) {
+                    const specialist = toolData.name.replace('transfer_to_', '');
+                    console.log(`\n🔀 Transferring to ${specialist}...`);
+                    selectedRole = specialist;
+                }
+            },
+        },
+    };
+    // Function to create the graph with supervisor having multiple handoff options
+    function createSupervisorGraph() {
+        console.log(`\nCreating graph with supervisor and 5 specialist agents.`);
+        console.log('All specialists share the same adaptive configuration.\n');
+        // Define the adaptive specialist configuration that will be reused
+        const specialistConfig = {
+            provider: Providers.OPENAI,
+            clientOptions: {
+                modelName: 'gpt-4.1',
+                apiKey: process.env.OPENAI_API_KEY,
+            },
+            instructions: `You are an Adaptive Specialist. Your agent ID indicates your role:
+      
+      - data_analyst: Focus on statistical analysis, metrics, ML evaluation, A/B testing
+      - security_expert: Focus on cybersecurity, vulnerability assessment, compliance  
+      - product_designer: Focus on UX/UI design, user research, accessibility
+      - devops_engineer: Focus on CI/CD, infrastructure, cloud platforms, monitoring
+      - legal_advisor: Focus on licensing, privacy laws, contracts, regulatory compliance
+      
+      The supervisor will provide specific instructions. Follow them while maintaining your expert perspective.`,
+            maxContextTokens: 8000,
+        };
+        // Create the graph with supervisor and all 5 specialists
+        // All specialists share the same adaptive configuration
+        const agents = [
+            {
+                agentId: 'supervisor',
+                provider: Providers.OPENAI,
+                clientOptions: {
+                    modelName: 'gpt-4.1-mini',
+                    apiKey: process.env.OPENAI_API_KEY,
+                },
+                instructions: `You are a Task Supervisor with access to 5 specialist agents:
+        1. transfer_to_data_analyst - For statistical analysis and metrics
+        2. transfer_to_security_expert - For cybersecurity and vulnerability assessment  
+        3. transfer_to_product_designer - For UX/UI design
+        4. transfer_to_devops_engineer - For infrastructure and deployment
+        5. transfer_to_legal_advisor - For compliance and licensing
+        
+        Your role is to:
+        1. Analyze the incoming request
+        2. Decide which specialist is best suited
+        3. Use the appropriate transfer tool (e.g., transfer_to_data_analyst)
+        4. Provide specific instructions to guide their work
+        
+        Be specific about what you need from the specialist.`,
+                maxContextTokens: 8000,
+            },
+            // Include all 5 specialists with the same adaptive configuration
+            {
+                agentId: 'data_analyst',
+                ...specialistConfig,
+            },
+            {
+                agentId: 'security_expert',
+                ...specialistConfig,
+            },
+            {
+                agentId: 'product_designer',
+                ...specialistConfig,
+            },
+            {
+                agentId: 'devops_engineer',
+                ...specialistConfig,
+            },
+            {
+                agentId: 'legal_advisor',
+                ...specialistConfig,
+            },
+        ];
+        // Create edges from supervisor to all 5 specialists
+        const edges = [
+            {
+                from: 'supervisor',
+                to: 'data_analyst',
+                description: 'Transfer to data analyst for statistical analysis and metrics',
+                edgeType: 'handoff',
+            },
+            {
+                from: 'supervisor',
+                to: 'security_expert',
+                description: 'Transfer to security expert for cybersecurity assessment',
+                edgeType: 'handoff',
+            },
+            {
+                from: 'supervisor',
+                to: 'product_designer',
+                description: 'Transfer to product designer for UX/UI design',
+                edgeType: 'handoff',
+            },
+            {
+                from: 'supervisor',
+                to: 'devops_engineer',
+                description: 'Transfer to DevOps engineer for infrastructure and deployment',
+                edgeType: 'handoff',
+            },
+            {
+                from: 'supervisor',
+                to: 'legal_advisor',
+                description: 'Transfer to legal advisor for compliance and licensing',
+                edgeType: 'handoff',
+            },
+        ];
+        return {
+            runId: `supervisor-multi-agent-${Date.now()}`,
+            graphConfig: {
+                type: 'multi-agent',
+                agents,
+                edges,
+            },
+            customHandlers,
+            returnContent: true,
+            skipCleanup: true,
+        };
+    }
+    try {
+        // Test with different queries
+        const testQueries = [
+            'How can we analyze user engagement metrics to improve our product?',
+            // 'What security measures should we implement for our new API?',
+            // 'Can you help design a better onboarding flow for our mobile app?',
+            // 'We need to set up a CI/CD pipeline for our microservices.',
+            // 'What are the legal implications of using GPL-licensed code in our product?',
+        ];
+        const config = {
+            configurable: {
+                thread_id: 'supervisor-conversation-1',
+            },
+            streamMode: 'values',
+            version: 'v2',
+        };
+        for (const query of testQueries) {
+            console.log(`\n${'='.repeat(60)}`);
+            console.log(`USER QUERY: "${query}"`);
+            console.log('='.repeat(60));
+            // Reset conversation
+            conversationHistory.length = 0;
+            conversationHistory.push(new HumanMessage(query));
+            // Create graph with supervisor having 5 handoff tools to 1 adaptive specialist
+            const runConfig = createSupervisorGraph();
+            const run = await Run.create(runConfig);
+            console.log('Processing request...');
+            // Process with streaming
+            const inputs = {
+                messages: conversationHistory,
+            };
+            const finalContentParts = await run.processStream(inputs, config);
+            const finalMessages = run.getRunMessages();
+            if (finalMessages) {
+                conversationHistory.push(...finalMessages);
+            }
+            // Show summary
+            console.log(`\n${'─'.repeat(60)}`);
+            console.log(`Agents in graph: 6 total (supervisor + 5 specialists)`);
+            console.log(`All specialists share the same adaptive configuration`);
+            console.log(`Supervisor tools: transfer_to_data_analyst, transfer_to_security_expert,`);
+            console.log(`                 transfer_to_product_designer, transfer_to_devops_engineer,`);
+            console.log(`                 transfer_to_legal_advisor`);
+            console.log('─'.repeat(60));
+            console.dir(contentParts, { depth: null });
+        }
+        await sleep(3000);
+    }
+    catch (error) {
+        console.error('Error in supervisor multi-agent test:', error);
+    }
+}
+// Run the test
+testSupervisorMultiAgent();
+//# sourceMappingURL=multi-agent-supervisor.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"multi-agent-supervisor.js","sourceRoot":"","sources":["multi-agent-supervisor.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,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,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;;;;;;;;;GASG;AACH,KAAK,UAAU,wBAAwB;IACrC,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;IAEhE,0EAA0E;IAC1E,qEAAqE;IACrE,2DAA2D;IAC3D,0DAA0D;IAC1D,EAAE;IACF,yEAAyE;IAEzE,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,qDAAqD;IACrD,MAAM,iBAAiB,GAAG;QACxB,YAAY,EAAE;YACZ,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;+DAO2C;YACzD,gBAAgB,EAAE,IAAI;SACvB;QACD,eAAe,EAAE;YACf,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;+DAO2C;YACzD,gBAAgB,EAAE,IAAI;SACvB;QACD,gBAAgB,EAAE;YAChB,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;+DAO2C;YACzD,gBAAgB,EAAE,IAAI;SACvB;QACD,eAAe,EAAE;YACf,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;+DAO2C;YACzD,gBAAgB,EAAE,IAAI;SACvB;QACD,aAAa,EAAE;YACb,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;+DAO2C;YACzD,gBAAgB,EAAE,IAAI;SACvB;KACF,CAAC;IAEF,2CAA2C;IAC3C,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,gBAAgB,GAAG,EAAE,CAAC;IAE1B,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,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,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,IAAI,QAAQ,EAAE,IAAI,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;oBAC7C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;oBAC7D,OAAO,CAAC,GAAG,CAAC,wBAAwB,UAAU,KAAK,CAAC,CAAC;oBACrD,YAAY,GAAG,UAAU,CAAC;gBAC5B,CAAC;YACH,CAAC;SACF;KACF,CAAC;IAEF,+EAA+E;IAC/E,SAAS,qBAAqB;QAC5B,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;QACzE,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;QAExE,mEAAmE;QACnE,MAAM,gBAAgB,GAAG;YACvB,QAAQ,EAAE,SAAS,CAAC,MAAM;YAC1B,aAAa,EAAE;gBACb,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;aACnC;YACD,YAAY,EAAE;;;;;;;;gHAQ4F;YAC1G,gBAAgB,EAAE,IAAI;SACvB,CAAC;QAEF,yDAAyD;QACzD,wDAAwD;QACxD,MAAM,MAAM,GAAoB;YAC9B;gBACE,OAAO,EAAE,YAAY;gBACrB,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,aAAa,EAAE;oBACb,SAAS,EAAE,cAAc;oBACzB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;iBACnC;gBACD,YAAY,EAAE;;;;;;;;;;;;;6DAauC;gBACrD,gBAAgB,EAAE,IAAI;aACvB;YACD,iEAAiE;YACjE;gBACE,OAAO,EAAE,cAAc;gBACvB,GAAG,gBAAgB;aACpB;YACD;gBACE,OAAO,EAAE,iBAAiB;gBAC1B,GAAG,gBAAgB;aACpB;YACD;gBACE,OAAO,EAAE,kBAAkB;gBAC3B,GAAG,gBAAgB;aACpB;YACD;gBACE,OAAO,EAAE,iBAAiB;gBAC1B,GAAG,gBAAgB;aACpB;YACD;gBACE,OAAO,EAAE,eAAe;gBACxB,GAAG,gBAAgB;aACpB;SACF,CAAC;QAEF,oDAAoD;QACpD,MAAM,KAAK,GAAkB;YAC3B;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,cAAc;gBAClB,WAAW,EACT,+DAA+D;gBACjE,QAAQ,EAAE,SAAS;aACpB;YACD;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,iBAAiB;gBACrB,WAAW,EAAE,0DAA0D;gBACvE,QAAQ,EAAE,SAAS;aACpB;YACD;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,kBAAkB;gBACtB,WAAW,EAAE,+CAA+C;gBAC5D,QAAQ,EAAE,SAAS;aACpB;YACD;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,iBAAiB;gBACrB,WAAW,EACT,+DAA+D;gBACjE,QAAQ,EAAE,SAAS;aACpB;YACD;gBACE,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,eAAe;gBACnB,WAAW,EAAE,wDAAwD;gBACrE,QAAQ,EAAE,SAAS;aACpB;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,0BAA0B,IAAI,CAAC,GAAG,EAAE,EAAE;YAC7C,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,8BAA8B;QAC9B,MAAM,WAAW,GAAG;YAClB,oEAAoE;YACpE,iEAAiE;YACjE,sEAAsE;YACtE,+DAA+D;YAC/D,gFAAgF;SACjF,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,2BAA2B;aACvC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,GAAG,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;YAE5B,qBAAqB;YACrB,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAAC;YAC/B,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;YAElD,+EAA+E;YAC/E,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAC1C,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAExC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAErC,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,eAAe;YACf,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;YACrE,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;YACrE,OAAO,CAAC,GAAG,CACT,0EAA0E,CAC3E,CAAC;YACF,OAAO,CAAC,GAAG,CACT,6EAA6E,CAC9E,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAC1D,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;IACpB,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-supervisor.ts

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