@illuma-ai/agents 1.1.20 → 1.1.22

package/src/scripts/parallel-full-metadata-test.js

@@ -1,176 +0,0 @@
-import { config } from 'dotenv';
-config();
-import { HumanMessage } from '@langchain/core/messages';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { Providers, GraphEvents } from '@/common';
-import { sleep } from '@/utils/run';
-import { Run } from '@/run';
-import { Calculator } from '@/tools/Calculator';
-const conversationHistory = [];
-/**
- * Dump ALL metadata fields to understand what LangSmith uses
- * to detect parallel execution
- */
-async function testFullMetadata() {
-    console.log('Dumping FULL metadata to find parallel detection fields...\n');
-    const { contentParts, aggregateContent } = createContentAggregator();
-    // Collect ALL metadata from all events
-    const allMetadata = [];
-    const startTime = Date.now();
-    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. Just say "Hello from A" in one sentence.`,
-        },
-        {
-            agentId: 'agent_b',
-            provider: Providers.ANTHROPIC,
-            clientOptions: {
-                modelName: 'claude-haiku-4-5',
-                apiKey: process.env.ANTHROPIC_API_KEY,
-            },
-            tools: [new Calculator()],
-            instructions: `You are Agent B. Calculate 2+2 using the calculator tool.`,
-        },
-    ];
-    const edges = [];
-    const captureMetadata = (eventName, metadata) => {
-        if (metadata) {
-            allMetadata.push({
-                event: eventName,
-                timestamp: Date.now() - startTime,
-                metadata: { ...metadata },
-            });
-        }
-    };
-    const customHandlers = {
-        [GraphEvents.TOOL_END]: {
-            handle: (_event, _data, metadata) => {
-                captureMetadata('TOOL_END', metadata);
-            },
-        },
-        [GraphEvents.TOOL_START]: {
-            handle: (_event, _data, metadata) => {
-                captureMetadata('TOOL_START', metadata);
-            },
-        },
-        [GraphEvents.CHAT_MODEL_END]: {
-            handle: (_event, _data, metadata) => {
-                captureMetadata('CHAT_MODEL_END', metadata);
-            },
-        },
-        [GraphEvents.CHAT_MODEL_START]: {
-            handle: (_event, _data, metadata) => {
-                captureMetadata('CHAT_MODEL_START', metadata);
-            },
-        },
-        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: (event, data, metadata) => {
-                captureMetadata('ON_RUN_STEP', metadata);
-                const runStep = data;
-                console.log(`\n🔍 ON_RUN_STEP: agentId=${runStep.agentId}, groupId=${runStep.groupId}`);
-                aggregateContent({ event, data: runStep });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: (event, data, metadata) => {
-                captureMetadata('ON_RUN_STEP_DELTA', metadata);
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: (event, data, metadata) => {
-                captureMetadata('ON_MESSAGE_DELTA', metadata);
-                aggregateContent({ event, data: data });
-            },
-        },
-    };
-    const runConfig = {
-        runId: `full-metadata-${Date.now()}`,
-        graphConfig: {
-            type: 'multi-agent',
-            agents,
-            edges,
-        },
-        customHandlers,
-        returnContent: true,
-        skipCleanup: true,
-    };
-    try {
-        const run = await Run.create(runConfig);
-        const userMessage = `Hi, and calculate 2+2`;
-        conversationHistory.push(new HumanMessage(userMessage));
-        const config = {
-            configurable: {
-                thread_id: 'full-metadata-test-1',
-            },
-            streamMode: 'values',
-            version: 'v2',
-        };
-        await run.processStream({ messages: conversationHistory }, config);
-        // Analysis - find ALL unique metadata keys
-        console.log('\n\n========== ALL UNIQUE METADATA KEYS ==========\n');
-        const allKeys = new Set();
-        for (const entry of allMetadata) {
-            for (const key of Object.keys(entry.metadata)) {
-                allKeys.add(key);
-            }
-        }
-        console.log('Keys found:', [...allKeys].sort());
-        // Print first CHAT_MODEL_START for each agent with FULL metadata
-        console.log('\n\n========== FULL METADATA FROM CHAT_MODEL_START ==========\n');
-        const seenAgents = new Set();
-        for (const entry of allMetadata) {
-            if (entry.event === 'CHAT_MODEL_START') {
-                const node = entry.metadata.langgraph_node;
-                if (!seenAgents.has(node)) {
-                    seenAgents.add(node);
-                    console.log(`\n--- ${node} ---`);
-                    console.dir(entry.metadata, { depth: null });
-                }
-            }
-        }
-        // Look specifically at checkpoint_ns and __pregel_task_id
-        console.log('\n\n========== POTENTIAL PARALLEL INDICATORS ==========\n');
-        console.log('Comparing checkpoint_ns and __pregel_task_id across agents:\n');
-        const agentMetadataMap = new Map();
-        for (const entry of allMetadata) {
-            if (entry.event === 'CHAT_MODEL_START') {
-                const node = entry.metadata.langgraph_node;
-                if (!agentMetadataMap.has(node)) {
-                    agentMetadataMap.set(node, entry.metadata);
-                }
-            }
-        }
-        for (const [node, meta] of agentMetadataMap) {
-            console.log(`${node}:`);
-            console.log(`  langgraph_step: ${meta.langgraph_step}`);
-            console.log(`  langgraph_triggers: ${JSON.stringify(meta.langgraph_triggers)}`);
-            console.log(`  checkpoint_ns: ${meta.checkpoint_ns}`);
-            console.log(`  __pregel_task_id: ${meta.__pregel_task_id}`);
-            console.log(`  langgraph_path: ${JSON.stringify(meta.langgraph_path)}`);
-            console.log(`  langgraph_checkpoint_ns: ${meta.langgraph_checkpoint_ns}`);
-            console.log();
-        }
-        // Check langgraph_triggers specifically
-        console.log('\n========== LANGGRAPH_TRIGGERS ANALYSIS ==========\n');
-        for (const [node, meta] of agentMetadataMap) {
-            const triggers = meta.langgraph_triggers;
-            console.log(`${node}: triggers = ${JSON.stringify(triggers)}`);
-        }
-        console.log('\n\nFinal content parts:');
-        console.dir(contentParts, { depth: null });
-        await sleep(1000);
-    }
-    catch (error) {
-        console.error('Error:', error);
-    }
-}
-testFullMetadata();
-//# sourceMappingURL=parallel-full-metadata-test.js.map

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

@@ -1,256 +0,0 @@
-import { config } from 'dotenv';
-config();
-import { HumanMessage } from '@langchain/core/messages';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { Providers, GraphEvents } from '@/common';
-import { sleep } from '@/utils/run';
-import { Run } from '@/run';
-import { Calculator } from '@/tools/Calculator';
-import { Tool } from '@langchain/core/tools';
-const conversationHistory = [];
-// Create a simple "WordCount" tool for the second agent
-class WordCounter extends Tool {
-    static lc_name() {
-        return 'WordCounter';
-    }
-    name = 'word_counter';
-    description = 'Useful for counting the number of words, characters, and sentences in a given text. Input should be the text to analyze.';
-    async _call(input) {
-        const words = input.trim().split(/\s+/).filter(Boolean).length;
-        const characters = input.length;
-        const sentences = input.split(/[.!?]+/).filter(Boolean).length;
-        return JSON.stringify({ words, characters, sentences });
-    }
-}
-/**
- * Example of parallel multi-agent system with tools
- *
- * Graph structure:
- * START -> [math_agent, text_agent] -> END (parallel from start, both run simultaneously)
- *
- * Both agents have tools they can use. This tests how langgraph_step behaves
- * when parallel agents call tools.
- */
-async function testParallelWithTools() {
-    console.log('Testing Parallel From Start Multi-Agent System WITH TOOLS...\n');
-    // Set up content aggregator
-    const { contentParts, aggregateContent } = createContentAggregator();
-    // Track metadata for analysis
-    const metadataLog = [];
-    const startTime = Date.now();
-    // Define two agents with different tools
-    const agents = [
-        {
-            agentId: 'math_agent',
-            provider: Providers.ANTHROPIC,
-            clientOptions: {
-                modelName: 'claude-haiku-4-5',
-                apiKey: process.env.ANTHROPIC_API_KEY,
-            },
-            tools: [new Calculator()],
-            instructions: `You are a MATH SPECIALIST. When asked about numbers or calculations, ALWAYS use the calculator tool to perform the calculation. Start your response with "🧮 MATH:". Keep your response concise.`,
-        },
-        {
-            agentId: 'text_agent',
-            provider: Providers.ANTHROPIC,
-            clientOptions: {
-                modelName: 'claude-haiku-4-5',
-                apiKey: process.env.ANTHROPIC_API_KEY,
-            },
-            tools: [new WordCounter()],
-            instructions: `You are a TEXT ANALYST. When asked about text or content, ALWAYS use the word_counter tool to analyze the text. Start your response with "📝 TEXT:". Keep your response concise.`,
-        },
-    ];
-    // No edges - both agents run in parallel from start
-    const edges = [];
-    // Track active agents and timing
-    const activeAgents = new Set();
-    const agentTimings = {};
-    // Helper to log metadata
-    const logMetadata = (eventName, metadata) => {
-        if (metadata) {
-            const entry = {
-                event: eventName,
-                langgraph_step: metadata.langgraph_step,
-                langgraph_node: metadata.langgraph_node,
-                timestamp: Date.now() - startTime,
-            };
-            metadataLog.push(entry);
-            console.log(`📊 [${entry.timestamp}ms] ${eventName}: step=${entry.langgraph_step}, node=${entry.langgraph_node}`);
-        }
-    };
-    // Create custom handlers with metadata logging
-    const customHandlers = {
-        [GraphEvents.TOOL_END]: {
-            handle: (_event, data, metadata) => {
-                console.log('\n====== TOOL_END ======');
-                logMetadata('TOOL_END', metadata);
-                console.dir(data, { depth: null });
-            },
-        },
-        [GraphEvents.TOOL_START]: {
-            handle: (_event, data, metadata) => {
-                console.log('\n====== TOOL_START ======');
-                logMetadata('TOOL_START', metadata);
-            },
-        },
-        [GraphEvents.CHAT_MODEL_END]: {
-            handle: (_event, _data, metadata) => {
-                console.log('\n====== CHAT_MODEL_END ======');
-                logMetadata('CHAT_MODEL_END', metadata);
-                const nodeName = metadata?.langgraph_node;
-                if (nodeName) {
-                    const elapsed = Date.now() - startTime;
-                    agentTimings[nodeName] = agentTimings[nodeName] || {};
-                    agentTimings[nodeName].end = elapsed;
-                }
-            },
-        },
-        [GraphEvents.CHAT_MODEL_START]: {
-            handle: (_event, _data, metadata) => {
-                console.log('\n====== CHAT_MODEL_START ======');
-                logMetadata('CHAT_MODEL_START', metadata);
-                const nodeName = metadata?.langgraph_node;
-                if (nodeName) {
-                    const elapsed = Date.now() - startTime;
-                    agentTimings[nodeName] = agentTimings[nodeName] || {};
-                    if (!agentTimings[nodeName].start) {
-                        agentTimings[nodeName].start = elapsed;
-                    }
-                    activeAgents.add(nodeName);
-                }
-            },
-        },
-        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-            handle: (event, data, metadata) => {
-                console.log('\n====== ON_RUN_STEP_COMPLETED ======');
-                logMetadata('ON_RUN_STEP_COMPLETED', metadata);
-                aggregateContent({
-                    event,
-                    data: data,
-                });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: (event, data, metadata) => {
-                console.log('\n====== ON_RUN_STEP ======');
-                logMetadata('ON_RUN_STEP', metadata);
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: (event, data, metadata) => {
-                logMetadata('ON_RUN_STEP_DELTA', metadata);
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: (event, data, metadata) => {
-                logMetadata('ON_MESSAGE_DELTA', metadata);
-                aggregateContent({ event, data: data });
-            },
-        },
-    };
-    // Create multi-agent run configuration
-    const runConfig = {
-        runId: `parallel-tools-${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);
-        // User message that should trigger both agents to use their tools
-        const userMessage = `I have two tasks:
-1. Calculate 1234 + 5678 * 2
-2. Analyze this text: "The quick brown fox jumps over the lazy dog"
-
-Please help with both!`;
-        conversationHistory.push(new HumanMessage(userMessage));
-        console.log('Invoking parallel-from-start multi-agent graph WITH TOOLS...\n');
-        console.log('Both math_agent and text_agent should start simultaneously and use tools!\n');
-        const config = {
-            configurable: {
-                thread_id: 'parallel-tools-test-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);
-        }
-        // Analysis output
-        console.log('\n\n========== METADATA ANALYSIS ==========');
-        console.log('\nAll metadata entries by timestamp:');
-        console.table(metadataLog);
-        // Group by langgraph_step
-        const stepGroups = new Map();
-        for (const entry of metadataLog) {
-            if (!stepGroups.has(entry.langgraph_step)) {
-                stepGroups.set(entry.langgraph_step, []);
-            }
-            stepGroups.get(entry.langgraph_step).push(entry);
-        }
-        console.log('\n\nGrouped by langgraph_step:');
-        for (const [step, entries] of stepGroups) {
-            const nodes = [...new Set(entries.map((e) => e.langgraph_node))];
-            console.log(`\n  Step ${step}: ${nodes.length} unique nodes: ${nodes.join(', ')}`);
-            console.log(`    Events: ${entries.map((e) => e.event).join(', ')}`);
-        }
-        // Identify parallel groups (same step, different nodes)
-        console.log('\n\n========== PARALLEL DETECTION ANALYSIS ==========');
-        for (const [step, entries] of stepGroups) {
-            const uniqueNodes = [...new Set(entries.map((e) => e.langgraph_node))];
-            if (uniqueNodes.length > 1) {
-                console.log(`✅ Step ${step} has MULTIPLE agents: ${uniqueNodes.join(', ')} - PARALLEL!`);
-            }
-            else {
-                console.log(`   Step ${step} has single agent: ${uniqueNodes[0]}`);
-            }
-        }
-        console.log('\n\n========== TIMING SUMMARY ==========');
-        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 overlap for parallel confirmation
-        const agentList = Object.keys(agentTimings);
-        if (agentList.length >= 2) {
-            const [a1, a2] = agentList;
-            const t1 = agentTimings[a1];
-            const t2 = agentTimings[a2];
-            if (t1.start && t2.start && t1.end && t2.end) {
-                const overlap = Math.min(t1.end, t2.end) - Math.max(t1.start, t2.start);
-                if (overlap > 0) {
-                    console.log(`\n✅ PARALLEL EXECUTION CONFIRMED: ${overlap}ms overlap`);
-                }
-                else {
-                    console.log(`\n❌ SEQUENTIAL EXECUTION: no overlap`);
-                }
-            }
-        }
-        console.log('====================================\n');
-        console.log('Final content parts:', contentParts.length, 'parts');
-        console.dir(contentParts, { depth: null });
-        await sleep(2000);
-    }
-    catch (error) {
-        console.error('Error in parallel-with-tools test:', error);
-    }
-}
-// Run the test
-testParallelWithTools();
-//# sourceMappingURL=parallel-tools-test.js.map