@illuma-ai/agents 1.0.89 → 1.0.93

package/src/scripts/programmatic_exec.js

@@ -0,0 +1,277 @@
+// src/scripts/programmatic_exec.ts
+/**
+ * Test script for Programmatic Tool Calling (PTC).
+ * Run with: npm run programmatic_exec
+ *
+ * Demonstrates:
+ * 1. Runtime toolMap injection - the tool map is passed at invocation time
+ * 2. Tool classification with allowed_callers (inspired by Anthropic's API)
+ *    - 'direct': Tool can only be called directly by the LLM (default)
+ *    - 'code_execution': Tool can only be called from within PTC
+ *    - Both: Tool can be called either way
+ *
+ * IMPORTANT: The Python code passed to PTC should NOT define the tool functions.
+ * The Code API automatically generates async function stubs from the tool definitions.
+ * The code should just CALL the tools as if they're already available:
+ *   - result = await get_weather(city="SF")
+ *   - results = await asyncio.gather(tool1(), tool2())
+ */
+import { config } from 'dotenv';
+config();
+import { createProgrammaticToolCallingTool } from '@/tools/ProgrammaticToolCalling';
+import { createGetTeamMembersTool, createGetExpensesTool, createGetWeatherTool, createCalculatorTool, createProgrammaticToolRegistry, } from '@/test/mockTools';
+async function runTest(ptcTool, testName, code, options) {
+    console.log(`\n${'='.repeat(70)}`);
+    console.log(`TEST: ${testName}`);
+    console.log('='.repeat(70));
+    console.log('\nCode:');
+    console.log('```python');
+    console.log(code.trim());
+    console.log('```\n');
+    try {
+        const startTime = Date.now();
+        // Manual testing: schema params + extras (LangChain moves extras to config.toolCall)
+        const result = await ptcTool.invoke({
+            code,
+            tools: options.tools,
+            session_id: options.session_id,
+            timeout: options.timeout,
+            toolMap: options.toolMap, // Non-schema field → config.toolCall.toolMap
+        });
+        const duration = Date.now() - startTime;
+        console.log(`Result (${duration}ms):`);
+        if (Array.isArray(result)) {
+            console.log(result[0]);
+            if (options.showArtifact) {
+                console.log('\n--- Artifact ---');
+                console.dir(result[1], { depth: null });
+            }
+        }
+        else {
+            console.log(result);
+        }
+    }
+    catch (error) {
+        console.error('Error:', error instanceof Error ? error.message : error);
+    }
+}
+// ============================================================================
+// Main
+// ============================================================================
+async function main() {
+    console.log('Programmatic Tool Calling (PTC) - Test Script');
+    console.log('==============================================');
+    console.log('Demonstrating runtime toolMap injection\n');
+    const apiKey = process.env.CODE_EXECUTOR_API_KEY;
+    if (!apiKey) {
+        console.error('Error: CODE_EXECUTOR_API_KEY environment variable is not set.');
+        console.log('Please set it in your .env file or environment.');
+        process.exit(1);
+    }
+    console.log('Creating mock tools...');
+    const mockTools = [
+        createGetTeamMembersTool(),
+        createGetExpensesTool(),
+        createGetWeatherTool(),
+        createCalculatorTool(),
+    ];
+    const toolMap = new Map(mockTools.map((t) => [t.name, t]));
+    const toolDefinitions = Array.from(createProgrammaticToolRegistry().values());
+    console.log(`ToolMap contains ${toolMap.size} tools: ${Array.from(toolMap.keys()).join(', ')}`);
+    console.log('\nCreating PTC tool (without toolMap)...');
+    const ptcTool = createProgrammaticToolCallingTool({ apiKey });
+    console.log('PTC tool created successfully!');
+    console.log('Note: toolMap will be passed at runtime with each invocation.\n');
+    const baseOptions = { toolMap, tools: toolDefinitions };
+    // =========================================================================
+    // Test 1: Simple async tool call
+    // =========================================================================
+    await runTest(ptcTool, 'Simple async tool call', `
+# Tools are auto-generated as async functions - just await them
+result = await get_weather(city="San Francisco")
+print(f"Weather in SF: {result['temperature']}°F, {result['condition']}")
+    `, { ...baseOptions, showArtifact: true });
+    // =========================================================================
+    // Test 2: Sequential loop with await
+    // =========================================================================
+    await runTest(ptcTool, 'Sequential loop - Process team expenses', `
+# Each tool call uses await
+team = await get_team_members()
+print("Team expense report:")
+print("-" * 30)
+total = 0
+for member in team:
+    expenses = await get_expenses(user_id=member['id'])
+    member_total = sum(e['amount'] for e in expenses)
+    total += member_total
+    print(f"{member['name']}: \${member_total:.2f}")
+print("-" * 30)
+print(f"Total: \${total:.2f}")
+    `, baseOptions);
+    // =========================================================================
+    // Test 3: Parallel execution with asyncio.gather
+    // =========================================================================
+    await runTest(ptcTool, 'Parallel execution - Weather for multiple cities', `
+# Use asyncio.gather for parallel tool calls - single round-trip!
+import asyncio
+
+cities = ["San Francisco", "New York", "London"]
+results = await asyncio.gather(*[
+    get_weather(city=city)
+    for city in cities
+])
+
+print("Weather report:")
+for city, weather in zip(cities, results):
+    print(f"  {city}: {weather['temperature']}°F, {weather['condition']}")
+    `, baseOptions);
+    // =========================================================================
+    // Test 4: Chained dependencies
+    // =========================================================================
+    await runTest(ptcTool, 'Chained dependencies - Get team then process each', `
+# Get team first, then fetch expenses for each
+team = await get_team_members()
+engineering = [m for m in team if m['department'] == 'Engineering']
+
+print(f"Engineering team ({len(engineering)} members):")
+for member in engineering:
+    expenses = await get_expenses(user_id=member['id'])
+    equipment = sum(e['amount'] for e in expenses if e['category'] == 'equipment')
+    print(f"  {member['name']}: \${equipment:.2f} on equipment")
+    `, baseOptions);
+    // =========================================================================
+    // Test 5: Conditional logic
+    // =========================================================================
+    await runTest(ptcTool, 'Conditional logic - Find high spenders', `
+team = await get_team_members()
+high_spenders = []
+
+for member in team:
+    expenses = await get_expenses(user_id=member['id'])
+    total = sum(e['amount'] for e in expenses)
+    if total > 300:
+        high_spenders.append((member['name'], total))
+
+if high_spenders:
+    print("High spenders (over $300):")
+    for name, amount in sorted(high_spenders, key=lambda x: x[1], reverse=True):
+        print(f"  {name}: \${amount:.2f}")
+else:
+    print("No high spenders found.")
+    `, baseOptions);
+    // =========================================================================
+    // Test 6: Mixed parallel and sequential
+    // =========================================================================
+    await runTest(ptcTool, 'Mixed - Parallel expense fetch after sequential team fetch', `
+import asyncio
+
+# Step 1: Get team (one tool call)
+team = await get_team_members()
+print(f"Fetched {len(team)} team members")
+
+# Step 2: Get all expenses in parallel (single round-trip for all!)
+all_expenses = await asyncio.gather(*[
+    get_expenses(user_id=member['id'])
+    for member in team
+])
+
+# Step 3: Process and output
+print("\\nExpense summary:")
+for member, expenses in zip(team, all_expenses):
+    total = sum(e['amount'] for e in expenses)
+    print(f"  {member['name']}: \${total:.2f} ({len(expenses)} items)")
+    `, baseOptions);
+    // =========================================================================
+    // Test 7: Calculator usage
+    // =========================================================================
+    await runTest(ptcTool, 'Calculator tool usage', `
+# All tools are async - use await
+result1 = await calculator(expression="2 + 2 * 3")
+result2 = await calculator(expression="(10 + 5) / 3")
+
+print(f"2 + 2 * 3 = {result1['result']}")
+print(f"(10 + 5) / 3 = {result2['result']:.2f}")
+    `, baseOptions);
+    // =========================================================================
+    // Test 8: Error handling in code
+    // =========================================================================
+    await runTest(ptcTool, 'Error handling - Invalid city', `
+# Tool errors become Python exceptions - handle with try/except
+cities = ["San Francisco", "InvalidCity", "New York"]
+
+for city in cities:
+    try:
+        weather = await get_weather(city=city)
+        print(f"{city}: {weather['temperature']}°F")
+    except Exception as e:
+        print(f"{city}: Error - {e}")
+    `, baseOptions);
+    // =========================================================================
+    // Test 9: Early termination
+    // =========================================================================
+    await runTest(ptcTool, 'Early termination - Stop when condition met', `
+# Stop as soon as we find what we need - no wasted tool calls
+team = await get_team_members()
+
+for member in team:
+    expenses = await get_expenses(user_id=member['id'])
+    if any(e['category'] == 'equipment' for e in expenses):
+        print(f"First team member with equipment expense: {member['name']}")
+        equipment_total = sum(e['amount'] for e in expenses if e['category'] == 'equipment')
+        print(f"Equipment total: \${equipment_total:.2f}")
+        break
+else:
+    print("No team member has equipment expenses")
+    `, baseOptions);
+    // =========================================================================
+    // Test 10: Subset of tools
+    // =========================================================================
+    await runTest(ptcTool, 'Subset of tools - Only weather', `
+# Only the weather tool is available in this execution
+import asyncio
+
+sf, nyc = await asyncio.gather(
+    get_weather(city="San Francisco"),
+    get_weather(city="New York")
+)
+print(f"SF: {sf['temperature']}°F vs NYC: {nyc['temperature']}°F")
+difference = abs(sf['temperature'] - nyc['temperature'])
+print(f"Temperature difference: {difference}°F")
+    `, {
+        ...baseOptions,
+        tools: [toolDefinitions.find((t) => t.name === 'get_weather')],
+    });
+    // =========================================================================
+    // Test 11: Note about ToolNode injection
+    // =========================================================================
+    console.log(`\n${'='.repeat(70)}`);
+    console.log('NOTE: ToolNode Runtime Injection');
+    console.log('='.repeat(70));
+    console.log('\nWhen PTC is invoked through ToolNode in a real agent:\n' +
+        '- ToolNode detects call.name === "run_tools_with_code"\n' +
+        '- ToolNode injects: { ...invokeParams, toolMap, toolDefs }\n' +
+        '- PTC tool extracts these from params (not from config)\n' +
+        '- No explicit tools parameter needed in schema\n\n' +
+        'This test demonstrates param injection with explicit tools:\n');
+    await runTest(ptcTool, 'Runtime injection with explicit tools', `
+# ToolNode would inject toolMap+toolDefs
+# For this test, we pass tools explicitly (same effect)
+team = await get_team_members()
+print(f"Team size: {len(team)}")
+for member in team:
+    print(f"- {member['name']} ({member['department']})")
+    `, baseOptions);
+    console.log('\n' + '='.repeat(70));
+    console.log('All tests completed!');
+    console.log('='.repeat(70) + '\n');
+    console.log('Summary of allowed_callers patterns:');
+    console.log('- get_team_members, get_expenses, calculator: code_execution only');
+    console.log('- get_weather: both direct and code_execution');
+    console.log('\nIn a real agent setup, the LLM would only see tools with allowed_callers');
+    console.log('including "direct", while PTC can call any tool with "code_execution".\n');
+}
+main().catch((err) => {
+    console.error('Fatal error:', err);
+    process.exit(1);
+});
+//# sourceMappingURL=programmatic_exec.js.map

package/src/scripts/programmatic_exec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"programmatic_exec.js","sourceRoot":"","sources":["programmatic_exec.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAIT,OAAO,EAAE,iCAAiC,EAAE,MAAM,iCAAiC,CAAC;AACpF,OAAO,EACL,wBAAwB,EACxB,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,8BAA8B,GAC/B,MAAM,kBAAkB,CAAC;AAc1B,KAAK,UAAU,OAAO,CACpB,OAA6D,EAC7D,QAAgB,EAChB,IAAY,EACZ,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,SAAS,QAAQ,EAAE,CAAC,CAAC;IACjC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvB,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAErB,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,qFAAqF;QACrF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC;YAClC,IAAI;YACJ,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,6CAA6C;SACxE,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAExC,OAAO,CAAC,GAAG,CAAC,WAAW,QAAQ,MAAM,CAAC,CAAC;QACvC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACvB,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,OAAO;AACP,+EAA+E;AAE/E,KAAK,UAAU,IAAI;IACjB,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IAEzD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IACjD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CACX,+DAA+D,CAChE,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IACtC,MAAM,SAAS,GAA8B;QAC3C,wBAAwB,EAAE;QAC1B,qBAAqB,EAAE;QACvB,oBAAoB,EAAE;QACtB,oBAAoB,EAAE;KACvB,CAAC;IAEF,MAAM,OAAO,GAAY,IAAI,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACpE,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,8BAA8B,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;IAE9E,OAAO,CAAC,GAAG,CACT,oBAAoB,OAAO,CAAC,IAAI,WAAW,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACnF,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IACxD,MAAM,OAAO,GAAG,iCAAiC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,CACT,iEAAiE,CAClE,CAAC;IAEF,MAAM,WAAW,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;IAExD,4EAA4E;IAC5E,iCAAiC;IACjC,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,wBAAwB,EACxB;;;;KAIC,EACD,EAAE,GAAG,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,CACvC,CAAC;IAEF,4EAA4E;IAC5E,qCAAqC;IACrC,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,yCAAyC,EACzC;;;;;;;;;;;;;KAaC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,iDAAiD;IACjD,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,kDAAkD,EAClD;;;;;;;;;;;;;KAaC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,+BAA+B;IAC/B,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,mDAAmD,EACnD;;;;;;;;;;KAUC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,4BAA4B;IAC5B,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,wCAAwC,EACxC;;;;;;;;;;;;;;;;KAgBC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,wCAAwC;IACxC,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,4DAA4D,EAC5D;;;;;;;;;;;;;;;;;;KAkBC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,2BAA2B;IAC3B,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,uBAAuB,EACvB;;;;;;;KAOC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,iCAAiC;IACjC,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,+BAA+B,EAC/B;;;;;;;;;;KAUC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,4BAA4B;IAC5B,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,6CAA6C,EAC7C;;;;;;;;;;;;;KAaC,EACD,WAAW,CACZ,CAAC;IAEF,4EAA4E;IAC5E,2BAA2B;IAC3B,4EAA4E;IAC5E,MAAM,OAAO,CACX,OAAO,EACP,gCAAgC,EAChC;;;;;;;;;;;KAWC,EACD;QACE,GAAG,WAAW;QACd,KAAK,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAE,CAAC;KAChE,CACF,CAAC;IAEF,4EAA4E;IAC5E,yCAAyC;IACzC,4EAA4E;IAC5E,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CACT,2DAA2D;QACzD,0DAA0D;QAC1D,8DAA8D;QAC9D,2DAA2D;QAC3D,oDAAoD;QACpD,+DAA+D,CAClE,CAAC;IAEF,MAAM,OAAO,CACX,OAAO,EACP,uCAAuC,EACvC;;;;;;;KAOC,EACD,WAAW,CACZ,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACpD,OAAO,CAAC,GAAG,CACT,mEAAmE,CACpE,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CACT,4EAA4E,CAC7E,CAAC;IACF,OAAO,CAAC,GAAG,CACT,0EAA0E,CAC3E,CAAC;AACJ,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/programmatic_exec_agent.js

@@ -0,0 +1,168 @@
+// src/scripts/programmatic_exec_agent.ts
+/**
+ * Test script for Programmatic Tool Calling (PTC) with agent integration.
+ * Run with: npm run programmatic_exec_agent
+ *
+ * Demonstrates:
+ * 1. Tool classification with allowed_callers:
+ *    - direct: Tool bound to LLM (can be called directly)
+ *    - code_execution: Tool available for PTC (not bound to LLM)
+ *    - Both: Tool bound to LLM AND available for PTC
+ * 2. Deferred loading with defer_loading: true (for tool search)
+ * 3. Agent-level tool configuration via toolRegistry
+ * 4. ToolNode runtime injection of programmatic tools
+ *
+ * This shows the real-world integration pattern with agents.
+ */
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+import { createProgrammaticToolCallingTool } from '@/tools/ProgrammaticToolCalling';
+import { createToolSearch } from '@/tools/ToolSearch';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { createGetTeamMembersTool, createGetExpensesTool, createGetWeatherTool, createProgrammaticToolRegistry, } from '@/test/mockTools';
+// ============================================================================
+// Tool Registry (Metadata)
+// ============================================================================
+/**
+ * Tool registry only needs business logic tools that require filtering.
+ * Special tools (execute_code, run_tools_with_code, tool_search)
+ * are always bound directly to the LLM and don't need registry entries.
+ */
+function createAgentToolRegistry() {
+    // Use shared programmatic tool registry (get_team_members, get_expenses, etc.)
+    return createProgrammaticToolRegistry();
+}
+// ============================================================================
+// Main
+// ============================================================================
+const conversationHistory = [];
+async function main() {
+    console.log('Programmatic Tool Calling - Agent Integration Test');
+    console.log('===================================================\n');
+    const { userName, location, provider } = await getArgs();
+    const llmConfig = getLLMConfig(provider);
+    // Create all tool instances
+    const mockTools = [
+        createGetTeamMembersTool(),
+        createGetExpensesTool(),
+        createGetWeatherTool(),
+    ];
+    const toolMap = new Map(mockTools.map((t) => [t.name, t]));
+    // Create special tools (PTC, code execution, tool search)
+    const codeExecTool = createCodeExecutionTool();
+    const ptcTool = createProgrammaticToolCallingTool();
+    const toolSearchTool = createToolSearch();
+    // Build complete tool list and map
+    const allTools = [...mockTools, codeExecTool, ptcTool, toolSearchTool];
+    const completeToolMap = new Map(allTools.map((t) => [t.name, t]));
+    // Create tool registry with allowed_callers configuration
+    // Only includes business logic tools (not special tools like execute_code, PTC, tool_search)
+    const toolRegistry = createAgentToolRegistry();
+    console.log('Tool configuration:');
+    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-only:', Array.from(toolRegistry.values())
+        .filter((t) => !t.allowed_callers ||
+        (t.allowed_callers.includes('direct') &&
+            !t.allowed_callers.includes('code_execution')))
+        .map((t) => t.name)
+        .join(', '));
+    console.log('- Both:', Array.from(toolRegistry.values())
+        .filter((t) => t.allowed_callers?.includes('direct') &&
+        t.allowed_callers?.includes('code_execution'))
+        .map((t) => t.name)
+        .join(', '));
+    // Create run with toolRegistry configuration
+    const run = await Run.create({
+        runId: 'ptc-agent-test',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            agents: [
+                {
+                    agentId: 'default',
+                    provider: llmConfig.provider,
+                    clientOptions: llmConfig,
+                    tools: allTools,
+                    toolMap: completeToolMap,
+                    toolRegistry, // Pass tool registry for programmatic/deferred tool config
+                    instructions: 'You are an AI assistant with access to programmatic tool calling. ' +
+                        'When you need to process multiple items or perform complex data operations, ' +
+                        'use the run_tools_with_code tool to write Python code that calls tools efficiently.',
+                },
+            ],
+        },
+        returnContent: true,
+        skipCleanup: true,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'ptc-test-conversation',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('\n' + '='.repeat(70));
+    console.log('Test: Process team expenses using PTC');
+    console.log('='.repeat(70) + '\n');
+    const userMessage = new HumanMessage(`Hi! I need you to analyze our team's expenses. Please:
+1. Get the list of team members
+2. For each member, get their expense records
+3. Calculate the total expenses per member
+4. Identify anyone who spent more than $300
+5. Show me the results in a nice format
+
+Use the run_tools_with_code tool to do this efficiently - don't call each tool separately!`);
+    conversationHistory.push(userMessage);
+    const inputs = {
+        messages: conversationHistory,
+    };
+    console.log('Running agent with PTC capability...\n');
+    const finalContentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+    }
+    console.log('\n' + '='.repeat(70));
+    console.log('Agent Response:');
+    console.log('='.repeat(70));
+    if (finalContentParts) {
+        for (const part of finalContentParts) {
+            if (part?.type === 'text' && part.text) {
+                console.log(part.text);
+            }
+        }
+    }
+    console.log('\n' + '='.repeat(70));
+    console.log('Test completed successfully!');
+    console.log('='.repeat(70));
+    console.log('\nKey observations:');
+    console.log('1. LLM only sees tools with allowed_callers including "direct" (get_weather, execute_code, run_tools_with_code, tool_search)');
+    console.log('2. When PTC is invoked, ToolNode automatically injects programmatic tools (get_team_members, get_expenses, get_weather)');
+    console.log('3. No need to manually configure runtime toolMap - handled by agent context');
+    console.log('4. Tool filtering based on allowed_callers happens automatically\n');
+}
+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);
+});
+main().catch((err) => {
+    console.error('Fatal error:', err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=programmatic_exec_agent.js.map

package/src/scripts/programmatic_exec_agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"programmatic_exec_agent.js","sourceRoot":"","sources":["programmatic_exec_agent.ts"],"names":[],"mappings":"AAAA,yCAAyC;AACzC;;;;;;;;;;;;;;GAcG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAGT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,iCAAiC,EAAE,MAAM,iCAAiC,CAAC;AACpF,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EACL,wBAAwB,EACxB,qBAAqB,EACrB,oBAAoB,EACpB,8BAA8B,GAC/B,MAAM,kBAAkB,CAAC;AAE1B,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E;;;;GAIG;AACH,SAAS,uBAAuB;IAC9B,+EAA+E;IAC/E,OAAO,8BAA8B,EAAE,CAAC;AAC1C,CAAC;AAED,+EAA+E;AAC/E,OAAO;AACP,+EAA+E;AAE/E,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C,KAAK,UAAU,IAAI;IACjB,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;IAClE,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;IAErE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACzD,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,4BAA4B;IAC5B,MAAM,SAAS,GAA8B;QAC3C,wBAAwB,EAAE;QAC1B,qBAAqB,EAAE;QACvB,oBAAoB,EAAE;KACvB,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE3D,0DAA0D;IAC1D,MAAM,YAAY,GAAG,uBAAuB,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,iCAAiC,EAAE,CAAC;IACpD,MAAM,cAAc,GAAG,gBAAgB,EAAE,CAAC;IAE1C,mCAAmC;IACnC,MAAM,QAAQ,GAAG,CAAC,GAAG,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;IACvE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAElE,0DAA0D;IAC1D,6FAA6F;IAC7F,MAAM,YAAY,GAAG,uBAAuB,EAAE,CAAC;IAE/C,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC/C,OAAO,CAAC,GAAG,CACT,yBAAyB,EACzB,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,gBAAgB,EAChB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;SAC9B,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,CAAC,eAAe;QAClB,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACnC,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,CACnD;SACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,IAAI,CAAC,CACd,CAAC;IACF,OAAO,CAAC,GAAG,CACT,SAAS,EACT,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;SAC9B,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,QAAQ,CAAC;QACrC,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAChD;SACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,IAAI,CAAC,CACd,CAAC;IAEF,6CAA6C;IAC7C,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,gBAAgB;QACvB,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,EAAE,eAAe;oBACxB,YAAY,EAAE,2DAA2D;oBACzE,YAAY,EACV,oEAAoE;wBACpE,8EAA8E;wBAC9E,qFAAqF;iBACxF;aACF;SACF;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,MAAM,MAAM,GAGR;QACF,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,uBAAuB;SACnC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;IACrD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAEnC,MAAM,WAAW,GAAG,IAAI,YAAY,CAClC;;;;;;;2FAOuF,CACxF,CAAC;IAEF,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAEtC,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IAEtD,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAE3C,IAAI,aAAa,EAAE,CAAC;QAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC/B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,IAAI,iBAAiB,EAAE,CAAC;QACtB,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;YACrC,IAAI,IAAI,EAAE,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACvC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,8HAA8H,CAC/H,CAAC;IACF,OAAO,CAAC,GAAG,CACT,yHAAyH,CAC1H,CAAC;IACF,OAAO,CAAC,GAAG,CACT,6EAA6E,CAC9E,CAAC;IACF,OAAO,CAAC,GAAG,CACT,oEAAoE,CACrE,CAAC;AACJ,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,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACnC,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/programmatic_exec_agent.ts

@@ -138,6 +138,7 @@ async function main(): Promise<void> {
       ],
     },
     returnContent: true,
+    skipCleanup: true,
   });
 
   const config: Partial<RunnableConfig> & {

package/src/scripts/search.js

@@ -0,0 +1,118 @@
+/* eslint-disable no-console */
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { createSearchTool } from '@/tools/search';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+const conversationHistory = [];
+async function testStandardStreaming() {
+    const { userName, location, provider, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                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);
+    if (llmConfig.provider === Providers.BEDROCK) {
+        llmConfig.promptCache = true;
+    }
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [
+                createSearchTool({
+                    scraperProvider: 'serper',
+                }),
+            ],
+            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 = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Search query (search tool test)');
+    // const userMessage = `
+    // Make a search for the weather in ${location} today, which is ${currentDate}.
+    // Before making the search, please let me know what you're about to do, then immediately start searching without hesitation.
+    // Make sure to always refer to me by name, which is ${userName}.
+    // After giving me a thorough summary, tell me a joke about the weather forecast we went over.
+    // `;
+    // const userMessage = 'Are massage guns good?';
+    // const userMessage = 'What is functional programming?';
+    const userMessage = "Get me today's trending news.";
+    // const userMessage = "search recent italy earthquake volcano activity";
+    // const userMessage =
+    //   "use 'Trump' as the exact search query and tell me what you find.";
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+        console.dir(conversationHistory, { depth: null });
+    }
+    // console.dir(finalContentParts, { depth: null });
+    console.log('\n\n====================\n\n');
+    // console.dir(contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.warn('Unhandled Rejection (non-fatal):', reason);
+});
+testStandardStreaming().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=search.js.map

package/src/scripts/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["search.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,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,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAElD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAY,SAAS,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,KAAK,UAAU,qBAAqB;IAClC,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,sCAAsC;gBACtC,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,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,sCAAsC;YACxC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,IAAI,SAAS,CAAC,QAAQ,KAAK,SAAS,CAAC,OAAO,EAAE,CAAC;QAC5C,SAAqC,CAAC,WAAW,GAAG,IAAI,CAAC;IAC5D,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE;gBACL,gBAAgB,CAAC;oBACf,eAAe,EAAE,QAAQ;iBAC1B,CAAC;aACH;YACD,YAAY,EACV,yEAAyE;YAC3E,kGAAkG;SACnG;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,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,yCAAyC,CAAC,CAAC;IAEvD,wBAAwB;IACxB,+EAA+E;IAC/E,6HAA6H;IAC7H,iEAAiE;IACjE,8FAA8F;IAC9F,KAAK;IACL,gDAAgD;IAChD,yDAAyD;IACzD,MAAM,WAAW,GAAG,+BAA+B,CAAC;IACpD,yEAAyE;IACzE,sBAAsB;IACtB,wEAAwE;IAExE,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,8CAA8C;AAChD,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE,MAAM,CAAC,CAAC;AAC3D,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/search.ts

@@ -91,6 +91,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,
   });