@illuma-ai/agents 1.0.90 → 1.0.93

package/src/scripts/code_exec_ptc.js

@@ -0,0 +1,265 @@
+// src/scripts/code_exec_ptc.ts
+/**
+ * Live LLM test for Programmatic Tool Calling (PTC).
+ * Run with: npm run code_exec_ptc
+ *
+ * Tests PTC with a real LLM in the loop, demonstrating:
+ * 1. LLM decides when to use PTC
+ * 2. LLM writes Python code that calls tools programmatically
+ * 3. ToolNode automatically injects programmatic tools
+ * 4. Tools filtered by allowed_callers
+ */
+import { config } from 'dotenv';
+config();
+import { randomUUID } from 'crypto';
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { 
+// createProgrammaticToolRegistry,
+createGetTeamMembersTool, createGetExpensesTool, createGetWeatherTool, } from '@/test/mockTools';
+import { createMetadataAggregator, ModelEndHandler, ToolEndHandler, } from '@/events';
+import { createProgrammaticToolCallingTool } from '@/tools/ProgrammaticToolCalling';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { GraphEvents } from '@/common';
+import { Run } from '@/run';
+const conversationHistory = [];
+/**
+ * Creates a tool registry where ALL business tools are code_execution ONLY.
+ * This forces the LLM to use PTC - it cannot call these tools directly.
+ */
+function createPTCOnlyToolRegistry() {
+    const toolDefs = [
+        {
+            name: 'get_team_members',
+            description: 'Get list of team members. Returns array of objects with id, name, and department fields.',
+            parameters: {
+                type: 'object',
+                properties: {},
+                required: [],
+            },
+            allowed_callers: ['code_execution'], // PTC ONLY - not direct
+        },
+        {
+            name: 'get_expenses',
+            description: 'Get expense records for a user. Returns array of objects with amount and category fields.',
+            parameters: {
+                type: 'object',
+                properties: {
+                    user_id: {
+                        type: 'string',
+                        description: 'The user ID to fetch expenses for',
+                    },
+                },
+                required: ['user_id'],
+            },
+            allowed_callers: ['code_execution'], // PTC ONLY - not direct
+        },
+        {
+            name: 'get_weather',
+            description: 'Get current weather for a city. Returns object with temperature (number) and condition (string) fields.',
+            parameters: {
+                type: 'object',
+                properties: {
+                    city: {
+                        type: 'string',
+                        description: 'City name',
+                    },
+                },
+                required: ['city'],
+            },
+            allowed_callers: ['code_execution'], // PTC ONLY - not direct (changed from ['direct', 'code_execution'])
+        },
+    ];
+    return new Map(toolDefs.map((def) => [def.name, def]));
+}
+async function testProgrammaticToolCalling() {
+    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) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+                console.dir(data, { depth: null });
+            },
+        },
+    };
+    const llmConfig = getLLMConfig(provider);
+    // Create mock tool instances
+    const teamTool = createGetTeamMembersTool();
+    const expensesTool = createGetExpensesTool();
+    const weatherTool = createGetWeatherTool();
+    // Create special tools
+    const codeExecTool = createCodeExecutionTool();
+    const ptcTool = createProgrammaticToolCallingTool();
+    // Build complete tool list and map
+    const allTools = [teamTool, expensesTool, weatherTool, codeExecTool, ptcTool];
+    const toolMap = new Map(allTools.map((t) => [t.name, t]));
+    // Create tool registry where ALL business tools are PTC-only
+    // This means the LLM CANNOT call get_team_members, get_expenses, get_weather directly
+    // It MUST use run_tools_with_code to invoke them
+    const toolRegistry = createPTCOnlyToolRegistry();
+    console.log('\n' + '='.repeat(70));
+    console.log('Tool Configuration Summary:');
+    console.log('='.repeat(70));
+    console.log('Total tools:', allTools.length);
+    console.log('Programmatic-allowed:', Array.from(toolRegistry.values())
+        .filter((t) => t.allowed_callers?.includes('code_execution'))
+        .map((t) => t.name)
+        .join(', '));
+    console.log('Direct-callable:', Array.from(toolRegistry.values())
+        .filter((t) => !t.allowed_callers || t.allowed_callers.includes('direct'))
+        .map((t) => t.name)
+        .join(', '));
+    console.log('='.repeat(70) + '\n');
+    const run = await Run.create({
+        runId: randomUUID(),
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            agents: [
+                {
+                    agentId: 'default',
+                    provider: llmConfig.provider,
+                    clientOptions: llmConfig,
+                    tools: allTools,
+                    toolMap,
+                    toolRegistry,
+                    instructions: 'You are a friendly AI assistant with advanced coding capabilities.\n\n' +
+                        'IMPORTANT: The tools get_team_members(), get_expenses(), and get_weather() are NOT available ' +
+                        'for direct function calling. You MUST use the run_tools_with_code tool to invoke them.\n\n' +
+                        'When you need to use these tools, write Python code using run_tools_with_code that calls:\n' +
+                        '- await get_team_members() - returns list of team members\n' +
+                        '- await get_expenses(user_id="...") - returns expenses for a user\n' +
+                        '- await get_weather(city="...") - returns weather data\n\n' +
+                        'Use asyncio.gather() for parallel execution when calling multiple tools.',
+                    additional_instructions: `The user's name is ${userName} and they are located in ${location}. Today is ${currentDate}.`,
+                },
+            ],
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'ptc-conversation-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Team Expense Analysis with PTC');
+    console.log('='.repeat(70) + '\n');
+    const userMessage1 = `Hi ${userName}! I need you to analyze our team's expenses. Please:
+
+1. Get the list of all team members
+2. For each member, fetch their expense records
+3. Calculate the total expenses per member
+4. Identify anyone who spent more than $500
+5. Show me a summary report
+
+IMPORTANT: Use the run_tools_with_code tool to do this efficiently. 
+Don't call each tool separately - write Python code that orchestrates all the calls!`;
+    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.log('Content Parts:');
+    console.dir(contentParts, { depth: null });
+    console.log('\n\n' + '='.repeat(70));
+    console.log('Test 2: Conditional Logic and Parallel Execution');
+    console.log('='.repeat(70) + '\n');
+    const userMessage2 = `Great job! Now let's test some advanced patterns. Please:
+
+1. Check the weather in both San Francisco and New York (in parallel!)
+2. Based on which city has better weather (warmer), fetch the team members
+3. For the Engineering team members only, calculate their travel expenses
+4. Show me the results
+
+Again, use run_tools_with_code for maximum efficiency. Use asyncio.gather() 
+to check both cities' weather at the same time!`;
+    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.log('Final Content Parts:');
+    console.dir(finalContentParts2, { depth: null });
+    console.log('\n\n' + '='.repeat(70));
+    console.log('Generating conversation title...');
+    console.log('='.repeat(70) + '\n');
+    const { handleLLMEnd, collected } = createMetadataAggregator();
+    const titleResult = await run.generateTitle({
+        provider,
+        inputText: userMessage1,
+        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);
+});
+testProgrammaticToolCalling().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=code_exec_ptc.js.map

package/src/scripts/code_exec_ptc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec_ptc.js","sourceRoot":"","sources":["code_exec_ptc.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B;;;;;;;;;GASG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO;AACL,kCAAkC;AAClC,wBAAwB,EACxB,qBAAqB,EACrB,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,wBAAwB,EACxB,eAAe,EACf,cAAc,GACf,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,iCAAiC,EAAE,MAAM,iCAAiC,CAAC;AACpF,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,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;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C;;;GAGG;AACH,SAAS,yBAAyB;IAChC,MAAM,QAAQ,GAAe;QAC3B;YACE,IAAI,EAAE,kBAAkB;YACxB,WAAW,EACT,0FAA0F;YAC5F,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,EAAE;aACb;YACD,eAAe,EAAE,CAAC,gBAAgB,CAAC,EAAE,wBAAwB;SAC9D;QACD;YACE,IAAI,EAAE,cAAc;YACpB,WAAW,EACT,2FAA2F;YAC7F,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,OAAO,EAAE;wBACP,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,mCAAmC;qBACjD;iBACF;gBACD,QAAQ,EAAE,CAAC,SAAS,CAAC;aACtB;YACD,eAAe,EAAE,CAAC,gBAAgB,CAAC,EAAE,wBAAwB;SAC9D;QACD;YACE,IAAI,EAAE,aAAa;YACnB,WAAW,EACT,yGAAyG;YAC3G,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,WAAW;qBACzB;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB;YACD,eAAe,EAAE,CAAC,gBAAgB,CAAC,EAAE,oEAAoE;SAC1G;KACF,CAAC;IAEF,OAAO,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,2BAA2B;IACxC,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;IAErE,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,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,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,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,wBAAwB,EAAE,CAAC;IAC5C,MAAM,YAAY,GAAG,qBAAqB,EAAE,CAAC;IAC7C,MAAM,WAAW,GAAG,oBAAoB,EAAE,CAAC;IAE3C,uBAAuB;IACvB,MAAM,YAAY,GAAG,uBAAuB,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,iCAAiC,EAAE,CAAC;IAEpD,mCAAmC;IACnC,MAAM,QAAQ,GAAG,CAAC,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAC9E,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE1D,6DAA6D;IAC7D,sFAAsF;IACtF,iDAAiD;IACjD,MAAM,YAAY,GAAG,yBAAyB,EAAE,CAAC;IAEjD,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CACT,uBAAuB,EACvB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;SAC5D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,IAAI,CAAC,CACd,CAAC;IACF,OAAO,CAAC,GAAG,CACT,kBAAkB,EAClB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;SACzE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,IAAI,CAAC,CACd,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAEnC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,UAAU,EAAE;QACnB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,MAAM,EAAE;gBACN;oBACE,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,SAAS,CAAC,QAAQ;oBAC5B,aAAa,EAAE,SAAS;oBACxB,KAAK,EAAE,QAAQ;oBACf,OAAO;oBACP,YAAY;oBACZ,YAAY,EACV,wEAAwE;wBACxE,+FAA+F;wBAC/F,4FAA4F;wBAC5F,6FAA6F;wBAC7F,6DAA6D;wBAC7D,qEAAqE;wBACrE,4DAA4D;wBAC5D,0EAA0E;oBAC5E,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,cAAc,WAAW,GAAG;iBACxH;aACF;SACF;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;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAEnC,MAAM,YAAY,GAAG,MAAM,QAAQ;;;;;;;;;qFASgD,CAAC;IAEpF,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,IAAI,MAAM,GAAG;QACX,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IAEF,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;IAED,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,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,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAEnC,MAAM,YAAY,GAAG;;;;;;;;gDAQyB,CAAC;IAE/C,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,MAAM,GAAG;QACP,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IAEF,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;IAED,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAEjD,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAEnC,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;IAEH,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,2BAA2B,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC1C,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_ptc.ts

@@ -217,6 +217,7 @@ async function testProgrammaticToolCalling(): Promise<void> {
       ],
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/code_exec_session.js

@@ -0,0 +1,217 @@
+// src/scripts/code_exec_session.ts
+/**
+ * Test script for automatic session tracking in code execution tools.
+ *
+ * This tests the automatic session_id injection feature where:
+ * 1. First code execution generates files and returns a session_id
+ * 2. Session context is stored in Graph.sessions
+ * 3. Subsequent code executions automatically have access to previous files
+ *    without the LLM needing to explicitly pass session_id
+ *
+ * Run with: npm run code_exec_session
+ */
+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 a formatted section header for test output
+ */
+function printSection(title) {
+    console.log('\n' + '='.repeat(60));
+    console.log(` ${title}`);
+    console.log('='.repeat(60) + '\n');
+}
+/**
+ * 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 testAutomaticSessionTracking() {
+    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) => {
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                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: 'session-tracking-test-1',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [createCodeExecutionTool()],
+            instructions: `You are an AI assistant testing automatic file persistence.
+When writing Python code:
+- Use print() for all outputs
+- Files from previous executions are automatically available in /mnt/data/
+- Files are READ-ONLY; write modifications to NEW filenames
+- IMPORTANT: Do NOT include session_id in your tool calls - it's handled automatically.`,
+            additional_instructions: `User: ${userName}, Location: ${location}, Date: ${currentDate}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const streamConfig = {
+        configurable: {
+            provider,
+            thread_id: 'session-tracking-test',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    // =========================================================================
+    // Test 1: Create initial file (establishes session)
+    // =========================================================================
+    printSection('Test 1: Create Initial File');
+    console.log('This test creates a file, which should establish a session context.\n');
+    const userMessage1 = `
+Create a Python file that writes a simple JSON config file named "app_config.json" with the following content:
+{
+  "app_name": "TestApp",
+  "version": "1.0.0",
+  "debug": true
+}
+
+After writing, print the contents to confirm it was created correctly.
+`;
+    conversationHistory.push(new HumanMessage(userMessage1));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const finalMessages1 = run.getRunMessages();
+    if (finalMessages1) {
+        conversationHistory.push(...finalMessages1);
+    }
+    printSection('Session Context After Test 1');
+    printSessionContext(run);
+    // =========================================================================
+    // Test 2: Access previously created file (uses automatic session injection)
+    // =========================================================================
+    printSection('Test 2: Access Previous File (Automatic Session)');
+    console.log('This test reads the file created in Test 1.');
+    console.log('The LLM does NOT need to provide session_id - it should be injected automatically.\n');
+    const userMessage2 = `
+Now read the app_config.json file that was just created and:
+1. Print its contents
+2. Confirm the version is "1.0.0"
+
+Note: You should be able to access this file from the previous execution automatically.
+`;
+    conversationHistory.push(new HumanMessage(userMessage2));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const finalMessages2 = run.getRunMessages();
+    if (finalMessages2) {
+        conversationHistory.push(...finalMessages2);
+    }
+    printSection('Session Context After Test 2');
+    printSessionContext(run);
+    // =========================================================================
+    // Test 3: Modify file (write to new filename)
+    // =========================================================================
+    printSection('Test 3: Modify File (Write to New Filename)');
+    console.log('This test modifies the config by reading the old file and writing a new one.\n');
+    const userMessage3 = `
+Read app_config.json, update the version to "2.0.0" and debug to false, 
+then save it as "app_config_v2.json". Print both the old and new contents.
+`;
+    conversationHistory.push(new HumanMessage(userMessage3));
+    await run.processStream({ messages: conversationHistory }, streamConfig);
+    const finalMessages3 = run.getRunMessages();
+    if (finalMessages3) {
+        conversationHistory.push(...finalMessages3);
+    }
+    printSection('Session Context After Test 3');
+    printSessionContext(run);
+    // =========================================================================
+    // Summary
+    // =========================================================================
+    printSection('Test Summary');
+    console.log('The automatic session tracking feature should have:');
+    console.log('1. Stored the session_id after the first code execution');
+    console.log('2. Automatically injected it into subsequent executions');
+    console.log('3. Accumulated file references across all executions');
+    console.log('\nCheck the session context output above to verify.\n');
+    // Generate title
+    const { handleLLMEnd, collected } = createMetadataAggregator();
+    const titleResult = await run.generateTitle({
+        provider,
+        inputText: 'Testing automatic session tracking for code execution',
+        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);
+});
+testAutomaticSessionTracking().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=code_exec_session.js.map

package/src/scripts/code_exec_session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec_session.js","sourceRoot":"","sources":["code_exec_session.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC;;;;;;;;;;GAUG;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,YAAY,CAAC,KAAa;IACjC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;AACrC,CAAC;AAED;;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,4BAA4B;IACzC,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;IAErE,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,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,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,yBAAyB;QAChC,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,YAAY,EAAE;;;;;wFAKoE;YAClF,uBAAuB,EAAE,SAAS,QAAQ,eAAe,QAAQ,WAAW,WAAW,GAAG;SAC3F;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,uBAAuB;SACnC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,4EAA4E;IAC5E,oDAAoD;IACpD,4EAA4E;IAC5E,YAAY,CAAC,6BAA6B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CACT,uEAAuE,CACxE,CAAC;IAEF,MAAM,YAAY,GAAG;;;;;;;;;CAStB,CAAC;IAEA,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IACzD,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IAEzE,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,YAAY,CAAC,8BAA8B,CAAC,CAAC;IAC7C,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAEzB,4EAA4E;IAC5E,4EAA4E;IAC5E,4EAA4E;IAC5E,YAAY,CAAC,kDAAkD,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CACT,sFAAsF,CACvF,CAAC;IAEF,MAAM,YAAY,GAAG;;;;;;CAMtB,CAAC;IAEA,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IACzD,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IAEzE,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,YAAY,CAAC,8BAA8B,CAAC,CAAC;IAC7C,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAEzB,4EAA4E;IAC5E,8CAA8C;IAC9C,4EAA4E;IAC5E,YAAY,CAAC,6CAA6C,CAAC,CAAC;IAC5D,OAAO,CAAC,GAAG,CACT,gFAAgF,CACjF,CAAC;IAEF,MAAM,YAAY,GAAG;;;CAGtB,CAAC;IAEA,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IACzD,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,EAAE,YAAY,CAAC,CAAC;IAEzE,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,YAAY,CAAC,8BAA8B,CAAC,CAAC;IAC7C,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAEzB,4EAA4E;IAC5E,UAAU;IACV,4EAA4E;IAC5E,YAAY,CAAC,cAAc,CAAC,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;IACnE,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;IACpE,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;IAErE,iBAAiB;IACjB,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,uDAAuD;QAClE,YAAY;QACZ,YAAY,EAAE;YACZ,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC;SAC9B;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,4BAA4B,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC3C,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_session.ts

@@ -141,6 +141,7 @@ When writing Python code:
       additional_instructions: `User: ${userName}, Location: ${location}, Date: ${currentDate}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/code_exec_simple.js

@@ -0,0 +1,120 @@
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { createTokenCounter } from '@/utils/tokens';
+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 instructions = 'You are a friendly AI assistant with coding capabilities. Always address the user by their name.';
+    const additional_instructions = `The user's name is ${userName} and they are located in ${location}.`;
+    // const userMessage1 = `how much memory is this (its in bytes) in MB? 31192000`;
+    // const userMessage1 = `can you show me a good use case for rscript by running some code`;
+    const userMessage1 = `Run hello world in french and in english, using python. please run 2 parallel code executions.`;
+    const humanMessage = new HumanMessage(userMessage1);
+    const tokenCounter = await createTokenCounter();
+    const indexTokenCountMap = {
+        0: tokenCounter(humanMessage),
+    };
+    const runConfig = {
+        runId: 'message-num-1',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [createCodeExecutionTool()],
+            instructions,
+            additional_instructions,
+            maxContextTokens: 8000,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+        indexTokenCountMap,
+        tokenCounter,
+    };
+    const run = await Run.create(runConfig);
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Simple Code Execution');
+    conversationHistory.push(humanMessage);
+    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 });
+}
+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=code_exec_simple.js.map

package/src/scripts/code_exec_simple.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code_exec_simple.js","sourceRoot":"","sources":["code_exec_simple.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,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;AAE5B,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,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,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;IACzC,MAAM,YAAY,GAChB,kGAAkG,CAAC;IACrG,MAAM,uBAAuB,GAAG,sBAAsB,QAAQ,4BAA4B,QAAQ,GAAG,CAAC;IAEtG,iFAAiF;IACjF,2FAA2F;IAC3F,MAAM,YAAY,GAAG,gGAAgG,CAAC;IACtH,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;IACpD,MAAM,YAAY,GAAG,MAAM,kBAAkB,EAAE,CAAC;IAChD,MAAM,kBAAkB,GAAG;QACzB,CAAC,EAAE,YAAY,CAAC,YAAY,CAAC;KAC9B,CAAC;IAEF,MAAM,SAAS,GAAgB;QAC7B,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,YAAY;YACZ,uBAAuB;YACvB,gBAAgB,EAAE,IAAI;SACvB;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;QACd,kBAAkB;QAClB,YAAY;KACb,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW,SAAS,CAAC,CAAC;IAElD,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAE7C,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAEvC,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;AAC7C,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,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_simple.ts

@@ -101,6 +101,7 @@ async function testCodeExecution(): Promise<void> {
       maxContextTokens: 8000,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
     indexTokenCountMap,
     tokenCounter,