@illuma-ai/agents 1.0.89 → 1.0.93

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

@@ -0,0 +1,143 @@
+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: 'direct' },
+    ];
+    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/sequential-full-metadata-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sequential-full-metadata-test.js","sourceRoot":"","sources":["sequential-full-metadata-test.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,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;;GAEG;AACH,KAAK,UAAU,sBAAsB;IACnC,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;IAEnE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAErE,MAAM,WAAW,GAIZ,EAAE,CAAC;IACR,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,uCAAuC;IACvC,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,2DAA2D;SAC1E;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,2DAA2D;SAC1E;KACF,CAAC;IAEF,oDAAoD;IACpD,MAAM,KAAK,GAAkB;QAC3B,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE;KACvD,CAAC;IAEF,MAAM,eAAe,GAAG,CACtB,SAAiB,EACjB,QAAkC,EAClC,EAAE;QACF,IAAI,QAAQ,EAAE,CAAC;YACb,WAAW,CAAC,IAAI,CAAC;gBACf,KAAK,EAAE,SAAS;gBAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBACjC,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE;aAC1B,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE;YAC5B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,eAAe,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;YAC9C,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,eAAe,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;YAChD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,eAAe,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;gBACzC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACvB,QAAkC,EAC5B,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,EACvB,QAAkC,EAC5B,EAAE;gBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,uBAAuB,IAAI,CAAC,GAAG,EAAE,EAAE;QAC1C,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,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAExC,MAAM,WAAW,GAAG,IAAI,CAAC;QACzB,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;QAExD,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,4BAA4B;aACxC;YACD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,IAAa;SACvB,CAAC;QAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,MAAM,CAAC,CAAC;QAEnE,qDAAqD;QACrD,OAAO,CAAC,GAAG,CACT,sEAAsE,CACvE,CAAC;QACF,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,IAAI,KAAK,CAAC,KAAK,KAAK,kBAAkB,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,cAAwB,CAAC;gBACrD,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,QAAQ,KAAK,CAAC,SAAS,SAAS,CAAC,CAAC;gBAC3D,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;QACtD,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CACT,IAAI,KAAK,CAAC,SAAS,OAAO,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,QAAQ,CAAC,cAAc,EAAE,CAC1E,CAAC;QACJ,CAAC;QAED,iBAAiB;QACjB,OAAO,CAAC,GAAG,CACT,gEAAgE,CACjE,CAAC;QAEF,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAmC,CAAC;QACpE,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,IAAI,KAAK,CAAC,KAAK,KAAK,kBAAkB,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,cAAwB,CAAC;gBACrD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAChC,gBAAgB,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,gBAAgB,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;YACxD,OAAO,CAAC,GAAG,CACT,yBAAyB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CACnE,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;YAC5D,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,EAAE,CAAC;QAChB,CAAC;QAED,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAED,sBAAsB,EAAE,CAAC"}

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

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

package/src/scripts/simple.js

@@ -0,0 +1,174 @@
+// 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/simple.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple.js","sourceRoot":"","sources":["simple.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,MAAM,EAAE,CAAC;AACT,OAAO,EACL,YAAY,GAGb,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EACL,cAAc,EACd,eAAe,EACf,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,WAAW,EAAE,SAAS,EAAe,MAAM,UAAU,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA8B,EAAE,CAAC;AAClD,IAAI,cAAc,GAAoB,EAAE,CAAC;AAEzC,KAAK,UAAU,qBAAqB;IAClC,MAAM,EACJ,QAAQ,EACR,QAAQ,EACR,QAAQ,EAAE,SAAS,EACnB,WAAW,GACZ,GAAG,MAAM,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAyC,CAAC;IAC1D,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,CAAC,cAAc,CAAC;QACjE,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;gBACnD,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,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,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,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,kBAAkB,CAAC,EAAE;YAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;gBAChD,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA6B,EAAE,CAAC,CAAC;YACnE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,sCAAsC;YACxC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IAC1C,IACE,eAAe,IAAI,SAAS;QAC3B,SAAmC,CAAC,aAAa,IAAI,IAAI,EAC1D,CAAC;QACD,MAAM,YAAY,GAAG,SAAkC,CAAC;QACxD,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;YAC/B,YAAY,CAAC,aAAa,CAAC,KAAK,GAAG,CACjC,GAA2B,EAC3B,IAAkB,EAClB,EAAE;gBACF,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;gBAC9B,OAAO,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAC1B,CAAC,CAAC;QACJ,CAAC;IACH,CAAC;IACD,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;IAEpC,IAAI,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE,CAAC;QACpC,SAAsC,CAAC,aAAa,GAAG;YACtD,cAAc,EAAE;gBACd,gBAAgB,EACd,mFAAmF;aACtF;SACF,CAAC;IACJ,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,MAAM,EAAE;QACf,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,aAAa;YACb,6BAA6B;YAC7B,YAAY,EACV,yEAAyE;YAC3E,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,GAAG;SAC/F;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,KAAK,EAAE,MAAM,EAAE;QACf,YAAY,EAAE;YACZ,OAAO,EAAE,UAAU;YACnB,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAE3C,MAAM,WAAW,GAAG,IAAI,CAAC;IAEzB,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,IAAI,aAAa,EAAE,CAAC;QAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,mDAAmD;IACnD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,wBAAwB,EAAE,CAAC;IAC/D,MAAM,YAAY,GAAsB;QACtC,QAAQ;QACR,SAAS,EAAE,WAAW;QACtB,YAAY;QACZ,uCAAuC;QACvC,YAAY,EAAE;YACZ,YAAY,EAAE;gBACZ,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,oBAAoB;aAChC;YACD,SAAS,EAAE;gBACT;oBACE,YAAY;iBACb;aACF;SACF;KACF,CAAC;IACF,IAAI,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE,CAAC;QACrC,YAAY,CAAC,aAAa,GAAG;YAC3B,KAAK,EAAE,kBAAkB;SAC1B,CAAC;IACJ,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IAC1D,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,cAAc,CAAC,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,SAAS,CAAC,CAAC;IAC1D,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;AACpB,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;IACtC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/simple.ts

@@ -141,6 +141,7 @@ async function testStandardStreaming(): Promise<void> {
       additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 
@@ -192,7 +193,7 @@ async function testStandardStreaming(): Promise<void> {
   };
   if (provider === Providers.ANTHROPIC) {
     titleOptions.clientOptions = {
-      model: 'claude-3-5-haiku-latest',
+      model: 'claude-haiku-4-5',
     };
   }
   const titleResult = await run.generateTitle(titleOptions);

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

@@ -0,0 +1,152 @@
+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/single-agent-metadata-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"single-agent-metadata-test.js","sourceRoot":"","sources":["single-agent-metadata-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAmB,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;;;GAGG;AACH,KAAK,UAAU,eAAe;IAC5B,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;IAE/D,4BAA4B;IAC5B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,OAAO,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAE9E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,yDAAyD;IACzD,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE;YAC5B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;gBACvD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACvC,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,IAAI,CAAC,CAAC;YAC/C,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,MAAc,EACd,KAAwB,EACxB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;gBACzD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACvC,OAAO,CAAC,GAAG,CAAC,kBAAkB,OAAO,IAAI,CAAC,CAAC;YAC7C,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;gBACrD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC3C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;gBAChD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;KACF,CAAC;IAEF,0EAA0E;IAC1E,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,EAAE;QACnC,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE;gBACT,QAAQ,EAAE,SAAS,CAAC,SAAS;gBAC7B,SAAS,EAAE,kBAAkB;gBAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;aACtC;YACD,YAAY,EAAE,4EAA4E;SAC3F;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,iCAAiC;QACjC,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAI,GAAW,CAAC,KAAK,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC5D,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;gBAC7D,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;oBACrD,OAAO,CAAC,GAAG,CAAC,YAAY,OAAO,EAAE,CAAC,CAAC;oBACnC,OAAO,CAAC,GAAG,CACT,UAAU,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,EAAE,CACrF,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAEnC,MAAM,WAAW,GAAG,sEAAsE,CAAC;QAC3F,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;QAExD,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;QAEhD,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,SAAS,EAAE,6BAA6B;aACzC;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,mCAAmC,CAAC,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClE,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,uDAAuD,CAAC,CAAC;QACrE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;QAEtD,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAED,eAAe;AACf,eAAe,EAAE,CAAC"}

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

@@ -20,8 +20,7 @@ async function testSingleAgent() {
   console.log('Testing Single Agent with Metadata Logging...\n');
 
   // Set up content aggregator
-  const { contentParts, aggregateContent, contentMetadataMap } =
-    createContentAggregator();
+  const { contentParts, aggregateContent, stepMap } = createContentAggregator();
 
   const startTime = Date.now();
 
@@ -128,6 +127,7 @@ async function testSingleAgent() {
     },
     customHandlers,
     returnContent: true,
+    skipCleanup: true,
   };
 
   try {
@@ -182,10 +182,8 @@ async function testSingleAgent() {
     console.log('Final content parts:', contentParts.length, 'parts');
     console.log('\n=== Content Parts (clean, no metadata) ===');
     console.dir(contentParts, { depth: null });
-    console.log(
-      '\n=== Content Metadata Map (should be empty for single-agent) ==='
-    );
-    console.dir(Object.fromEntries(contentMetadataMap), { 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);