illuma-agents 1.0.10 → 1.0.11

package/dist/esm/tools/ProgrammaticToolCalling.mjs

@@ -0,0 +1,430 @@
+import { z } from 'zod';
+import { config } from 'dotenv';
+import fetch from 'node-fetch';
+import { HttpsProxyAgent } from 'https-proxy-agent';
+import { getEnvironmentVariable } from '@langchain/core/utils/env';
+import { tool } from '@langchain/core/tools';
+import { imageExtRegex, getCodeBaseURL } from './CodeExecutor.mjs';
+import { Constants, EnvVar } from '../common/enum.mjs';
+
+// src/tools/ProgrammaticToolCalling.ts
+config();
+// ============================================================================
+// Constants
+// ============================================================================
+const imageMessage = 'Image is already displayed to the user';
+const otherMessage = 'File is already downloaded by the user';
+const accessMessage = 'Note: Files are READ-ONLY. Save changes to NEW filenames. To access these files in future executions, provide the `session_id` as a parameter (not in your code).';
+const emptyOutputMessage = 'stdout: Empty. Ensure you\'re writing output explicitly.\n';
+/** Default max round-trips to prevent infinite loops */
+const DEFAULT_MAX_ROUND_TRIPS = 20;
+/** Default execution timeout in milliseconds */
+const DEFAULT_TIMEOUT = 60000;
+// ============================================================================
+// Schema
+// ============================================================================
+const ProgrammaticToolCallingSchema = z.object({
+    code: z
+        .string()
+        .min(1)
+        .describe(`Python code that calls tools programmatically. Tools are automatically available as async Python functions - DO NOT define them yourself.
+
+The Code API generates async function stubs from the tool definitions. Just call them directly:
+
+Example (Simple call):
+  result = await get_weather(city="San Francisco")
+  print(result)
+
+Example (Parallel - Fastest):
+  results = await asyncio.gather(
+      get_weather(city="SF"),
+      get_weather(city="NYC"),
+      get_weather(city="London")
+  )
+  for city, weather in zip(["SF", "NYC", "London"], results):
+      print(f"{city}: {weather['temperature']}°F")
+
+Example (Loop with processing):
+  team = await get_team_members()
+  for member in team:
+      expenses = await get_expenses(user_id=member['id'])
+      total = sum(e['amount'] for e in expenses)
+      print(f"{member['name']}: \${total:.2f}")
+
+Example (Conditional logic):
+  data = await fetch_data(source="primary")
+  if not data:
+      data = await fetch_data(source="backup")
+  print(f"Got {len(data)} records")
+
+Requirements:
+- Tools are pre-defined as async functions - DO NOT write function definitions
+- Use await for all tool calls
+- Use asyncio.gather() for parallel execution of independent calls
+- Only print() output flows back to the context window
+- Tool results from programmatic calls do NOT consume context tokens`),
+    session_id: z
+        .string()
+        .optional()
+        .describe('Session ID for file access (same as regular code execution). Files load into /mnt/data/ and are READ-ONLY.'),
+    timeout: z
+        .number()
+        .int()
+        .min(1000)
+        .max(300000)
+        .optional()
+        .default(DEFAULT_TIMEOUT)
+        .describe('Maximum execution time in milliseconds. Default: 60 seconds. Max: 5 minutes.'),
+});
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/** Python reserved keywords that get `_tool` suffix in Code API */
+const PYTHON_KEYWORDS = new Set([
+    'False',
+    'None',
+    'True',
+    'and',
+    'as',
+    'assert',
+    'async',
+    'await',
+    'break',
+    'class',
+    'continue',
+    'def',
+    'del',
+    'elif',
+    'else',
+    'except',
+    'finally',
+    'for',
+    'from',
+    'global',
+    'if',
+    'import',
+    'in',
+    'is',
+    'lambda',
+    'nonlocal',
+    'not',
+    'or',
+    'pass',
+    'raise',
+    'return',
+    'try',
+    'while',
+    'with',
+    'yield',
+]);
+/**
+ * Normalizes a tool name to Python identifier format.
+ * Must match the Code API's `normalizePythonFunctionName` exactly:
+ * 1. Replace hyphens and spaces with underscores
+ * 2. Remove any other invalid characters
+ * 3. Prefix with underscore if starts with number
+ * 4. Append `_tool` if it's a Python keyword
+ * @param name - The tool name to normalize
+ * @returns Normalized Python-safe identifier
+ */
+function normalizeToPythonIdentifier(name) {
+    let normalized = name.replace(/[-\s]/g, '_');
+    normalized = normalized.replace(/[^a-zA-Z0-9_]/g, '');
+    if (/^[0-9]/.test(normalized)) {
+        normalized = '_' + normalized;
+    }
+    if (PYTHON_KEYWORDS.has(normalized)) {
+        normalized = normalized + '_tool';
+    }
+    return normalized;
+}
+/**
+ * Extracts tool names that are actually called in the Python code.
+ * Handles hyphen/underscore conversion since Python identifiers use underscores.
+ * @param code - The Python code to analyze
+ * @param toolNameMap - Map from normalized Python name to original tool name
+ * @returns Set of original tool names found in the code
+ */
+function extractUsedToolNames(code, toolNameMap) {
+    const usedTools = new Set();
+    for (const [pythonName, originalName] of toolNameMap) {
+        const escapedName = pythonName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+        const pattern = new RegExp(`\\b${escapedName}\\s*\\(`, 'g');
+        if (pattern.test(code)) {
+            usedTools.add(originalName);
+        }
+    }
+    return usedTools;
+}
+/**
+ * Filters tool definitions to only include tools actually used in the code.
+ * Handles the hyphen-to-underscore conversion for Python compatibility.
+ * @param toolDefs - All available tool definitions
+ * @param code - The Python code to analyze
+ * @param debug - Enable debug logging
+ * @returns Filtered array of tool definitions
+ */
+function filterToolsByUsage(toolDefs, code, debug = false) {
+    const toolNameMap = new Map();
+    for (const tool of toolDefs) {
+        const pythonName = normalizeToPythonIdentifier(tool.name);
+        toolNameMap.set(pythonName, tool.name);
+    }
+    const usedToolNames = extractUsedToolNames(code, toolNameMap);
+    if (debug) {
+        // eslint-disable-next-line no-console
+        console.log(`[PTC Debug] Tool filtering: found ${usedToolNames.size}/${toolDefs.length} tools in code`);
+        if (usedToolNames.size > 0) {
+            // eslint-disable-next-line no-console
+            console.log(`[PTC Debug] Matched tools: ${Array.from(usedToolNames).join(', ')}`);
+        }
+    }
+    if (usedToolNames.size === 0) {
+        if (debug) {
+            // eslint-disable-next-line no-console
+            console.log('[PTC Debug] No tools detected in code - sending all tools as fallback');
+        }
+        return toolDefs;
+    }
+    return toolDefs.filter((tool) => usedToolNames.has(tool.name));
+}
+/**
+ * Makes an HTTP request to the Code API.
+ * @param endpoint - The API endpoint URL
+ * @param apiKey - The API key for authentication
+ * @param body - The request body
+ * @param proxy - Optional HTTP proxy URL
+ * @returns The parsed API response
+ */
+async function makeRequest(endpoint, apiKey, body, proxy) {
+    const fetchOptions = {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'User-Agent': 'Illuma/1.0',
+            'X-API-Key': apiKey,
+        },
+        body: JSON.stringify(body),
+    };
+    if (proxy != null && proxy !== '') {
+        fetchOptions.agent = new HttpsProxyAgent(proxy);
+    }
+    const response = await fetch(endpoint, fetchOptions);
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`HTTP error! status: ${response.status}, body: ${errorText}`);
+    }
+    return (await response.json());
+}
+/**
+ * Executes tools in parallel when requested by the API.
+ * Uses Promise.all for parallel execution, catching individual errors.
+ * @param toolCalls - Array of tool calls from the API
+ * @param toolMap - Map of tool names to executable tools
+ * @returns Array of tool results
+ */
+async function executeTools(toolCalls, toolMap) {
+    const executions = toolCalls.map(async (call) => {
+        const tool = toolMap.get(call.name);
+        if (!tool) {
+            return {
+                call_id: call.id,
+                result: null,
+                is_error: true,
+                error_message: `Tool '${call.name}' not found. Available tools: ${Array.from(toolMap.keys()).join(', ')}`,
+            };
+        }
+        try {
+            const result = await tool.invoke(call.input, {
+                metadata: { [Constants.PROGRAMMATIC_TOOL_CALLING]: true },
+            });
+            return {
+                call_id: call.id,
+                result,
+                is_error: false,
+            };
+        }
+        catch (error) {
+            return {
+                call_id: call.id,
+                result: null,
+                is_error: true,
+                error_message: error.message || 'Tool execution failed',
+            };
+        }
+    });
+    return await Promise.all(executions);
+}
+/**
+ * Formats the completed response for the agent.
+ * @param response - The completed API response
+ * @returns Tuple of [formatted string, artifact]
+ */
+function formatCompletedResponse(response) {
+    let formatted = '';
+    if (response.stdout != null && response.stdout !== '') {
+        formatted += `stdout:\n${response.stdout}\n`;
+    }
+    else {
+        formatted += emptyOutputMessage;
+    }
+    if (response.stderr != null && response.stderr !== '') {
+        formatted += `stderr:\n${response.stderr}\n`;
+    }
+    if (response.files && response.files.length > 0) {
+        formatted += 'Generated files:\n';
+        const fileCount = response.files.length;
+        for (let i = 0; i < fileCount; i++) {
+            const file = response.files[i];
+            const isImage = imageExtRegex.test(file.name);
+            formatted += `- /mnt/data/${file.name} | ${isImage ? imageMessage : otherMessage}`;
+            if (i < fileCount - 1) {
+                formatted += fileCount <= 3 ? ', ' : ',\n';
+            }
+        }
+        formatted += `\nsession_id: ${response.session_id}\n\n${accessMessage}`;
+    }
+    return [
+        formatted.trim(),
+        {
+            session_id: response.session_id,
+            files: response.files,
+        },
+    ];
+}
+// ============================================================================
+// Tool Factory
+// ============================================================================
+/**
+ * Creates a Programmatic Tool Calling tool for complex multi-tool workflows.
+ *
+ * This tool enables AI agents to write Python code that orchestrates multiple
+ * tool calls programmatically, reducing LLM round-trips and token usage.
+ *
+ * The tool map must be provided at runtime via config.configurable.toolMap.
+ *
+ * @param params - Configuration parameters (apiKey, baseUrl, maxRoundTrips, proxy)
+ * @returns A LangChain DynamicStructuredTool for programmatic tool calling
+ *
+ * @example
+ * const ptcTool = createProgrammaticToolCallingTool({
+ *   apiKey: process.env.CODE_API_KEY,
+ *   maxRoundTrips: 20
+ * });
+ *
+ * const [output, artifact] = await ptcTool.invoke(
+ *   { code, tools },
+ *   { configurable: { toolMap } }
+ * );
+ */
+function createProgrammaticToolCallingTool(initParams = {}) {
+    const apiKey = initParams[EnvVar.CODE_API_KEY] ??
+        initParams.apiKey ??
+        getEnvironmentVariable(EnvVar.CODE_API_KEY) ??
+        '';
+    if (!apiKey) {
+        throw new Error('No API key provided for programmatic tool calling. ' +
+            'Set CODE_API_KEY environment variable or pass apiKey in initParams.');
+    }
+    const baseUrl = initParams.baseUrl ?? getCodeBaseURL();
+    const maxRoundTrips = initParams.maxRoundTrips ?? DEFAULT_MAX_ROUND_TRIPS;
+    const proxy = initParams.proxy ?? process.env.PROXY;
+    const debug = initParams.debug ?? process.env.PTC_DEBUG === 'true';
+    const EXEC_ENDPOINT = `${baseUrl}/exec/programmatic`;
+    const description = `
+Run tools by writing Python code. Tools are available as async functions - just call them with await.
+
+This is different from execute_code: here you can call your tools (like get_weather, get_expenses, etc.) directly in Python code.
+
+Usage:
+- Tools are pre-defined as async functions - call them with await
+- Use asyncio.gather() to run multiple tools in parallel
+- Only print() output is returned - tool results stay in Python
+
+Examples:
+- Simple: result = await get_weather(city="NYC")
+- Loop: for user in users: data = await get_expenses(user_id=user['id'])
+- Parallel: sf, ny = await asyncio.gather(get_weather(city="SF"), get_weather(city="NY"))
+
+When to use this instead of calling tools directly:
+- You need to call tools in a loop (process many items)
+- You want parallel execution (asyncio.gather)
+- You need conditionals based on tool results
+- You want to aggregate/filter data before returning
+`.trim();
+    return tool(async (params, config) => {
+        const { code, session_id, timeout = DEFAULT_TIMEOUT } = params;
+        // Extra params injected by ToolNode (follows web_search pattern)
+        const { toolMap, toolDefs } = (config.toolCall ?? {});
+        if (toolMap == null || toolMap.size === 0) {
+            throw new Error('No toolMap provided. ' +
+                'ToolNode should inject this from AgentContext when invoked through the graph.');
+        }
+        if (toolDefs == null || toolDefs.length === 0) {
+            throw new Error('No tool definitions provided. ' +
+                'Either pass tools in the input or ensure ToolNode injects toolDefs.');
+        }
+        let roundTrip = 0;
+        try {
+            // ====================================================================
+            // Phase 1: Filter tools and make initial request
+            // ====================================================================
+            const effectiveTools = filterToolsByUsage(toolDefs, code, debug);
+            if (debug) {
+                // eslint-disable-next-line no-console
+                console.log(`[PTC Debug] Sending ${effectiveTools.length} tools to API ` +
+                    `(filtered from ${toolDefs.length})`);
+            }
+            let response = await makeRequest(EXEC_ENDPOINT, apiKey, {
+                code,
+                tools: effectiveTools,
+                session_id,
+                timeout,
+            }, proxy);
+            // ====================================================================
+            // Phase 2: Handle response loop
+            // ====================================================================
+            while (response.status === 'tool_call_required') {
+                roundTrip++;
+                if (roundTrip > maxRoundTrips) {
+                    throw new Error(`Exceeded maximum round trips (${maxRoundTrips}). ` +
+                        'This may indicate an infinite loop, excessive tool calls, ' +
+                        'or a logic error in your code.');
+                }
+                if (debug) {
+                    // eslint-disable-next-line no-console
+                    console.log(`[PTC Debug] Round trip ${roundTrip}: ${response.tool_calls?.length ?? 0} tool(s) to execute`);
+                }
+                const toolResults = await executeTools(response.tool_calls ?? [], toolMap);
+                response = await makeRequest(EXEC_ENDPOINT, apiKey, {
+                    continuation_token: response.continuation_token,
+                    tool_results: toolResults,
+                }, proxy);
+            }
+            // ====================================================================
+            // Phase 3: Handle final state
+            // ====================================================================
+            if (response.status === 'completed') {
+                return formatCompletedResponse(response);
+            }
+            if (response.status === 'error') {
+                throw new Error(`Execution error: ${response.error}` +
+                    (response.stderr != null && response.stderr !== ''
+                        ? `\n\nStderr:\n${response.stderr}`
+                        : ''));
+            }
+            throw new Error(`Unexpected response status: ${response.status}`);
+        }
+        catch (error) {
+            throw new Error(`Programmatic execution failed: ${error.message}`);
+        }
+    }, {
+        name: Constants.PROGRAMMATIC_TOOL_CALLING,
+        description,
+        schema: ProgrammaticToolCallingSchema,
+        responseFormat: Constants.CONTENT_AND_ARTIFACT,
+    });
+}
+
+export { createProgrammaticToolCallingTool, executeTools, extractUsedToolNames, filterToolsByUsage, formatCompletedResponse, makeRequest, normalizeToPythonIdentifier };
+//# sourceMappingURL=ProgrammaticToolCalling.mjs.map

package/dist/esm/tools/ProgrammaticToolCalling.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProgrammaticToolCalling.mjs","sources":["../../../src/tools/ProgrammaticToolCalling.ts"],"sourcesContent":["// src/tools/ProgrammaticToolCalling.ts\nimport { z } from 'zod';\nimport { config } from 'dotenv';\nimport fetch, { RequestInit } from 'node-fetch';\nimport { HttpsProxyAgent } from 'https-proxy-agent';\nimport { getEnvironmentVariable } from '@langchain/core/utils/env';\nimport { tool, DynamicStructuredTool } from '@langchain/core/tools';\nimport type { ToolCall } from '@langchain/core/messages/tool';\nimport type * as t from '@/types';\nimport { imageExtRegex, getCodeBaseURL } from './CodeExecutor';\nimport { EnvVar, Constants } from '@/common';\n\nconfig();\n\n// ============================================================================\n// Constants\n// ============================================================================\n\nconst imageMessage = 'Image is already displayed to the user';\nconst otherMessage = 'File is already downloaded by the user';\nconst accessMessage =\n  'Note: Files are READ-ONLY. Save changes to NEW filenames. To access these files in future executions, provide the `session_id` as a parameter (not in your code).';\nconst emptyOutputMessage =\n  'stdout: Empty. Ensure you\\'re writing output explicitly.\\n';\n\n/** Default max round-trips to prevent infinite loops */\nconst DEFAULT_MAX_ROUND_TRIPS = 20;\n\n/** Default execution timeout in milliseconds */\nconst DEFAULT_TIMEOUT = 60000;\n\n// ============================================================================\n// Schema\n// ============================================================================\n\nconst ProgrammaticToolCallingSchema = z.object({\n  code: z\n    .string()\n    .min(1)\n    .describe(\n      `Python code that calls tools programmatically. Tools are automatically available as async Python functions - DO NOT define them yourself.\n\nThe Code API generates async function stubs from the tool definitions. Just call them directly:\n\nExample (Simple call):\n  result = await get_weather(city=\"San Francisco\")\n  print(result)\n\nExample (Parallel - Fastest):\n  results = await asyncio.gather(\n      get_weather(city=\"SF\"),\n      get_weather(city=\"NYC\"),\n      get_weather(city=\"London\")\n  )\n  for city, weather in zip([\"SF\", \"NYC\", \"London\"], results):\n      print(f\"{city}: {weather['temperature']}°F\")\n\nExample (Loop with processing):\n  team = await get_team_members()\n  for member in team:\n      expenses = await get_expenses(user_id=member['id'])\n      total = sum(e['amount'] for e in expenses)\n      print(f\"{member['name']}: \\${total:.2f}\")\n\nExample (Conditional logic):\n  data = await fetch_data(source=\"primary\")\n  if not data:\n      data = await fetch_data(source=\"backup\")\n  print(f\"Got {len(data)} records\")\n\nRequirements:\n- Tools are pre-defined as async functions - DO NOT write function definitions\n- Use await for all tool calls\n- Use asyncio.gather() for parallel execution of independent calls\n- Only print() output flows back to the context window\n- Tool results from programmatic calls do NOT consume context tokens`\n    ),\n  session_id: z\n    .string()\n    .optional()\n    .describe(\n      'Session ID for file access (same as regular code execution). Files load into /mnt/data/ and are READ-ONLY.'\n    ),\n  timeout: z\n    .number()\n    .int()\n    .min(1000)\n    .max(300000)\n    .optional()\n    .default(DEFAULT_TIMEOUT)\n    .describe(\n      'Maximum execution time in milliseconds. Default: 60 seconds. Max: 5 minutes.'\n    ),\n});\n\n// ============================================================================\n// Helper Functions\n// ============================================================================\n\n/** Python reserved keywords that get `_tool` suffix in Code API */\nconst PYTHON_KEYWORDS = new Set([\n  'False',\n  'None',\n  'True',\n  'and',\n  'as',\n  'assert',\n  'async',\n  'await',\n  'break',\n  'class',\n  'continue',\n  'def',\n  'del',\n  'elif',\n  'else',\n  'except',\n  'finally',\n  'for',\n  'from',\n  'global',\n  'if',\n  'import',\n  'in',\n  'is',\n  'lambda',\n  'nonlocal',\n  'not',\n  'or',\n  'pass',\n  'raise',\n  'return',\n  'try',\n  'while',\n  'with',\n  'yield',\n]);\n\n/**\n * Normalizes a tool name to Python identifier format.\n * Must match the Code API's `normalizePythonFunctionName` exactly:\n * 1. Replace hyphens and spaces with underscores\n * 2. Remove any other invalid characters\n * 3. Prefix with underscore if starts with number\n * 4. Append `_tool` if it's a Python keyword\n * @param name - The tool name to normalize\n * @returns Normalized Python-safe identifier\n */\nexport function normalizeToPythonIdentifier(name: string): string {\n  let normalized = name.replace(/[-\\s]/g, '_');\n\n  normalized = normalized.replace(/[^a-zA-Z0-9_]/g, '');\n\n  if (/^[0-9]/.test(normalized)) {\n    normalized = '_' + normalized;\n  }\n\n  if (PYTHON_KEYWORDS.has(normalized)) {\n    normalized = normalized + '_tool';\n  }\n\n  return normalized;\n}\n\n/**\n * Extracts tool names that are actually called in the Python code.\n * Handles hyphen/underscore conversion since Python identifiers use underscores.\n * @param code - The Python code to analyze\n * @param toolNameMap - Map from normalized Python name to original tool name\n * @returns Set of original tool names found in the code\n */\nexport function extractUsedToolNames(\n  code: string,\n  toolNameMap: Map<string, string>\n): Set<string> {\n  const usedTools = new Set<string>();\n\n  for (const [pythonName, originalName] of toolNameMap) {\n    const escapedName = pythonName.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n    const pattern = new RegExp(`\\\\b${escapedName}\\\\s*\\\\(`, 'g');\n\n    if (pattern.test(code)) {\n      usedTools.add(originalName);\n    }\n  }\n\n  return usedTools;\n}\n\n/**\n * Filters tool definitions to only include tools actually used in the code.\n * Handles the hyphen-to-underscore conversion for Python compatibility.\n * @param toolDefs - All available tool definitions\n * @param code - The Python code to analyze\n * @param debug - Enable debug logging\n * @returns Filtered array of tool definitions\n */\nexport function filterToolsByUsage(\n  toolDefs: t.LCTool[],\n  code: string,\n  debug = false\n): t.LCTool[] {\n  const toolNameMap = new Map<string, string>();\n  for (const tool of toolDefs) {\n    const pythonName = normalizeToPythonIdentifier(tool.name);\n    toolNameMap.set(pythonName, tool.name);\n  }\n\n  const usedToolNames = extractUsedToolNames(code, toolNameMap);\n\n  if (debug) {\n    // eslint-disable-next-line no-console\n    console.log(\n      `[PTC Debug] Tool filtering: found ${usedToolNames.size}/${toolDefs.length} tools in code`\n    );\n    if (usedToolNames.size > 0) {\n      // eslint-disable-next-line no-console\n      console.log(\n        `[PTC Debug] Matched tools: ${Array.from(usedToolNames).join(', ')}`\n      );\n    }\n  }\n\n  if (usedToolNames.size === 0) {\n    if (debug) {\n      // eslint-disable-next-line no-console\n      console.log(\n        '[PTC Debug] No tools detected in code - sending all tools as fallback'\n      );\n    }\n    return toolDefs;\n  }\n\n  return toolDefs.filter((tool) => usedToolNames.has(tool.name));\n}\n\n/**\n * Makes an HTTP request to the Code API.\n * @param endpoint - The API endpoint URL\n * @param apiKey - The API key for authentication\n * @param body - The request body\n * @param proxy - Optional HTTP proxy URL\n * @returns The parsed API response\n */\nexport async function makeRequest(\n  endpoint: string,\n  apiKey: string,\n  body: Record<string, unknown>,\n  proxy?: string\n): Promise<t.ProgrammaticExecutionResponse> {\n  const fetchOptions: RequestInit = {\n    method: 'POST',\n    headers: {\n      'Content-Type': 'application/json',\n      'User-Agent': 'Illuma/1.0',\n      'X-API-Key': apiKey,\n    },\n    body: JSON.stringify(body),\n  };\n\n  if (proxy != null && proxy !== '') {\n    fetchOptions.agent = new HttpsProxyAgent(proxy);\n  }\n\n  const response = await fetch(endpoint, fetchOptions);\n\n  if (!response.ok) {\n    const errorText = await response.text();\n    throw new Error(\n      `HTTP error! status: ${response.status}, body: ${errorText}`\n    );\n  }\n\n  return (await response.json()) as t.ProgrammaticExecutionResponse;\n}\n\n/**\n * Executes tools in parallel when requested by the API.\n * Uses Promise.all for parallel execution, catching individual errors.\n * @param toolCalls - Array of tool calls from the API\n * @param toolMap - Map of tool names to executable tools\n * @returns Array of tool results\n */\nexport async function executeTools(\n  toolCalls: t.PTCToolCall[],\n  toolMap: t.ToolMap\n): Promise<t.PTCToolResult[]> {\n  const executions = toolCalls.map(async (call): Promise<t.PTCToolResult> => {\n    const tool = toolMap.get(call.name);\n\n    if (!tool) {\n      return {\n        call_id: call.id,\n        result: null,\n        is_error: true,\n        error_message: `Tool '${call.name}' not found. Available tools: ${Array.from(toolMap.keys()).join(', ')}`,\n      };\n    }\n\n    try {\n      const result = await tool.invoke(call.input, {\n        metadata: { [Constants.PROGRAMMATIC_TOOL_CALLING]: true },\n      });\n      return {\n        call_id: call.id,\n        result,\n        is_error: false,\n      };\n    } catch (error) {\n      return {\n        call_id: call.id,\n        result: null,\n        is_error: true,\n        error_message: (error as Error).message || 'Tool execution failed',\n      };\n    }\n  });\n\n  return await Promise.all(executions);\n}\n\n/**\n * Formats the completed response for the agent.\n * @param response - The completed API response\n * @returns Tuple of [formatted string, artifact]\n */\nexport function formatCompletedResponse(\n  response: t.ProgrammaticExecutionResponse\n): [string, t.ProgrammaticExecutionArtifact] {\n  let formatted = '';\n\n  if (response.stdout != null && response.stdout !== '') {\n    formatted += `stdout:\\n${response.stdout}\\n`;\n  } else {\n    formatted += emptyOutputMessage;\n  }\n\n  if (response.stderr != null && response.stderr !== '') {\n    formatted += `stderr:\\n${response.stderr}\\n`;\n  }\n\n  if (response.files && response.files.length > 0) {\n    formatted += 'Generated files:\\n';\n\n    const fileCount = response.files.length;\n    for (let i = 0; i < fileCount; i++) {\n      const file = response.files[i];\n      const isImage = imageExtRegex.test(file.name);\n      formatted += `- /mnt/data/${file.name} | ${isImage ? imageMessage : otherMessage}`;\n\n      if (i < fileCount - 1) {\n        formatted += fileCount <= 3 ? ', ' : ',\\n';\n      }\n    }\n\n    formatted += `\\nsession_id: ${response.session_id}\\n\\n${accessMessage}`;\n  }\n\n  return [\n    formatted.trim(),\n    {\n      session_id: response.session_id,\n      files: response.files,\n    },\n  ];\n}\n\n// ============================================================================\n// Tool Factory\n// ============================================================================\n\n/**\n * Creates a Programmatic Tool Calling tool for complex multi-tool workflows.\n *\n * This tool enables AI agents to write Python code that orchestrates multiple\n * tool calls programmatically, reducing LLM round-trips and token usage.\n *\n * The tool map must be provided at runtime via config.configurable.toolMap.\n *\n * @param params - Configuration parameters (apiKey, baseUrl, maxRoundTrips, proxy)\n * @returns A LangChain DynamicStructuredTool for programmatic tool calling\n *\n * @example\n * const ptcTool = createProgrammaticToolCallingTool({\n *   apiKey: process.env.CODE_API_KEY,\n *   maxRoundTrips: 20\n * });\n *\n * const [output, artifact] = await ptcTool.invoke(\n *   { code, tools },\n *   { configurable: { toolMap } }\n * );\n */\nexport function createProgrammaticToolCallingTool(\n  initParams: t.ProgrammaticToolCallingParams = {}\n): DynamicStructuredTool<typeof ProgrammaticToolCallingSchema> {\n  const apiKey =\n    (initParams[EnvVar.CODE_API_KEY] as string | undefined) ??\n    initParams.apiKey ??\n    getEnvironmentVariable(EnvVar.CODE_API_KEY) ??\n    '';\n\n  if (!apiKey) {\n    throw new Error(\n      'No API key provided for programmatic tool calling. ' +\n        'Set CODE_API_KEY environment variable or pass apiKey in initParams.'\n    );\n  }\n\n  const baseUrl = initParams.baseUrl ?? getCodeBaseURL();\n  const maxRoundTrips = initParams.maxRoundTrips ?? DEFAULT_MAX_ROUND_TRIPS;\n  const proxy = initParams.proxy ?? process.env.PROXY;\n  const debug = initParams.debug ?? process.env.PTC_DEBUG === 'true';\n  const EXEC_ENDPOINT = `${baseUrl}/exec/programmatic`;\n\n  const description = `\nRun tools by writing Python code. Tools are available as async functions - just call them with await.\n\nThis is different from execute_code: here you can call your tools (like get_weather, get_expenses, etc.) directly in Python code.\n\nUsage:\n- Tools are pre-defined as async functions - call them with await\n- Use asyncio.gather() to run multiple tools in parallel\n- Only print() output is returned - tool results stay in Python\n\nExamples:\n- Simple: result = await get_weather(city=\"NYC\")\n- Loop: for user in users: data = await get_expenses(user_id=user['id'])\n- Parallel: sf, ny = await asyncio.gather(get_weather(city=\"SF\"), get_weather(city=\"NY\"))\n\nWhen to use this instead of calling tools directly:\n- You need to call tools in a loop (process many items)\n- You want parallel execution (asyncio.gather)\n- You need conditionals based on tool results\n- You want to aggregate/filter data before returning\n`.trim();\n\n  return tool<typeof ProgrammaticToolCallingSchema>(\n    async (params, config) => {\n      const { code, session_id, timeout = DEFAULT_TIMEOUT } = params;\n\n      // Extra params injected by ToolNode (follows web_search pattern)\n      const { toolMap, toolDefs } = (config.toolCall ?? {}) as ToolCall &\n        Partial<t.ProgrammaticCache>;\n\n      if (toolMap == null || toolMap.size === 0) {\n        throw new Error(\n          'No toolMap provided. ' +\n            'ToolNode should inject this from AgentContext when invoked through the graph.'\n        );\n      }\n\n      if (toolDefs == null || toolDefs.length === 0) {\n        throw new Error(\n          'No tool definitions provided. ' +\n            'Either pass tools in the input or ensure ToolNode injects toolDefs.'\n        );\n      }\n\n      let roundTrip = 0;\n\n      try {\n        // ====================================================================\n        // Phase 1: Filter tools and make initial request\n        // ====================================================================\n\n        const effectiveTools = filterToolsByUsage(toolDefs, code, debug);\n\n        if (debug) {\n          // eslint-disable-next-line no-console\n          console.log(\n            `[PTC Debug] Sending ${effectiveTools.length} tools to API ` +\n              `(filtered from ${toolDefs.length})`\n          );\n        }\n\n        let response = await makeRequest(\n          EXEC_ENDPOINT,\n          apiKey,\n          {\n            code,\n            tools: effectiveTools,\n            session_id,\n            timeout,\n          },\n          proxy\n        );\n\n        // ====================================================================\n        // Phase 2: Handle response loop\n        // ====================================================================\n\n        while (response.status === 'tool_call_required') {\n          roundTrip++;\n\n          if (roundTrip > maxRoundTrips) {\n            throw new Error(\n              `Exceeded maximum round trips (${maxRoundTrips}). ` +\n                'This may indicate an infinite loop, excessive tool calls, ' +\n                'or a logic error in your code.'\n            );\n          }\n\n          if (debug) {\n            // eslint-disable-next-line no-console\n            console.log(\n              `[PTC Debug] Round trip ${roundTrip}: ${response.tool_calls?.length ?? 0} tool(s) to execute`\n            );\n          }\n\n          const toolResults = await executeTools(\n            response.tool_calls ?? [],\n            toolMap\n          );\n\n          response = await makeRequest(\n            EXEC_ENDPOINT,\n            apiKey,\n            {\n              continuation_token: response.continuation_token,\n              tool_results: toolResults,\n            },\n            proxy\n          );\n        }\n\n        // ====================================================================\n        // Phase 3: Handle final state\n        // ====================================================================\n\n        if (response.status === 'completed') {\n          return formatCompletedResponse(response);\n        }\n\n        if (response.status === 'error') {\n          throw new Error(\n            `Execution error: ${response.error}` +\n              (response.stderr != null && response.stderr !== ''\n                ? `\\n\\nStderr:\\n${response.stderr}`\n                : '')\n          );\n        }\n\n        throw new Error(`Unexpected response status: ${response.status}`);\n      } catch (error) {\n        throw new Error(\n          `Programmatic execution failed: ${(error as Error).message}`\n        );\n      }\n    },\n    {\n      name: Constants.PROGRAMMATIC_TOOL_CALLING,\n      description,\n      schema: ProgrammaticToolCallingSchema,\n      responseFormat: Constants.CONTENT_AND_ARTIFACT,\n    }\n  );\n}\n"],"names":[],"mappings":";;;;;;;;;AAAA;AAYA,MAAM,EAAE;AAER;AACA;AACA;AAEA,MAAM,YAAY,GAAG,wCAAwC;AAC7D,MAAM,YAAY,GAAG,wCAAwC;AAC7D,MAAM,aAAa,GACjB,mKAAmK;AACrK,MAAM,kBAAkB,GACtB,4DAA4D;AAE9D;AACA,MAAM,uBAAuB,GAAG,EAAE;AAElC;AACA,MAAM,eAAe,GAAG,KAAK;AAE7B;AACA;AACA;AAEA,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;AAC7C,IAAA,IAAI,EAAE;AACH,SAAA,MAAM;SACN,GAAG,CAAC,CAAC;AACL,SAAA,QAAQ,CACP,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qEAmC+D,CAChE;AACH,IAAA,UAAU,EAAE;AACT,SAAA,MAAM;AACN,SAAA,QAAQ;SACR,QAAQ,CACP,4GAA4G,CAC7G;AACH,IAAA,OAAO,EAAE;AACN,SAAA,MAAM;AACN,SAAA,GAAG;SACH,GAAG,CAAC,IAAI;SACR,GAAG,CAAC,MAAM;AACV,SAAA,QAAQ;SACR,OAAO,CAAC,eAAe;SACvB,QAAQ,CACP,8EAA8E,CAC/E;AACJ,CAAA,CAAC;AAEF;AACA;AACA;AAEA;AACA,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;IAC9B,OAAO;IACP,MAAM;IACN,MAAM;IACN,KAAK;IACL,IAAI;IACJ,QAAQ;IACR,OAAO;IACP,OAAO;IACP,OAAO;IACP,OAAO;IACP,UAAU;IACV,KAAK;IACL,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,SAAS;IACT,KAAK;IACL,MAAM;IACN,QAAQ;IACR,IAAI;IACJ,QAAQ;IACR,IAAI;IACJ,IAAI;IACJ,QAAQ;IACR,UAAU;IACV,KAAK;IACL,IAAI;IACJ,MAAM;IACN,OAAO;IACP,QAAQ;IACR,KAAK;IACL,OAAO;IACP,MAAM;IACN,OAAO;AACR,CAAA,CAAC;AAEF;;;;;;;;;AASG;AACG,SAAU,2BAA2B,CAAC,IAAY,EAAA;IACtD,IAAI,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;IAE5C,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC;AAErD,IAAA,IAAI,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE;AAC7B,QAAA,UAAU,GAAG,GAAG,GAAG,UAAU;;AAG/B,IAAA,IAAI,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;AACnC,QAAA,UAAU,GAAG,UAAU,GAAG,OAAO;;AAGnC,IAAA,OAAO,UAAU;AACnB;AAEA;;;;;;AAMG;AACa,SAAA,oBAAoB,CAClC,IAAY,EACZ,WAAgC,EAAA;AAEhC,IAAA,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU;IAEnC,KAAK,MAAM,CAAC,UAAU,EAAE,YAAY,CAAC,IAAI,WAAW,EAAE;QACpD,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;QACrE,MAAM,OAAO,GAAG,IAAI,MAAM,CAAC,CAAM,GAAA,EAAA,WAAW,CAAS,OAAA,CAAA,EAAE,GAAG,CAAC;AAE3D,QAAA,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;AACtB,YAAA,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC;;;AAI/B,IAAA,OAAO,SAAS;AAClB;AAEA;;;;;;;AAOG;AACG,SAAU,kBAAkB,CAChC,QAAoB,EACpB,IAAY,EACZ,KAAK,GAAG,KAAK,EAAA;AAEb,IAAA,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB;AAC7C,IAAA,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;QAC3B,MAAM,UAAU,GAAG,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;QACzD,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC;;IAGxC,MAAM,aAAa,GAAG,oBAAoB,CAAC,IAAI,EAAE,WAAW,CAAC;IAE7D,IAAI,KAAK,EAAE;;AAET,QAAA,OAAO,CAAC,GAAG,CACT,CAAA,kCAAA,EAAqC,aAAa,CAAC,IAAI,CAAA,CAAA,EAAI,QAAQ,CAAC,MAAM,CAAA,cAAA,CAAgB,CAC3F;AACD,QAAA,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE;;AAE1B,YAAA,OAAO,CAAC,GAAG,CACT,CAA8B,2BAAA,EAAA,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,CAAE,CACrE;;;AAIL,IAAA,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE;QAC5B,IAAI,KAAK,EAAE;;AAET,YAAA,OAAO,CAAC,GAAG,CACT,uEAAuE,CACxE;;AAEH,QAAA,OAAO,QAAQ;;AAGjB,IAAA,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChE;AAEA;;;;;;;AAOG;AACI,eAAe,WAAW,CAC/B,QAAgB,EAChB,MAAc,EACd,IAA6B,EAC7B,KAAc,EAAA;AAEd,IAAA,MAAM,YAAY,GAAgB;AAChC,QAAA,MAAM,EAAE,MAAM;AACd,QAAA,OAAO,EAAE;AACP,YAAA,cAAc,EAAE,kBAAkB;AAClC,YAAA,YAAY,EAAE,YAAY;AAC1B,YAAA,WAAW,EAAE,MAAM;AACpB,SAAA;AACD,QAAA,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B;IAED,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,KAAK,EAAE,EAAE;QACjC,YAAY,CAAC,KAAK,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC;;IAGjD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,YAAY,CAAC;AAEpD,IAAA,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;AAChB,QAAA,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE;QACvC,MAAM,IAAI,KAAK,CACb,CAAuB,oBAAA,EAAA,QAAQ,CAAC,MAAM,CAAW,QAAA,EAAA,SAAS,CAAE,CAAA,CAC7D;;AAGH,IAAA,QAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE;AAC/B;AAEA;;;;;;AAMG;AACI,eAAe,YAAY,CAChC,SAA0B,EAC1B,OAAkB,EAAA;IAElB,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,IAAI,KAA8B;QACxE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;QAEnC,IAAI,CAAC,IAAI,EAAE;YACT,OAAO;gBACL,OAAO,EAAE,IAAI,CAAC,EAAE;AAChB,gBAAA,MAAM,EAAE,IAAI;AACZ,gBAAA,QAAQ,EAAE,IAAI;gBACd,aAAa,EAAE,SAAS,IAAI,CAAC,IAAI,CAAiC,8BAAA,EAAA,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAE,CAAA;aAC1G;;AAGH,QAAA,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC3C,QAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,yBAAyB,GAAG,IAAI,EAAE;AAC1D,aAAA,CAAC;YACF,OAAO;gBACL,OAAO,EAAE,IAAI,CAAC,EAAE;gBAChB,MAAM;AACN,gBAAA,QAAQ,EAAE,KAAK;aAChB;;QACD,OAAO,KAAK,EAAE;YACd,OAAO;gBACL,OAAO,EAAE,IAAI,CAAC,EAAE;AAChB,gBAAA,MAAM,EAAE,IAAI;AACZ,gBAAA,QAAQ,EAAE,IAAI;AACd,gBAAA,aAAa,EAAG,KAAe,CAAC,OAAO,IAAI,uBAAuB;aACnE;;AAEL,KAAC,CAAC;AAEF,IAAA,OAAO,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;AACtC;AAEA;;;;AAIG;AACG,SAAU,uBAAuB,CACrC,QAAyC,EAAA;IAEzC,IAAI,SAAS,GAAG,EAAE;AAElB,IAAA,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,EAAE,EAAE;AACrD,QAAA,SAAS,IAAI,CAAY,SAAA,EAAA,QAAQ,CAAC,MAAM,IAAI;;SACvC;QACL,SAAS,IAAI,kBAAkB;;AAGjC,IAAA,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,EAAE,EAAE;AACrD,QAAA,SAAS,IAAI,CAAY,SAAA,EAAA,QAAQ,CAAC,MAAM,IAAI;;AAG9C,IAAA,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;QAC/C,SAAS,IAAI,oBAAoB;AAEjC,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM;AACvC,QAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE;YAClC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;AAC7C,YAAA,SAAS,IAAI,CAAe,YAAA,EAAA,IAAI,CAAC,IAAI,MAAM,OAAO,GAAG,YAAY,GAAG,YAAY,EAAE;AAElF,YAAA,IAAI,CAAC,GAAG,SAAS,GAAG,CAAC,EAAE;AACrB,gBAAA,SAAS,IAAI,SAAS,IAAI,CAAC,GAAG,IAAI,GAAG,KAAK;;;QAI9C,SAAS,IAAI,iBAAiB,QAAQ,CAAC,UAAU,CAAO,IAAA,EAAA,aAAa,EAAE;;IAGzE,OAAO;QACL,SAAS,CAAC,IAAI,EAAE;AAChB,QAAA;YACE,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,KAAK,EAAE,QAAQ,CAAC,KAAK;AACtB,SAAA;KACF;AACH;AAEA;AACA;AACA;AAEA;;;;;;;;;;;;;;;;;;;;;AAqBG;AACa,SAAA,iCAAiC,CAC/C,UAAA,GAA8C,EAAE,EAAA;AAEhD,IAAA,MAAM,MAAM,GACT,UAAU,CAAC,MAAM,CAAC,YAAY,CAAwB;AACvD,QAAA,UAAU,CAAC,MAAM;AACjB,QAAA,sBAAsB,CAAC,MAAM,CAAC,YAAY,CAAC;AAC3C,QAAA,EAAE;IAEJ,IAAI,CAAC,MAAM,EAAE;QACX,MAAM,IAAI,KAAK,CACb,qDAAqD;AACnD,YAAA,qEAAqE,CACxE;;IAGH,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,IAAI,cAAc,EAAE;AACtD,IAAA,MAAM,aAAa,GAAG,UAAU,CAAC,aAAa,IAAI,uBAAuB;IACzE,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK;AACnD,IAAA,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,KAAK,MAAM;AAClE,IAAA,MAAM,aAAa,GAAG,CAAG,EAAA,OAAO,oBAAoB;AAEpD,IAAA,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;CAoBrB,CAAC,IAAI,EAAE;IAEN,OAAO,IAAI,CACT,OAAO,MAAM,EAAE,MAAM,KAAI;QACvB,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,GAAG,eAAe,EAAE,GAAG,MAAM;;AAG9D,QAAA,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,MAAM,CAAC,QAAQ,IAAI,EAAE,CACtB;QAE9B,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE;YACzC,MAAM,IAAI,KAAK,CACb,uBAAuB;AACrB,gBAAA,+EAA+E,CAClF;;QAGH,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7C,MAAM,IAAI,KAAK,CACb,gCAAgC;AAC9B,gBAAA,qEAAqE,CACxE;;QAGH,IAAI,SAAS,GAAG,CAAC;AAEjB,QAAA,IAAI;;;;YAKF,MAAM,cAAc,GAAG,kBAAkB,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC;YAEhE,IAAI,KAAK,EAAE;;AAET,gBAAA,OAAO,CAAC,GAAG,CACT,uBAAuB,cAAc,CAAC,MAAM,CAAgB,cAAA,CAAA;AAC1D,oBAAA,CAAA,eAAA,EAAkB,QAAQ,CAAC,MAAM,CAAA,CAAA,CAAG,CACvC;;YAGH,IAAI,QAAQ,GAAG,MAAM,WAAW,CAC9B,aAAa,EACb,MAAM,EACN;gBACE,IAAI;AACJ,gBAAA,KAAK,EAAE,cAAc;gBACrB,UAAU;gBACV,OAAO;aACR,EACD,KAAK,CACN;;;;AAMD,YAAA,OAAO,QAAQ,CAAC,MAAM,KAAK,oBAAoB,EAAE;AAC/C,gBAAA,SAAS,EAAE;AAEX,gBAAA,IAAI,SAAS,GAAG,aAAa,EAAE;AAC7B,oBAAA,MAAM,IAAI,KAAK,CACb,CAAA,8BAAA,EAAiC,aAAa,CAAK,GAAA,CAAA;wBACjD,4DAA4D;AAC5D,wBAAA,gCAAgC,CACnC;;gBAGH,IAAI,KAAK,EAAE;;AAET,oBAAA,OAAO,CAAC,GAAG,CACT,CAAA,uBAAA,EAA0B,SAAS,CAAK,EAAA,EAAA,QAAQ,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAA,mBAAA,CAAqB,CAC9F;;AAGH,gBAAA,MAAM,WAAW,GAAG,MAAM,YAAY,CACpC,QAAQ,CAAC,UAAU,IAAI,EAAE,EACzB,OAAO,CACR;AAED,gBAAA,QAAQ,GAAG,MAAM,WAAW,CAC1B,aAAa,EACb,MAAM,EACN;oBACE,kBAAkB,EAAE,QAAQ,CAAC,kBAAkB;AAC/C,oBAAA,YAAY,EAAE,WAAW;iBAC1B,EACD,KAAK,CACN;;;;;AAOH,YAAA,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,EAAE;AACnC,gBAAA,OAAO,uBAAuB,CAAC,QAAQ,CAAC;;AAG1C,YAAA,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE;AAC/B,gBAAA,MAAM,IAAI,KAAK,CACb,oBAAoB,QAAQ,CAAC,KAAK,CAAE,CAAA;qBACjC,QAAQ,CAAC,MAAM,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK;AAC9C,0BAAE,CAAA,aAAA,EAAgB,QAAQ,CAAC,MAAM,CAAE;AACnC,0BAAE,EAAE,CAAC,CACV;;YAGH,MAAM,IAAI,KAAK,CAAC,CAAA,4BAAA,EAA+B,QAAQ,CAAC,MAAM,CAAE,CAAA,CAAC;;QACjE,OAAO,KAAK,EAAE;YACd,MAAM,IAAI,KAAK,CACb,CAAA,+BAAA,EAAmC,KAAe,CAAC,OAAO,CAAE,CAAA,CAC7D;;AAEL,KAAC,EACD;QACE,IAAI,EAAE,SAAS,CAAC,yBAAyB;QACzC,WAAW;AACX,QAAA,MAAM,EAAE,6BAA6B;QACrC,cAAc,EAAE,SAAS,CAAC,oBAAoB;AAC/C,KAAA,CACF;AACH;;;;"}

package/dist/esm/tools/ToolNode.mjs

@@ -1,6 +1,6 @@
 import { isBaseMessage, ToolMessage, isAIMessage } from '@langchain/core/messages';
 import { isCommand, isGraphInterrupt, Command, END, Send } from '@langchain/langgraph';
-import '../common/enum.mjs';
+import { Constants } from '../common/enum.mjs';
 import 'nanoid';
 import '../messages/core.mjs';
 import { RunnableCallable } from '../utils/run.mjs';
@@ -14,7 +14,6 @@ function isSend(value) {
 }
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 class ToolNode extends RunnableCallable {
-    tools;
     toolMap;
     loadRuntimeTools;
     handleToolErrors = true;
@@ -22,15 +21,41 @@ class ToolNode extends RunnableCallable {
     toolCallStepIds;
     errorHandler;
     toolUsageCount;
-    constructor({ tools, toolMap, name, tags, errorHandler, toolCallStepIds, handleToolErrors, loadRuntimeTools, }) {
+    /** Tool registry for filtering (lazy computation of programmatic maps) */
+    toolRegistry;
+    /** Cached programmatic tools (computed once on first PTC call) */
+    programmaticCache;
+    constructor({ tools, toolMap, name, tags, errorHandler, toolCallStepIds, handleToolErrors, loadRuntimeTools, toolRegistry, }) {
         super({ name, tags, func: (input, config) => this.run(input, config) });
-        this.tools = tools;
         this.toolMap = toolMap ?? new Map(tools.map((tool) => [tool.name, tool]));
         this.toolCallStepIds = toolCallStepIds;
         this.handleToolErrors = handleToolErrors ?? this.handleToolErrors;
         this.loadRuntimeTools = loadRuntimeTools;
         this.errorHandler = errorHandler;
         this.toolUsageCount = new Map();
+        this.toolRegistry = toolRegistry;
+    }
+    /**
+     * Returns cached programmatic tools, computing once on first access.
+     * Single iteration builds both toolMap and toolDefs simultaneously.
+     */
+    getProgrammaticTools() {
+        if (this.programmaticCache)
+            return this.programmaticCache;
+        const toolMap = new Map();
+        const toolDefs = [];
+        if (this.toolRegistry) {
+            for (const [name, toolDef] of this.toolRegistry) {
+                if ((toolDef.allowed_callers ?? ['direct']).includes('code_execution')) {
+                    toolDefs.push(toolDef);
+                    const tool = this.toolMap.get(name);
+                    if (tool)
+                        toolMap.set(name, tool);
+                }
+            }
+        }
+        this.programmaticCache = { toolMap, toolDefs };
+        return this.programmaticCache;
     }
     /**
      * Returns a snapshot of the current tool usage counts.
@@ -52,17 +77,30 @@ class ToolNode extends RunnableCallable {
             this.toolUsageCount.set(call.name, turn + 1);
             const args = call.args;
             const stepId = this.toolCallStepIds?.get(call.id);
-            const output = await tool.invoke({ ...call, args, type: 'tool_call', stepId, turn }, config);
-            // Debug logging for image generation
-            if (call.name === 'image_generation') {
-                console.log('[ToolNode] image_generation output:', {
-                    isBaseMessage: isBaseMessage(output),
-                    messageType: isBaseMessage(output) ? output._getType() : 'not a message',
-                    isCommand: isCommand(output),
-                    hasArtifact: isBaseMessage(output) && output.artifact !== undefined,
-                    outputType: typeof output,
-                });
+            // Build invoke params - LangChain extracts non-schema fields to config.toolCall
+            let invokeParams = {
+                ...call,
+                args,
+                type: 'tool_call',
+                stepId,
+                turn,
+            };
+            // Inject runtime data for special tools (becomes available at config.toolCall)
+            if (call.name === Constants.PROGRAMMATIC_TOOL_CALLING) {
+                const { toolMap, toolDefs } = this.getProgrammaticTools();
+                invokeParams = {
+                    ...invokeParams,
+                    toolMap,
+                    toolDefs,
+                };
+            }
+            else if (call.name === Constants.TOOL_SEARCH_REGEX) {
+                invokeParams = {
+                    ...invokeParams,
+                    toolRegistry: this.toolRegistry,
+                };
             }
+            const output = await tool.invoke(invokeParams, config);
             if ((isBaseMessage(output) && output._getType() === 'tool') ||
                 isCommand(output)) {
                 return output;
@@ -158,9 +196,9 @@ class ToolNode extends RunnableCallable {
             }
             if (this.loadRuntimeTools) {
                 const { tools, toolMap } = this.loadRuntimeTools(aiMessage.tool_calls ?? []);
-                this.tools = tools;
                 this.toolMap =
                     toolMap ?? new Map(tools.map((tool) => [tool.name, tool]));
+                this.programmaticCache = undefined; // Invalidate cache on toolMap change
             }
             outputs = await Promise.all(aiMessage.tool_calls
                 ?.filter((call) => {