@illuma-ai/agents 1.1.21 → 1.1.23

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

@@ -1,143 +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';
-const conversationHistory = [];
-/**
- * Dump ALL metadata for SEQUENTIAL execution to compare with parallel
- */
-async function testSequentialMetadata() {
-    console.log('Dumping FULL metadata for SEQUENTIAL execution...\n');
-    const { contentParts, aggregateContent } = createContentAggregator();
-    const allMetadata = [];
-    const startTime = Date.now();
-    // Sequential chain: agent_a -> agent_b
-    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,
-            },
-            instructions: `You are Agent B. Just say "Hello from B" in one sentence.`,
-        },
-    ];
-    // Sequential edge: A -> B (using edgeType not type)
-    const edges = [
-        { from: 'agent_a', to: 'agent_b', edgeType: 'sequence' },
-    ];
-    const captureMetadata = (eventName, metadata) => {
-        if (metadata) {
-            allMetadata.push({
-                event: eventName,
-                timestamp: Date.now() - startTime,
-                metadata: { ...metadata },
-            });
-        }
-    };
-    const customHandlers = {
-        [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);
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: (event, data, metadata) => {
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: (event, data, metadata) => {
-                aggregateContent({ event, data: data });
-            },
-        },
-    };
-    const runConfig = {
-        runId: `sequential-metadata-${Date.now()}`,
-        graphConfig: {
-            type: 'multi-agent',
-            agents,
-            edges,
-        },
-        customHandlers,
-        returnContent: true,
-        skipCleanup: true,
-    };
-    try {
-        const run = await Run.create(runConfig);
-        const userMessage = `Hi`;
-        conversationHistory.push(new HumanMessage(userMessage));
-        const config = {
-            configurable: {
-                thread_id: 'sequential-metadata-test-1',
-            },
-            streamMode: 'values',
-            version: 'v2',
-        };
-        await run.processStream({ messages: conversationHistory }, config);
-        // Print ALL CHAT_MODEL_START metadata (don't dedupe)
-        console.log('\n\n========== ALL CHAT_MODEL_START EVENTS (SEQUENTIAL) ==========\n');
-        for (const entry of allMetadata) {
-            if (entry.event === 'CHAT_MODEL_START') {
-                const node = entry.metadata.langgraph_node;
-                console.log(`\n--- ${node} (at ${entry.timestamp}ms) ---`);
-                console.dir(entry.metadata, { depth: null });
-            }
-        }
-        console.log('\n\n========== ALL EVENTS ==========\n');
-        for (const entry of allMetadata) {
-            console.log(`[${entry.timestamp}ms] ${entry.event}: ${entry.metadata.langgraph_node}`);
-        }
-        // Key comparison
-        console.log('\n\n========== KEY FIELDS COMPARISON (SEQUENTIAL) ==========\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();
-        }
-        await sleep(1000);
-    }
-    catch (error) {
-        console.error('Error:', error);
-    }
-}
-testSequentialMetadata();
-//# sourceMappingURL=sequential-full-metadata-test.js.map

package/src/scripts/simple.js

@@ -1,174 +0,0 @@
-// src/scripts/cli.ts
-import { config } from 'dotenv';
-import { v4 as uuidv4 } from 'uuid';
-config();
-import { HumanMessage, } from '@langchain/core/messages';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler, createMetadataAggregator, } from '@/events';
-import { GraphEvents, Providers } from '@/common';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { getArgs } from '@/scripts/args';
-import { sleep } from '@/utils/run';
-import { Run } from '@/run';
-const conversationHistory = [];
-let _contentParts = [];
-let collectedUsage = [];
-async function testStandardStreaming() {
-    const { userName, location, provider: _provider, currentDate, } = await getArgs();
-    const { contentParts, aggregateContent } = createContentAggregator();
-    _contentParts = contentParts;
-    const customHandlers = {
-        [GraphEvents.TOOL_END]: new ToolEndHandler(),
-        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
-        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-            handle: (event, data) => {
-                console.log('====== ON_RUN_STEP_COMPLETED ======');
-                console.dir(data, { depth: null });
-                aggregateContent({
-                    event,
-                    data: data,
-                });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: (event, data) => {
-                console.log('====== ON_RUN_STEP ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: (event, data) => {
-                console.log('====== ON_RUN_STEP_DELTA ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: (event, data) => {
-                console.log('====== ON_MESSAGE_DELTA ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_REASONING_DELTA]: {
-            handle: (event, data) => {
-                console.log('====== ON_REASONING_DELTA ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.TOOL_START]: {
-            handle: (_event, data, metadata) => {
-                console.log('====== TOOL_START ======');
-                // console.dir(data, { depth: null });
-            },
-        },
-    };
-    const llmConfig = getLLMConfig(_provider);
-    if ('configuration' in llmConfig &&
-        llmConfig.configuration != null) {
-        const openAIConfig = llmConfig;
-        if (openAIConfig.configuration) {
-            openAIConfig.configuration.fetch = (url, init) => {
-                console.log('Fetching:', url);
-                return fetch(url, init);
-            };
-        }
-    }
-    const provider = llmConfig.provider;
-    if (provider === Providers.ANTHROPIC) {
-        llmConfig.clientOptions = {
-            defaultHeaders: {
-                'anthropic-beta': 'token-efficient-tools-2025-02-19,output-128k-2025-02-19,prompt-caching-2024-07-31',
-            },
-        };
-    }
-    const run = await Run.create({
-        runId: uuidv4(),
-        graphConfig: {
-            type: 'standard',
-            llmConfig,
-            // tools: [],
-            // reasoningKey: 'reasoning',
-            instructions: 'You are a friendly AI assistant. Always address the user by their name.',
-            additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
-        },
-        returnContent: true,
-        skipCleanup: true,
-        customHandlers,
-    });
-    const config = {
-        runId: uuidv4(),
-        configurable: {
-            user_id: 'user-123',
-            thread_id: 'conversation-num-1',
-        },
-        streamMode: 'values',
-        version: 'v2',
-    };
-    console.log('Test 1: Simple message test');
-    const userMessage = `hi`;
-    conversationHistory.push(new HumanMessage(userMessage));
-    const inputs = {
-        messages: conversationHistory,
-    };
-    const finalContentParts = await run.processStream(inputs, config);
-    const finalMessages = run.getRunMessages();
-    if (finalMessages) {
-        conversationHistory.push(...finalMessages);
-        console.dir(conversationHistory, { depth: null });
-    }
-    // console.dir(finalContentParts, { depth: null });
-    console.log('\n\n====================\n\n');
-    console.dir(contentParts, { depth: null });
-    const { handleLLMEnd, collected } = createMetadataAggregator();
-    const titleOptions = {
-        provider,
-        inputText: userMessage,
-        contentParts,
-        // titleMethod: TitleMethod.STRUCTURED,
-        chainOptions: {
-            configurable: {
-                user_id: 'user-123',
-                thread_id: 'conversation-num-1',
-            },
-            callbacks: [
-                {
-                    handleLLMEnd,
-                },
-            ],
-        },
-    };
-    if (provider === Providers.ANTHROPIC) {
-        titleOptions.clientOptions = {
-            model: 'claude-haiku-4-5',
-        };
-    }
-    const titleResult = await run.generateTitle(titleOptions);
-    console.log('Collected usage metadata:', collectedUsage);
-    console.log('Generated Title:', titleResult);
-    console.log('Collected title usage metadata:', collected);
-    await sleep(5000);
-}
-process.on('unhandledRejection', (reason, promise) => {
-    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-    console.log('Conversation history:');
-    console.dir(conversationHistory, { depth: null });
-    console.log('Content parts:');
-    console.dir(_contentParts, { depth: null });
-    process.exit(1);
-});
-process.on('uncaughtException', (err) => {
-    console.error('Uncaught Exception:', err);
-});
-testStandardStreaming().catch((err) => {
-    console.error(err);
-    console.log('Conversation history:');
-    console.dir(conversationHistory, { depth: null });
-    console.log('Content parts:');
-    console.dir(_contentParts, { depth: null });
-    process.exit(1);
-});
-//# sourceMappingURL=simple.js.map

package/src/scripts/single-agent-metadata-test.js

@@ -1,152 +0,0 @@
-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 = [];
-/**
- * Single agent test with extensive metadata logging
- * Compare with multi-agent-parallel-start.ts to see metadata differences
- */
-async function testSingleAgent() {
-    console.log('Testing Single Agent with Metadata Logging...\n');
-    // Set up content aggregator
-    const { contentParts, aggregateContent, stepMap } = createContentAggregator();
-    const startTime = Date.now();
-    // 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_COMPLETED]: {
-            handle: (event, data, metadata) => {
-                console.log('\n====== ON_RUN_STEP_COMPLETED ======');
-                console.log('DATA:');
-                console.dir(data, { depth: null });
-                console.log('METADATA:');
-                console.dir(metadata, { depth: null });
-                aggregateContent({
-                    event,
-                    data: data,
-                });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: (event, data, metadata) => {
-                console.log('\n====== ON_RUN_STEP ======');
-                console.log('DATA:');
-                console.dir(data, { depth: null });
-                console.log('METADATA:');
-                console.dir(metadata, { depth: null });
-                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) => {
-                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 single-agent run configuration (standard graph, not multi-agent)
-    const runConfig = {
-        runId: `single-agent-${Date.now()}`,
-        graphConfig: {
-            type: 'standard',
-            llmConfig: {
-                provider: Providers.ANTHROPIC,
-                modelName: 'claude-haiku-4-5',
-                apiKey: process.env.ANTHROPIC_API_KEY,
-            },
-            instructions: `You are a helpful AI assistant. Keep your response concise (50-100 words).`,
-        },
-        customHandlers,
-        returnContent: true,
-        skipCleanup: true,
-    };
-    try {
-        // Create and execute the run
-        const run = await Run.create(runConfig);
-        // Debug: Log the graph structure
-        console.log('=== DEBUG: Graph Structure ===');
-        const graph = run.Graph;
-        console.log('Graph exists:', !!graph);
-        if (graph) {
-            console.log('Graph type:', graph.constructor.name);
-            console.log('AgentContexts exists:', !!graph.agentContexts);
-            if (graph.agentContexts) {
-                console.log('AgentContexts size:', graph.agentContexts.size);
-                for (const [agentId, context] of graph.agentContexts) {
-                    console.log(`\nAgent: ${agentId}`);
-                    console.log(`Tools: ${context.tools?.map((t) => t.name || 'unnamed').join(', ') || 'none'}`);
-                }
-            }
-        }
-        console.log('=== END DEBUG ===\n');
-        const userMessage = `What are the best approaches to learning a new programming language?`;
-        conversationHistory.push(new HumanMessage(userMessage));
-        console.log('Invoking single-agent graph...\n');
-        const config = {
-            configurable: {
-                thread_id: 'single-agent-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========== SUMMARY ==========');
-        console.log('Final content parts:', contentParts.length, 'parts');
-        console.log('\n=== Content Parts (clean, no metadata) ===');
-        console.dir(contentParts, { depth: null });
-        console.log('\n=== Step Map (should be empty for single-agent) ===');
-        console.dir(Object.fromEntries(stepMap), { depth: null });
-        console.log('====================================\n');
-        await sleep(3000);
-    }
-    catch (error) {
-        console.error('Error in single-agent test:', error);
-    }
-}
-// Run the test
-testSingleAgent();
-//# sourceMappingURL=single-agent-metadata-test.js.map

package/src/scripts/stream.js

@@ -1,113 +0,0 @@
-/* eslint-disable no-console */
-// src/scripts/cli.ts
-import { config } from 'dotenv';
-config();
-import { HumanMessage } from '@langchain/core/messages';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler } from '@/events';
-import { getArgs } from '@/scripts/args';
-import { Run } from '@/run';
-import { GraphEvents } from '@/common';
-import { getLLMConfig } from '@/utils/llmConfig';
-const conversationHistory = [];
-async function testStandardStreaming() {
-    const { userName, location, provider, currentDate } = await getArgs();
-    const { contentParts, aggregateContent } = createContentAggregator();
-    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) => {
-                console.log('====== ON_RUN_STEP_COMPLETED ======');
-                // console.dir(data, { depth: null });
-                aggregateContent({
-                    event,
-                    data: data,
-                });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: (event, data) => {
-                console.log('====== ON_RUN_STEP ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: (event, data) => {
-                console.log('====== ON_RUN_STEP_DELTA ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: (event, data) => {
-                console.log('====== ON_MESSAGE_DELTA ======');
-                console.dir(data, { depth: null });
-                aggregateContent({ event, data: data });
-            },
-        },
-        [GraphEvents.TOOL_START]: {
-            handle: (_event, data, metadata) => {
-                console.log('====== TOOL_START ======');
-                // console.dir(data, { depth: null });
-            },
-        },
-    };
-    const llmConfig = getLLMConfig(provider);
-    const run = await Run.create({
-        runId: 'test-run-id',
-        graphConfig: {
-            type: 'standard',
-            llmConfig,
-            tools: [],
-            instructions: 'You are a friendly AI assistant. Always address the user by their name.',
-            additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
-            streamBuffer: 3000,
-        },
-        returnContent: true,
-        skipCleanup: true,
-        customHandlers,
-    });
-    const config = {
-        configurable: {
-            provider,
-            thread_id: 'conversation-num-1',
-        },
-        streamMode: 'values',
-        version: 'v2',
-    };
-    console.log('Test 1: Weather query (content parts test)');
-    const userMessage = `
-  Make a search for the weather in ${location} today, which is ${currentDate}.
-  Before making the search, please let me know what you're about to do, then immediately start searching without hesitation.
-  Make sure to always refer to me by name, which is ${userName}.
-  After giving me a thorough summary, tell me a joke about the weather forecast we went over.
-  `;
-    conversationHistory.push(new HumanMessage(userMessage));
-    const inputs = {
-        messages: conversationHistory,
-    };
-    const finalContentParts = await run.processStream(inputs, config);
-    const finalMessages = run.getRunMessages();
-    if (finalMessages) {
-        conversationHistory.push(...finalMessages);
-        console.dir(conversationHistory, { depth: null });
-    }
-    // console.dir(finalContentParts, { depth: null });
-    console.log('\n\n====================\n\n');
-    // console.dir(contentParts, { depth: null });
-}
-process.on('unhandledRejection', (reason, promise) => {
-    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-    console.log('Conversation history:');
-    process.exit(1);
-});
-testStandardStreaming().catch((err) => {
-    console.error(err);
-    console.log('Conversation history:');
-    console.dir(conversationHistory, { depth: null });
-    process.exit(1);
-});
-//# sourceMappingURL=stream.js.map