@illuma-ai/agents 1.0.90 → 1.0.94

package/src/scripts/code_exec.js

@@ -0,0 +1,171 @@
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler, createMetadataAggregator, } from '@/events';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { GraphEvents } from '@/common';
+import { Run } from '@/run';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+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: [createCodeExecutionTool()],
+            instructions: 'You are a friendly AI assistant with coding 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',
+        // recursionLimit: 3,
+    };
+    console.log('Test 1: Sorting Algorithm Comparison');
+    const userMessage1 = `
+  Hi ${userName} here. I need a Python script that compares different sorting algorithms. Can you write a script that:
+  1. Implements three sorting algorithms: Bubble Sort, Insertion Sort, and Merge Sort
+  2. Generates three random lists of integers:
+     - A small list (20 elements)
+     - A medium list (100 elements)
+     - A large list (1000 elements)
+  3. Applies each sorting algorithm to each list and measures the execution time
+  4. Prints a comparison table showing the time taken by each algorithm for each list size
+  5. Determines and announces the fastest algorithm for each list size
+  Please write the script and then execute it to show the results.
+  `;
+    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: Text File Analysis and Processing');
+    const userMessage2 = `
+  Great job on the sorting algorithms! Now, let's solve a popular LeetCode problem using the Merge Sort algorithm we implemented. The problem is "Merge Intervals". Here's what I need:
+
+  1. Implement a solution to the Merge Intervals problem:
+     - Given an array of intervals where intervals[i] = [starti, endi], merge all overlapping intervals.
+     - Return an array of the non-overlapping intervals that cover all the intervals in the input.
+
+  2. Use the Merge Sort algorithm as part of the solution to sort the intervals based on their start times.
+
+  3. Implement a function to generate a random list of intervals for testing.
+
+  4. Create test cases:
+     - A small case with 5 intervals
+     - A medium case with 20 intervals
+     - A large case with 100 intervals
+
+  5. Apply your solution to each test case and print:
+     - The original intervals
+     - The merged intervals
+     - The time taken to solve each case
+
+  Please write the script and execute it to demonstrate the results for all three test cases.
+  `;
+    conversationHistory.push(new HumanMessage(userMessage2));
+    inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts2 = await run.processStream(inputs, config);
+    const finalMessages2 = run.getRunMessages();
+    if (finalMessages2) {
+        conversationHistory.push(...finalMessages2);
+    }
+    console.log('\n\n====================\n\n');
+    console.dir(finalContentParts2, { 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);
+});
+process.on('uncaughtException', (err) => {
+    console.error('Uncaught Exception:', err);
+});
+testCodeExecution().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=code_exec.js.map

package/src/scripts/code_exec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec.js","sourceRoot":"","sources":["code_exec.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAA0B,MAAM,0BAA0B,CAAC;AAGhF,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EACL,cAAc,EACd,eAAe,EACf,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C,KAAK,UAAU,iBAAiB;IAC9B,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACtE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,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,kDAAkD;gBAClD,sCAAsC;gBACtC,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,iDAAiD;gBACjD,sCAAsC;gBACtC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,YAAY,EACV,kGAAkG;YACpG,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,GAIR;QACF,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;QACtB,qBAAqB;KACtB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IAEpD,MAAM,YAAY,GAAG;OAChB,QAAQ;;;;;;;;;;GAUZ,CAAC;IAEF,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,IAAI,MAAM,GAAG;QACX,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,kBAAkB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnE,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IAEzD,MAAM,YAAY,GAAG;;;;;;;;;;;;;;;;;;;;;;GAsBpB,CAAC;IAEF,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,MAAM,GAAG;QACP,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,kBAAkB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnE,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAEjD,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,wBAAwB,EAAE,CAAC;IAC/D,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC;QAC1C,QAAQ;QACR,SAAS,EAAE,YAAY;QACvB,YAAY;QACZ,YAAY,EAAE;YACZ,SAAS,EAAE;gBACT;oBACE,YAAY;iBACb;aACF;SACF;KACF,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;AAChD,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,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,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAChC,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/code_exec.ts

@@ -93,6 +93,7 @@ async function testCodeExecution(): Promise<void> {
       additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/code_exec_files.js

@@ -0,0 +1,180 @@
+// src/scripts/code_exec_files.ts
+/**
+ * Tests automatic session tracking for code execution file persistence.
+ * Files created in one execution are automatically available in subsequent executions
+ * without the LLM needing to track or pass session_id.
+ *
+ * Run with: npm run code_exec_files
+ */
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler, createMetadataAggregator, } from '@/events';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Constants, GraphEvents } from '@/common';
+import { Run } from '@/run';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+const conversationHistory = [];
+/**
+ * Prints session context from the graph for debugging
+ */
+function printSessionContext(run) {
+    const graph = run.Graph;
+    if (!graph) {
+        console.log('[Session] No graph available');
+        return;
+    }
+    const session = graph.sessions.get(Constants.EXECUTE_CODE);
+    if (!session) {
+        console.log('[Session] No session context stored yet');
+        return;
+    }
+    console.log('[Session] Current session context:');
+    console.log(`  - session_id: ${session.session_id}`);
+    console.log(`  - files: ${JSON.stringify(session.files, null, 2)}`);
+    console.log(`  - lastUpdated: ${new Date(session.lastUpdated).toISOString()}`);
+}
+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: [createCodeExecutionTool()],
+            instructions: 'You are a friendly AI assistant with coding capabilities. Always address the user by their name.',
+            additional_instructions: `The user's name is ${userName} and they are located in ${location}. The current date is ${currentDate}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const streamConfig = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('\n========== Test 1: Create Project Plan ==========\n');
+    console.log('Creating initial file - this establishes the session context.\n');
+    const userMessage1 = `
+  Hi ${userName} here. We are testing your file capabilities.
+  
+  1. Create a text file named "project_plan.txt" that contains: "This is a project plan for a new software development project."
+  
+  Please generate this file so I can review it.
+  `;
+    conversationHistory.push(new HumanMessage(userMessage1));
+    let inputs = {
+        messages: conversationHistory,
+    };
+    await run.processStream(inputs, streamConfig);
+    const finalMessages1 = run.getRunMessages();
+    if (finalMessages1) {
+        conversationHistory.push(...finalMessages1);
+    }
+    console.log('\n\n========== Session Context After Test 1 ==========\n');
+    printSessionContext(run);
+    console.dir(contentParts, { depth: null });
+    console.log('\n========== Test 2: Edit Project Plan ==========\n');
+    console.log('Editing the file from Test 1 - session_id is automatically injected.\n');
+    const userMessage2 = `
+  Thanks for creating the project plan. Now I'd like you to edit the same plan to:
+  
+  1. Read the existing project_plan.txt file
+  2. Add a new section called "Technology Stack" that contains: "The technology stack for this project includes the following technologies" and nothing more.
+  3. Save this as a new file called "project_plan_v2.txt" (remember files are read-only)
+  4. Print the contents of both files to verify
+`;
+    conversationHistory.push(new HumanMessage(userMessage2));
+    inputs = {
+        messages: conversationHistory,
+    };
+    await run.processStream(inputs, streamConfig);
+    const finalMessages2 = run.getRunMessages();
+    if (finalMessages2) {
+        conversationHistory.push(...finalMessages2);
+    }
+    console.log('\n\n========== Session Context After Test 2 ==========\n');
+    printSessionContext(run);
+    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);
+});
+process.on('uncaughtException', (err) => {
+    console.error('Uncaught Exception:', err);
+});
+testCodeExecution().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=code_exec_files.js.map

package/src/scripts/code_exec_files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec_files.js","sourceRoot":"","sources":["code_exec_files.ts"],"names":[],"mappings":"AAAA,iCAAiC;AACjC;;;;;;GAMG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EACL,cAAc,EACd,eAAe,EACf,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;GAEG;AACH,SAAS,mBAAmB,CAAC,GAAkB;IAC7C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAE5C,CAAC;IAEd,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;QACvD,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACrD,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IACpE,OAAO,CAAC,GAAG,CACT,oBAAoB,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,EAAE,CAClE,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,iBAAiB;IAC9B,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACtE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,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,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,SAAmC,EAC7B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,YAAY,EACV,kGAAkG;YACpG,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,yBAAyB,WAAW,GAAG;SACnI;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,YAAY,GAId;QACF,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;IACrE,OAAO,CAAC,GAAG,CACT,iEAAiE,CAClE,CAAC;IAEF,MAAM,YAAY,GAAG;OAChB,QAAQ;;;;;GAKZ,CAAC;IAEF,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,IAAI,MAAM,GAAG;QACX,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC9C,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IACxE,mBAAmB,CAAC,GAAG,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;IACnE,OAAO,CAAC,GAAG,CACT,wEAAwE,CACzE,CAAC;IAEF,MAAM,YAAY,GAAG;;;;;;;CAOtB,CAAC;IAEA,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,MAAM,GAAG;QACP,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC9C,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IACxE,mBAAmB,CAAC,GAAG,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,wBAAwB,EAAE,CAAC;IAC/D,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC;QAC1C,QAAQ;QACR,SAAS,EAAE,YAAY;QACvB,YAAY;QACZ,YAAY,EAAE;YACZ,SAAS,EAAE;gBACT;oBACE,YAAY;iBACb;aACF;SACF;KACF,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;AAChD,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,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,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAChC,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/code_exec_files.ts

@@ -127,6 +127,7 @@ async function testCodeExecution(): Promise<void> {
       additional_instructions: `The user's name is ${userName} and they are located in ${location}. The current date is ${currentDate}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/code_exec_multi_session.js

@@ -0,0 +1,185 @@
+// src/scripts/code_exec_multi_session.ts
+/**
+ * Tests multi-session file tracking for code execution.
+ * Verifies that:
+ * 1. Files from multiple executions are accumulated
+ * 2. Each file tracks its source session_id
+ * 3. Edited/recreated files replace older versions (latest preferred)
+ *
+ * Run with: npm run code_exec_multi_session
+ */
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Constants, GraphEvents } from '@/common';
+import { Run } from '@/run';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+const conversationHistory = [];
+/**
+ * Prints session context from the graph
+ */
+function printSessionContext(run, label) {
+    const graph = run.Graph;
+    if (!graph) {
+        console.log(`\n[${label}] No graph available`);
+        return;
+    }
+    const session = graph.sessions.get(Constants.EXECUTE_CODE);
+    console.log(`\n========== ${label} ==========`);
+    if (!session) {
+        console.log('  No session context stored yet');
+        return;
+    }
+    console.log(`  Latest session_id: ${session.session_id}`);
+    console.log(`  Files tracked: ${session.files?.length ?? 0}`);
+    for (const file of session.files ?? []) {
+        console.log(`    - ${file.name} (session: ${file.session_id})`);
+    }
+}
+async function testMultiSessionFiles() {
+    const { provider } = await getArgs();
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+    };
+    const llmConfig = getLLMConfig(provider);
+    const run = await Run.create({
+        runId: 'multi-session-test',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [createCodeExecutionTool()],
+            instructions: `You are a coding assistant. Execute code exactly as requested.
+When asked to create files, use Python and save to /mnt/data/.
+When reading files, print their contents.
+Be concise in responses.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const streamConfig = {
+        configurable: {
+            provider,
+            thread_id: 'multi-session-test',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    // ========== TEST 1: Create first file ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST 1: Create first file (file_a.txt)');
+    console.log('='.repeat(60));
+    conversationHistory.push(new HumanMessage(`
+Create a file called "file_a.txt" with the content:
+"This is file A, version 1"
+Print confirmation when done.
+`));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const messages1 = run.getRunMessages();
+    if (messages1)
+        conversationHistory.push(...messages1);
+    printSessionContext(run, 'After Test 1');
+    // ========== TEST 2: Create second file (different session) ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST 2: Create second file (file_b.txt)');
+    console.log('Expecting: Both file_a.txt and file_b.txt tracked');
+    console.log('='.repeat(60));
+    conversationHistory.push(new HumanMessage(`
+Create a NEW file called "file_b.txt" with the content:
+"This is file B"
+Print confirmation when done.
+`));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const messages2 = run.getRunMessages();
+    if (messages2)
+        conversationHistory.push(...messages2);
+    printSessionContext(run, 'After Test 2');
+    // ========== TEST 3: Read BOTH files (verifies accumulation) ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST 3: Read BOTH files from previous executions');
+    console.log('This verifies multi-session file accumulation works');
+    console.log('='.repeat(60));
+    conversationHistory.push(new HumanMessage(`
+Read and print the contents of BOTH files:
+1. file_a.txt
+2. file_b.txt
+
+Show me what's in each file.
+`));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const messages3 = run.getRunMessages();
+    if (messages3)
+        conversationHistory.push(...messages3);
+    printSessionContext(run, 'After Test 3');
+    // ========== TEST 4: Edit file_a.txt (verifies latest-wins) ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST 4: Edit file_a.txt (create new version)');
+    console.log('Expecting: Old file_a.txt replaced with new version');
+    console.log('='.repeat(60));
+    conversationHistory.push(new HumanMessage(`
+Create an UPDATED version of "file_a.txt" with the content:
+"This is file A, version 2 - UPDATED"
+Print confirmation when done.
+`));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const messages4 = run.getRunMessages();
+    if (messages4)
+        conversationHistory.push(...messages4);
+    printSessionContext(run, 'After Test 4');
+    // ========== TEST 5: Read file_a.txt (verifies latest version) ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST 5: Read file_a.txt to verify it has the UPDATED content');
+    console.log('Expected: "version 2 - UPDATED" NOT "version 1"');
+    console.log('='.repeat(60));
+    conversationHistory.push(new HumanMessage(`
+Read and print the contents of file_a.txt.
+Tell me what version it shows.
+`));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const messages5 = run.getRunMessages();
+    if (messages5)
+        conversationHistory.push(...messages5);
+    printSessionContext(run, 'Final Session State');
+    // ========== SUMMARY ==========
+    console.log('\n\n' + '='.repeat(60));
+    console.log('TEST SUMMARY');
+    console.log('='.repeat(60));
+    const finalSession = run.Graph?.sessions.get(Constants.EXECUTE_CODE);
+    if (finalSession) {
+        const files = finalSession.files ?? [];
+        const uniqueSessionIds = new Set(files.map((f) => f.session_id));
+        console.log(`\nTotal files tracked: ${files.length}`);
+        console.log(`Unique session_ids: ${uniqueSessionIds.size}`);
+        console.log('\nFiles:');
+        for (const file of files) {
+            console.log(`  - ${file.name} (session: ${file.session_id?.slice(0, 20)}...)`);
+        }
+        // Verify expectations
+        const fileACount = files.filter((f) => f.name === 'file_a.txt').length;
+        const fileBCount = files.filter((f) => f.name === 'file_b.txt').length;
+        console.log('\n✓ Checks:');
+        console.log(`  file_a.txt count: ${fileACount} (expected: 1, latest wins)`);
+        console.log(`  file_b.txt count: ${fileBCount} (expected: 1)`);
+        if (fileACount === 1 && fileBCount === 1) {
+            console.log('\n✅ All tests passed! Multi-session tracking works.');
+        }
+        else {
+            console.log('\n❌ Test failed - unexpected file counts');
+        }
+    }
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    process.exit(1);
+});
+testMultiSessionFiles().catch((err) => {
+    console.error('Test failed:', err);
+    process.exit(1);
+});
+//# sourceMappingURL=code_exec_multi_session.js.map

package/src/scripts/code_exec_multi_session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec_multi_session.js","sourceRoot":"","sources":["code_exec_multi_session.ts"],"names":[],"mappings":"AAAA,yCAAyC;AACzC;;;;;;;;GAQG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;GAEG;AACH,SAAS,mBAAmB,CAAC,GAAkB,EAAE,KAAa;IAC5D,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,sBAAsB,CAAC,CAAC;QAC/C,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAE5C,CAAC;IAEd,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,aAAa,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QAC/C,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IAC1D,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9D,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,IAAI,cAAc,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IAErC,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;KAC9D,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,oBAAoB;QAC3B,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,YAAY,EAAE;;;yBAGK;SACpB;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,YAAY,GAGd;QACF,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,kDAAkD;IAClD,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,mBAAmB,CAAC,IAAI,CACtB,IAAI,YAAY,CAAC;;;;CAIpB,CAAC,CACC,CAAC;IAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACvC,IAAI,SAAS;QAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAEtD,mBAAmB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;IAEzC,uEAAuE;IACvE,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,mBAAmB,CAAC,IAAI,CACtB,IAAI,YAAY,CAAC;;;;CAIpB,CAAC,CACC,CAAC;IAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACvC,IAAI,SAAS;QAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAEtD,mBAAmB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;IAEzC,wEAAwE;IACxE,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;IAChE,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;IACnE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,mBAAmB,CAAC,IAAI,CACtB,IAAI,YAAY,CAAC;;;;;;CAMpB,CAAC,CACC,CAAC;IAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACvC,IAAI,SAAS;QAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAEtD,mBAAmB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;IAEzC,uEAAuE;IACvE,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;IAC5D,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;IACnE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,mBAAmB,CAAC,IAAI,CACtB,IAAI,YAAY,CAAC;;;;CAIpB,CAAC,CACC,CAAC;IAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACvC,IAAI,SAAS;QAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAEtD,mBAAmB,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;IAEzC,0EAA0E;IAC1E,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,8DAA8D,CAAC,CAAC;IAC5E,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,mBAAmB,CAAC,IAAI,CACtB,IAAI,YAAY,CAAC;;;CAGpB,CAAC,CACC,CAAC;IAEF,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACvC,IAAI,SAAS;QAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAEtD,mBAAmB,CAAC,GAAG,EAAE,qBAAqB,CAAC,CAAC;IAEhD,gCAAgC;IAChC,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,YAAY,GAAG,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAEtD,CAAC;IAEd,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC;QACvC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,0BAA0B,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,OAAO,CAAC,GAAG,CAAC,uBAAuB,gBAAgB,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CACT,OAAO,IAAI,CAAC,IAAI,cAAc,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAClE,CAAC;QACJ,CAAC;QAED,sBAAsB;QACtB,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,MAAM,CAAC;QACvE,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,MAAM,CAAC;QAEvE,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,uBAAuB,UAAU,6BAA6B,CAAC,CAAC;QAC5E,OAAO,CAAC,GAAG,CAAC,uBAAuB,UAAU,gBAAgB,CAAC,CAAC;QAE/D,IAAI,UAAU,KAAK,CAAC,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;QACrE,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;AACH,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/code_exec_multi_session.ts

@@ -44,8 +44,8 @@ function printSessionContext(run: Run<t.IState>, label: string): void {
   }
 
   console.log(`  Latest session_id: ${session.session_id}`);
-  console.log(`  Files tracked: ${session.files.length}`);
-  for (const file of session.files) {
+  console.log(`  Files tracked: ${session.files?.length ?? 0}`);
+  for (const file of session.files ?? []) {
     console.log(`    - ${file.name} (session: ${file.session_id})`);
   }
 }
@@ -73,6 +73,7 @@ When reading files, print their contents.
 Be concise in responses.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 
@@ -198,25 +199,20 @@ Tell me what version it shows.
     | undefined;
 
   if (finalSession) {
-    const uniqueSessionIds = new Set(
-      finalSession.files.map((f) => f.session_id)
-    );
-    console.log(`\nTotal files tracked: ${finalSession.files.length}`);
+    const files = finalSession.files ?? [];
+    const uniqueSessionIds = new Set(files.map((f) => f.session_id));
+    console.log(`\nTotal files tracked: ${files.length}`);
     console.log(`Unique session_ids: ${uniqueSessionIds.size}`);
     console.log('\nFiles:');
-    for (const file of finalSession.files) {
+    for (const file of files) {
       console.log(
         `  - ${file.name} (session: ${file.session_id?.slice(0, 20)}...)`
       );
     }
 
     // Verify expectations
-    const fileACount = finalSession.files.filter(
-      (f) => f.name === 'file_a.txt'
-    ).length;
-    const fileBCount = finalSession.files.filter(
-      (f) => f.name === 'file_b.txt'
-    ).length;
+    const fileACount = files.filter((f) => f.name === 'file_a.txt').length;
+    const fileBCount = files.filter((f) => f.name === 'file_b.txt').length;
 
     console.log('\n✓ Checks:');
     console.log(`  file_a.txt count: ${fileACount} (expected: 1, latest wins)`);