@librechat/agents 3.0.35 → 3.0.40

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (68) hide show
  1. package/dist/cjs/agents/AgentContext.cjs +71 -2
  2. package/dist/cjs/agents/AgentContext.cjs.map +1 -1
  3. package/dist/cjs/common/enum.cjs +2 -0
  4. package/dist/cjs/common/enum.cjs.map +1 -1
  5. package/dist/cjs/events.cjs +3 -0
  6. package/dist/cjs/events.cjs.map +1 -1
  7. package/dist/cjs/graphs/Graph.cjs +7 -2
  8. package/dist/cjs/graphs/Graph.cjs.map +1 -1
  9. package/dist/cjs/instrumentation.cjs +1 -1
  10. package/dist/cjs/instrumentation.cjs.map +1 -1
  11. package/dist/cjs/main.cjs +12 -0
  12. package/dist/cjs/main.cjs.map +1 -1
  13. package/dist/cjs/tools/ProgrammaticToolCalling.cjs +329 -0
  14. package/dist/cjs/tools/ProgrammaticToolCalling.cjs.map +1 -0
  15. package/dist/cjs/tools/ToolNode.cjs +34 -3
  16. package/dist/cjs/tools/ToolNode.cjs.map +1 -1
  17. package/dist/cjs/tools/ToolSearchRegex.cjs +455 -0
  18. package/dist/cjs/tools/ToolSearchRegex.cjs.map +1 -0
  19. package/dist/esm/agents/AgentContext.mjs +71 -2
  20. package/dist/esm/agents/AgentContext.mjs.map +1 -1
  21. package/dist/esm/common/enum.mjs +2 -0
  22. package/dist/esm/common/enum.mjs.map +1 -1
  23. package/dist/esm/events.mjs +4 -1
  24. package/dist/esm/events.mjs.map +1 -1
  25. package/dist/esm/graphs/Graph.mjs +7 -2
  26. package/dist/esm/graphs/Graph.mjs.map +1 -1
  27. package/dist/esm/instrumentation.mjs +1 -1
  28. package/dist/esm/instrumentation.mjs.map +1 -1
  29. package/dist/esm/main.mjs +2 -0
  30. package/dist/esm/main.mjs.map +1 -1
  31. package/dist/esm/tools/ProgrammaticToolCalling.mjs +324 -0
  32. package/dist/esm/tools/ProgrammaticToolCalling.mjs.map +1 -0
  33. package/dist/esm/tools/ToolNode.mjs +34 -3
  34. package/dist/esm/tools/ToolNode.mjs.map +1 -1
  35. package/dist/esm/tools/ToolSearchRegex.mjs +448 -0
  36. package/dist/esm/tools/ToolSearchRegex.mjs.map +1 -0
  37. package/dist/types/agents/AgentContext.d.ts +25 -1
  38. package/dist/types/common/enum.d.ts +2 -0
  39. package/dist/types/graphs/Graph.d.ts +2 -1
  40. package/dist/types/index.d.ts +2 -0
  41. package/dist/types/test/mockTools.d.ts +28 -0
  42. package/dist/types/tools/ProgrammaticToolCalling.d.ts +86 -0
  43. package/dist/types/tools/ToolNode.d.ts +7 -1
  44. package/dist/types/tools/ToolSearchRegex.d.ts +80 -0
  45. package/dist/types/types/graph.d.ts +7 -1
  46. package/dist/types/types/tools.d.ts +136 -0
  47. package/package.json +5 -1
  48. package/src/agents/AgentContext.ts +86 -0
  49. package/src/common/enum.ts +2 -0
  50. package/src/events.ts +5 -1
  51. package/src/graphs/Graph.ts +8 -1
  52. package/src/index.ts +2 -0
  53. package/src/instrumentation.ts +1 -1
  54. package/src/llm/google/llm.spec.ts +3 -1
  55. package/src/scripts/code_exec_ptc.ts +277 -0
  56. package/src/scripts/programmatic_exec.ts +396 -0
  57. package/src/scripts/programmatic_exec_agent.ts +231 -0
  58. package/src/scripts/tool_search_regex.ts +162 -0
  59. package/src/test/mockTools.ts +366 -0
  60. package/src/tools/ProgrammaticToolCalling.ts +423 -0
  61. package/src/tools/ToolNode.ts +38 -4
  62. package/src/tools/ToolSearchRegex.ts +535 -0
  63. package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.ts +318 -0
  64. package/src/tools/__tests__/ProgrammaticToolCalling.test.ts +613 -0
  65. package/src/tools/__tests__/ToolSearchRegex.integration.test.ts +161 -0
  66. package/src/tools/__tests__/ToolSearchRegex.test.ts +232 -0
  67. package/src/types/graph.ts +7 -1
  68. package/src/types/tools.ts +166 -0
@@ -0,0 +1,455 @@
1
+ 'use strict';
2
+
3
+ var zod = require('zod');
4
+ var dotenv = require('dotenv');
5
+ var fetch = require('node-fetch');
6
+ var httpsProxyAgent = require('https-proxy-agent');
7
+ var env = require('@langchain/core/utils/env');
8
+ var tools = require('@langchain/core/tools');
9
+ var CodeExecutor = require('./CodeExecutor.cjs');
10
+ var _enum = require('../common/enum.cjs');
11
+
12
+ // src/tools/ToolSearchRegex.ts
13
+ dotenv.config();
14
+ /** Maximum allowed regex pattern length */
15
+ const MAX_PATTERN_LENGTH = 200;
16
+ /** Maximum allowed regex nesting depth */
17
+ const MAX_REGEX_COMPLEXITY = 5;
18
+ /** Default search timeout in milliseconds */
19
+ const SEARCH_TIMEOUT = 5000;
20
+ const ToolSearchRegexSchema = zod.z.object({
21
+ query: zod.z
22
+ .string()
23
+ .min(1)
24
+ .max(MAX_PATTERN_LENGTH)
25
+ .describe('Regex pattern to search tool names and descriptions. Special regex characters will be sanitized for safety.'),
26
+ fields: zod.z
27
+ .array(zod.z.enum(['name', 'description', 'parameters']))
28
+ .optional()
29
+ .default(['name', 'description'])
30
+ .describe('Which fields to search. Default: name and description'),
31
+ max_results: zod.z
32
+ .number()
33
+ .int()
34
+ .min(1)
35
+ .max(50)
36
+ .optional()
37
+ .default(10)
38
+ .describe('Maximum number of matching tools to return'),
39
+ });
40
+ /**
41
+ * Escapes special regex characters in a string to use as a literal pattern.
42
+ * @param pattern - The string to escape
43
+ * @returns The escaped string safe for use in a RegExp
44
+ */
45
+ function escapeRegexSpecialChars(pattern) {
46
+ return pattern.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
47
+ }
48
+ /**
49
+ * Counts the maximum nesting depth of groups in a regex pattern.
50
+ * @param pattern - The regex pattern to analyze
51
+ * @returns The maximum nesting depth
52
+ */
53
+ function countNestedGroups(pattern) {
54
+ let maxDepth = 0;
55
+ let currentDepth = 0;
56
+ for (let i = 0; i < pattern.length; i++) {
57
+ if (pattern[i] === '(' && (i === 0 || pattern[i - 1] !== '\\')) {
58
+ currentDepth++;
59
+ maxDepth = Math.max(maxDepth, currentDepth);
60
+ }
61
+ else if (pattern[i] === ')' && (i === 0 || pattern[i - 1] !== '\\')) {
62
+ currentDepth = Math.max(0, currentDepth - 1);
63
+ }
64
+ }
65
+ return maxDepth;
66
+ }
67
+ /**
68
+ * Detects nested quantifiers that can cause catastrophic backtracking.
69
+ * Patterns like (a+)+, (a*)*, (a+)*, etc.
70
+ * @param pattern - The regex pattern to check
71
+ * @returns True if nested quantifiers are detected
72
+ */
73
+ function hasNestedQuantifiers(pattern) {
74
+ const nestedQuantifierPattern = /\([^)]*[+*][^)]*\)[+*?]/;
75
+ return nestedQuantifierPattern.test(pattern);
76
+ }
77
+ /**
78
+ * Checks if a regex pattern contains potentially dangerous constructs.
79
+ * @param pattern - The regex pattern to validate
80
+ * @returns True if the pattern is dangerous
81
+ */
82
+ function isDangerousPattern(pattern) {
83
+ if (hasNestedQuantifiers(pattern)) {
84
+ return true;
85
+ }
86
+ if (countNestedGroups(pattern) > MAX_REGEX_COMPLEXITY) {
87
+ return true;
88
+ }
89
+ const dangerousPatterns = [
90
+ /\.\{1000,\}/, // Excessive wildcards
91
+ /\(\?=\.\{100,\}\)/, // Runaway lookaheads
92
+ /\([^)]*\|\s*\){20,}/, // Excessive alternation (rough check)
93
+ /\(\.\*\)\+/, // (.*)+
94
+ /\(\.\+\)\+/, // (.+)+
95
+ /\(\.\*\)\*/, // (.*)*
96
+ /\(\.\+\)\*/, // (.+)*
97
+ ];
98
+ for (const dangerous of dangerousPatterns) {
99
+ if (dangerous.test(pattern)) {
100
+ return true;
101
+ }
102
+ }
103
+ return false;
104
+ }
105
+ /**
106
+ * Sanitizes a regex pattern for safe execution.
107
+ * If the pattern is dangerous, it will be escaped to a literal string search.
108
+ * @param pattern - The regex pattern to sanitize
109
+ * @returns Object containing the safe pattern and whether it was escaped
110
+ */
111
+ function sanitizeRegex(pattern) {
112
+ if (isDangerousPattern(pattern)) {
113
+ return {
114
+ safe: escapeRegexSpecialChars(pattern),
115
+ wasEscaped: true,
116
+ };
117
+ }
118
+ try {
119
+ new RegExp(pattern);
120
+ return { safe: pattern, wasEscaped: false };
121
+ }
122
+ catch {
123
+ return {
124
+ safe: escapeRegexSpecialChars(pattern),
125
+ wasEscaped: true,
126
+ };
127
+ }
128
+ }
129
+ /**
130
+ * Simplifies tool parameters for search purposes.
131
+ * Extracts only the essential structure needed for parameter name searching.
132
+ * @param parameters - The tool's JSON schema parameters
133
+ * @returns Simplified parameters object
134
+ */
135
+ function simplifyParametersForSearch(parameters) {
136
+ if (!parameters) {
137
+ return undefined;
138
+ }
139
+ if (parameters.properties) {
140
+ return {
141
+ type: parameters.type,
142
+ properties: Object.fromEntries(Object.entries(parameters.properties).map(([key, value]) => [
143
+ key,
144
+ { type: value.type },
145
+ ])),
146
+ };
147
+ }
148
+ return { type: parameters.type };
149
+ }
150
+ /**
151
+ * Generates the JavaScript search script to be executed in the sandbox.
152
+ * Uses plain JavaScript for maximum compatibility with the Code API.
153
+ * @param deferredTools - Array of tool metadata to search through
154
+ * @param fields - Which fields to search
155
+ * @param maxResults - Maximum number of results to return
156
+ * @param sanitizedPattern - The sanitized regex pattern
157
+ * @returns The JavaScript code string
158
+ */
159
+ function generateSearchScript(deferredTools, fields, maxResults, sanitizedPattern) {
160
+ const lines = [
161
+ '// Tool definitions (injected)',
162
+ 'var tools = ' + JSON.stringify(deferredTools) + ';',
163
+ 'var searchFields = ' + JSON.stringify(fields) + ';',
164
+ 'var maxResults = ' + maxResults + ';',
165
+ 'var pattern = ' + JSON.stringify(sanitizedPattern) + ';',
166
+ '',
167
+ '// Compile regex (pattern is sanitized client-side)',
168
+ 'var regex;',
169
+ 'try {',
170
+ ' regex = new RegExp(pattern, \'i\');',
171
+ '} catch (e) {',
172
+ ' regex = new RegExp(pattern.replace(/[.*+?^${}()[\\]\\\\|]/g, "\\\\$&"), "i");',
173
+ '}',
174
+ '',
175
+ '// Search logic',
176
+ 'var results = [];',
177
+ '',
178
+ 'for (var j = 0; j < tools.length; j++) {',
179
+ ' var tool = tools[j];',
180
+ ' var bestScore = 0;',
181
+ ' var matchedField = \'\';',
182
+ ' var snippet = \'\';',
183
+ '',
184
+ ' // Search name (highest priority)',
185
+ ' if (searchFields.indexOf(\'name\') >= 0 && regex.test(tool.name)) {',
186
+ ' bestScore = 0.95;',
187
+ ' matchedField = \'name\';',
188
+ ' snippet = tool.name;',
189
+ ' }',
190
+ '',
191
+ ' // Search description (medium priority)',
192
+ ' if (searchFields.indexOf(\'description\') >= 0 && tool.description && regex.test(tool.description)) {',
193
+ ' if (bestScore === 0) {',
194
+ ' bestScore = 0.75;',
195
+ ' matchedField = \'description\';',
196
+ ' snippet = tool.description.substring(0, 100);',
197
+ ' }',
198
+ ' }',
199
+ '',
200
+ ' // Search parameter names (lower priority)',
201
+ ' if (searchFields.indexOf(\'parameters\') >= 0 && tool.parameters && tool.parameters.properties) {',
202
+ ' var paramNames = Object.keys(tool.parameters.properties).join(\' \');',
203
+ ' if (regex.test(paramNames)) {',
204
+ ' if (bestScore === 0) {',
205
+ ' bestScore = 0.60;',
206
+ ' matchedField = \'parameters\';',
207
+ ' snippet = paramNames;',
208
+ ' }',
209
+ ' }',
210
+ ' }',
211
+ '',
212
+ ' if (bestScore > 0) {',
213
+ ' results.push({',
214
+ ' tool_name: tool.name,',
215
+ ' match_score: bestScore,',
216
+ ' matched_field: matchedField,',
217
+ ' snippet: snippet',
218
+ ' });',
219
+ ' }',
220
+ '}',
221
+ '',
222
+ '// Sort by score (descending) and limit results',
223
+ 'results.sort(function(a, b) { return b.match_score - a.match_score; });',
224
+ 'var topResults = results.slice(0, maxResults);',
225
+ '',
226
+ '// Output as JSON',
227
+ 'console.log(JSON.stringify({',
228
+ ' tool_references: topResults.map(function(r) {',
229
+ ' return {',
230
+ ' tool_name: r.tool_name,',
231
+ ' match_score: r.match_score,',
232
+ ' matched_field: r.matched_field,',
233
+ ' snippet: r.snippet',
234
+ ' };',
235
+ ' }),',
236
+ ' total_tools_searched: tools.length,',
237
+ ' pattern_used: pattern',
238
+ '}));',
239
+ ];
240
+ return lines.join('\n');
241
+ }
242
+ /**
243
+ * Parses the search results from stdout JSON.
244
+ * @param stdout - The stdout string containing JSON results
245
+ * @returns Parsed search response
246
+ */
247
+ function parseSearchResults(stdout) {
248
+ const jsonMatch = stdout.trim();
249
+ const parsed = JSON.parse(jsonMatch);
250
+ return parsed;
251
+ }
252
+ /**
253
+ * Formats search results into a human-readable string.
254
+ * @param searchResponse - The parsed search response
255
+ * @returns Formatted string for LLM consumption
256
+ */
257
+ function formatSearchResults(searchResponse) {
258
+ const { tool_references, total_tools_searched, pattern_used } = searchResponse;
259
+ if (tool_references.length === 0) {
260
+ return `No tools matched the pattern "${pattern_used}".\nTotal tools searched: ${total_tools_searched}`;
261
+ }
262
+ let response = `Found ${tool_references.length} matching tools:\n\n`;
263
+ for (const ref of tool_references) {
264
+ response += `- ${ref.tool_name} (score: ${ref.match_score.toFixed(2)})\n`;
265
+ response += ` Matched in: ${ref.matched_field}\n`;
266
+ response += ` Snippet: ${ref.snippet}\n\n`;
267
+ }
268
+ response += `Total tools searched: ${total_tools_searched}\n`;
269
+ response += `Pattern used: ${pattern_used}`;
270
+ return response;
271
+ }
272
+ /**
273
+ * Creates a Tool Search Regex tool for discovering tools from a large registry.
274
+ *
275
+ * This tool enables AI agents to dynamically discover tools from a large library
276
+ * without loading all tool definitions into the LLM context window. The agent
277
+ * can search for relevant tools on-demand using regex patterns.
278
+ *
279
+ * The tool registry can be provided either:
280
+ * 1. At initialization time via params.toolRegistry
281
+ * 2. At runtime via config.configurable.toolRegistry when invoking
282
+ *
283
+ * @param params - Configuration parameters for the tool (toolRegistry is optional)
284
+ * @returns A LangChain DynamicStructuredTool for tool searching
285
+ *
286
+ * @example
287
+ * // Option 1: Registry at initialization
288
+ * const tool = createToolSearchRegexTool({ apiKey, toolRegistry });
289
+ * await tool.invoke({ query: 'expense' });
290
+ *
291
+ * @example
292
+ * // Option 2: Registry at runtime
293
+ * const tool = createToolSearchRegexTool({ apiKey });
294
+ * await tool.invoke(
295
+ * { query: 'expense' },
296
+ * { configurable: { toolRegistry, onlyDeferred: true } }
297
+ * );
298
+ */
299
+ function createToolSearchRegexTool(initParams = {}) {
300
+ const apiKey = initParams[_enum.EnvVar.CODE_API_KEY] ??
301
+ initParams.apiKey ??
302
+ env.getEnvironmentVariable(_enum.EnvVar.CODE_API_KEY) ??
303
+ '';
304
+ if (!apiKey) {
305
+ throw new Error('No API key provided for tool search regex tool.');
306
+ }
307
+ const baseEndpoint = initParams.baseUrl ?? CodeExecutor.getCodeBaseURL();
308
+ const EXEC_ENDPOINT = `${baseEndpoint}/exec`;
309
+ const defaultOnlyDeferred = initParams.onlyDeferred ?? true;
310
+ const description = `
311
+ Searches through available tools to find ones matching your query pattern.
312
+
313
+ Usage:
314
+ - Provide a regex pattern to search tool names and descriptions.
315
+ - Use this when you need to discover tools for a specific task.
316
+ - Results include tool names, match quality scores, and snippets showing where the match occurred.
317
+ - Higher scores (0.9+) indicate name matches, medium scores (0.7+) indicate description matches.
318
+ `.trim();
319
+ return tools.tool(async (params, config) => {
320
+ const { query, fields = ['name', 'description'], max_results = 10, } = params;
321
+ // Extra params injected by ToolNode (follows web_search pattern)
322
+ const { toolRegistry: paramToolRegistry, onlyDeferred: paramOnlyDeferred, } = config.toolCall ?? {};
323
+ const { safe: sanitizedPattern, wasEscaped } = sanitizeRegex(query);
324
+ let warningMessage = '';
325
+ if (wasEscaped) {
326
+ warningMessage =
327
+ 'Note: The provided pattern was converted to a literal search for safety.\n\n';
328
+ }
329
+ // Priority: ToolNode injection (via config.toolCall) > initialization params
330
+ const toolRegistry = paramToolRegistry ?? initParams.toolRegistry;
331
+ const onlyDeferred = paramOnlyDeferred !== undefined
332
+ ? paramOnlyDeferred
333
+ : defaultOnlyDeferred;
334
+ if (toolRegistry == null) {
335
+ return [
336
+ `${warningMessage}Error: No tool registry provided. Configure toolRegistry at agent level or initialization.`,
337
+ {
338
+ tool_references: [],
339
+ metadata: {
340
+ total_searched: 0,
341
+ pattern: sanitizedPattern,
342
+ error: 'No tool registry provided',
343
+ },
344
+ },
345
+ ];
346
+ }
347
+ const toolsArray = Array.from(toolRegistry.values());
348
+ const deferredTools = toolsArray
349
+ .filter((lcTool) => {
350
+ if (onlyDeferred === true) {
351
+ return lcTool.defer_loading === true;
352
+ }
353
+ return true;
354
+ })
355
+ .map((lcTool) => ({
356
+ name: lcTool.name,
357
+ description: lcTool.description ?? '',
358
+ parameters: simplifyParametersForSearch(lcTool.parameters),
359
+ }));
360
+ if (deferredTools.length === 0) {
361
+ return [
362
+ `${warningMessage}No tools available to search. The tool registry is empty or no deferred tools are registered.`,
363
+ {
364
+ tool_references: [],
365
+ metadata: {
366
+ total_searched: 0,
367
+ pattern: sanitizedPattern,
368
+ },
369
+ },
370
+ ];
371
+ }
372
+ const searchScript = generateSearchScript(deferredTools, fields, max_results, sanitizedPattern);
373
+ const postData = {
374
+ lang: 'js',
375
+ code: searchScript,
376
+ timeout: SEARCH_TIMEOUT,
377
+ };
378
+ try {
379
+ const fetchOptions = {
380
+ method: 'POST',
381
+ headers: {
382
+ 'Content-Type': 'application/json',
383
+ 'User-Agent': 'LibreChat/1.0',
384
+ 'X-API-Key': apiKey,
385
+ },
386
+ body: JSON.stringify(postData),
387
+ };
388
+ if (process.env.PROXY != null && process.env.PROXY !== '') {
389
+ fetchOptions.agent = new httpsProxyAgent.HttpsProxyAgent(process.env.PROXY);
390
+ }
391
+ const response = await fetch(EXEC_ENDPOINT, fetchOptions);
392
+ if (!response.ok) {
393
+ throw new Error(`HTTP error! status: ${response.status}`);
394
+ }
395
+ const result = await response.json();
396
+ if (result.stderr && result.stderr.trim()) {
397
+ // eslint-disable-next-line no-console
398
+ console.warn('[ToolSearchRegex] stderr:', result.stderr);
399
+ }
400
+ if (!result.stdout || !result.stdout.trim()) {
401
+ return [
402
+ `${warningMessage}No tools matched the pattern "${sanitizedPattern}".\nTotal tools searched: ${deferredTools.length}`,
403
+ {
404
+ tool_references: [],
405
+ metadata: {
406
+ total_searched: deferredTools.length,
407
+ pattern: sanitizedPattern,
408
+ },
409
+ },
410
+ ];
411
+ }
412
+ const searchResponse = parseSearchResults(result.stdout);
413
+ const formattedOutput = `${warningMessage}${formatSearchResults(searchResponse)}`;
414
+ return [
415
+ formattedOutput,
416
+ {
417
+ tool_references: searchResponse.tool_references,
418
+ metadata: {
419
+ total_searched: searchResponse.total_tools_searched,
420
+ pattern: searchResponse.pattern_used,
421
+ },
422
+ },
423
+ ];
424
+ }
425
+ catch (error) {
426
+ // eslint-disable-next-line no-console
427
+ console.error('[ToolSearchRegex] Error:', error);
428
+ const errorMessage = error instanceof Error ? error.message : String(error);
429
+ return [
430
+ `Tool search failed: ${errorMessage}\n\nSuggestion: Try a simpler search pattern or search for specific tool names.`,
431
+ {
432
+ tool_references: [],
433
+ metadata: {
434
+ total_searched: 0,
435
+ pattern: sanitizedPattern,
436
+ error: errorMessage,
437
+ },
438
+ },
439
+ ];
440
+ }
441
+ }, {
442
+ name: _enum.Constants.TOOL_SEARCH_REGEX,
443
+ description,
444
+ schema: ToolSearchRegexSchema,
445
+ responseFormat: _enum.Constants.CONTENT_AND_ARTIFACT,
446
+ });
447
+ }
448
+
449
+ exports.countNestedGroups = countNestedGroups;
450
+ exports.createToolSearchRegexTool = createToolSearchRegexTool;
451
+ exports.escapeRegexSpecialChars = escapeRegexSpecialChars;
452
+ exports.hasNestedQuantifiers = hasNestedQuantifiers;
453
+ exports.isDangerousPattern = isDangerousPattern;
454
+ exports.sanitizeRegex = sanitizeRegex;
455
+ //# sourceMappingURL=ToolSearchRegex.cjs.map
@@ -0,0 +1 @@
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': 'LibreChat/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,eAAe;AAC7B,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;;;;;;;;;"}
@@ -5,6 +5,16 @@ import { ContentTypes, Providers } from '../common/enum.mjs';
5
5
 
6
6
  /* eslint-disable no-console */
7
7
  // src/agents/AgentContext.ts
8
+ /**
9
+ * Checks if a tool allows the specified caller type.
10
+ * Default is 'direct' only if allowed_callers is not specified.
11
+ */
12
+ function toolAllowsCaller(toolDef, caller) {
13
+ if (!toolDef)
14
+ return false;
15
+ const allowedCallers = toolDef.allowed_callers ?? ['direct'];
16
+ return allowedCallers.includes(caller);
17
+ }
8
18
  /**
9
19
  * Encapsulates agent-specific state that can vary between agents in a multi-agent system
10
20
  */
@@ -13,7 +23,7 @@ class AgentContext {
13
23
  * Create an AgentContext from configuration with token accounting initialization
14
24
  */
15
25
  static fromConfig(agentConfig, tokenCounter, indexTokenCountMap) {
16
- const { agentId, provider, clientOptions, tools, toolMap, toolEnd, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, } = agentConfig;
26
+ const { agentId, provider, clientOptions, tools, toolMap, toolEnd, toolRegistry, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, } = agentConfig;
17
27
  const agentContext = new AgentContext({
18
28
  agentId,
19
29
  provider,
@@ -22,6 +32,7 @@ class AgentContext {
22
32
  streamBuffer,
23
33
  tools,
24
34
  toolMap,
35
+ toolRegistry,
25
36
  instructions,
26
37
  additionalInstructions: additional_instructions,
27
38
  reasoningKey,
@@ -74,6 +85,11 @@ class AgentContext {
74
85
  tools;
75
86
  /** Tool map for this agent */
76
87
  toolMap;
88
+ /**
89
+ * Tool definitions registry (includes deferred and programmatic tool metadata).
90
+ * Used for tool search and programmatic tool calling.
91
+ */
92
+ toolRegistry;
77
93
  /** Instructions for this agent */
78
94
  instructions;
79
95
  /** Additional instructions for this agent */
@@ -94,7 +110,7 @@ class AgentContext {
94
110
  tokenCalculationPromise;
95
111
  /** Format content blocks as strings (for legacy compatibility) */
96
112
  useLegacyContent = false;
97
- constructor({ agentId, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, }) {
113
+ constructor({ agentId, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, toolRegistry, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, }) {
98
114
  this.agentId = agentId;
99
115
  this.provider = provider;
100
116
  this.clientOptions = clientOptions;
@@ -103,6 +119,7 @@ class AgentContext {
103
119
  this.tokenCounter = tokenCounter;
104
120
  this.tools = tools;
105
121
  this.toolMap = toolMap;
122
+ this.toolRegistry = toolRegistry;
106
123
  this.instructions = instructions;
107
124
  this.additionalInstructions = additionalInstructions;
108
125
  if (reasoningKey) {
@@ -214,6 +231,58 @@ class AgentContext {
214
231
  // Add tool tokens to existing instruction tokens (which may already include system message tokens)
215
232
  this.instructionTokens += toolTokens;
216
233
  }
234
+ /**
235
+ * Gets a map of tools that allow programmatic (code_execution) calling.
236
+ * Filters toolMap based on toolRegistry's allowed_callers settings.
237
+ * @returns ToolMap containing only tools that allow code_execution
238
+ */
239
+ getProgrammaticToolMap() {
240
+ const programmaticMap = new Map();
241
+ if (!this.toolMap) {
242
+ return programmaticMap;
243
+ }
244
+ for (const [name, tool] of this.toolMap) {
245
+ const toolDef = this.toolRegistry?.get(name);
246
+ if (toolAllowsCaller(toolDef, 'code_execution')) {
247
+ programmaticMap.set(name, tool);
248
+ }
249
+ }
250
+ return programmaticMap;
251
+ }
252
+ /**
253
+ * Gets tool definitions for tools that allow programmatic calling.
254
+ * Used to send to the Code API for stub generation.
255
+ * @returns Array of LCTool definitions for programmatic tools
256
+ */
257
+ getProgrammaticToolDefs() {
258
+ if (!this.toolRegistry) {
259
+ return [];
260
+ }
261
+ const defs = [];
262
+ for (const [_name, toolDef] of this.toolRegistry) {
263
+ if (toolAllowsCaller(toolDef, 'code_execution')) {
264
+ defs.push(toolDef);
265
+ }
266
+ }
267
+ return defs;
268
+ }
269
+ /**
270
+ * Gets the tool registry for deferred tools (for tool search).
271
+ * @param onlyDeferred If true, only returns tools with defer_loading=true
272
+ * @returns LCToolRegistry with tool definitions
273
+ */
274
+ getDeferredToolRegistry(onlyDeferred = true) {
275
+ const registry = new Map();
276
+ if (!this.toolRegistry) {
277
+ return registry;
278
+ }
279
+ for (const [name, toolDef] of this.toolRegistry) {
280
+ if (!onlyDeferred || toolDef.defer_loading === true) {
281
+ registry.set(name, toolDef);
282
+ }
283
+ }
284
+ return registry;
285
+ }
217
286
  }
218
287
 
219
288
  export { AgentContext };