@illuma-ai/agents 1.1.21 → 1.1.22

package/src/scripts/image.js

@@ -1,138 +0,0 @@
-// src/scripts/image.ts
-import { config } from 'dotenv';
-config();
-import { HumanMessage } from '@langchain/core/messages';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler, createMetadataAggregator, } from '@/events';
-// @ts-expect-error — example module not in current codebase
-import { fetchRandomImageTool } from '@/tools/example';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { getArgs } from '@/scripts/args';
-import { GraphEvents } from '@/common';
-import { Run } from '@/run';
-const conversationHistory = [];
-async function testCodeExecution() {
-    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: 'message-num-1',
-        graphConfig: {
-            type: 'standard',
-            llmConfig,
-            tools: [fetchRandomImageTool],
-            // tools: [fetchRandomImageURL],
-            instructions: 'You are a friendly AI assistant with internet capabilities. 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 = {
-        configurable: {
-            provider,
-            thread_id: 'conversation-num-1',
-        },
-        streamMode: 'values',
-        version: 'v2',
-    };
-    console.log('Fetch Random Image');
-    const userMessage1 = `Hi ${userName} here. Please get me 2 random images. Describe them after you receive them.`;
-    conversationHistory.push(new HumanMessage(userMessage1));
-    let inputs = {
-        messages: conversationHistory,
-    };
-    const finalContentParts1 = await run.processStream(inputs, config);
-    const finalMessages1 = run.getRunMessages();
-    if (finalMessages1) {
-        conversationHistory.push(...finalMessages1);
-    }
-    console.log('\n\n====================\n\n');
-    console.dir(contentParts, { depth: null });
-    console.log('Test 2: Follow up with another message');
-    const userMessage2 = `thanks, you're the best!`;
-    conversationHistory.push(new HumanMessage(userMessage2));
-    inputs = {
-        messages: conversationHistory,
-    };
-    const finalContentParts2 = await run.processStream(inputs, config, {
-        keepContent: true,
-    });
-    const finalMessages2 = run.getRunMessages();
-    if (finalMessages2) {
-        conversationHistory.push(...finalMessages2);
-    }
-    console.log('\n\n====================\n\n');
-    console.dir(contentParts, { depth: null });
-    const { handleLLMEnd, collected } = createMetadataAggregator();
-    const titleResult = await run.generateTitle({
-        provider,
-        inputText: userMessage2,
-        contentParts,
-        chainOptions: {
-            callbacks: [
-                {
-                    handleLLMEnd,
-                },
-            ],
-        },
-    });
-    console.log('Generated Title:', titleResult);
-    console.log('Collected metadata:', collected);
-}
-process.on('unhandledRejection', (reason, promise) => {
-    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-    console.log('Conversation history:');
-    console.dir(conversationHistory, { depth: null });
-    process.exit(1);
-});
-testCodeExecution().catch((err) => {
-    console.error(err);
-    console.log('Conversation history:');
-    console.dir(conversationHistory, { depth: null });
-    process.exit(1);
-});
-//# sourceMappingURL=image.js.map

package/src/scripts/memory.js

@@ -1,83 +0,0 @@
-// src/scripts/cli.ts
-import { config } from 'dotenv';
-config();
-import z from 'zod';
-import { tool } from '@langchain/core/tools';
-import { HumanMessage } from '@langchain/core/messages';
-import { getArgs } from '@/scripts/args';
-import { Providers } from '@/common';
-import { Run } from '@/run';
-const conversationHistory = [];
-const memoryKv = {};
-const setMemory = tool(async ({ key, value }) => {
-    if (!/^[a-z_]+$/.test(key)) {
-        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'),
-    }),
-});
-async function testStandardStreaming() {
-    const { userName, provider } = await getArgs();
-    const run = await Run.create({
-        runId: 'memory-run',
-        graphConfig: {
-            type: 'standard',
-            llmConfig: {
-                provider: Providers.OPENAI,
-                model: 'gpt-4o-mini',
-                temperature: 0.5,
-                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`.',
-            toolEnd: true,
-        },
-        returnContent: true,
-        skipCleanup: true,
-    });
-    const config = {
-        configurable: {
-            provider,
-            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,
-    };
-    await run.processStream(inputs, config);
-    const finalMessages = run.getRunMessages();
-    if (finalMessages) {
-        conversationHistory.push(...finalMessages);
-        console.dir(conversationHistory, { depth: null });
-    }
-    console.log('\n\n====================\n\n');
-    console.dir(memoryKv, { 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=memory.js.map

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

@@ -1,271 +0,0 @@
-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: 'sequence',
-            // 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-conditional.js

@@ -1,185 +0,0 @@
-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