illuma-agents 1.0.10 → 1.0.11

package/dist/cjs/tools/ToolSearchRegex.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolSearchRegex.cjs","sources":["../../../src/tools/ToolSearchRegex.ts"],"sourcesContent":["// src/tools/ToolSearchRegex.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 * as t from '@/types';\nimport { getCodeBaseURL } from './CodeExecutor';\nimport { EnvVar, Constants } from '@/common';\n\nconfig();\n\n/** Maximum allowed regex pattern length */\nconst MAX_PATTERN_LENGTH = 200;\n\n/** Maximum allowed regex nesting depth */\nconst MAX_REGEX_COMPLEXITY = 5;\n\n/** Default search timeout in milliseconds */\nconst SEARCH_TIMEOUT = 5000;\n\nconst ToolSearchRegexSchema = z.object({\n  query: z\n    .string()\n    .min(1)\n    .max(MAX_PATTERN_LENGTH)\n    .describe(\n      'Regex pattern to search tool names and descriptions. Special regex characters will be sanitized for safety.'\n    ),\n  fields: z\n    .array(z.enum(['name', 'description', 'parameters']))\n    .optional()\n    .default(['name', 'description'])\n    .describe('Which fields to search. Default: name and description'),\n  max_results: z\n    .number()\n    .int()\n    .min(1)\n    .max(50)\n    .optional()\n    .default(10)\n    .describe('Maximum number of matching tools to return'),\n});\n\n/**\n * Escapes special regex characters in a string to use as a literal pattern.\n * @param pattern - The string to escape\n * @returns The escaped string safe for use in a RegExp\n */\nfunction escapeRegexSpecialChars(pattern: string): string {\n  return pattern.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n}\n\n/**\n * Counts the maximum nesting depth of groups in a regex pattern.\n * @param pattern - The regex pattern to analyze\n * @returns The maximum nesting depth\n */\nfunction countNestedGroups(pattern: string): number {\n  let maxDepth = 0;\n  let currentDepth = 0;\n\n  for (let i = 0; i < pattern.length; i++) {\n    if (pattern[i] === '(' && (i === 0 || pattern[i - 1] !== '\\\\')) {\n      currentDepth++;\n      maxDepth = Math.max(maxDepth, currentDepth);\n    } else if (pattern[i] === ')' && (i === 0 || pattern[i - 1] !== '\\\\')) {\n      currentDepth = Math.max(0, currentDepth - 1);\n    }\n  }\n\n  return maxDepth;\n}\n\n/**\n * Detects nested quantifiers that can cause catastrophic backtracking.\n * Patterns like (a+)+, (a*)*, (a+)*, etc.\n * @param pattern - The regex pattern to check\n * @returns True if nested quantifiers are detected\n */\nfunction hasNestedQuantifiers(pattern: string): boolean {\n  const nestedQuantifierPattern = /\\([^)]*[+*][^)]*\\)[+*?]/;\n  return nestedQuantifierPattern.test(pattern);\n}\n\n/**\n * Checks if a regex pattern contains potentially dangerous constructs.\n * @param pattern - The regex pattern to validate\n * @returns True if the pattern is dangerous\n */\nfunction isDangerousPattern(pattern: string): boolean {\n  if (hasNestedQuantifiers(pattern)) {\n    return true;\n  }\n\n  if (countNestedGroups(pattern) > MAX_REGEX_COMPLEXITY) {\n    return true;\n  }\n\n  const dangerousPatterns = [\n    /\\.\\{1000,\\}/, // Excessive wildcards\n    /\\(\\?=\\.\\{100,\\}\\)/, // Runaway lookaheads\n    /\\([^)]*\\|\\s*\\){20,}/, // Excessive alternation (rough check)\n    /\\(\\.\\*\\)\\+/, // (.*)+\n    /\\(\\.\\+\\)\\+/, // (.+)+\n    /\\(\\.\\*\\)\\*/, // (.*)*\n    /\\(\\.\\+\\)\\*/, // (.+)*\n  ];\n\n  for (const dangerous of dangerousPatterns) {\n    if (dangerous.test(pattern)) {\n      return true;\n    }\n  }\n\n  return false;\n}\n\n/**\n * Sanitizes a regex pattern for safe execution.\n * If the pattern is dangerous, it will be escaped to a literal string search.\n * @param pattern - The regex pattern to sanitize\n * @returns Object containing the safe pattern and whether it was escaped\n */\nfunction sanitizeRegex(pattern: string): { safe: string; wasEscaped: boolean } {\n  if (isDangerousPattern(pattern)) {\n    return {\n      safe: escapeRegexSpecialChars(pattern),\n      wasEscaped: true,\n    };\n  }\n\n  try {\n    new RegExp(pattern);\n    return { safe: pattern, wasEscaped: false };\n  } catch {\n    return {\n      safe: escapeRegexSpecialChars(pattern),\n      wasEscaped: true,\n    };\n  }\n}\n\n/**\n * Simplifies tool parameters for search purposes.\n * Extracts only the essential structure needed for parameter name searching.\n * @param parameters - The tool's JSON schema parameters\n * @returns Simplified parameters object\n */\nfunction simplifyParametersForSearch(\n  parameters?: t.JsonSchemaType\n): t.JsonSchemaType | undefined {\n  if (!parameters) {\n    return undefined;\n  }\n\n  if (parameters.properties) {\n    return {\n      type: parameters.type,\n      properties: Object.fromEntries(\n        Object.entries(parameters.properties).map(([key, value]) => [\n          key,\n          { type: (value as t.JsonSchemaType).type },\n        ])\n      ),\n    } as t.JsonSchemaType;\n  }\n\n  return { type: parameters.type };\n}\n\n/**\n * Generates the JavaScript search script to be executed in the sandbox.\n * Uses plain JavaScript for maximum compatibility with the Code API.\n * @param deferredTools - Array of tool metadata to search through\n * @param fields - Which fields to search\n * @param maxResults - Maximum number of results to return\n * @param sanitizedPattern - The sanitized regex pattern\n * @returns The JavaScript code string\n */\nfunction generateSearchScript(\n  deferredTools: t.ToolMetadata[],\n  fields: string[],\n  maxResults: number,\n  sanitizedPattern: string\n): string {\n  const lines = [\n    '// Tool definitions (injected)',\n    'var tools = ' + JSON.stringify(deferredTools) + ';',\n    'var searchFields = ' + JSON.stringify(fields) + ';',\n    'var maxResults = ' + maxResults + ';',\n    'var pattern = ' + JSON.stringify(sanitizedPattern) + ';',\n    '',\n    '// Compile regex (pattern is sanitized client-side)',\n    'var regex;',\n    'try {',\n    '  regex = new RegExp(pattern, \\'i\\');',\n    '} catch (e) {',\n    '  regex = new RegExp(pattern.replace(/[.*+?^${}()[\\\\]\\\\\\\\|]/g, \"\\\\\\\\$&\"), \"i\");',\n    '}',\n    '',\n    '// Search logic',\n    'var results = [];',\n    '',\n    'for (var j = 0; j < tools.length; j++) {',\n    '  var tool = tools[j];',\n    '  var bestScore = 0;',\n    '  var matchedField = \\'\\';',\n    '  var snippet = \\'\\';',\n    '',\n    '  // Search name (highest priority)',\n    '  if (searchFields.indexOf(\\'name\\') >= 0 && regex.test(tool.name)) {',\n    '    bestScore = 0.95;',\n    '    matchedField = \\'name\\';',\n    '    snippet = tool.name;',\n    '  }',\n    '',\n    '  // Search description (medium priority)',\n    '  if (searchFields.indexOf(\\'description\\') >= 0 && tool.description && regex.test(tool.description)) {',\n    '    if (bestScore === 0) {',\n    '      bestScore = 0.75;',\n    '      matchedField = \\'description\\';',\n    '      snippet = tool.description.substring(0, 100);',\n    '    }',\n    '  }',\n    '',\n    '  // Search parameter names (lower priority)',\n    '  if (searchFields.indexOf(\\'parameters\\') >= 0 && tool.parameters && tool.parameters.properties) {',\n    '    var paramNames = Object.keys(tool.parameters.properties).join(\\' \\');',\n    '    if (regex.test(paramNames)) {',\n    '      if (bestScore === 0) {',\n    '        bestScore = 0.60;',\n    '        matchedField = \\'parameters\\';',\n    '        snippet = paramNames;',\n    '      }',\n    '    }',\n    '  }',\n    '',\n    '  if (bestScore > 0) {',\n    '    results.push({',\n    '      tool_name: tool.name,',\n    '      match_score: bestScore,',\n    '      matched_field: matchedField,',\n    '      snippet: snippet',\n    '    });',\n    '  }',\n    '}',\n    '',\n    '// Sort by score (descending) and limit results',\n    'results.sort(function(a, b) { return b.match_score - a.match_score; });',\n    'var topResults = results.slice(0, maxResults);',\n    '',\n    '// Output as JSON',\n    'console.log(JSON.stringify({',\n    '  tool_references: topResults.map(function(r) {',\n    '    return {',\n    '      tool_name: r.tool_name,',\n    '      match_score: r.match_score,',\n    '      matched_field: r.matched_field,',\n    '      snippet: r.snippet',\n    '    };',\n    '  }),',\n    '  total_tools_searched: tools.length,',\n    '  pattern_used: pattern',\n    '}));',\n  ];\n  return lines.join('\\n');\n}\n\n/**\n * Parses the search results from stdout JSON.\n * @param stdout - The stdout string containing JSON results\n * @returns Parsed search response\n */\nfunction parseSearchResults(stdout: string): t.ToolSearchResponse {\n  const jsonMatch = stdout.trim();\n  const parsed = JSON.parse(jsonMatch) as t.ToolSearchResponse;\n  return parsed;\n}\n\n/**\n * Formats search results into a human-readable string.\n * @param searchResponse - The parsed search response\n * @returns Formatted string for LLM consumption\n */\nfunction formatSearchResults(searchResponse: t.ToolSearchResponse): string {\n  const { tool_references, total_tools_searched, pattern_used } =\n    searchResponse;\n\n  if (tool_references.length === 0) {\n    return `No tools matched the pattern \"${pattern_used}\".\\nTotal tools searched: ${total_tools_searched}`;\n  }\n\n  let response = `Found ${tool_references.length} matching tools:\\n\\n`;\n\n  for (const ref of tool_references) {\n    response += `- ${ref.tool_name} (score: ${ref.match_score.toFixed(2)})\\n`;\n    response += `  Matched in: ${ref.matched_field}\\n`;\n    response += `  Snippet: ${ref.snippet}\\n\\n`;\n  }\n\n  response += `Total tools searched: ${total_tools_searched}\\n`;\n  response += `Pattern used: ${pattern_used}`;\n\n  return response;\n}\n\n/**\n * Creates a Tool Search Regex tool for discovering tools from a large registry.\n *\n * This tool enables AI agents to dynamically discover tools from a large library\n * without loading all tool definitions into the LLM context window. The agent\n * can search for relevant tools on-demand using regex patterns.\n *\n * The tool registry can be provided either:\n * 1. At initialization time via params.toolRegistry\n * 2. At runtime via config.configurable.toolRegistry when invoking\n *\n * @param params - Configuration parameters for the tool (toolRegistry is optional)\n * @returns A LangChain DynamicStructuredTool for tool searching\n *\n * @example\n * // Option 1: Registry at initialization\n * const tool = createToolSearchRegexTool({ apiKey, toolRegistry });\n * await tool.invoke({ query: 'expense' });\n *\n * @example\n * // Option 2: Registry at runtime\n * const tool = createToolSearchRegexTool({ apiKey });\n * await tool.invoke(\n *   { query: 'expense' },\n *   { configurable: { toolRegistry, onlyDeferred: true } }\n * );\n */\nfunction createToolSearchRegexTool(\n  initParams: t.ToolSearchRegexParams = {}\n): DynamicStructuredTool<typeof ToolSearchRegexSchema> {\n  const apiKey: string =\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('No API key provided for tool search regex tool.');\n  }\n\n  const baseEndpoint = initParams.baseUrl ?? getCodeBaseURL();\n  const EXEC_ENDPOINT = `${baseEndpoint}/exec`;\n  const defaultOnlyDeferred = initParams.onlyDeferred ?? true;\n\n  const description = `\nSearches through available tools to find ones matching your query pattern.\n\nUsage:\n- Provide a regex pattern to search tool names and descriptions.\n- Use this when you need to discover tools for a specific task.\n- Results include tool names, match quality scores, and snippets showing where the match occurred.\n- Higher scores (0.9+) indicate name matches, medium scores (0.7+) indicate description matches.\n`.trim();\n\n  return tool<typeof ToolSearchRegexSchema>(\n    async (params, config) => {\n      const {\n        query,\n        fields = ['name', 'description'],\n        max_results = 10,\n      } = params;\n\n      // Extra params injected by ToolNode (follows web_search pattern)\n      const {\n        toolRegistry: paramToolRegistry,\n        onlyDeferred: paramOnlyDeferred,\n      } = config.toolCall ?? {};\n\n      const { safe: sanitizedPattern, wasEscaped } = sanitizeRegex(query);\n\n      let warningMessage = '';\n      if (wasEscaped) {\n        warningMessage =\n          'Note: The provided pattern was converted to a literal search for safety.\\n\\n';\n      }\n\n      // Priority: ToolNode injection (via config.toolCall) > initialization params\n      const toolRegistry = paramToolRegistry ?? initParams.toolRegistry;\n      const onlyDeferred =\n        paramOnlyDeferred !== undefined\n          ? paramOnlyDeferred\n          : defaultOnlyDeferred;\n\n      if (toolRegistry == null) {\n        return [\n          `${warningMessage}Error: No tool registry provided. Configure toolRegistry at agent level or initialization.`,\n          {\n            tool_references: [],\n            metadata: {\n              total_searched: 0,\n              pattern: sanitizedPattern,\n              error: 'No tool registry provided',\n            },\n          },\n        ];\n      }\n\n      const toolsArray: t.LCTool[] = Array.from(toolRegistry.values());\n\n      const deferredTools: t.ToolMetadata[] = toolsArray\n        .filter((lcTool) => {\n          if (onlyDeferred === true) {\n            return lcTool.defer_loading === true;\n          }\n          return true;\n        })\n        .map((lcTool) => ({\n          name: lcTool.name,\n          description: lcTool.description ?? '',\n          parameters: simplifyParametersForSearch(lcTool.parameters),\n        }));\n\n      if (deferredTools.length === 0) {\n        return [\n          `${warningMessage}No tools available to search. The tool registry is empty or no deferred tools are registered.`,\n          {\n            tool_references: [],\n            metadata: {\n              total_searched: 0,\n              pattern: sanitizedPattern,\n            },\n          },\n        ];\n      }\n\n      const searchScript = generateSearchScript(\n        deferredTools,\n        fields,\n        max_results,\n        sanitizedPattern\n      );\n\n      const postData = {\n        lang: 'js',\n        code: searchScript,\n        timeout: SEARCH_TIMEOUT,\n      };\n\n      try {\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(postData),\n        };\n\n        if (process.env.PROXY != null && process.env.PROXY !== '') {\n          fetchOptions.agent = new HttpsProxyAgent(process.env.PROXY);\n        }\n\n        const response = await fetch(EXEC_ENDPOINT, fetchOptions);\n        if (!response.ok) {\n          throw new Error(`HTTP error! status: ${response.status}`);\n        }\n\n        const result: t.ExecuteResult = await response.json();\n\n        if (result.stderr && result.stderr.trim()) {\n          // eslint-disable-next-line no-console\n          console.warn('[ToolSearchRegex] stderr:', result.stderr);\n        }\n\n        if (!result.stdout || !result.stdout.trim()) {\n          return [\n            `${warningMessage}No tools matched the pattern \"${sanitizedPattern}\".\\nTotal tools searched: ${deferredTools.length}`,\n            {\n              tool_references: [],\n              metadata: {\n                total_searched: deferredTools.length,\n                pattern: sanitizedPattern,\n              },\n            },\n          ];\n        }\n\n        const searchResponse = parseSearchResults(result.stdout);\n        const formattedOutput = `${warningMessage}${formatSearchResults(searchResponse)}`;\n\n        return [\n          formattedOutput,\n          {\n            tool_references: searchResponse.tool_references,\n            metadata: {\n              total_searched: searchResponse.total_tools_searched,\n              pattern: searchResponse.pattern_used,\n            },\n          },\n        ];\n      } catch (error) {\n        // eslint-disable-next-line no-console\n        console.error('[ToolSearchRegex] Error:', error);\n\n        const errorMessage =\n          error instanceof Error ? error.message : String(error);\n        return [\n          `Tool search failed: ${errorMessage}\\n\\nSuggestion: Try a simpler search pattern or search for specific tool names.`,\n          {\n            tool_references: [],\n            metadata: {\n              total_searched: 0,\n              pattern: sanitizedPattern,\n              error: errorMessage,\n            },\n          },\n        ];\n      }\n    },\n    {\n      name: Constants.TOOL_SEARCH_REGEX,\n      description,\n      schema: ToolSearchRegexSchema,\n      responseFormat: Constants.CONTENT_AND_ARTIFACT,\n    }\n  );\n}\n\nexport {\n  createToolSearchRegexTool,\n  sanitizeRegex,\n  escapeRegexSpecialChars,\n  isDangerousPattern,\n  countNestedGroups,\n  hasNestedQuantifiers,\n};\n"],"names":["config","z","EnvVar","getEnvironmentVariable","getCodeBaseURL","tool","HttpsProxyAgent","Constants"],"mappings":";;;;;;;;;;;AAAA;AAWAA,aAAM,EAAE;AAER;AACA,MAAM,kBAAkB,GAAG,GAAG;AAE9B;AACA,MAAM,oBAAoB,GAAG,CAAC;AAE9B;AACA,MAAM,cAAc,GAAG,IAAI;AAE3B,MAAM,qBAAqB,GAAGC,KAAC,CAAC,MAAM,CAAC;AACrC,IAAA,KAAK,EAAEA;AACJ,SAAA,MAAM;SACN,GAAG,CAAC,CAAC;SACL,GAAG,CAAC,kBAAkB;SACtB,QAAQ,CACP,6GAA6G,CAC9G;AACH,IAAA,MAAM,EAAEA;AACL,SAAA,KAAK,CAACA,KAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;AACnD,SAAA,QAAQ;AACR,SAAA,OAAO,CAAC,CAAC,MAAM,EAAE,aAAa,CAAC;SAC/B,QAAQ,CAAC,uDAAuD,CAAC;AACpE,IAAA,WAAW,EAAEA;AACV,SAAA,MAAM;AACN,SAAA,GAAG;SACH,GAAG,CAAC,CAAC;SACL,GAAG,CAAC,EAAE;AACN,SAAA,QAAQ;SACR,OAAO,CAAC,EAAE;SACV,QAAQ,CAAC,4CAA4C,CAAC;AAC1D,CAAA,CAAC;AAEF;;;;AAIG;AACH,SAAS,uBAAuB,CAAC,OAAe,EAAA;IAC9C,OAAO,OAAO,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;AACvD;AAEA;;;;AAIG;AACH,SAAS,iBAAiB,CAAC,OAAe,EAAA;IACxC,IAAI,QAAQ,GAAG,CAAC;IAChB,IAAI,YAAY,GAAG,CAAC;AAEpB,IAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACvC,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE;AAC9D,YAAA,YAAY,EAAE;YACd,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,YAAY,CAAC;;aACtC,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE;YACrE,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,GAAG,CAAC,CAAC;;;AAIhD,IAAA,OAAO,QAAQ;AACjB;AAEA;;;;;AAKG;AACH,SAAS,oBAAoB,CAAC,OAAe,EAAA;IAC3C,MAAM,uBAAuB,GAAG,yBAAyB;AACzD,IAAA,OAAO,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC;AAC9C;AAEA;;;;AAIG;AACH,SAAS,kBAAkB,CAAC,OAAe,EAAA;AACzC,IAAA,IAAI,oBAAoB,CAAC,OAAO,CAAC,EAAE;AACjC,QAAA,OAAO,IAAI;;AAGb,IAAA,IAAI,iBAAiB,CAAC,OAAO,CAAC,GAAG,oBAAoB,EAAE;AACrD,QAAA,OAAO,IAAI;;AAGb,IAAA,MAAM,iBAAiB,GAAG;AACxB,QAAA,aAAa;AACb,QAAA,mBAAmB;AACnB,QAAA,qBAAqB;AACrB,QAAA,YAAY;AACZ,QAAA,YAAY;AACZ,QAAA,YAAY;AACZ,QAAA,YAAY;KACb;AAED,IAAA,KAAK,MAAM,SAAS,IAAI,iBAAiB,EAAE;AACzC,QAAA,IAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;AAC3B,YAAA,OAAO,IAAI;;;AAIf,IAAA,OAAO,KAAK;AACd;AAEA;;;;;AAKG;AACH,SAAS,aAAa,CAAC,OAAe,EAAA;AACpC,IAAA,IAAI,kBAAkB,CAAC,OAAO,CAAC,EAAE;QAC/B,OAAO;AACL,YAAA,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC;AACtC,YAAA,UAAU,EAAE,IAAI;SACjB;;AAGH,IAAA,IAAI;AACF,QAAA,IAAI,MAAM,CAAC,OAAO,CAAC;QACnB,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE;;AAC3C,IAAA,MAAM;QACN,OAAO;AACL,YAAA,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC;AACtC,YAAA,UAAU,EAAE,IAAI;SACjB;;AAEL;AAEA;;;;;AAKG;AACH,SAAS,2BAA2B,CAClC,UAA6B,EAAA;IAE7B,IAAI,CAAC,UAAU,EAAE;AACf,QAAA,OAAO,SAAS;;AAGlB,IAAA,IAAI,UAAU,CAAC,UAAU,EAAE;QACzB,OAAO;YACL,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,UAAU,EAAE,MAAM,CAAC,WAAW,CAC5B,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK;gBAC1D,GAAG;AACH,gBAAA,EAAE,IAAI,EAAG,KAA0B,CAAC,IAAI,EAAE;AAC3C,aAAA,CAAC,CACH;SACkB;;AAGvB,IAAA,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE;AAClC;AAEA;;;;;;;;AAQG;AACH,SAAS,oBAAoB,CAC3B,aAA+B,EAC/B,MAAgB,EAChB,UAAkB,EAClB,gBAAwB,EAAA;AAExB,IAAA,MAAM,KAAK,GAAG;QACZ,gCAAgC;QAChC,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,GAAG;QACpD,qBAAqB,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,GAAG;QACpD,mBAAmB,GAAG,UAAU,GAAG,GAAG;QACtC,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,GAAG,GAAG;QACzD,EAAE;QACF,qDAAqD;QACrD,YAAY;QACZ,OAAO;QACP,uCAAuC;QACvC,eAAe;QACf,iFAAiF;QACjF,GAAG;QACH,EAAE;QACF,iBAAiB;QACjB,mBAAmB;QACnB,EAAE;QACF,0CAA0C;QAC1C,wBAAwB;QACxB,sBAAsB;QACtB,4BAA4B;QAC5B,uBAAuB;QACvB,EAAE;QACF,qCAAqC;QACrC,uEAAuE;QACvE,uBAAuB;QACvB,8BAA8B;QAC9B,0BAA0B;QAC1B,KAAK;QACL,EAAE;QACF,2CAA2C;QAC3C,yGAAyG;QACzG,4BAA4B;QAC5B,yBAAyB;QACzB,uCAAuC;QACvC,qDAAqD;QACrD,OAAO;QACP,KAAK;QACL,EAAE;QACF,8CAA8C;QAC9C,qGAAqG;QACrG,2EAA2E;QAC3E,mCAAmC;QACnC,8BAA8B;QAC9B,2BAA2B;QAC3B,wCAAwC;QACxC,+BAA+B;QAC/B,SAAS;QACT,OAAO;QACP,KAAK;QACL,EAAE;QACF,wBAAwB;QACxB,oBAAoB;QACpB,6BAA6B;QAC7B,+BAA+B;QAC/B,oCAAoC;QACpC,wBAAwB;QACxB,SAAS;QACT,KAAK;QACL,GAAG;QACH,EAAE;QACF,iDAAiD;QACjD,yEAAyE;QACzE,gDAAgD;QAChD,EAAE;QACF,mBAAmB;QACnB,8BAA8B;QAC9B,iDAAiD;QACjD,cAAc;QACd,+BAA+B;QAC/B,mCAAmC;QACnC,uCAAuC;QACvC,0BAA0B;QAC1B,QAAQ;QACR,OAAO;QACP,uCAAuC;QACvC,yBAAyB;QACzB,MAAM;KACP;AACD,IAAA,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;AACzB;AAEA;;;;AAIG;AACH,SAAS,kBAAkB,CAAC,MAAc,EAAA;AACxC,IAAA,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,EAAE;IAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAyB;AAC5D,IAAA,OAAO,MAAM;AACf;AAEA;;;;AAIG;AACH,SAAS,mBAAmB,CAAC,cAAoC,EAAA;IAC/D,MAAM,EAAE,eAAe,EAAE,oBAAoB,EAAE,YAAY,EAAE,GAC3D,cAAc;AAEhB,IAAA,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;AAChC,QAAA,OAAO,CAAiC,8BAAA,EAAA,YAAY,CAA6B,0BAAA,EAAA,oBAAoB,EAAE;;AAGzG,IAAA,IAAI,QAAQ,GAAG,CAAA,MAAA,EAAS,eAAe,CAAC,MAAM,sBAAsB;AAEpE,IAAA,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE;AACjC,QAAA,QAAQ,IAAI,CAAA,EAAA,EAAK,GAAG,CAAC,SAAS,CAAY,SAAA,EAAA,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK;AACzE,QAAA,QAAQ,IAAI,CAAiB,cAAA,EAAA,GAAG,CAAC,aAAa,IAAI;AAClD,QAAA,QAAQ,IAAI,CAAc,WAAA,EAAA,GAAG,CAAC,OAAO,MAAM;;AAG7C,IAAA,QAAQ,IAAI,CAAA,sBAAA,EAAyB,oBAAoB,CAAA,EAAA,CAAI;AAC7D,IAAA,QAAQ,IAAI,CAAA,cAAA,EAAiB,YAAY,CAAA,CAAE;AAE3C,IAAA,OAAO,QAAQ;AACjB;AAEA;;;;;;;;;;;;;;;;;;;;;;;;;;AA0BG;AACH,SAAS,yBAAyB,CAChC,UAAA,GAAsC,EAAE,EAAA;AAExC,IAAA,MAAM,MAAM,GACT,UAAU,CAACC,YAAM,CAAC,YAAY,CAAwB;AACvD,QAAA,UAAU,CAAC,MAAM;AACjB,QAAAC,0BAAsB,CAACD,YAAM,CAAC,YAAY,CAAC;AAC3C,QAAA,EAAE;IAEJ,IAAI,CAAC,MAAM,EAAE;AACX,QAAA,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC;;IAGpE,MAAM,YAAY,GAAG,UAAU,CAAC,OAAO,IAAIE,2BAAc,EAAE;AAC3D,IAAA,MAAM,aAAa,GAAG,CAAG,EAAA,YAAY,OAAO;AAC5C,IAAA,MAAM,mBAAmB,GAAG,UAAU,CAAC,YAAY,IAAI,IAAI;AAE3D,IAAA,MAAM,WAAW,GAAG;;;;;;;;CAQrB,CAAC,IAAI,EAAE;IAEN,OAAOC,UAAI,CACT,OAAO,MAAM,EAAE,MAAM,KAAI;AACvB,QAAA,MAAM,EACJ,KAAK,EACL,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,EAChC,WAAW,GAAG,EAAE,GACjB,GAAG,MAAM;;AAGV,QAAA,MAAM,EACJ,YAAY,EAAE,iBAAiB,EAC/B,YAAY,EAAE,iBAAiB,GAChC,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE;AAEzB,QAAA,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC,KAAK,CAAC;QAEnE,IAAI,cAAc,GAAG,EAAE;QACvB,IAAI,UAAU,EAAE;YACd,cAAc;AACZ,gBAAA,8EAA8E;;;AAIlF,QAAA,MAAM,YAAY,GAAG,iBAAiB,IAAI,UAAU,CAAC,YAAY;AACjE,QAAA,MAAM,YAAY,GAChB,iBAAiB,KAAK;AACpB,cAAE;cACA,mBAAmB;AAEzB,QAAA,IAAI,YAAY,IAAI,IAAI,EAAE;YACxB,OAAO;AACL,gBAAA,CAAA,EAAG,cAAc,CAA4F,0FAAA,CAAA;AAC7G,gBAAA;AACE,oBAAA,eAAe,EAAE,EAAE;AACnB,oBAAA,QAAQ,EAAE;AACR,wBAAA,cAAc,EAAE,CAAC;AACjB,wBAAA,OAAO,EAAE,gBAAgB;AACzB,wBAAA,KAAK,EAAE,2BAA2B;AACnC,qBAAA;AACF,iBAAA;aACF;;QAGH,MAAM,UAAU,GAAe,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;QAEhE,MAAM,aAAa,GAAqB;AACrC,aAAA,MAAM,CAAC,CAAC,MAAM,KAAI;AACjB,YAAA,IAAI,YAAY,KAAK,IAAI,EAAE;AACzB,gBAAA,OAAO,MAAM,CAAC,aAAa,KAAK,IAAI;;AAEtC,YAAA,OAAO,IAAI;AACb,SAAC;AACA,aAAA,GAAG,CAAC,CAAC,MAAM,MAAM;YAChB,IAAI,EAAE,MAAM,CAAC,IAAI;AACjB,YAAA,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE;AACrC,YAAA,UAAU,EAAE,2BAA2B,CAAC,MAAM,CAAC,UAAU,CAAC;AAC3D,SAAA,CAAC,CAAC;AAEL,QAAA,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE;YAC9B,OAAO;AACL,gBAAA,CAAA,EAAG,cAAc,CAA+F,6FAAA,CAAA;AAChH,gBAAA;AACE,oBAAA,eAAe,EAAE,EAAE;AACnB,oBAAA,QAAQ,EAAE;AACR,wBAAA,cAAc,EAAE,CAAC;AACjB,wBAAA,OAAO,EAAE,gBAAgB;AAC1B,qBAAA;AACF,iBAAA;aACF;;AAGH,QAAA,MAAM,YAAY,GAAG,oBAAoB,CACvC,aAAa,EACb,MAAM,EACN,WAAW,EACX,gBAAgB,CACjB;AAED,QAAA,MAAM,QAAQ,GAAG;AACf,YAAA,IAAI,EAAE,IAAI;AACV,YAAA,IAAI,EAAE,YAAY;AAClB,YAAA,OAAO,EAAE,cAAc;SACxB;AAED,QAAA,IAAI;AACF,YAAA,MAAM,YAAY,GAAgB;AAChC,gBAAA,MAAM,EAAE,MAAM;AACd,gBAAA,OAAO,EAAE;AACP,oBAAA,cAAc,EAAE,kBAAkB;AAClC,oBAAA,YAAY,EAAE,YAAY;AAC1B,oBAAA,WAAW,EAAE,MAAM;AACpB,iBAAA;AACD,gBAAA,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;aAC/B;AAED,YAAA,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,EAAE,EAAE;AACzD,gBAAA,YAAY,CAAC,KAAK,GAAG,IAAIC,+BAAe,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;;YAG7D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,aAAa,EAAE,YAAY,CAAC;AACzD,YAAA,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;gBAChB,MAAM,IAAI,KAAK,CAAC,CAAA,oBAAA,EAAuB,QAAQ,CAAC,MAAM,CAAE,CAAA,CAAC;;AAG3D,YAAA,MAAM,MAAM,GAAoB,MAAM,QAAQ,CAAC,IAAI,EAAE;YAErD,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE;;gBAEzC,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE,MAAM,CAAC,MAAM,CAAC;;AAG1D,YAAA,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE;gBAC3C,OAAO;AACL,oBAAA,CAAA,EAAG,cAAc,CAAiC,8BAAA,EAAA,gBAAgB,6BAA6B,aAAa,CAAC,MAAM,CAAE,CAAA;AACrH,oBAAA;AACE,wBAAA,eAAe,EAAE,EAAE;AACnB,wBAAA,QAAQ,EAAE;4BACR,cAAc,EAAE,aAAa,CAAC,MAAM;AACpC,4BAAA,OAAO,EAAE,gBAAgB;AAC1B,yBAAA;AACF,qBAAA;iBACF;;YAGH,MAAM,cAAc,GAAG,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC;YACxD,MAAM,eAAe,GAAG,CAAA,EAAG,cAAc,CAAA,EAAG,mBAAmB,CAAC,cAAc,CAAC,CAAA,CAAE;YAEjF,OAAO;gBACL,eAAe;AACf,gBAAA;oBACE,eAAe,EAAE,cAAc,CAAC,eAAe;AAC/C,oBAAA,QAAQ,EAAE;wBACR,cAAc,EAAE,cAAc,CAAC,oBAAoB;wBACnD,OAAO,EAAE,cAAc,CAAC,YAAY;AACrC,qBAAA;AACF,iBAAA;aACF;;QACD,OAAO,KAAK,EAAE;;AAEd,YAAA,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC;AAEhD,YAAA,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YACxD,OAAO;AACL,gBAAA,CAAA,oBAAA,EAAuB,YAAY,CAAiF,+EAAA,CAAA;AACpH,gBAAA;AACE,oBAAA,eAAe,EAAE,EAAE;AACnB,oBAAA,QAAQ,EAAE;AACR,wBAAA,cAAc,EAAE,CAAC;AACjB,wBAAA,OAAO,EAAE,gBAAgB;AACzB,wBAAA,KAAK,EAAE,YAAY;AACpB,qBAAA;AACF,iBAAA;aACF;;AAEL,KAAC,EACD;QACE,IAAI,EAAEC,eAAS,CAAC,iBAAiB;QACjC,WAAW;AACX,QAAA,MAAM,EAAE,qBAAqB;QAC7B,cAAc,EAAEA,eAAS,CAAC,oBAAoB;AAC/C,KAAA,CACF;AACH;;;;;;;;;"}

package/dist/cjs/tools/search/schema.cjs

@@ -11,17 +11,10 @@ exports.DATE_RANGE = void 0;
     DATE_RANGE["PAST_YEAR"] = "y";
 })(exports.DATE_RANGE || (exports.DATE_RANGE = {}));
 const DEFAULT_QUERY_DESCRIPTION = `
-Search query OR a URL for direct content extraction.
-
-**IMPORTANT - URL REQUESTS:**
-If the user provides a URL to read, summarize, or analyze, you MUST include the complete URL in this field.
-- User says: "Summarize https://example.com/article" → query: "https://example.com/article"
-- User says: "What does this page say? https://news.site/story" → query: "https://news.site/story"
-DO NOT convert URLs into search keywords. Pass the URL directly.
-
-**SEARCH MODE GUIDELINES:**
+GUIDELINES:
 - Start broad, then narrow: Begin with key concepts, then refine with specifics
 - Think like sources: Use terminology experts would use in the field
+- Consider perspective: Frame queries from different viewpoints for better results
 - Quality over quantity: A precise 3-4 word query often beats lengthy sentences
 
 TECHNIQUES (combine for power searches):
@@ -30,6 +23,11 @@ TECHNIQUES (combine for power searches):
 - SITE-SPECIFIC: Restrict to websites (site:edu research)
 - FILETYPE: Find specific documents (filetype:pdf study)
 - OR OPERATOR: Find alternatives (electric OR hybrid cars)
+- DATE RANGE: Recent information (data after:2020)
+- WILDCARDS: Use * for unknown terms (how to * bread)
+- SPECIFIC QUESTIONS: Use who/what/when/where/why/how
+- DOMAIN TERMS: Include technical terminology for specialized topics
+- CONCISE TERMS: Prioritize keywords over sentences
 `.trim();
 const DEFAULT_COUNTRY_DESCRIPTION = `Country code to localize search results.
 Use standard 2-letter country codes: "us", "uk", "ca", "de", "fr", "jp", "br", etc.

package/dist/cjs/tools/search/schema.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"schema.cjs","sources":["../../../../src/tools/search/schema.ts"],"sourcesContent":["import { z } from 'zod';\n\nexport enum DATE_RANGE {\n  PAST_HOUR = 'h',\n  PAST_24_HOURS = 'd',\n  PAST_WEEK = 'w',\n  PAST_MONTH = 'm',\n  PAST_YEAR = 'y',\n}\n\nexport const DEFAULT_QUERY_DESCRIPTION = `\nSearch query OR a URL for direct content extraction.\n\n**IMPORTANT - URL REQUESTS:**\nIf the user provides a URL to read, summarize, or analyze, you MUST include the complete URL in this field.\n- User says: \"Summarize https://example.com/article\" → query: \"https://example.com/article\"\n- User says: \"What does this page say? https://news.site/story\" → query: \"https://news.site/story\"\nDO NOT convert URLs into search keywords. Pass the URL directly.\n\n**SEARCH MODE GUIDELINES:**\n- Start broad, then narrow: Begin with key concepts, then refine with specifics\n- Think like sources: Use terminology experts would use in the field\n- Quality over quantity: A precise 3-4 word query often beats lengthy sentences\n\nTECHNIQUES (combine for power searches):\n- EXACT PHRASES: Use quotes (\"climate change report\")\n- EXCLUDE TERMS: Use minus to remove unwanted results (-wikipedia)\n- SITE-SPECIFIC: Restrict to websites (site:edu research)\n- FILETYPE: Find specific documents (filetype:pdf study)\n- OR OPERATOR: Find alternatives (electric OR hybrid cars)\n`.trim();\n\nexport const DEFAULT_COUNTRY_DESCRIPTION =\n  `Country code to localize search results.\nUse standard 2-letter country codes: \"us\", \"uk\", \"ca\", \"de\", \"fr\", \"jp\", \"br\", etc.\nProvide this when the search should return results specific to a particular country.\nExamples:\n- \"us\" for United States (default)\n- \"de\" for Germany\n- \"in\" for India\n`.trim();\n\nexport const querySchema = z.string().describe(DEFAULT_QUERY_DESCRIPTION);\nexport const dateSchema = z\n  .nativeEnum(DATE_RANGE)\n  .optional()\n  .describe('Date range for search results.');\nexport const countrySchema = z\n  .string()\n  .optional()\n  .describe(DEFAULT_COUNTRY_DESCRIPTION);\nexport const imagesSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run an image search.');\n\nexport const videosSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run a video search.');\n\nexport const newsSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run a news search.');\n"],"names":["DATE_RANGE","z"],"mappings":";;;;AAEYA;AAAZ,CAAA,UAAY,UAAU,EAAA;AACpB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,eAAA,CAAA,GAAA,GAAmB;AACnB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,YAAA,CAAA,GAAA,GAAgB;AAChB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACjB,CAAC,EANWA,kBAAU,KAAVA,kBAAU,GAMrB,EAAA,CAAA,CAAA;AAEY,MAAA,yBAAyB,GAAG;;;;;;;;;;;;;;;;;;;;CAoBxC,CAAC,IAAI;AAEO,MAAA,2BAA2B,GACtC,CAAA;;;;;;;CAOD,CAAC,IAAI;AAEC,MAAM,WAAW,GAAGC,KAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB;AACjE,MAAM,UAAU,GAAGA;KACvB,UAAU,CAACD,kBAAU;AACrB,KAAA,QAAQ;KACR,QAAQ,CAAC,gCAAgC;AACrC,MAAM,aAAa,GAAGC;AAC1B,KAAA,MAAM;AACN,KAAA,QAAQ;KACR,QAAQ,CAAC,2BAA2B;AAChC,MAAM,YAAY,GAAGA;AACzB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,sCAAsC;AAE3C,MAAM,YAAY,GAAGA;AACzB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,qCAAqC;AAE1C,MAAM,UAAU,GAAGA;AACvB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,oCAAoC;;;;;;;;;;;"}
+{"version":3,"file":"schema.cjs","sources":["../../../../src/tools/search/schema.ts"],"sourcesContent":["import { z } from 'zod';\n\nexport enum DATE_RANGE {\n  PAST_HOUR = 'h',\n  PAST_24_HOURS = 'd',\n  PAST_WEEK = 'w',\n  PAST_MONTH = 'm',\n  PAST_YEAR = 'y',\n}\n\nexport const DEFAULT_QUERY_DESCRIPTION = `\nGUIDELINES:\n- Start broad, then narrow: Begin with key concepts, then refine with specifics\n- Think like sources: Use terminology experts would use in the field\n- Consider perspective: Frame queries from different viewpoints for better results\n- Quality over quantity: A precise 3-4 word query often beats lengthy sentences\n\nTECHNIQUES (combine for power searches):\n- EXACT PHRASES: Use quotes (\"climate change report\")\n- EXCLUDE TERMS: Use minus to remove unwanted results (-wikipedia)\n- SITE-SPECIFIC: Restrict to websites (site:edu research)\n- FILETYPE: Find specific documents (filetype:pdf study)\n- OR OPERATOR: Find alternatives (electric OR hybrid cars)\n- DATE RANGE: Recent information (data after:2020)\n- WILDCARDS: Use * for unknown terms (how to * bread)\n- SPECIFIC QUESTIONS: Use who/what/when/where/why/how\n- DOMAIN TERMS: Include technical terminology for specialized topics\n- CONCISE TERMS: Prioritize keywords over sentences\n`.trim();\n\nexport const DEFAULT_COUNTRY_DESCRIPTION =\n  `Country code to localize search results.\nUse standard 2-letter country codes: \"us\", \"uk\", \"ca\", \"de\", \"fr\", \"jp\", \"br\", etc.\nProvide this when the search should return results specific to a particular country.\nExamples:\n- \"us\" for United States (default)\n- \"de\" for Germany\n- \"in\" for India\n`.trim();\n\nexport const querySchema = z.string().describe(DEFAULT_QUERY_DESCRIPTION);\nexport const dateSchema = z\n  .nativeEnum(DATE_RANGE)\n  .optional()\n  .describe('Date range for search results.');\nexport const countrySchema = z\n  .string()\n  .optional()\n  .describe(DEFAULT_COUNTRY_DESCRIPTION);\nexport const imagesSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run an image search.');\n\nexport const videosSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run a video search.');\n\nexport const newsSchema = z\n  .boolean()\n  .optional()\n  .describe('Whether to also run a news search.');\n"],"names":["DATE_RANGE","z"],"mappings":";;;;AAEYA;AAAZ,CAAA,UAAY,UAAU,EAAA;AACpB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,eAAA,CAAA,GAAA,GAAmB;AACnB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,YAAA,CAAA,GAAA,GAAgB;AAChB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACjB,CAAC,EANWA,kBAAU,KAAVA,kBAAU,GAMrB,EAAA,CAAA,CAAA;AAEY,MAAA,yBAAyB,GAAG;;;;;;;;;;;;;;;;;;CAkBxC,CAAC,IAAI;AAEO,MAAA,2BAA2B,GACtC,CAAA;;;;;;;CAOD,CAAC,IAAI;AAEC,MAAM,WAAW,GAAGC,KAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB;AACjE,MAAM,UAAU,GAAGA;KACvB,UAAU,CAACD,kBAAU;AACrB,KAAA,QAAQ;KACR,QAAQ,CAAC,gCAAgC;AACrC,MAAM,aAAa,GAAGC;AAC1B,KAAA,MAAM;AACN,KAAA,QAAQ;KACR,QAAQ,CAAC,2BAA2B;AAChC,MAAM,YAAY,GAAGA;AACzB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,sCAAsC;AAE3C,MAAM,YAAY,GAAGA;AACzB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,qCAAqC;AAE1C,MAAM,UAAU,GAAGA;AACvB,KAAA,OAAO;AACP,KAAA,QAAQ;KACR,QAAQ,CAAC,oCAAoC;;;;;;;;;;;"}

package/dist/cjs/utils/run.cjs

@@ -30,8 +30,12 @@ class RunnableCallable extends runnables.Runnable {
     }
     async _tracedInvoke(input, config, runManager) {
         return new Promise((resolve, reject) => {
+            // Defensive check: ensure runManager has getChild method before calling
+            const childCallbacks = typeof runManager?.getChild === 'function'
+                ? runManager.getChild()
+                : undefined;
             const childConfig = runnables.patchConfig(config, {
-                callbacks: runManager?.getChild(),
+                callbacks: childCallbacks,
             });
             void singletons.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
                 try {

package/dist/cjs/utils/run.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"run.cjs","sources":["../../../src/utils/run.ts"],"sourcesContent":["import { CallbackManagerForChainRun } from '@langchain/core/callbacks/manager';\nimport {\n  mergeConfigs,\n  patchConfig,\n  Runnable,\n  RunnableConfig,\n} from '@langchain/core/runnables';\nimport { AsyncLocalStorageProviderSingleton } from '@langchain/core/singletons';\n\n/**\n * Delays the execution for a specified number of milliseconds.\n *\n * @param {number} ms - The number of milliseconds to delay.\n * @return {Promise<void>} A promise that resolves after the specified delay.\n */\nexport function sleep(ms: number): Promise<void> {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface RunnableCallableArgs extends Partial<any> {\n  name?: string;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  func: (...args: any[]) => any;\n  tags?: string[];\n  trace?: boolean;\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = ['langgraph'];\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  func: (...args: any[]) => any;\n\n  tags?: string[];\n\n  config?: RunnableConfig;\n\n  trace: boolean = true;\n\n  recurse: boolean = true;\n\n  constructor(fields: RunnableCallableArgs) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.trace = fields.trace ?? this.trace;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  protected async _tracedInvoke(\n    input: I,\n    config?: Partial<RunnableConfig>,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<O> {\n    return new Promise<O>((resolve, reject) => {\n      const childConfig = patchConfig(config, {\n        callbacks: runManager?.getChild(),\n      });\n      void AsyncLocalStorageProviderSingleton.runWithConfig(\n        childConfig,\n        async () => {\n          try {\n            const output = await this.func(input, childConfig);\n            resolve(output);\n          } catch (e) {\n            reject(e);\n          }\n        }\n      );\n    });\n  }\n\n  async invoke(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    options?: Partial<RunnableConfig> | undefined\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  ): Promise<any> {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    let returnValue: any;\n\n    if (this.trace) {\n      returnValue = await this._callWithConfig(\n        this._tracedInvoke,\n        input,\n        mergeConfigs(this.config, options)\n      );\n    } else {\n      returnValue = await this.func(input, mergeConfigs(this.config, options));\n    }\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await returnValue.invoke(input, options);\n    }\n\n    return returnValue;\n  }\n}"],"names":["Runnable","patchConfig","AsyncLocalStorageProviderSingleton","mergeConfigs"],"mappings":";;;;;AASA;;;;;AAKG;AACG,SAAU,KAAK,CAAC,EAAU,EAAA;AAC9B,IAAA,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC1D;AAYM,MAAO,gBAA2C,SAAQA,kBAAc,CAAA;AAC5E,IAAA,YAAY,GAAa,CAAC,WAAW,CAAC;;AAGtC,IAAA,IAAI;AAEJ,IAAA,IAAI;AAEJ,IAAA,MAAM;IAEN,KAAK,GAAY,IAAI;IAErB,OAAO,GAAY,IAAI;AAEvB,IAAA,WAAA,CAAY,MAA4B,EAAA;AACtC,QAAA,KAAK,EAAE;AACP,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI;AAC3C,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,SAAS;QAC7D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;QACvC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;;AAGrC,IAAA,MAAM,aAAa,CAC3B,KAAQ,EACR,MAAgC,EAChC,UAAuC,EAAA;QAEvC,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,KAAI;AACxC,YAAA,MAAM,WAAW,GAAGC,qBAAW,CAAC,MAAM,EAAE;AACtC,gBAAA,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE;AAClC,aAAA,CAAC;YACF,KAAKC,6CAAkC,CAAC,aAAa,CACnD,WAAW,EACX,YAAW;AACT,gBAAA,IAAI;oBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC;oBAClD,OAAO,CAAC,MAAM,CAAC;;gBACf,OAAO,CAAC,EAAE;oBACV,MAAM,CAAC,CAAC,CAAC;;AAEb,aAAC,CACF;AACH,SAAC,CAAC;;AAGJ,IAAA,MAAM,MAAM;;AAEV,IAAA,KAAU,EACV;;;;AAIA,QAAA,IAAI,WAAgB;AAEpB,QAAA,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CACtC,IAAI,CAAC,aAAa,EAClB,KAAK,EACLC,sBAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CACnC;;aACI;AACL,YAAA,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAEA,sBAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;;QAG1E,IAAIH,kBAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;YACpD,OAAO,MAAM,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;;AAGjD,QAAA,OAAO,WAAW;;AAErB;;;;;"}
+{"version":3,"file":"run.cjs","sources":["../../../src/utils/run.ts"],"sourcesContent":["import { CallbackManagerForChainRun } from '@langchain/core/callbacks/manager';\nimport {\n  mergeConfigs,\n  patchConfig,\n  Runnable,\n  RunnableConfig,\n} from '@langchain/core/runnables';\nimport { AsyncLocalStorageProviderSingleton } from '@langchain/core/singletons';\n\n/**\n * Delays the execution for a specified number of milliseconds.\n *\n * @param {number} ms - The number of milliseconds to delay.\n * @return {Promise<void>} A promise that resolves after the specified delay.\n */\nexport function sleep(ms: number): Promise<void> {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface RunnableCallableArgs extends Partial<any> {\n  name?: string;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  func: (...args: any[]) => any;\n  tags?: string[];\n  trace?: boolean;\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = ['langgraph'];\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  func: (...args: any[]) => any;\n\n  tags?: string[];\n\n  config?: RunnableConfig;\n\n  trace: boolean = true;\n\n  recurse: boolean = true;\n\n  constructor(fields: RunnableCallableArgs) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.trace = fields.trace ?? this.trace;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  protected async _tracedInvoke(\n    input: I,\n    config?: Partial<RunnableConfig>,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<O> {\n    return new Promise<O>((resolve, reject) => {\n      // Defensive check: ensure runManager has getChild method before calling\n      const childCallbacks = typeof runManager?.getChild === 'function' \n        ? runManager.getChild() \n        : undefined;\n      const childConfig = patchConfig(config, {\n        callbacks: childCallbacks,\n      });\n      void AsyncLocalStorageProviderSingleton.runWithConfig(\n        childConfig,\n        async () => {\n          try {\n            const output = await this.func(input, childConfig);\n            resolve(output);\n          } catch (e) {\n            reject(e);\n          }\n        }\n      );\n    });\n  }\n\n  async invoke(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    options?: Partial<RunnableConfig> | undefined\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  ): Promise<any> {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    let returnValue: any;\n\n    if (this.trace) {\n      returnValue = await this._callWithConfig(\n        this._tracedInvoke,\n        input,\n        mergeConfigs(this.config, options)\n      );\n    } else {\n      returnValue = await this.func(input, mergeConfigs(this.config, options));\n    }\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await returnValue.invoke(input, options);\n    }\n\n    return returnValue;\n  }\n}"],"names":["Runnable","patchConfig","AsyncLocalStorageProviderSingleton","mergeConfigs"],"mappings":";;;;;AASA;;;;;AAKG;AACG,SAAU,KAAK,CAAC,EAAU,EAAA;AAC9B,IAAA,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC1D;AAYM,MAAO,gBAA2C,SAAQA,kBAAc,CAAA;AAC5E,IAAA,YAAY,GAAa,CAAC,WAAW,CAAC;;AAGtC,IAAA,IAAI;AAEJ,IAAA,IAAI;AAEJ,IAAA,MAAM;IAEN,KAAK,GAAY,IAAI;IAErB,OAAO,GAAY,IAAI;AAEvB,IAAA,WAAA,CAAY,MAA4B,EAAA;AACtC,QAAA,KAAK,EAAE;AACP,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI;AAC3C,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,SAAS;QAC7D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;QACvC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;;AAGrC,IAAA,MAAM,aAAa,CAC3B,KAAQ,EACR,MAAgC,EAChC,UAAuC,EAAA;QAEvC,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,KAAI;;AAExC,YAAA,MAAM,cAAc,GAAG,OAAO,UAAU,EAAE,QAAQ,KAAK;AACrD,kBAAE,UAAU,CAAC,QAAQ;kBACnB,SAAS;AACb,YAAA,MAAM,WAAW,GAAGC,qBAAW,CAAC,MAAM,EAAE;AACtC,gBAAA,SAAS,EAAE,cAAc;AAC1B,aAAA,CAAC;YACF,KAAKC,6CAAkC,CAAC,aAAa,CACnD,WAAW,EACX,YAAW;AACT,gBAAA,IAAI;oBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC;oBAClD,OAAO,CAAC,MAAM,CAAC;;gBACf,OAAO,CAAC,EAAE;oBACV,MAAM,CAAC,CAAC,CAAC;;AAEb,aAAC,CACF;AACH,SAAC,CAAC;;AAGJ,IAAA,MAAM,MAAM;;AAEV,IAAA,KAAU,EACV;;;;AAIA,QAAA,IAAI,WAAgB;AAEpB,QAAA,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CACtC,IAAI,CAAC,aAAa,EAClB,KAAK,EACLC,sBAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CACnC;;aACI;AACL,YAAA,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAEA,sBAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;;QAG1E,IAAIH,kBAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;YACpD,OAAO,MAAM,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;;AAGjD,QAAA,OAAO,WAAW;;AAErB;;;;;"}

package/dist/esm/agents/AgentContext.mjs

@@ -13,7 +13,7 @@ class AgentContext {
      * Create an AgentContext from configuration with token accounting initialization
      */
     static fromConfig(agentConfig, tokenCounter, indexTokenCountMap) {
-        const { agentId, provider, clientOptions, tools, toolMap, toolEnd, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, } = agentConfig;
+        const { agentId, provider, clientOptions, tools, toolMap, toolEnd, toolRegistry, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, } = agentConfig;
         const agentContext = new AgentContext({
             agentId,
             provider,
@@ -22,6 +22,7 @@ class AgentContext {
             streamBuffer,
             tools,
             toolMap,
+            toolRegistry,
             instructions,
             additionalInstructions: additional_instructions,
             reasoningKey,
@@ -31,12 +32,16 @@ class AgentContext {
             useLegacyContent,
         });
         if (tokenCounter) {
+            // Initialize system runnable BEFORE async tool token calculation
+            // This ensures system message tokens are in instructionTokens before
+            // updateTokenMapWithInstructions is called
+            agentContext.initializeSystemRunnable();
             const tokenMap = indexTokenCountMap || {};
             agentContext.indexTokenCountMap = tokenMap;
             agentContext.tokenCalculationPromise = agentContext
                 .calculateInstructionTokens(tokenCounter)
                 .then(() => {
-                // Update token map with instruction tokens
+                // Update token map with instruction tokens (includes system + tool tokens)
                 agentContext.updateTokenMapWithInstructions(tokenMap);
             })
                 .catch((err) => {
@@ -74,6 +79,13 @@ class AgentContext {
     tools;
     /** Tool map for this agent */
     toolMap;
+    /**
+     * Tool definitions registry (includes deferred and programmatic tool metadata).
+     * Used for tool search and programmatic tool calling.
+     */
+    toolRegistry;
+    /** Set of tool names discovered via tool search (to be loaded) */
+    discoveredToolNames = new Set();
     /** Instructions for this agent */
     instructions;
     /** Additional instructions for this agent */
@@ -88,13 +100,17 @@ class AgentContext {
     currentTokenType = ContentTypes.TEXT;
     /** Whether tools should end the workflow */
     toolEnd = false;
-    /** System runnable for this agent */
-    systemRunnable;
+    /** Cached system runnable (created lazily) */
+    cachedSystemRunnable;
+    /** Whether system runnable needs rebuild (set when discovered tools change) */
+    systemRunnableStale = true;
+    /** Cached system message token count (separate from tool tokens) */
+    systemMessageTokens = 0;
     /** Promise for token calculation initialization */
     tokenCalculationPromise;
     /** Format content blocks as strings (for legacy compatibility) */
     useLegacyContent = false;
-    constructor({ agentId, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, }) {
+    constructor({ agentId, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, toolRegistry, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, }) {
         this.agentId = agentId;
         this.provider = provider;
         this.clientOptions = clientOptions;
@@ -103,6 +119,7 @@ class AgentContext {
         this.tokenCounter = tokenCounter;
         this.tools = tools;
         this.toolMap = toolMap;
+        this.toolRegistry = toolRegistry;
         this.instructions = instructions;
         this.additionalInstructions = additionalInstructions;
         if (reasoningKey) {
@@ -115,24 +132,112 @@ class AgentContext {
             this.instructionTokens = instructionTokens;
         }
         this.useLegacyContent = useLegacyContent ?? false;
-        this.systemRunnable = this.createSystemRunnable();
     }
     /**
-     * Create system runnable from instructions and calculate tokens if tokenCounter is available
+     * Builds instructions text for tools that are ONLY callable via programmatic code execution.
+     * These tools cannot be called directly by the LLM but are available through the
+     * run_tools_with_code tool.
+     *
+     * Includes:
+     * - Code_execution-only tools that are NOT deferred
+     * - Code_execution-only tools that ARE deferred but have been discovered via tool search
+     */
+    buildProgrammaticOnlyToolsInstructions() {
+        if (!this.toolRegistry)
+            return '';
+        const programmaticOnlyTools = [];
+        for (const [name, toolDef] of this.toolRegistry) {
+            const allowedCallers = toolDef.allowed_callers ?? ['direct'];
+            const isCodeExecutionOnly = allowedCallers.includes('code_execution') &&
+                !allowedCallers.includes('direct');
+            if (!isCodeExecutionOnly)
+                continue;
+            // Include if: not deferred OR deferred but discovered
+            const isDeferred = toolDef.defer_loading === true;
+            const isDiscovered = this.discoveredToolNames.has(name);
+            if (!isDeferred || isDiscovered) {
+                programmaticOnlyTools.push(toolDef);
+            }
+        }
+        if (programmaticOnlyTools.length === 0)
+            return '';
+        const toolDescriptions = programmaticOnlyTools
+            .map((tool) => {
+            let desc = `- **${tool.name}**`;
+            if (tool.description != null && tool.description !== '') {
+                desc += `: ${tool.description}`;
+            }
+            if (tool.parameters) {
+                desc += `\n  Parameters: ${JSON.stringify(tool.parameters, null, 2).replace(/\n/g, '\n  ')}`;
+            }
+            return desc;
+        })
+            .join('\n\n');
+        return ('\n\n## Programmatic-Only Tools\n\n' +
+            'The following tools are available exclusively through the `run_tools_with_code` tool. ' +
+            'You cannot call these tools directly; instead, use `run_tools_with_code` with Python code that invokes them.\n\n' +
+            toolDescriptions);
+    }
+    /**
+     * Gets the system runnable, creating it lazily if needed.
+     * Includes instructions, additional instructions, and programmatic-only tools documentation.
+     * Only rebuilds when marked stale (via markToolsAsDiscovered).
      */
-    createSystemRunnable() {
-        let finalInstructions = this.instructions;
+    get systemRunnable() {
+        // Return cached if not stale
+        if (!this.systemRunnableStale && this.cachedSystemRunnable !== undefined) {
+            return this.cachedSystemRunnable;
+        }
+        // Stale or first access - rebuild
+        const instructionsString = this.buildInstructionsString();
+        this.cachedSystemRunnable = this.buildSystemRunnable(instructionsString);
+        this.systemRunnableStale = false;
+        return this.cachedSystemRunnable;
+    }
+    /**
+     * Explicitly initializes the system runnable.
+     * Call this before async token calculation to ensure system message tokens are counted first.
+     */
+    initializeSystemRunnable() {
+        if (this.systemRunnableStale || this.cachedSystemRunnable === undefined) {
+            const instructionsString = this.buildInstructionsString();
+            this.cachedSystemRunnable = this.buildSystemRunnable(instructionsString);
+            this.systemRunnableStale = false;
+        }
+    }
+    /**
+     * Builds the raw instructions string (without creating SystemMessage).
+     */
+    buildInstructionsString() {
+        let result = this.instructions ?? '';
         if (this.additionalInstructions != null &&
             this.additionalInstructions !== '') {
-            finalInstructions =
-                finalInstructions != null && finalInstructions
-                    ? `${finalInstructions}\n\n${this.additionalInstructions}`
-                    : this.additionalInstructions;
-        }
-        // Handle Anthropic prompt caching
-        if (finalInstructions != null &&
-            finalInstructions !== '' &&
-            this.provider === Providers.ANTHROPIC) {
+            result = result
+                ? `${result}\n\n${this.additionalInstructions}`
+                : this.additionalInstructions;
+        }
+        const programmaticToolsDoc = this.buildProgrammaticOnlyToolsInstructions();
+        if (programmaticToolsDoc) {
+            result = result
+                ? `${result}${programmaticToolsDoc}`
+                : programmaticToolsDoc;
+        }
+        return result;
+    }
+    /**
+     * Build system runnable from pre-built instructions string.
+     * Only called when content has actually changed.
+     */
+    buildSystemRunnable(instructionsString) {
+        if (!instructionsString) {
+            // Remove previous tokens if we had a system message before
+            this.instructionTokens -= this.systemMessageTokens;
+            this.systemMessageTokens = 0;
+            return undefined;
+        }
+        let finalInstructions = instructionsString;
+        // Handle Anthropic prompt caching (Direct API)
+        if (this.provider === Providers.ANTHROPIC) {
             const anthropicOptions = this.clientOptions;
             const defaultHeaders = anthropicOptions?.clientOptions?.defaultHeaders;
             const anthropicBeta = defaultHeaders?.['anthropic-beta'];
@@ -142,29 +247,56 @@ class AgentContext {
                     content: [
                         {
                             type: 'text',
-                            text: this.instructions,
+                            text: instructionsString,
                             cache_control: { type: 'ephemeral' },
                         },
                     ],
                 };
             }
         }
-        if (finalInstructions != null && finalInstructions !== '') {
-            const systemMessage = new SystemMessage(finalInstructions);
-            if (this.tokenCounter) {
-                this.instructionTokens += this.tokenCounter(systemMessage);
+        // Handle Bedrock prompt caching (Converse API)
+        // Adds cachePoint block after text content for system message caching
+        // NOTE: Both Claude and Nova models support cachePoint in system and messages
+        // (Nova does NOT support cachePoint in tools - that check is in bedrock/index.ts)
+        if (this.provider === Providers.BEDROCK) {
+            const bedrockOptions = this.clientOptions;
+            const modelId = bedrockOptions?.model?.toLowerCase() ?? '';
+            const supportsCaching = modelId.includes('claude') || modelId.includes('anthropic') || modelId.includes('nova');
+            if (bedrockOptions?.promptCache === true && supportsCaching) {
+                // Always log system cache structure
+                console.log(`[Cache] 📝 System | chars=${instructionsString.length} | tokens=${this.systemMessageTokens} | model=${modelId}`);
+                finalInstructions = {
+                    content: [
+                        {
+                            type: 'text',
+                            text: instructionsString,
+                        },
+                        {
+                            cachePoint: { type: 'default' },
+                        },
+                    ],
+                };
             }
-            return RunnableLambda.from((messages) => {
-                return [systemMessage, ...messages];
-            }).withConfig({ runName: 'prompt' });
         }
-        return undefined;
+        const systemMessage = new SystemMessage(finalInstructions);
+        // Update token counts (subtract old, add new)
+        if (this.tokenCounter) {
+            this.instructionTokens -= this.systemMessageTokens;
+            this.systemMessageTokens = this.tokenCounter(systemMessage);
+            this.instructionTokens += this.systemMessageTokens;
+        }
+        return RunnableLambda.from((messages) => {
+            return [systemMessage, ...messages];
+        }).withConfig({ runName: 'prompt' });
     }
     /**
      * Reset context for a new run
      */
     reset() {
         this.instructionTokens = 0;
+        this.systemMessageTokens = 0;
+        this.cachedSystemRunnable = undefined;
+        this.systemRunnableStale = true;
         this.lastToken = undefined;
         this.indexTokenCountMap = {};
         this.currentUsage = undefined;
@@ -172,6 +304,7 @@ class AgentContext {
         this.lastStreamCall = undefined;
         this.tokenTypeSwitch = undefined;
         this.currentTokenType = ContentTypes.TEXT;
+        this.discoveredToolNames.clear();
     }
     /**
      * Update the token count map with instruction tokens
@@ -214,6 +347,74 @@ class AgentContext {
         // Add tool tokens to existing instruction tokens (which may already include system message tokens)
         this.instructionTokens += toolTokens;
     }
+    /**
+     * Gets the tool registry for deferred tools (for tool search).
+     * @param onlyDeferred If true, only returns tools with defer_loading=true
+     * @returns LCToolRegistry with tool definitions
+     */
+    getDeferredToolRegistry(onlyDeferred = true) {
+        const registry = new Map();
+        if (!this.toolRegistry) {
+            return registry;
+        }
+        for (const [name, toolDef] of this.toolRegistry) {
+            if (!onlyDeferred || toolDef.defer_loading === true) {
+                registry.set(name, toolDef);
+            }
+        }
+        return registry;
+    }
+    /**
+     * Marks tools as discovered via tool search.
+     * Discovered tools will be included in the next model binding.
+     * Only marks system runnable stale if NEW tools were actually added.
+     * @param toolNames - Array of discovered tool names
+     * @returns true if any new tools were discovered
+     */
+    markToolsAsDiscovered(toolNames) {
+        let hasNewDiscoveries = false;
+        for (const name of toolNames) {
+            if (!this.discoveredToolNames.has(name)) {
+                this.discoveredToolNames.add(name);
+                hasNewDiscoveries = true;
+            }
+        }
+        if (hasNewDiscoveries) {
+            this.systemRunnableStale = true;
+        }
+        return hasNewDiscoveries;
+    }
+    /**
+     * Gets tools that should be bound to the LLM.
+     * Includes:
+     * 1. Non-deferred tools with allowed_callers: ['direct']
+     * 2. Discovered tools (from tool search)
+     * @returns Array of tools to bind to model
+     */
+    getToolsForBinding() {
+        if (!this.tools || !this.toolRegistry) {
+            return this.tools;
+        }
+        const toolsToInclude = this.tools.filter((tool) => {
+            if (!('name' in tool)) {
+                return true; // No name, include by default
+            }
+            const toolDef = this.toolRegistry?.get(tool.name);
+            if (!toolDef) {
+                return true; // Not in registry, include by default
+            }
+            // Check if discovered (overrides defer_loading)
+            if (this.discoveredToolNames.has(tool.name)) {
+                // Discovered tools must still have allowed_callers: ['direct']
+                const allowedCallers = toolDef.allowed_callers ?? ['direct'];
+                return allowedCallers.includes('direct');
+            }
+            // Not discovered: must be direct-callable AND not deferred
+            const allowedCallers = toolDef.allowed_callers ?? ['direct'];
+            return (allowedCallers.includes('direct') && toolDef.defer_loading !== true);
+        });
+        return toolsToInclude;
+    }
 }
 
 export { AgentContext };

package/dist/esm/agents/AgentContext.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"AgentContext.mjs","sources":["../../../src/agents/AgentContext.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/agents/AgentContext.ts\nimport { zodToJsonSchema } from 'zod-to-json-schema';\nimport { SystemMessage } from '@langchain/core/messages';\nimport { RunnableLambda } from '@langchain/core/runnables';\nimport type {\n  UsageMetadata,\n  BaseMessage,\n  BaseMessageFields,\n} from '@langchain/core/messages';\nimport type { RunnableConfig, Runnable } from '@langchain/core/runnables';\nimport type * as t from '@/types';\nimport type { createPruneMessages } from '@/messages';\nimport { ContentTypes, Providers } from '@/common';\n\n/**\n * Encapsulates agent-specific state that can vary between agents in a multi-agent system\n */\nexport class AgentContext {\n  /**\n   * Create an AgentContext from configuration with token accounting initialization\n   */\n  static fromConfig(\n    agentConfig: t.AgentInputs,\n    tokenCounter?: t.TokenCounter,\n    indexTokenCountMap?: Record<string, number>\n  ): AgentContext {\n    const {\n      agentId,\n      provider,\n      clientOptions,\n      tools,\n      toolMap,\n      toolEnd,\n      instructions,\n      additional_instructions,\n      streamBuffer,\n      maxContextTokens,\n      reasoningKey,\n      useLegacyContent,\n    } = agentConfig;\n\n    const agentContext = new AgentContext({\n      agentId,\n      provider,\n      clientOptions,\n      maxContextTokens,\n      streamBuffer,\n      tools,\n      toolMap,\n      instructions,\n      additionalInstructions: additional_instructions,\n      reasoningKey,\n      toolEnd,\n      instructionTokens: 0,\n      tokenCounter,\n      useLegacyContent,\n    });\n\n    if (tokenCounter) {\n      const tokenMap = indexTokenCountMap || {};\n      agentContext.indexTokenCountMap = tokenMap;\n      agentContext.tokenCalculationPromise = agentContext\n        .calculateInstructionTokens(tokenCounter)\n        .then(() => {\n          // Update token map with instruction tokens\n          agentContext.updateTokenMapWithInstructions(tokenMap);\n        })\n        .catch((err) => {\n          console.error('Error calculating instruction tokens:', err);\n        });\n    } else if (indexTokenCountMap) {\n      agentContext.indexTokenCountMap = indexTokenCountMap;\n    }\n\n    return agentContext;\n  }\n\n  /** Agent identifier */\n  agentId: string;\n  /** Provider for this specific agent */\n  provider: Providers;\n  /** Client options for this agent */\n  clientOptions?: t.ClientOptions;\n  /** Token count map indexed by message position */\n  indexTokenCountMap: Record<string, number | undefined> = {};\n  /** Maximum context tokens for this agent */\n  maxContextTokens?: number;\n  /** Current usage metadata for this agent */\n  currentUsage?: Partial<UsageMetadata>;\n  /** Prune messages function configured for this agent */\n  pruneMessages?: ReturnType<typeof createPruneMessages>;\n  /** Token counter function for this agent */\n  tokenCounter?: t.TokenCounter;\n  /** Instructions/system message token count */\n  instructionTokens: number = 0;\n  /** The amount of time that should pass before another consecutive API call */\n  streamBuffer?: number;\n  /** Last stream call timestamp for rate limiting */\n  lastStreamCall?: number;\n  /** Tools available to this agent */\n  tools?: t.GraphTools;\n  /** Tool map for this agent */\n  toolMap?: t.ToolMap;\n  /** Instructions for this agent */\n  instructions?: string;\n  /** Additional instructions for this agent */\n  additionalInstructions?: string;\n  /** Reasoning key for this agent */\n  reasoningKey: 'reasoning_content' | 'reasoning' = 'reasoning_content';\n  /** Last token for reasoning detection */\n  lastToken?: string;\n  /** Token type switch state */\n  tokenTypeSwitch?: 'reasoning' | 'content';\n  /** Current token type being processed */\n  currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text' =\n    ContentTypes.TEXT;\n  /** Whether tools should end the workflow */\n  toolEnd: boolean = false;\n  /** System runnable for this agent */\n  systemRunnable?: Runnable<\n    BaseMessage[],\n    (BaseMessage | SystemMessage)[],\n    RunnableConfig<Record<string, unknown>>\n  >;\n  /** Promise for token calculation initialization */\n  tokenCalculationPromise?: Promise<void>;\n  /** Format content blocks as strings (for legacy compatibility) */\n  useLegacyContent: boolean = false;\n\n  constructor({\n    agentId,\n    provider,\n    clientOptions,\n    maxContextTokens,\n    streamBuffer,\n    tokenCounter,\n    tools,\n    toolMap,\n    instructions,\n    additionalInstructions,\n    reasoningKey,\n    toolEnd,\n    instructionTokens,\n    useLegacyContent,\n  }: {\n    agentId: string;\n    provider: Providers;\n    clientOptions?: t.ClientOptions;\n    maxContextTokens?: number;\n    streamBuffer?: number;\n    tokenCounter?: t.TokenCounter;\n    tools?: t.GraphTools;\n    toolMap?: t.ToolMap;\n    instructions?: string;\n    additionalInstructions?: string;\n    reasoningKey?: 'reasoning_content' | 'reasoning';\n    toolEnd?: boolean;\n    instructionTokens?: number;\n    useLegacyContent?: boolean;\n  }) {\n    this.agentId = agentId;\n    this.provider = provider;\n    this.clientOptions = clientOptions;\n    this.maxContextTokens = maxContextTokens;\n    this.streamBuffer = streamBuffer;\n    this.tokenCounter = tokenCounter;\n    this.tools = tools;\n    this.toolMap = toolMap;\n    this.instructions = instructions;\n    this.additionalInstructions = additionalInstructions;\n    if (reasoningKey) {\n      this.reasoningKey = reasoningKey;\n    }\n    if (toolEnd !== undefined) {\n      this.toolEnd = toolEnd;\n    }\n    if (instructionTokens !== undefined) {\n      this.instructionTokens = instructionTokens;\n    }\n\n    this.useLegacyContent = useLegacyContent ?? false;\n\n    this.systemRunnable = this.createSystemRunnable();\n  }\n\n  /**\n   * Create system runnable from instructions and calculate tokens if tokenCounter is available\n   */\n  private createSystemRunnable():\n    | Runnable<\n        BaseMessage[],\n        (BaseMessage | SystemMessage)[],\n        RunnableConfig<Record<string, unknown>>\n      >\n    | undefined {\n    let finalInstructions: string | BaseMessageFields | undefined =\n      this.instructions;\n\n    if (\n      this.additionalInstructions != null &&\n      this.additionalInstructions !== ''\n    ) {\n      finalInstructions =\n        finalInstructions != null && finalInstructions\n          ? `${finalInstructions}\\n\\n${this.additionalInstructions}`\n          : this.additionalInstructions;\n    }\n\n    // Handle Anthropic prompt caching\n    if (\n      finalInstructions != null &&\n      finalInstructions !== '' &&\n      this.provider === Providers.ANTHROPIC\n    ) {\n      const anthropicOptions = this.clientOptions as\n        | t.AnthropicClientOptions\n        | undefined;\n      const defaultHeaders = anthropicOptions?.clientOptions?.defaultHeaders as\n        | Record<string, string>\n        | undefined;\n      const anthropicBeta = defaultHeaders?.['anthropic-beta'];\n      if (\n        typeof anthropicBeta === 'string' &&\n        anthropicBeta.includes('prompt-caching')\n      ) {\n        finalInstructions = {\n          content: [\n            {\n              type: 'text',\n              text: this.instructions,\n              cache_control: { type: 'ephemeral' },\n            },\n          ],\n        };\n      }\n    }\n\n    if (finalInstructions != null && finalInstructions !== '') {\n      const systemMessage = new SystemMessage(finalInstructions);\n\n      if (this.tokenCounter) {\n        this.instructionTokens += this.tokenCounter(systemMessage);\n      }\n\n      return RunnableLambda.from((messages: BaseMessage[]) => {\n        return [systemMessage, ...messages];\n      }).withConfig({ runName: 'prompt' });\n    }\n\n    return undefined;\n  }\n\n  /**\n   * Reset context for a new run\n   */\n  reset(): void {\n    this.instructionTokens = 0;\n    this.lastToken = undefined;\n    this.indexTokenCountMap = {};\n    this.currentUsage = undefined;\n    this.pruneMessages = undefined;\n    this.lastStreamCall = undefined;\n    this.tokenTypeSwitch = undefined;\n    this.currentTokenType = ContentTypes.TEXT;\n  }\n\n  /**\n   * Update the token count map with instruction tokens\n   */\n  updateTokenMapWithInstructions(baseTokenMap: Record<string, number>): void {\n    if (this.instructionTokens > 0) {\n      // Shift all indices by the instruction token count\n      const shiftedMap: Record<string, number> = {};\n      for (const [key, value] of Object.entries(baseTokenMap)) {\n        const index = parseInt(key, 10);\n        if (!isNaN(index)) {\n          shiftedMap[String(index)] =\n            value + (index === 0 ? this.instructionTokens : 0);\n        }\n      }\n      this.indexTokenCountMap = shiftedMap;\n    } else {\n      this.indexTokenCountMap = { ...baseTokenMap };\n    }\n  }\n\n  /**\n   * Calculate tool tokens and add to instruction tokens\n   * Note: System message tokens are calculated during systemRunnable creation\n   */\n  async calculateInstructionTokens(\n    tokenCounter: t.TokenCounter\n  ): Promise<void> {\n    let toolTokens = 0;\n    if (this.tools && this.tools.length > 0) {\n      for (const tool of this.tools) {\n        const genericTool = tool as Record<string, unknown>;\n        if (\n          genericTool.schema != null &&\n          typeof genericTool.schema === 'object'\n        ) {\n          const schema = genericTool.schema as {\n            describe: (desc: string) => unknown;\n          };\n          const describedSchema = schema.describe(\n            (genericTool.description as string) || ''\n          );\n          const jsonSchema = zodToJsonSchema(\n            describedSchema as Parameters<typeof zodToJsonSchema>[0],\n            (genericTool.name as string) || ''\n          );\n          toolTokens += tokenCounter(\n            new SystemMessage(JSON.stringify(jsonSchema))\n          );\n        }\n      }\n    }\n\n    // Add tool tokens to existing instruction tokens (which may already include system message tokens)\n    this.instructionTokens += toolTokens;\n  }\n}\n"],"names":[],"mappings":";;;;;AAAA;AACA;AAcA;;AAEG;MACU,YAAY,CAAA;AACvB;;AAEG;AACH,IAAA,OAAO,UAAU,CACf,WAA0B,EAC1B,YAA6B,EAC7B,kBAA2C,EAAA;QAE3C,MAAM,EACJ,OAAO,EACP,QAAQ,EACR,aAAa,EACb,KAAK,EACL,OAAO,EACP,OAAO,EACP,YAAY,EACZ,uBAAuB,EACvB,YAAY,EACZ,gBAAgB,EAChB,YAAY,EACZ,gBAAgB,GACjB,GAAG,WAAW;AAEf,QAAA,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC;YACpC,OAAO;YACP,QAAQ;YACR,aAAa;YACb,gBAAgB;YAChB,YAAY;YACZ,KAAK;YACL,OAAO;YACP,YAAY;AACZ,YAAA,sBAAsB,EAAE,uBAAuB;YAC/C,YAAY;YACZ,OAAO;AACP,YAAA,iBAAiB,EAAE,CAAC;YACpB,YAAY;YACZ,gBAAgB;AACjB,SAAA,CAAC;QAEF,IAAI,YAAY,EAAE;AAChB,YAAA,MAAM,QAAQ,GAAG,kBAAkB,IAAI,EAAE;AACzC,YAAA,YAAY,CAAC,kBAAkB,GAAG,QAAQ;YAC1C,YAAY,CAAC,uBAAuB,GAAG;iBACpC,0BAA0B,CAAC,YAAY;iBACvC,IAAI,CAAC,MAAK;;AAET,gBAAA,YAAY,CAAC,8BAA8B,CAAC,QAAQ,CAAC;AACvD,aAAC;AACA,iBAAA,KAAK,CAAC,CAAC,GAAG,KAAI;AACb,gBAAA,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC;AAC7D,aAAC,CAAC;;aACC,IAAI,kBAAkB,EAAE;AAC7B,YAAA,YAAY,CAAC,kBAAkB,GAAG,kBAAkB;;AAGtD,QAAA,OAAO,YAAY;;;AAIrB,IAAA,OAAO;;AAEP,IAAA,QAAQ;;AAER,IAAA,aAAa;;IAEb,kBAAkB,GAAuC,EAAE;;AAE3D,IAAA,gBAAgB;;AAEhB,IAAA,YAAY;;AAEZ,IAAA,aAAa;;AAEb,IAAA,YAAY;;IAEZ,iBAAiB,GAAW,CAAC;;AAE7B,IAAA,YAAY;;AAEZ,IAAA,cAAc;;AAEd,IAAA,KAAK;;AAEL,IAAA,OAAO;;AAEP,IAAA,YAAY;;AAEZ,IAAA,sBAAsB;;IAEtB,YAAY,GAAsC,mBAAmB;;AAErE,IAAA,SAAS;;AAET,IAAA,eAAe;;AAEf,IAAA,gBAAgB,GACd,YAAY,CAAC,IAAI;;IAEnB,OAAO,GAAY,KAAK;;AAExB,IAAA,cAAc;;AAMd,IAAA,uBAAuB;;IAEvB,gBAAgB,GAAY,KAAK;AAEjC,IAAA,WAAA,CAAY,EACV,OAAO,EACP,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,YAAY,EACZ,YAAY,EACZ,KAAK,EACL,OAAO,EACP,YAAY,EACZ,sBAAsB,EACtB,YAAY,EACZ,OAAO,EACP,iBAAiB,EACjB,gBAAgB,GAgBjB,EAAA;AACC,QAAA,IAAI,CAAC,OAAO,GAAG,OAAO;AACtB,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;AACxB,QAAA,IAAI,CAAC,aAAa,GAAG,aAAa;AAClC,QAAA,IAAI,CAAC,gBAAgB,GAAG,gBAAgB;AACxC,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,KAAK,GAAG,KAAK;AAClB,QAAA,IAAI,CAAC,OAAO,GAAG,OAAO;AACtB,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,sBAAsB,GAAG,sBAAsB;QACpD,IAAI,YAAY,EAAE;AAChB,YAAA,IAAI,CAAC,YAAY,GAAG,YAAY;;AAElC,QAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,YAAA,IAAI,CAAC,OAAO,GAAG,OAAO;;AAExB,QAAA,IAAI,iBAAiB,KAAK,SAAS,EAAE;AACnC,YAAA,IAAI,CAAC,iBAAiB,GAAG,iBAAiB;;AAG5C,QAAA,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,IAAI,KAAK;AAEjD,QAAA,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,oBAAoB,EAAE;;AAGnD;;AAEG;IACK,oBAAoB,GAAA;AAO1B,QAAA,IAAI,iBAAiB,GACnB,IAAI,CAAC,YAAY;AAEnB,QAAA,IACE,IAAI,CAAC,sBAAsB,IAAI,IAAI;AACnC,YAAA,IAAI,CAAC,sBAAsB,KAAK,EAAE,EAClC;YACA,iBAAiB;gBACf,iBAAiB,IAAI,IAAI,IAAI;AAC3B,sBAAE,CAAG,EAAA,iBAAiB,OAAO,IAAI,CAAC,sBAAsB,CAAE;AAC1D,sBAAE,IAAI,CAAC,sBAAsB;;;QAInC,IACE,iBAAiB,IAAI,IAAI;AACzB,YAAA,iBAAiB,KAAK,EAAE;AACxB,YAAA,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,SAAS,EACrC;AACA,YAAA,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAEjB;AACb,YAAA,MAAM,cAAc,GAAG,gBAAgB,EAAE,aAAa,EAAE,cAE3C;AACb,YAAA,MAAM,aAAa,GAAG,cAAc,GAAG,gBAAgB,CAAC;YACxD,IACE,OAAO,aAAa,KAAK,QAAQ;AACjC,gBAAA,aAAa,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EACxC;AACA,gBAAA,iBAAiB,GAAG;AAClB,oBAAA,OAAO,EAAE;AACP,wBAAA;AACE,4BAAA,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,YAAY;AACvB,4BAAA,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;AACrC,yBAAA;AACF,qBAAA;iBACF;;;QAIL,IAAI,iBAAiB,IAAI,IAAI,IAAI,iBAAiB,KAAK,EAAE,EAAE;AACzD,YAAA,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,iBAAiB,CAAC;AAE1D,YAAA,IAAI,IAAI,CAAC,YAAY,EAAE;gBACrB,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;;AAG5D,YAAA,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,QAAuB,KAAI;AACrD,gBAAA,OAAO,CAAC,aAAa,EAAE,GAAG,QAAQ,CAAC;aACpC,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;;AAGtC,QAAA,OAAO,SAAS;;AAGlB;;AAEG;IACH,KAAK,GAAA;AACH,QAAA,IAAI,CAAC,iBAAiB,GAAG,CAAC;AAC1B,QAAA,IAAI,CAAC,SAAS,GAAG,SAAS;AAC1B,QAAA,IAAI,CAAC,kBAAkB,GAAG,EAAE;AAC5B,QAAA,IAAI,CAAC,YAAY,GAAG,SAAS;AAC7B,QAAA,IAAI,CAAC,aAAa,GAAG,SAAS;AAC9B,QAAA,IAAI,CAAC,cAAc,GAAG,SAAS;AAC/B,QAAA,IAAI,CAAC,eAAe,GAAG,SAAS;AAChC,QAAA,IAAI,CAAC,gBAAgB,GAAG,YAAY,CAAC,IAAI;;AAG3C;;AAEG;AACH,IAAA,8BAA8B,CAAC,YAAoC,EAAA;AACjE,QAAA,IAAI,IAAI,CAAC,iBAAiB,GAAG,CAAC,EAAE;;YAE9B,MAAM,UAAU,GAA2B,EAAE;AAC7C,YAAA,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE;gBACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC;AAC/B,gBAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;AACjB,oBAAA,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,wBAAA,KAAK,IAAI,KAAK,KAAK,CAAC,GAAG,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;;;AAGxD,YAAA,IAAI,CAAC,kBAAkB,GAAG,UAAU;;aAC/B;AACL,YAAA,IAAI,CAAC,kBAAkB,GAAG,EAAE,GAAG,YAAY,EAAE;;;AAIjD;;;AAGG;IACH,MAAM,0BAA0B,CAC9B,YAA4B,EAAA;QAE5B,IAAI,UAAU,GAAG,CAAC;AAClB,QAAA,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE;gBAC7B,MAAM,WAAW,GAAG,IAA+B;AACnD,gBAAA,IACE,WAAW,CAAC,MAAM,IAAI,IAAI;AAC1B,oBAAA,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC;AACA,oBAAA,MAAM,MAAM,GAAG,WAAW,CAAC,MAE1B;AACD,oBAAA,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CACpC,WAAW,CAAC,WAAsB,IAAI,EAAE,CAC1C;AACD,oBAAA,MAAM,UAAU,GAAG,eAAe,CAChC,eAAwD,EACvD,WAAW,CAAC,IAAe,IAAI,EAAE,CACnC;AACD,oBAAA,UAAU,IAAI,YAAY,CACxB,IAAI,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAC9C;;;;;AAMP,QAAA,IAAI,CAAC,iBAAiB,IAAI,UAAU;;AAEvC;;;;"}
+{"version":3,"file":"AgentContext.mjs","sources":["../../../src/agents/AgentContext.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/agents/AgentContext.ts\nimport { zodToJsonSchema } from 'zod-to-json-schema';\nimport { SystemMessage } from '@langchain/core/messages';\nimport { RunnableLambda } from '@langchain/core/runnables';\nimport type {\n  UsageMetadata,\n  BaseMessage,\n  BaseMessageFields,\n} from '@langchain/core/messages';\nimport type { RunnableConfig, Runnable } from '@langchain/core/runnables';\nimport type * as t from '@/types';\nimport type { createPruneMessages } from '@/messages';\nimport { ContentTypes, Providers } from '@/common';\n\n/**\n * Encapsulates agent-specific state that can vary between agents in a multi-agent system\n */\nexport class AgentContext {\n  /**\n   * Create an AgentContext from configuration with token accounting initialization\n   */\n  static fromConfig(\n    agentConfig: t.AgentInputs,\n    tokenCounter?: t.TokenCounter,\n    indexTokenCountMap?: Record<string, number>\n  ): AgentContext {\n    const {\n      agentId,\n      provider,\n      clientOptions,\n      tools,\n      toolMap,\n      toolEnd,\n      toolRegistry,\n      instructions,\n      additional_instructions,\n      streamBuffer,\n      maxContextTokens,\n      reasoningKey,\n      useLegacyContent,\n    } = agentConfig;\n\n    const agentContext = new AgentContext({\n      agentId,\n      provider,\n      clientOptions,\n      maxContextTokens,\n      streamBuffer,\n      tools,\n      toolMap,\n      toolRegistry,\n      instructions,\n      additionalInstructions: additional_instructions,\n      reasoningKey,\n      toolEnd,\n      instructionTokens: 0,\n      tokenCounter,\n      useLegacyContent,\n    });\n\n    if (tokenCounter) {\n      // Initialize system runnable BEFORE async tool token calculation\n      // This ensures system message tokens are in instructionTokens before\n      // updateTokenMapWithInstructions is called\n      agentContext.initializeSystemRunnable();\n\n      const tokenMap = indexTokenCountMap || {};\n      agentContext.indexTokenCountMap = tokenMap;\n      agentContext.tokenCalculationPromise = agentContext\n        .calculateInstructionTokens(tokenCounter)\n        .then(() => {\n          // Update token map with instruction tokens (includes system + tool tokens)\n          agentContext.updateTokenMapWithInstructions(tokenMap);\n        })\n        .catch((err) => {\n          console.error('Error calculating instruction tokens:', err);\n        });\n    } else if (indexTokenCountMap) {\n      agentContext.indexTokenCountMap = indexTokenCountMap;\n    }\n\n    return agentContext;\n  }\n\n  /** Agent identifier */\n  agentId: string;\n  /** Provider for this specific agent */\n  provider: Providers;\n  /** Client options for this agent */\n  clientOptions?: t.ClientOptions;\n  /** Token count map indexed by message position */\n  indexTokenCountMap: Record<string, number | undefined> = {};\n  /** Maximum context tokens for this agent */\n  maxContextTokens?: number;\n  /** Current usage metadata for this agent */\n  currentUsage?: Partial<UsageMetadata>;\n  /** Prune messages function configured for this agent */\n  pruneMessages?: ReturnType<typeof createPruneMessages>;\n  /** Token counter function for this agent */\n  tokenCounter?: t.TokenCounter;\n  /** Instructions/system message token count */\n  instructionTokens: number = 0;\n  /** The amount of time that should pass before another consecutive API call */\n  streamBuffer?: number;\n  /** Last stream call timestamp for rate limiting */\n  lastStreamCall?: number;\n  /** Tools available to this agent */\n  tools?: t.GraphTools;\n  /** Tool map for this agent */\n  toolMap?: t.ToolMap;\n  /**\n   * Tool definitions registry (includes deferred and programmatic tool metadata).\n   * Used for tool search and programmatic tool calling.\n   */\n  toolRegistry?: t.LCToolRegistry;\n  /** Set of tool names discovered via tool search (to be loaded) */\n  discoveredToolNames: Set<string> = new Set();\n  /** Instructions for this agent */\n  instructions?: string;\n  /** Additional instructions for this agent */\n  additionalInstructions?: string;\n  /** Reasoning key for this agent */\n  reasoningKey: 'reasoning_content' | 'reasoning' = 'reasoning_content';\n  /** Last token for reasoning detection */\n  lastToken?: string;\n  /** Token type switch state */\n  tokenTypeSwitch?: 'reasoning' | 'content';\n  /** Current token type being processed */\n  currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text' =\n    ContentTypes.TEXT;\n  /** Whether tools should end the workflow */\n  toolEnd: boolean = false;\n  /** Cached system runnable (created lazily) */\n  private cachedSystemRunnable?: Runnable<\n    BaseMessage[],\n    (BaseMessage | SystemMessage)[],\n    RunnableConfig<Record<string, unknown>>\n  >;\n  /** Whether system runnable needs rebuild (set when discovered tools change) */\n  private systemRunnableStale: boolean = true;\n  /** Cached system message token count (separate from tool tokens) */\n  private systemMessageTokens: number = 0;\n  /** Promise for token calculation initialization */\n  tokenCalculationPromise?: Promise<void>;\n  /** Format content blocks as strings (for legacy compatibility) */\n  useLegacyContent: boolean = false;\n\n  constructor({\n    agentId,\n    provider,\n    clientOptions,\n    maxContextTokens,\n    streamBuffer,\n    tokenCounter,\n    tools,\n    toolMap,\n    toolRegistry,\n    instructions,\n    additionalInstructions,\n    reasoningKey,\n    toolEnd,\n    instructionTokens,\n    useLegacyContent,\n  }: {\n    agentId: string;\n    provider: Providers;\n    clientOptions?: t.ClientOptions;\n    maxContextTokens?: number;\n    streamBuffer?: number;\n    tokenCounter?: t.TokenCounter;\n    tools?: t.GraphTools;\n    toolMap?: t.ToolMap;\n    toolRegistry?: t.LCToolRegistry;\n    instructions?: string;\n    additionalInstructions?: string;\n    reasoningKey?: 'reasoning_content' | 'reasoning';\n    toolEnd?: boolean;\n    instructionTokens?: number;\n    useLegacyContent?: boolean;\n  }) {\n    this.agentId = agentId;\n    this.provider = provider;\n    this.clientOptions = clientOptions;\n    this.maxContextTokens = maxContextTokens;\n    this.streamBuffer = streamBuffer;\n    this.tokenCounter = tokenCounter;\n    this.tools = tools;\n    this.toolMap = toolMap;\n    this.toolRegistry = toolRegistry;\n    this.instructions = instructions;\n    this.additionalInstructions = additionalInstructions;\n    if (reasoningKey) {\n      this.reasoningKey = reasoningKey;\n    }\n    if (toolEnd !== undefined) {\n      this.toolEnd = toolEnd;\n    }\n    if (instructionTokens !== undefined) {\n      this.instructionTokens = instructionTokens;\n    }\n\n    this.useLegacyContent = useLegacyContent ?? false;\n  }\n\n  /**\n   * Builds instructions text for tools that are ONLY callable via programmatic code execution.\n   * These tools cannot be called directly by the LLM but are available through the\n   * run_tools_with_code tool.\n   *\n   * Includes:\n   * - Code_execution-only tools that are NOT deferred\n   * - Code_execution-only tools that ARE deferred but have been discovered via tool search\n   */\n  private buildProgrammaticOnlyToolsInstructions(): string {\n    if (!this.toolRegistry) return '';\n\n    const programmaticOnlyTools: t.LCTool[] = [];\n    for (const [name, toolDef] of this.toolRegistry) {\n      const allowedCallers = toolDef.allowed_callers ?? ['direct'];\n      const isCodeExecutionOnly =\n        allowedCallers.includes('code_execution') &&\n        !allowedCallers.includes('direct');\n\n      if (!isCodeExecutionOnly) continue;\n\n      // Include if: not deferred OR deferred but discovered\n      const isDeferred = toolDef.defer_loading === true;\n      const isDiscovered = this.discoveredToolNames.has(name);\n      if (!isDeferred || isDiscovered) {\n        programmaticOnlyTools.push(toolDef);\n      }\n    }\n\n    if (programmaticOnlyTools.length === 0) return '';\n\n    const toolDescriptions = programmaticOnlyTools\n      .map((tool) => {\n        let desc = `- **${tool.name}**`;\n        if (tool.description != null && tool.description !== '') {\n          desc += `: ${tool.description}`;\n        }\n        if (tool.parameters) {\n          desc += `\\n  Parameters: ${JSON.stringify(tool.parameters, null, 2).replace(/\\n/g, '\\n  ')}`;\n        }\n        return desc;\n      })\n      .join('\\n\\n');\n\n    return (\n      '\\n\\n## Programmatic-Only Tools\\n\\n' +\n      'The following tools are available exclusively through the `run_tools_with_code` tool. ' +\n      'You cannot call these tools directly; instead, use `run_tools_with_code` with Python code that invokes them.\\n\\n' +\n      toolDescriptions\n    );\n  }\n\n  /**\n   * Gets the system runnable, creating it lazily if needed.\n   * Includes instructions, additional instructions, and programmatic-only tools documentation.\n   * Only rebuilds when marked stale (via markToolsAsDiscovered).\n   */\n  get systemRunnable():\n    | Runnable<\n        BaseMessage[],\n        (BaseMessage | SystemMessage)[],\n        RunnableConfig<Record<string, unknown>>\n      >\n    | undefined {\n    // Return cached if not stale\n    if (!this.systemRunnableStale && this.cachedSystemRunnable !== undefined) {\n      return this.cachedSystemRunnable;\n    }\n\n    // Stale or first access - rebuild\n    const instructionsString = this.buildInstructionsString();\n    this.cachedSystemRunnable = this.buildSystemRunnable(instructionsString);\n    this.systemRunnableStale = false;\n    return this.cachedSystemRunnable;\n  }\n\n  /**\n   * Explicitly initializes the system runnable.\n   * Call this before async token calculation to ensure system message tokens are counted first.\n   */\n  initializeSystemRunnable(): void {\n    if (this.systemRunnableStale || this.cachedSystemRunnable === undefined) {\n      const instructionsString = this.buildInstructionsString();\n      this.cachedSystemRunnable = this.buildSystemRunnable(instructionsString);\n      this.systemRunnableStale = false;\n    }\n  }\n\n  /**\n   * Builds the raw instructions string (without creating SystemMessage).\n   */\n  private buildInstructionsString(): string {\n    let result = this.instructions ?? '';\n\n    if (\n      this.additionalInstructions != null &&\n      this.additionalInstructions !== ''\n    ) {\n      result = result\n        ? `${result}\\n\\n${this.additionalInstructions}`\n        : this.additionalInstructions;\n    }\n\n    const programmaticToolsDoc = this.buildProgrammaticOnlyToolsInstructions();\n    if (programmaticToolsDoc) {\n      result = result\n        ? `${result}${programmaticToolsDoc}`\n        : programmaticToolsDoc;\n    }\n\n    return result;\n  }\n\n  /**\n   * Build system runnable from pre-built instructions string.\n   * Only called when content has actually changed.\n   */\n  private buildSystemRunnable(\n    instructionsString: string\n  ):\n    | Runnable<\n        BaseMessage[],\n        (BaseMessage | SystemMessage)[],\n        RunnableConfig<Record<string, unknown>>\n      >\n    | undefined {\n    if (!instructionsString) {\n      // Remove previous tokens if we had a system message before\n      this.instructionTokens -= this.systemMessageTokens;\n      this.systemMessageTokens = 0;\n      return undefined;\n    }\n\n    let finalInstructions: string | BaseMessageFields = instructionsString;\n\n    // Handle Anthropic prompt caching (Direct API)\n    if (this.provider === Providers.ANTHROPIC) {\n      const anthropicOptions = this.clientOptions as\n        | t.AnthropicClientOptions\n        | undefined;\n      const defaultHeaders = anthropicOptions?.clientOptions?.defaultHeaders as\n        | Record<string, string>\n        | undefined;\n      const anthropicBeta = defaultHeaders?.['anthropic-beta'];\n      if (\n        typeof anthropicBeta === 'string' &&\n        anthropicBeta.includes('prompt-caching')\n      ) {\n        finalInstructions = {\n          content: [\n            {\n              type: 'text',\n              text: instructionsString,\n              cache_control: { type: 'ephemeral' },\n            },\n          ],\n        };\n      }\n    }\n\n    // Handle Bedrock prompt caching (Converse API)\n    // Adds cachePoint block after text content for system message caching\n    // NOTE: Both Claude and Nova models support cachePoint in system and messages\n    // (Nova does NOT support cachePoint in tools - that check is in bedrock/index.ts)\n    if (this.provider === Providers.BEDROCK) {\n      const bedrockOptions = this.clientOptions as\n        | t.BedrockAnthropicInput\n        | undefined;\n      const modelId = bedrockOptions?.model?.toLowerCase() ?? '';\n      const supportsCaching = modelId.includes('claude') || modelId.includes('anthropic') || modelId.includes('nova');\n\n      if (bedrockOptions?.promptCache === true && supportsCaching) {\n        // Always log system cache structure\n        console.log(`[Cache] 📝 System | chars=${instructionsString.length} | tokens=${this.systemMessageTokens} | model=${modelId}`);\n        \n        finalInstructions = {\n          content: [\n            {\n              type: 'text',\n              text: instructionsString,\n            },\n            {\n              cachePoint: { type: 'default' },\n            },\n          ],\n        };\n      }\n    }\n\n    const systemMessage = new SystemMessage(finalInstructions);\n\n    // Update token counts (subtract old, add new)\n    if (this.tokenCounter) {\n      this.instructionTokens -= this.systemMessageTokens;\n      this.systemMessageTokens = this.tokenCounter(systemMessage);\n      this.instructionTokens += this.systemMessageTokens;\n    }\n\n    return RunnableLambda.from((messages: BaseMessage[]) => {\n      return [systemMessage, ...messages];\n    }).withConfig({ runName: 'prompt' });\n  }\n\n  /**\n   * Reset context for a new run\n   */\n  reset(): void {\n    this.instructionTokens = 0;\n    this.systemMessageTokens = 0;\n    this.cachedSystemRunnable = undefined;\n    this.systemRunnableStale = true;\n    this.lastToken = undefined;\n    this.indexTokenCountMap = {};\n    this.currentUsage = undefined;\n    this.pruneMessages = undefined;\n    this.lastStreamCall = undefined;\n    this.tokenTypeSwitch = undefined;\n    this.currentTokenType = ContentTypes.TEXT;\n    this.discoveredToolNames.clear();\n  }\n\n  /**\n   * Update the token count map with instruction tokens\n   */\n  updateTokenMapWithInstructions(baseTokenMap: Record<string, number>): void {\n    if (this.instructionTokens > 0) {\n      // Shift all indices by the instruction token count\n      const shiftedMap: Record<string, number> = {};\n      for (const [key, value] of Object.entries(baseTokenMap)) {\n        const index = parseInt(key, 10);\n        if (!isNaN(index)) {\n          shiftedMap[String(index)] =\n            value + (index === 0 ? this.instructionTokens : 0);\n        }\n      }\n      this.indexTokenCountMap = shiftedMap;\n    } else {\n      this.indexTokenCountMap = { ...baseTokenMap };\n    }\n  }\n\n  /**\n   * Calculate tool tokens and add to instruction tokens\n   * Note: System message tokens are calculated during systemRunnable creation\n   */\n  async calculateInstructionTokens(\n    tokenCounter: t.TokenCounter\n  ): Promise<void> {\n    let toolTokens = 0;\n    if (this.tools && this.tools.length > 0) {\n      for (const tool of this.tools) {\n        const genericTool = tool as Record<string, unknown>;\n        if (\n          genericTool.schema != null &&\n          typeof genericTool.schema === 'object'\n        ) {\n          const schema = genericTool.schema as {\n            describe: (desc: string) => unknown;\n          };\n          const describedSchema = schema.describe(\n            (genericTool.description as string) || ''\n          );\n          const jsonSchema = zodToJsonSchema(\n            describedSchema as Parameters<typeof zodToJsonSchema>[0],\n            (genericTool.name as string) || ''\n          );\n          toolTokens += tokenCounter(\n            new SystemMessage(JSON.stringify(jsonSchema))\n          );\n        }\n      }\n    }\n\n    // Add tool tokens to existing instruction tokens (which may already include system message tokens)\n    this.instructionTokens += toolTokens;\n  }\n\n  /**\n   * Gets the tool registry for deferred tools (for tool search).\n   * @param onlyDeferred If true, only returns tools with defer_loading=true\n   * @returns LCToolRegistry with tool definitions\n   */\n  getDeferredToolRegistry(onlyDeferred: boolean = true): t.LCToolRegistry {\n    const registry: t.LCToolRegistry = new Map();\n\n    if (!this.toolRegistry) {\n      return registry;\n    }\n\n    for (const [name, toolDef] of this.toolRegistry) {\n      if (!onlyDeferred || toolDef.defer_loading === true) {\n        registry.set(name, toolDef);\n      }\n    }\n\n    return registry;\n  }\n\n  /**\n   * Marks tools as discovered via tool search.\n   * Discovered tools will be included in the next model binding.\n   * Only marks system runnable stale if NEW tools were actually added.\n   * @param toolNames - Array of discovered tool names\n   * @returns true if any new tools were discovered\n   */\n  markToolsAsDiscovered(toolNames: string[]): boolean {\n    let hasNewDiscoveries = false;\n    for (const name of toolNames) {\n      if (!this.discoveredToolNames.has(name)) {\n        this.discoveredToolNames.add(name);\n        hasNewDiscoveries = true;\n      }\n    }\n    if (hasNewDiscoveries) {\n      this.systemRunnableStale = true;\n    }\n    return hasNewDiscoveries;\n  }\n\n  /**\n   * Gets tools that should be bound to the LLM.\n   * Includes:\n   * 1. Non-deferred tools with allowed_callers: ['direct']\n   * 2. Discovered tools (from tool search)\n   * @returns Array of tools to bind to model\n   */\n  getToolsForBinding(): t.GraphTools | undefined {\n    if (!this.tools || !this.toolRegistry) {\n      return this.tools;\n    }\n\n    const toolsToInclude = this.tools.filter((tool) => {\n      if (!('name' in tool)) {\n        return true; // No name, include by default\n      }\n\n      const toolDef = this.toolRegistry?.get(tool.name);\n      if (!toolDef) {\n        return true; // Not in registry, include by default\n      }\n\n      // Check if discovered (overrides defer_loading)\n      if (this.discoveredToolNames.has(tool.name)) {\n        // Discovered tools must still have allowed_callers: ['direct']\n        const allowedCallers = toolDef.allowed_callers ?? ['direct'];\n        return allowedCallers.includes('direct');\n      }\n\n      // Not discovered: must be direct-callable AND not deferred\n      const allowedCallers = toolDef.allowed_callers ?? ['direct'];\n      return (\n        allowedCallers.includes('direct') && toolDef.defer_loading !== true\n      );\n    });\n\n    return toolsToInclude;\n  }\n}\n"],"names":[],"mappings":";;;;;AAAA;AACA;AAcA;;AAEG;MACU,YAAY,CAAA;AACvB;;AAEG;AACH,IAAA,OAAO,UAAU,CACf,WAA0B,EAC1B,YAA6B,EAC7B,kBAA2C,EAAA;AAE3C,QAAA,MAAM,EACJ,OAAO,EACP,QAAQ,EACR,aAAa,EACb,KAAK,EACL,OAAO,EACP,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,uBAAuB,EACvB,YAAY,EACZ,gBAAgB,EAChB,YAAY,EACZ,gBAAgB,GACjB,GAAG,WAAW;AAEf,QAAA,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC;YACpC,OAAO;YACP,QAAQ;YACR,aAAa;YACb,gBAAgB;YAChB,YAAY;YACZ,KAAK;YACL,OAAO;YACP,YAAY;YACZ,YAAY;AACZ,YAAA,sBAAsB,EAAE,uBAAuB;YAC/C,YAAY;YACZ,OAAO;AACP,YAAA,iBAAiB,EAAE,CAAC;YACpB,YAAY;YACZ,gBAAgB;AACjB,SAAA,CAAC;QAEF,IAAI,YAAY,EAAE;;;;YAIhB,YAAY,CAAC,wBAAwB,EAAE;AAEvC,YAAA,MAAM,QAAQ,GAAG,kBAAkB,IAAI,EAAE;AACzC,YAAA,YAAY,CAAC,kBAAkB,GAAG,QAAQ;YAC1C,YAAY,CAAC,uBAAuB,GAAG;iBACpC,0BAA0B,CAAC,YAAY;iBACvC,IAAI,CAAC,MAAK;;AAET,gBAAA,YAAY,CAAC,8BAA8B,CAAC,QAAQ,CAAC;AACvD,aAAC;AACA,iBAAA,KAAK,CAAC,CAAC,GAAG,KAAI;AACb,gBAAA,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC;AAC7D,aAAC,CAAC;;aACC,IAAI,kBAAkB,EAAE;AAC7B,YAAA,YAAY,CAAC,kBAAkB,GAAG,kBAAkB;;AAGtD,QAAA,OAAO,YAAY;;;AAIrB,IAAA,OAAO;;AAEP,IAAA,QAAQ;;AAER,IAAA,aAAa;;IAEb,kBAAkB,GAAuC,EAAE;;AAE3D,IAAA,gBAAgB;;AAEhB,IAAA,YAAY;;AAEZ,IAAA,aAAa;;AAEb,IAAA,YAAY;;IAEZ,iBAAiB,GAAW,CAAC;;AAE7B,IAAA,YAAY;;AAEZ,IAAA,cAAc;;AAEd,IAAA,KAAK;;AAEL,IAAA,OAAO;AACP;;;AAGG;AACH,IAAA,YAAY;;AAEZ,IAAA,mBAAmB,GAAgB,IAAI,GAAG,EAAE;;AAE5C,IAAA,YAAY;;AAEZ,IAAA,sBAAsB;;IAEtB,YAAY,GAAsC,mBAAmB;;AAErE,IAAA,SAAS;;AAET,IAAA,eAAe;;AAEf,IAAA,gBAAgB,GACd,YAAY,CAAC,IAAI;;IAEnB,OAAO,GAAY,KAAK;;AAEhB,IAAA,oBAAoB;;IAMpB,mBAAmB,GAAY,IAAI;;IAEnC,mBAAmB,GAAW,CAAC;;AAEvC,IAAA,uBAAuB;;IAEvB,gBAAgB,GAAY,KAAK;AAEjC,IAAA,WAAA,CAAY,EACV,OAAO,EACP,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,YAAY,EACZ,YAAY,EACZ,KAAK,EACL,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,sBAAsB,EACtB,YAAY,EACZ,OAAO,EACP,iBAAiB,EACjB,gBAAgB,GAiBjB,EAAA;AACC,QAAA,IAAI,CAAC,OAAO,GAAG,OAAO;AACtB,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;AACxB,QAAA,IAAI,CAAC,aAAa,GAAG,aAAa;AAClC,QAAA,IAAI,CAAC,gBAAgB,GAAG,gBAAgB;AACxC,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,KAAK,GAAG,KAAK;AAClB,QAAA,IAAI,CAAC,OAAO,GAAG,OAAO;AACtB,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,sBAAsB,GAAG,sBAAsB;QACpD,IAAI,YAAY,EAAE;AAChB,YAAA,IAAI,CAAC,YAAY,GAAG,YAAY;;AAElC,QAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,YAAA,IAAI,CAAC,OAAO,GAAG,OAAO;;AAExB,QAAA,IAAI,iBAAiB,KAAK,SAAS,EAAE;AACnC,YAAA,IAAI,CAAC,iBAAiB,GAAG,iBAAiB;;AAG5C,QAAA,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,IAAI,KAAK;;AAGnD;;;;;;;;AAQG;IACK,sCAAsC,GAAA;QAC5C,IAAI,CAAC,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,EAAE;QAEjC,MAAM,qBAAqB,GAAe,EAAE;QAC5C,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE;YAC/C,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,IAAI,CAAC,QAAQ,CAAC;AAC5D,YAAA,MAAM,mBAAmB,GACvB,cAAc,CAAC,QAAQ,CAAC,gBAAgB,CAAC;AACzC,gBAAA,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC;AAEpC,YAAA,IAAI,CAAC,mBAAmB;gBAAE;;AAG1B,YAAA,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,KAAK,IAAI;YACjD,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC;AACvD,YAAA,IAAI,CAAC,UAAU,IAAI,YAAY,EAAE;AAC/B,gBAAA,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC;;;AAIvC,QAAA,IAAI,qBAAqB,CAAC,MAAM,KAAK,CAAC;AAAE,YAAA,OAAO,EAAE;QAEjD,MAAM,gBAAgB,GAAG;AACtB,aAAA,GAAG,CAAC,CAAC,IAAI,KAAI;AACZ,YAAA,IAAI,IAAI,GAAG,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,IAAI;AAC/B,YAAA,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,IAAI,IAAI,CAAC,WAAW,KAAK,EAAE,EAAE;AACvD,gBAAA,IAAI,IAAI,CAAK,EAAA,EAAA,IAAI,CAAC,WAAW,EAAE;;AAEjC,YAAA,IAAI,IAAI,CAAC,UAAU,EAAE;gBACnB,IAAI,IAAI,mBAAmB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA,CAAE;;AAE9F,YAAA,OAAO,IAAI;AACb,SAAC;aACA,IAAI,CAAC,MAAM,CAAC;AAEf,QAAA,QACE,oCAAoC;YACpC,wFAAwF;YACxF,kHAAkH;AAClH,YAAA,gBAAgB;;AAIpB;;;;AAIG;AACH,IAAA,IAAI,cAAc,GAAA;;QAQhB,IAAI,CAAC,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,oBAAoB,KAAK,SAAS,EAAE;YACxE,OAAO,IAAI,CAAC,oBAAoB;;;AAIlC,QAAA,MAAM,kBAAkB,GAAG,IAAI,CAAC,uBAAuB,EAAE;QACzD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;AACxE,QAAA,IAAI,CAAC,mBAAmB,GAAG,KAAK;QAChC,OAAO,IAAI,CAAC,oBAAoB;;AAGlC;;;AAGG;IACH,wBAAwB,GAAA;QACtB,IAAI,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,oBAAoB,KAAK,SAAS,EAAE;AACvE,YAAA,MAAM,kBAAkB,GAAG,IAAI,CAAC,uBAAuB,EAAE;YACzD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;AACxE,YAAA,IAAI,CAAC,mBAAmB,GAAG,KAAK;;;AAIpC;;AAEG;IACK,uBAAuB,GAAA;AAC7B,QAAA,IAAI,MAAM,GAAG,IAAI,CAAC,YAAY,IAAI,EAAE;AAEpC,QAAA,IACE,IAAI,CAAC,sBAAsB,IAAI,IAAI;AACnC,YAAA,IAAI,CAAC,sBAAsB,KAAK,EAAE,EAClC;AACA,YAAA,MAAM,GAAG;AACP,kBAAE,CAAG,EAAA,MAAM,OAAO,IAAI,CAAC,sBAAsB,CAAE;AAC/C,kBAAE,IAAI,CAAC,sBAAsB;;AAGjC,QAAA,MAAM,oBAAoB,GAAG,IAAI,CAAC,sCAAsC,EAAE;QAC1E,IAAI,oBAAoB,EAAE;AACxB,YAAA,MAAM,GAAG;AACP,kBAAE,CAAA,EAAG,MAAM,CAAA,EAAG,oBAAoB,CAAE;kBAClC,oBAAoB;;AAG1B,QAAA,OAAO,MAAM;;AAGf;;;AAGG;AACK,IAAA,mBAAmB,CACzB,kBAA0B,EAAA;QAQ1B,IAAI,CAAC,kBAAkB,EAAE;;AAEvB,YAAA,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,mBAAmB;AAClD,YAAA,IAAI,CAAC,mBAAmB,GAAG,CAAC;AAC5B,YAAA,OAAO,SAAS;;QAGlB,IAAI,iBAAiB,GAA+B,kBAAkB;;QAGtE,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE;AACzC,YAAA,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAEjB;AACb,YAAA,MAAM,cAAc,GAAG,gBAAgB,EAAE,aAAa,EAAE,cAE3C;AACb,YAAA,MAAM,aAAa,GAAG,cAAc,GAAG,gBAAgB,CAAC;YACxD,IACE,OAAO,aAAa,KAAK,QAAQ;AACjC,gBAAA,aAAa,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EACxC;AACA,gBAAA,iBAAiB,GAAG;AAClB,oBAAA,OAAO,EAAE;AACP,wBAAA;AACE,4BAAA,IAAI,EAAE,MAAM;AACZ,4BAAA,IAAI,EAAE,kBAAkB;AACxB,4BAAA,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;AACrC,yBAAA;AACF,qBAAA;iBACF;;;;;;;QAQL,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,OAAO,EAAE;AACvC,YAAA,MAAM,cAAc,GAAG,IAAI,CAAC,aAEf;YACb,MAAM,OAAO,GAAG,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE;YAC1D,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;YAE/G,IAAI,cAAc,EAAE,WAAW,KAAK,IAAI,IAAI,eAAe,EAAE;;AAE3D,gBAAA,OAAO,CAAC,GAAG,CAAC,CAAA,0BAAA,EAA6B,kBAAkB,CAAC,MAAM,CAAa,UAAA,EAAA,IAAI,CAAC,mBAAmB,CAAA,SAAA,EAAY,OAAO,CAAA,CAAE,CAAC;AAE7H,gBAAA,iBAAiB,GAAG;AAClB,oBAAA,OAAO,EAAE;AACP,wBAAA;AACE,4BAAA,IAAI,EAAE,MAAM;AACZ,4BAAA,IAAI,EAAE,kBAAkB;AACzB,yBAAA;AACD,wBAAA;AACE,4BAAA,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;AAChC,yBAAA;AACF,qBAAA;iBACF;;;AAIL,QAAA,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,iBAAiB,CAAC;;AAG1D,QAAA,IAAI,IAAI,CAAC,YAAY,EAAE;AACrB,YAAA,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,mBAAmB;YAClD,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;AAC3D,YAAA,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,mBAAmB;;AAGpD,QAAA,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,QAAuB,KAAI;AACrD,YAAA,OAAO,CAAC,aAAa,EAAE,GAAG,QAAQ,CAAC;SACpC,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;;AAGtC;;AAEG;IACH,KAAK,GAAA;AACH,QAAA,IAAI,CAAC,iBAAiB,GAAG,CAAC;AAC1B,QAAA,IAAI,CAAC,mBAAmB,GAAG,CAAC;AAC5B,QAAA,IAAI,CAAC,oBAAoB,GAAG,SAAS;AACrC,QAAA,IAAI,CAAC,mBAAmB,GAAG,IAAI;AAC/B,QAAA,IAAI,CAAC,SAAS,GAAG,SAAS;AAC1B,QAAA,IAAI,CAAC,kBAAkB,GAAG,EAAE;AAC5B,QAAA,IAAI,CAAC,YAAY,GAAG,SAAS;AAC7B,QAAA,IAAI,CAAC,aAAa,GAAG,SAAS;AAC9B,QAAA,IAAI,CAAC,cAAc,GAAG,SAAS;AAC/B,QAAA,IAAI,CAAC,eAAe,GAAG,SAAS;AAChC,QAAA,IAAI,CAAC,gBAAgB,GAAG,YAAY,CAAC,IAAI;AACzC,QAAA,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE;;AAGlC;;AAEG;AACH,IAAA,8BAA8B,CAAC,YAAoC,EAAA;AACjE,QAAA,IAAI,IAAI,CAAC,iBAAiB,GAAG,CAAC,EAAE;;YAE9B,MAAM,UAAU,GAA2B,EAAE;AAC7C,YAAA,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE;gBACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC;AAC/B,gBAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;AACjB,oBAAA,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,wBAAA,KAAK,IAAI,KAAK,KAAK,CAAC,GAAG,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;;;AAGxD,YAAA,IAAI,CAAC,kBAAkB,GAAG,UAAU;;aAC/B;AACL,YAAA,IAAI,CAAC,kBAAkB,GAAG,EAAE,GAAG,YAAY,EAAE;;;AAIjD;;;AAGG;IACH,MAAM,0BAA0B,CAC9B,YAA4B,EAAA;QAE5B,IAAI,UAAU,GAAG,CAAC;AAClB,QAAA,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE;gBAC7B,MAAM,WAAW,GAAG,IAA+B;AACnD,gBAAA,IACE,WAAW,CAAC,MAAM,IAAI,IAAI;AAC1B,oBAAA,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC;AACA,oBAAA,MAAM,MAAM,GAAG,WAAW,CAAC,MAE1B;AACD,oBAAA,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CACpC,WAAW,CAAC,WAAsB,IAAI,EAAE,CAC1C;AACD,oBAAA,MAAM,UAAU,GAAG,eAAe,CAChC,eAAwD,EACvD,WAAW,CAAC,IAAe,IAAI,EAAE,CACnC;AACD,oBAAA,UAAU,IAAI,YAAY,CACxB,IAAI,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAC9C;;;;;AAMP,QAAA,IAAI,CAAC,iBAAiB,IAAI,UAAU;;AAGtC;;;;AAIG;IACH,uBAAuB,CAAC,eAAwB,IAAI,EAAA;AAClD,QAAA,MAAM,QAAQ,GAAqB,IAAI,GAAG,EAAE;AAE5C,QAAA,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;AACtB,YAAA,OAAO,QAAQ;;QAGjB,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE;YAC/C,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC,aAAa,KAAK,IAAI,EAAE;AACnD,gBAAA,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC;;;AAI/B,QAAA,OAAO,QAAQ;;AAGjB;;;;;;AAMG;AACH,IAAA,qBAAqB,CAAC,SAAmB,EAAA;QACvC,IAAI,iBAAiB,GAAG,KAAK;AAC7B,QAAA,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE;YAC5B,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;AACvC,gBAAA,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC;gBAClC,iBAAiB,GAAG,IAAI;;;QAG5B,IAAI,iBAAiB,EAAE;AACrB,YAAA,IAAI,CAAC,mBAAmB,GAAG,IAAI;;AAEjC,QAAA,OAAO,iBAAiB;;AAG1B;;;;;;AAMG;IACH,kBAAkB,GAAA;QAChB,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACrC,OAAO,IAAI,CAAC,KAAK;;QAGnB,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,KAAI;AAChD,YAAA,IAAI,EAAE,MAAM,IAAI,IAAI,CAAC,EAAE;gBACrB,OAAO,IAAI,CAAC;;AAGd,YAAA,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YACjD,IAAI,CAAC,OAAO,EAAE;gBACZ,OAAO,IAAI,CAAC;;;YAId,IAAI,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;;gBAE3C,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,IAAI,CAAC,QAAQ,CAAC;AAC5D,gBAAA,OAAO,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC;;;YAI1C,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,IAAI,CAAC,QAAQ,CAAC;AAC5D,YAAA,QACE,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,aAAa,KAAK,IAAI;AAEvE,SAAC,CAAC;AAEF,QAAA,OAAO,cAAc;;AAExB;;;;"}

package/dist/esm/common/enum.mjs

@@ -143,6 +143,8 @@ var Constants;
 (function (Constants) {
     Constants["OFFICIAL_CODE_BASEURL"] = "https://api.illuma.ai/v1";
     Constants["EXECUTE_CODE"] = "execute_code";
+    Constants["TOOL_SEARCH_REGEX"] = "tool_search_regex";
+    Constants["PROGRAMMATIC_TOOL_CALLING"] = "run_tools_with_code";
     Constants["WEB_SEARCH"] = "web_search";
     Constants["CONTENT_AND_ARTIFACT"] = "content_and_artifact";
     Constants["LC_TRANSFER_TO_"] = "lc_transfer_to_";