@librechat/agents 3.2.32 → 3.2.33
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.
- package/dist/cjs/_virtual/_rolldown/runtime.cjs +23 -0
- package/dist/cjs/agents/AgentContext.cjs +844 -1046
- package/dist/cjs/agents/AgentContext.cjs.map +1 -1
- package/dist/cjs/common/constants.cjs +13 -13
- package/dist/cjs/common/constants.cjs.map +1 -1
- package/dist/cjs/common/enum.cjs +233 -240
- package/dist/cjs/common/enum.cjs.map +1 -1
- package/dist/cjs/common/index.cjs +2 -0
- package/dist/cjs/events.cjs +121 -169
- package/dist/cjs/events.cjs.map +1 -1
- package/dist/cjs/graphs/Graph.cjs +1389 -1807
- package/dist/cjs/graphs/Graph.cjs.map +1 -1
- package/dist/cjs/graphs/MultiAgentGraph.cjs +713 -945
- package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
- package/dist/cjs/graphs/index.cjs +2 -0
- package/dist/cjs/hitl/askUserQuestion.cjs +60 -62
- package/dist/cjs/hitl/askUserQuestion.cjs.map +1 -1
- package/dist/cjs/hitl/index.cjs +1 -0
- package/dist/cjs/hooks/HookRegistry.cjs +176 -202
- package/dist/cjs/hooks/HookRegistry.cjs.map +1 -1
- package/dist/cjs/hooks/createToolPolicyHook.cjs +71 -101
- package/dist/cjs/hooks/createToolPolicyHook.cjs.map +1 -1
- package/dist/cjs/hooks/createWorkspacePolicyHook.cjs +170 -273
- package/dist/cjs/hooks/createWorkspacePolicyHook.cjs.map +1 -1
- package/dist/cjs/hooks/executeHooks.cjs +227 -282
- package/dist/cjs/hooks/executeHooks.cjs.map +1 -1
- package/dist/cjs/hooks/index.cjs +6 -0
- package/dist/cjs/hooks/matchers.cjs +196 -230
- package/dist/cjs/hooks/matchers.cjs.map +1 -1
- package/dist/cjs/hooks/types.cjs +24 -24
- package/dist/cjs/hooks/types.cjs.map +1 -1
- package/dist/cjs/instrumentation.cjs +110 -137
- package/dist/cjs/instrumentation.cjs.map +1 -1
- package/dist/cjs/langchain/google-common.cjs +0 -3
- package/dist/cjs/langchain/index.cjs +80 -43
- package/dist/cjs/langchain/language_models/chat_models.cjs +0 -3
- package/dist/cjs/langchain/messages/tool.cjs +0 -3
- package/dist/cjs/langchain/messages.cjs +35 -18
- package/dist/cjs/langchain/openai.cjs +0 -3
- package/dist/cjs/langchain/prompts.cjs +5 -8
- package/dist/cjs/langchain/runnables.cjs +11 -10
- package/dist/cjs/langchain/tools.cjs +14 -11
- package/dist/cjs/langchain/utils/env.cjs +5 -8
- package/dist/cjs/langfuse.cjs +60 -79
- package/dist/cjs/langfuse.cjs.map +1 -1
- package/dist/cjs/langfuseToolOutputTracing.cjs +267 -399
- package/dist/cjs/langfuseToolOutputTracing.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/index.cjs +432 -562
- package/dist/cjs/llm/anthropic/index.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/types.cjs +23 -47
- package/dist/cjs/llm/anthropic/types.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs +441 -731
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs +171 -256
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/output_parsers.cjs +2 -0
- package/dist/cjs/llm/anthropic/utils/tools.cjs +12 -26
- package/dist/cjs/llm/anthropic/utils/tools.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/index.cjs +195 -240
- package/dist/cjs/llm/bedrock/index.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/toolCache.cjs +84 -106
- package/dist/cjs/llm/bedrock/toolCache.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/utils/index.cjs +2 -0
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs +357 -620
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs +105 -149
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs.map +1 -1
- package/dist/cjs/llm/fake.cjs +86 -96
- package/dist/cjs/llm/fake.cjs.map +1 -1
- package/dist/cjs/llm/google/index.cjs +183 -237
- package/dist/cjs/llm/google/index.cjs.map +1 -1
- package/dist/cjs/llm/google/utils/common.cjs +398 -674
- package/dist/cjs/llm/google/utils/common.cjs.map +1 -1
- package/dist/cjs/llm/google/utils/zod_to_genai_parameters.cjs +2 -0
- package/dist/cjs/llm/init.cjs +44 -53
- package/dist/cjs/llm/init.cjs.map +1 -1
- package/dist/cjs/llm/invoke.cjs +142 -182
- package/dist/cjs/llm/invoke.cjs.map +1 -1
- package/dist/cjs/llm/openai/index.cjs +991 -1276
- package/dist/cjs/llm/openai/index.cjs.map +1 -1
- package/dist/cjs/llm/openai/utils/index.cjs +189 -316
- package/dist/cjs/llm/openai/utils/index.cjs.map +1 -1
- package/dist/cjs/llm/openrouter/index.cjs +102 -153
- package/dist/cjs/llm/openrouter/index.cjs.map +1 -1
- package/dist/cjs/llm/openrouter/toolCache.cjs +35 -44
- package/dist/cjs/llm/openrouter/toolCache.cjs.map +1 -1
- package/dist/cjs/llm/providers.cjs +29 -37
- package/dist/cjs/llm/providers.cjs.map +1 -1
- package/dist/cjs/llm/request.cjs +20 -33
- package/dist/cjs/llm/request.cjs.map +1 -1
- package/dist/cjs/llm/vertexai/index.cjs +427 -453
- package/dist/cjs/llm/vertexai/index.cjs.map +1 -1
- package/dist/cjs/main.cjs +547 -528
- package/dist/cjs/messages/anthropicToolCache.cjs +68 -119
- package/dist/cjs/messages/anthropicToolCache.cjs.map +1 -1
- package/dist/cjs/messages/cache.cjs +305 -418
- package/dist/cjs/messages/cache.cjs.map +1 -1
- package/dist/cjs/messages/content.cjs +36 -49
- package/dist/cjs/messages/content.cjs.map +1 -1
- package/dist/cjs/messages/contextPruning.cjs +112 -145
- package/dist/cjs/messages/contextPruning.cjs.map +1 -1
- package/dist/cjs/messages/contextPruningSettings.cjs +36 -46
- package/dist/cjs/messages/contextPruningSettings.cjs.map +1 -1
- package/dist/cjs/messages/core.cjs +256 -397
- package/dist/cjs/messages/core.cjs.map +1 -1
- package/dist/cjs/messages/format.cjs +904 -1387
- package/dist/cjs/messages/format.cjs.map +1 -1
- package/dist/cjs/messages/ids.cjs +16 -20
- package/dist/cjs/messages/ids.cjs.map +1 -1
- package/dist/cjs/messages/index.cjs +12 -0
- package/dist/cjs/messages/langchain.cjs +18 -18
- package/dist/cjs/messages/langchain.cjs.map +1 -1
- package/dist/cjs/messages/prune.cjs +1054 -1517
- package/dist/cjs/messages/prune.cjs.map +1 -1
- package/dist/cjs/messages/recency.cjs +77 -95
- package/dist/cjs/messages/recency.cjs.map +1 -1
- package/dist/cjs/messages/reducer.cjs +63 -78
- package/dist/cjs/messages/reducer.cjs.map +1 -1
- package/dist/cjs/messages/tools.cjs +51 -79
- package/dist/cjs/messages/tools.cjs.map +1 -1
- package/dist/cjs/openai/index.cjs +171 -217
- package/dist/cjs/openai/index.cjs.map +1 -1
- package/dist/cjs/responses/index.cjs +302 -391
- package/dist/cjs/responses/index.cjs.map +1 -1
- package/dist/cjs/run.cjs +903 -1113
- package/dist/cjs/run.cjs.map +1 -1
- package/dist/cjs/session/AgentSession.cjs +805 -986
- package/dist/cjs/session/AgentSession.cjs.map +1 -1
- package/dist/cjs/session/JsonlSessionStore.cjs +327 -410
- package/dist/cjs/session/JsonlSessionStore.cjs.map +1 -1
- package/dist/cjs/session/handlers.cjs +192 -208
- package/dist/cjs/session/handlers.cjs.map +1 -1
- package/dist/cjs/session/ids.cjs +9 -10
- package/dist/cjs/session/ids.cjs.map +1 -1
- package/dist/cjs/session/index.cjs +4 -0
- package/dist/cjs/session/messageSerialization.cjs +94 -156
- package/dist/cjs/session/messageSerialization.cjs.map +1 -1
- package/dist/cjs/splitStream.cjs +147 -206
- package/dist/cjs/splitStream.cjs.map +1 -1
- package/dist/cjs/stream.cjs +856 -1344
- package/dist/cjs/stream.cjs.map +1 -1
- package/dist/cjs/summarization/index.cjs +57 -101
- package/dist/cjs/summarization/index.cjs.map +1 -1
- package/dist/cjs/summarization/node.cjs +643 -796
- package/dist/cjs/summarization/node.cjs.map +1 -1
- package/dist/cjs/tools/BashExecutor.cjs +110 -136
- package/dist/cjs/tools/BashExecutor.cjs.map +1 -1
- package/dist/cjs/tools/BashProgrammaticToolCalling.cjs +165 -245
- package/dist/cjs/tools/BashProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/Calculator.cjs +36 -57
- package/dist/cjs/tools/Calculator.cjs.map +1 -1
- package/dist/cjs/tools/CodeExecutor.cjs +126 -168
- package/dist/cjs/tools/CodeExecutor.cjs.map +1 -1
- package/dist/cjs/tools/CodeSessionFileSummary.cjs +36 -46
- package/dist/cjs/tools/CodeSessionFileSummary.cjs.map +1 -1
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs +459 -649
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/ReadFile.cjs +17 -20
- package/dist/cjs/tools/ReadFile.cjs.map +1 -1
- package/dist/cjs/tools/SkillTool.cjs +26 -27
- package/dist/cjs/tools/SkillTool.cjs.map +1 -1
- package/dist/cjs/tools/SubagentTool.cjs +59 -61
- package/dist/cjs/tools/SubagentTool.cjs.map +1 -1
- package/dist/cjs/tools/ToolNode.cjs +2109 -2686
- package/dist/cjs/tools/ToolNode.cjs.map +1 -1
- package/dist/cjs/tools/ToolSearch.cjs +663 -825
- package/dist/cjs/tools/ToolSearch.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareBridgeRuntime.cjs +248 -340
- package/dist/cjs/tools/cloudflare/CloudflareBridgeRuntime.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareProgrammaticToolCalling.cjs +170 -197
- package/dist/cjs/tools/cloudflare/CloudflareProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareSandboxExecutionEngine.cjs +425 -520
- package/dist/cjs/tools/cloudflare/CloudflareSandboxExecutionEngine.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareSandboxTools.cjs +91 -124
- package/dist/cjs/tools/cloudflare/CloudflareSandboxTools.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/index.cjs +4 -0
- package/dist/cjs/tools/eagerEventExecution.cjs +75 -99
- package/dist/cjs/tools/eagerEventExecution.cjs.map +1 -1
- package/dist/cjs/tools/handlers.cjs +200 -262
- package/dist/cjs/tools/handlers.cjs.map +1 -1
- package/dist/cjs/tools/local/CompileCheckTool.cjs +150 -212
- package/dist/cjs/tools/local/CompileCheckTool.cjs.map +1 -1
- package/dist/cjs/tools/local/FileCheckpointer.cjs +77 -85
- package/dist/cjs/tools/local/FileCheckpointer.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalCodingTools.cjs +763 -1022
- package/dist/cjs/tools/local/LocalCodingTools.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalExecutionEngine.cjs +666 -941
- package/dist/cjs/tools/local/LocalExecutionEngine.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalExecutionTools.cjs +49 -92
- package/dist/cjs/tools/local/LocalExecutionTools.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalProgrammaticToolCalling.cjs +286 -354
- package/dist/cjs/tools/local/LocalProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/local/attachments.cjs +108 -165
- package/dist/cjs/tools/local/attachments.cjs.map +1 -1
- package/dist/cjs/tools/local/bashAst.cjs +99 -113
- package/dist/cjs/tools/local/bashAst.cjs.map +1 -1
- package/dist/cjs/tools/local/editStrategies.cjs +126 -169
- package/dist/cjs/tools/local/editStrategies.cjs.map +1 -1
- package/dist/cjs/tools/local/index.cjs +12 -0
- package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs +136 -218
- package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs.map +1 -1
- package/dist/cjs/tools/local/syntaxCheck.cjs +142 -161
- package/dist/cjs/tools/local/syntaxCheck.cjs.map +1 -1
- package/dist/cjs/tools/local/textEncoding.cjs +25 -23
- package/dist/cjs/tools/local/textEncoding.cjs.map +1 -1
- package/dist/cjs/tools/local/workspaceFS.cjs +38 -46
- package/dist/cjs/tools/local/workspaceFS.cjs.map +1 -1
- package/dist/cjs/tools/ptcTimeout.cjs +27 -47
- package/dist/cjs/tools/ptcTimeout.cjs.map +1 -1
- package/dist/cjs/tools/schema.cjs +24 -23
- package/dist/cjs/tools/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/anthropic.cjs +24 -33
- package/dist/cjs/tools/search/anthropic.cjs.map +1 -1
- package/dist/cjs/tools/search/content.cjs +95 -137
- package/dist/cjs/tools/search/content.cjs.map +1 -1
- package/dist/cjs/tools/search/firecrawl.cjs +141 -172
- package/dist/cjs/tools/search/firecrawl.cjs.map +1 -1
- package/dist/cjs/tools/search/format.cjs +128 -196
- package/dist/cjs/tools/search/format.cjs.map +1 -1
- package/dist/cjs/tools/search/highlights.cjs +165 -232
- package/dist/cjs/tools/search/highlights.cjs.map +1 -1
- package/dist/cjs/tools/search/index.cjs +2 -0
- package/dist/cjs/tools/search/rerankers.cjs +151 -174
- package/dist/cjs/tools/search/rerankers.cjs.map +1 -1
- package/dist/cjs/tools/search/schema.cjs +40 -39
- package/dist/cjs/tools/search/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/search.cjs +428 -530
- package/dist/cjs/tools/search/search.cjs.map +1 -1
- package/dist/cjs/tools/search/serper-scraper.cjs +106 -127
- package/dist/cjs/tools/search/serper-scraper.cjs.map +1 -1
- package/dist/cjs/tools/search/tavily-scraper.cjs +129 -181
- package/dist/cjs/tools/search/tavily-scraper.cjs.map +1 -1
- package/dist/cjs/tools/search/tavily-search.cjs +295 -359
- package/dist/cjs/tools/search/tavily-search.cjs.map +1 -1
- package/dist/cjs/tools/search/tool.cjs +260 -299
- package/dist/cjs/tools/search/tool.cjs.map +1 -1
- package/dist/cjs/tools/search/utils.cjs +74 -117
- package/dist/cjs/tools/search/utils.cjs.map +1 -1
- package/dist/cjs/tools/skillCatalog.cjs +54 -72
- package/dist/cjs/tools/skillCatalog.cjs.map +1 -1
- package/dist/cjs/tools/streamedToolCallSeals.cjs +19 -36
- package/dist/cjs/tools/streamedToolCallSeals.cjs.map +1 -1
- package/dist/cjs/tools/subagent/SubagentExecutor.cjs +612 -771
- package/dist/cjs/tools/subagent/SubagentExecutor.cjs.map +1 -1
- package/dist/cjs/tools/subagent/index.cjs +1 -0
- package/dist/cjs/tools/toolOutputReferences.cjs +523 -630
- package/dist/cjs/tools/toolOutputReferences.cjs.map +1 -1
- package/dist/cjs/utils/callbacks.cjs +11 -21
- package/dist/cjs/utils/callbacks.cjs.map +1 -1
- package/dist/cjs/utils/errors.cjs +70 -95
- package/dist/cjs/utils/errors.cjs.map +1 -1
- package/dist/cjs/utils/events.cjs +32 -42
- package/dist/cjs/utils/events.cjs.map +1 -1
- package/dist/cjs/utils/graph.cjs +8 -12
- package/dist/cjs/utils/graph.cjs.map +1 -1
- package/dist/cjs/utils/handlers.cjs +60 -82
- package/dist/cjs/utils/handlers.cjs.map +1 -1
- package/dist/cjs/utils/index.cjs +9 -0
- package/dist/cjs/utils/llm.cjs +19 -27
- package/dist/cjs/utils/llm.cjs.map +1 -1
- package/dist/cjs/utils/misc.cjs +30 -46
- package/dist/cjs/utils/misc.cjs.map +1 -1
- package/dist/cjs/utils/run.cjs +50 -66
- package/dist/cjs/utils/run.cjs.map +1 -1
- package/dist/cjs/utils/schema.cjs +11 -19
- package/dist/cjs/utils/schema.cjs.map +1 -1
- package/dist/cjs/utils/title.cjs +71 -106
- package/dist/cjs/utils/title.cjs.map +1 -1
- package/dist/cjs/utils/tokens.cjs +186 -283
- package/dist/cjs/utils/tokens.cjs.map +1 -1
- package/dist/cjs/utils/truncation.cjs +95 -114
- package/dist/cjs/utils/truncation.cjs.map +1 -1
- package/dist/esm/agents/AgentContext.mjs +844 -1044
- package/dist/esm/agents/AgentContext.mjs.map +1 -1
- package/dist/esm/common/constants.mjs +13 -11
- package/dist/esm/common/constants.mjs.map +1 -1
- package/dist/esm/common/enum.mjs +221 -238
- package/dist/esm/common/enum.mjs.map +1 -1
- package/dist/esm/common/index.mjs +3 -0
- package/dist/esm/events.mjs +121 -167
- package/dist/esm/events.mjs.map +1 -1
- package/dist/esm/graphs/Graph.mjs +1388 -1804
- package/dist/esm/graphs/Graph.mjs.map +1 -1
- package/dist/esm/graphs/MultiAgentGraph.mjs +713 -943
- package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
- package/dist/esm/graphs/index.mjs +3 -0
- package/dist/esm/hitl/askUserQuestion.mjs +60 -60
- package/dist/esm/hitl/askUserQuestion.mjs.map +1 -1
- package/dist/esm/hitl/index.mjs +2 -0
- package/dist/esm/hooks/HookRegistry.mjs +176 -200
- package/dist/esm/hooks/HookRegistry.mjs.map +1 -1
- package/dist/esm/hooks/createToolPolicyHook.mjs +71 -99
- package/dist/esm/hooks/createToolPolicyHook.mjs.map +1 -1
- package/dist/esm/hooks/createWorkspacePolicyHook.mjs +170 -271
- package/dist/esm/hooks/createWorkspacePolicyHook.mjs.map +1 -1
- package/dist/esm/hooks/executeHooks.mjs +227 -280
- package/dist/esm/hooks/executeHooks.mjs.map +1 -1
- package/dist/esm/hooks/index.mjs +7 -0
- package/dist/esm/hooks/matchers.mjs +196 -228
- package/dist/esm/hooks/matchers.mjs.map +1 -1
- package/dist/esm/hooks/types.mjs +24 -22
- package/dist/esm/hooks/types.mjs.map +1 -1
- package/dist/esm/instrumentation.mjs +109 -132
- package/dist/esm/instrumentation.mjs.map +1 -1
- package/dist/esm/langchain/google-common.mjs +1 -2
- package/dist/esm/langchain/index.mjs +5 -5
- package/dist/esm/langchain/language_models/chat_models.mjs +1 -2
- package/dist/esm/langchain/messages/tool.mjs +1 -2
- package/dist/esm/langchain/messages.mjs +2 -2
- package/dist/esm/langchain/openai.mjs +1 -2
- package/dist/esm/langchain/prompts.mjs +2 -2
- package/dist/esm/langchain/runnables.mjs +2 -2
- package/dist/esm/langchain/tools.mjs +2 -2
- package/dist/esm/langchain/utils/env.mjs +2 -2
- package/dist/esm/langfuse.mjs +60 -76
- package/dist/esm/langfuse.mjs.map +1 -1
- package/dist/esm/langfuseToolOutputTracing.mjs +267 -395
- package/dist/esm/langfuseToolOutputTracing.mjs.map +1 -1
- package/dist/esm/llm/anthropic/index.mjs +432 -559
- package/dist/esm/llm/anthropic/index.mjs.map +1 -1
- package/dist/esm/llm/anthropic/types.mjs +23 -45
- package/dist/esm/llm/anthropic/types.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs +439 -725
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs +171 -253
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/output_parsers.mjs +3 -0
- package/dist/esm/llm/anthropic/utils/tools.mjs +12 -24
- package/dist/esm/llm/anthropic/utils/tools.mjs.map +1 -1
- package/dist/esm/llm/bedrock/index.mjs +195 -238
- package/dist/esm/llm/bedrock/index.mjs.map +1 -1
- package/dist/esm/llm/bedrock/toolCache.mjs +84 -104
- package/dist/esm/llm/bedrock/toolCache.mjs.map +1 -1
- package/dist/esm/llm/bedrock/utils/index.mjs +3 -0
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs +357 -618
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs.map +1 -1
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs +105 -147
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs.map +1 -1
- package/dist/esm/llm/fake.mjs +86 -94
- package/dist/esm/llm/fake.mjs.map +1 -1
- package/dist/esm/llm/google/index.mjs +183 -235
- package/dist/esm/llm/google/index.mjs.map +1 -1
- package/dist/esm/llm/google/utils/common.mjs +397 -666
- package/dist/esm/llm/google/utils/common.mjs.map +1 -1
- package/dist/esm/llm/google/utils/zod_to_genai_parameters.mjs +3 -0
- package/dist/esm/llm/init.mjs +44 -51
- package/dist/esm/llm/init.mjs.map +1 -1
- package/dist/esm/llm/invoke.mjs +142 -180
- package/dist/esm/llm/invoke.mjs.map +1 -1
- package/dist/esm/llm/openai/index.mjs +991 -1271
- package/dist/esm/llm/openai/index.mjs.map +1 -1
- package/dist/esm/llm/openai/utils/index.mjs +188 -312
- package/dist/esm/llm/openai/utils/index.mjs.map +1 -1
- package/dist/esm/llm/openrouter/index.mjs +102 -151
- package/dist/esm/llm/openrouter/index.mjs.map +1 -1
- package/dist/esm/llm/openrouter/toolCache.mjs +35 -42
- package/dist/esm/llm/openrouter/toolCache.mjs.map +1 -1
- package/dist/esm/llm/providers.mjs +29 -34
- package/dist/esm/llm/providers.mjs.map +1 -1
- package/dist/esm/llm/request.mjs +20 -31
- package/dist/esm/llm/request.mjs.map +1 -1
- package/dist/esm/llm/vertexai/index.mjs +427 -449
- package/dist/esm/llm/vertexai/index.mjs.map +1 -1
- package/dist/esm/main.mjs +99 -87
- package/dist/esm/messages/anthropicToolCache.mjs +68 -117
- package/dist/esm/messages/anthropicToolCache.mjs.map +1 -1
- package/dist/esm/messages/cache.mjs +305 -416
- package/dist/esm/messages/cache.mjs.map +1 -1
- package/dist/esm/messages/content.mjs +36 -47
- package/dist/esm/messages/content.mjs.map +1 -1
- package/dist/esm/messages/contextPruning.mjs +112 -143
- package/dist/esm/messages/contextPruning.mjs.map +1 -1
- package/dist/esm/messages/contextPruningSettings.mjs +36 -44
- package/dist/esm/messages/contextPruningSettings.mjs.map +1 -1
- package/dist/esm/messages/core.mjs +254 -393
- package/dist/esm/messages/core.mjs.map +1 -1
- package/dist/esm/messages/format.mjs +902 -1383
- package/dist/esm/messages/format.mjs.map +1 -1
- package/dist/esm/messages/ids.mjs +16 -18
- package/dist/esm/messages/ids.mjs.map +1 -1
- package/dist/esm/messages/index.mjs +13 -0
- package/dist/esm/messages/langchain.mjs +18 -16
- package/dist/esm/messages/langchain.mjs.map +1 -1
- package/dist/esm/messages/prune.mjs +1053 -1514
- package/dist/esm/messages/prune.mjs.map +1 -1
- package/dist/esm/messages/recency.mjs +77 -93
- package/dist/esm/messages/recency.mjs.map +1 -1
- package/dist/esm/messages/reducer.mjs +63 -76
- package/dist/esm/messages/reducer.mjs.map +1 -1
- package/dist/esm/messages/tools.mjs +49 -75
- package/dist/esm/messages/tools.mjs.map +1 -1
- package/dist/esm/openai/index.mjs +170 -215
- package/dist/esm/openai/index.mjs.map +1 -1
- package/dist/esm/responses/index.mjs +301 -389
- package/dist/esm/responses/index.mjs.map +1 -1
- package/dist/esm/run.mjs +903 -1111
- package/dist/esm/run.mjs.map +1 -1
- package/dist/esm/session/AgentSession.mjs +806 -985
- package/dist/esm/session/AgentSession.mjs.map +1 -1
- package/dist/esm/session/JsonlSessionStore.mjs +326 -407
- package/dist/esm/session/JsonlSessionStore.mjs.map +1 -1
- package/dist/esm/session/handlers.mjs +192 -206
- package/dist/esm/session/handlers.mjs.map +1 -1
- package/dist/esm/session/ids.mjs +9 -8
- package/dist/esm/session/ids.mjs.map +1 -1
- package/dist/esm/session/index.mjs +5 -0
- package/dist/esm/session/messageSerialization.mjs +94 -154
- package/dist/esm/session/messageSerialization.mjs.map +1 -1
- package/dist/esm/splitStream.mjs +147 -204
- package/dist/esm/splitStream.mjs.map +1 -1
- package/dist/esm/stream.mjs +854 -1341
- package/dist/esm/stream.mjs.map +1 -1
- package/dist/esm/summarization/index.mjs +57 -99
- package/dist/esm/summarization/index.mjs.map +1 -1
- package/dist/esm/summarization/node.mjs +640 -790
- package/dist/esm/summarization/node.mjs.map +1 -1
- package/dist/esm/tools/BashExecutor.mjs +103 -129
- package/dist/esm/tools/BashExecutor.mjs.map +1 -1
- package/dist/esm/tools/BashProgrammaticToolCalling.mjs +162 -239
- package/dist/esm/tools/BashProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/Calculator.mjs +34 -36
- package/dist/esm/tools/Calculator.mjs.map +1 -1
- package/dist/esm/tools/CodeExecutor.mjs +123 -164
- package/dist/esm/tools/CodeExecutor.mjs.map +1 -1
- package/dist/esm/tools/CodeSessionFileSummary.mjs +36 -44
- package/dist/esm/tools/CodeSessionFileSummary.mjs.map +1 -1
- package/dist/esm/tools/ProgrammaticToolCalling.mjs +454 -644
- package/dist/esm/tools/ProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/ReadFile.mjs +17 -18
- package/dist/esm/tools/ReadFile.mjs.map +1 -1
- package/dist/esm/tools/SkillTool.mjs +26 -25
- package/dist/esm/tools/SkillTool.mjs.map +1 -1
- package/dist/esm/tools/SubagentTool.mjs +59 -59
- package/dist/esm/tools/SubagentTool.mjs.map +1 -1
- package/dist/esm/tools/ToolNode.mjs +2107 -2684
- package/dist/esm/tools/ToolNode.mjs.map +1 -1
- package/dist/esm/tools/ToolSearch.mjs +659 -804
- package/dist/esm/tools/ToolSearch.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareBridgeRuntime.mjs +248 -338
- package/dist/esm/tools/cloudflare/CloudflareBridgeRuntime.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareProgrammaticToolCalling.mjs +170 -195
- package/dist/esm/tools/cloudflare/CloudflareProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareSandboxExecutionEngine.mjs +424 -517
- package/dist/esm/tools/cloudflare/CloudflareSandboxExecutionEngine.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareSandboxTools.mjs +91 -122
- package/dist/esm/tools/cloudflare/CloudflareSandboxTools.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/index.mjs +5 -0
- package/dist/esm/tools/eagerEventExecution.mjs +75 -96
- package/dist/esm/tools/eagerEventExecution.mjs.map +1 -1
- package/dist/esm/tools/handlers.mjs +200 -260
- package/dist/esm/tools/handlers.mjs.map +1 -1
- package/dist/esm/tools/local/CompileCheckTool.mjs +150 -210
- package/dist/esm/tools/local/CompileCheckTool.mjs.map +1 -1
- package/dist/esm/tools/local/FileCheckpointer.mjs +77 -83
- package/dist/esm/tools/local/FileCheckpointer.mjs.map +1 -1
- package/dist/esm/tools/local/LocalCodingTools.mjs +760 -1017
- package/dist/esm/tools/local/LocalCodingTools.mjs.map +1 -1
- package/dist/esm/tools/local/LocalExecutionEngine.mjs +663 -936
- package/dist/esm/tools/local/LocalExecutionEngine.mjs.map +1 -1
- package/dist/esm/tools/local/LocalExecutionTools.mjs +49 -90
- package/dist/esm/tools/local/LocalExecutionTools.mjs.map +1 -1
- package/dist/esm/tools/local/LocalProgrammaticToolCalling.mjs +283 -349
- package/dist/esm/tools/local/LocalProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/local/attachments.mjs +108 -163
- package/dist/esm/tools/local/attachments.mjs.map +1 -1
- package/dist/esm/tools/local/bashAst.mjs +99 -111
- package/dist/esm/tools/local/bashAst.mjs.map +1 -1
- package/dist/esm/tools/local/editStrategies.mjs +126 -167
- package/dist/esm/tools/local/editStrategies.mjs.map +1 -1
- package/dist/esm/tools/local/index.mjs +13 -0
- package/dist/esm/tools/local/resolveLocalExecutionTools.mjs +136 -216
- package/dist/esm/tools/local/resolveLocalExecutionTools.mjs.map +1 -1
- package/dist/esm/tools/local/syntaxCheck.mjs +138 -155
- package/dist/esm/tools/local/syntaxCheck.mjs.map +1 -1
- package/dist/esm/tools/local/textEncoding.mjs +25 -21
- package/dist/esm/tools/local/textEncoding.mjs.map +1 -1
- package/dist/esm/tools/local/workspaceFS.mjs +38 -44
- package/dist/esm/tools/local/workspaceFS.mjs.map +1 -1
- package/dist/esm/tools/ptcTimeout.mjs +27 -42
- package/dist/esm/tools/ptcTimeout.mjs.map +1 -1
- package/dist/esm/tools/schema.mjs +24 -21
- package/dist/esm/tools/schema.mjs.map +1 -1
- package/dist/esm/tools/search/anthropic.mjs +24 -31
- package/dist/esm/tools/search/anthropic.mjs.map +1 -1
- package/dist/esm/tools/search/content.mjs +93 -116
- package/dist/esm/tools/search/content.mjs.map +1 -1
- package/dist/esm/tools/search/firecrawl.mjs +139 -169
- package/dist/esm/tools/search/firecrawl.mjs.map +1 -1
- package/dist/esm/tools/search/format.mjs +128 -194
- package/dist/esm/tools/search/format.mjs.map +1 -1
- package/dist/esm/tools/search/highlights.mjs +165 -230
- package/dist/esm/tools/search/highlights.mjs.map +1 -1
- package/dist/esm/tools/search/index.mjs +3 -0
- package/dist/esm/tools/search/rerankers.mjs +149 -168
- package/dist/esm/tools/search/rerankers.mjs.map +1 -1
- package/dist/esm/tools/search/schema.mjs +39 -37
- package/dist/esm/tools/search/schema.mjs.map +1 -1
- package/dist/esm/tools/search/search.mjs +426 -528
- package/dist/esm/tools/search/search.mjs.map +1 -1
- package/dist/esm/tools/search/serper-scraper.mjs +104 -124
- package/dist/esm/tools/search/serper-scraper.mjs.map +1 -1
- package/dist/esm/tools/search/tavily-scraper.mjs +127 -178
- package/dist/esm/tools/search/tavily-scraper.mjs.map +1 -1
- package/dist/esm/tools/search/tavily-search.mjs +293 -357
- package/dist/esm/tools/search/tavily-search.mjs.map +1 -1
- package/dist/esm/tools/search/tool.mjs +259 -297
- package/dist/esm/tools/search/tool.mjs.map +1 -1
- package/dist/esm/tools/search/utils.mjs +74 -115
- package/dist/esm/tools/search/utils.mjs.map +1 -1
- package/dist/esm/tools/skillCatalog.mjs +54 -70
- package/dist/esm/tools/skillCatalog.mjs.map +1 -1
- package/dist/esm/tools/streamedToolCallSeals.mjs +19 -31
- package/dist/esm/tools/streamedToolCallSeals.mjs.map +1 -1
- package/dist/esm/tools/subagent/SubagentExecutor.mjs +612 -768
- package/dist/esm/tools/subagent/SubagentExecutor.mjs.map +1 -1
- package/dist/esm/tools/subagent/index.mjs +2 -0
- package/dist/esm/tools/toolOutputReferences.mjs +523 -624
- package/dist/esm/tools/toolOutputReferences.mjs.map +1 -1
- package/dist/esm/utils/callbacks.mjs +11 -19
- package/dist/esm/utils/callbacks.mjs.map +1 -1
- package/dist/esm/utils/errors.mjs +70 -93
- package/dist/esm/utils/errors.mjs.map +1 -1
- package/dist/esm/utils/events.mjs +32 -40
- package/dist/esm/utils/events.mjs.map +1 -1
- package/dist/esm/utils/graph.mjs +8 -10
- package/dist/esm/utils/graph.mjs.map +1 -1
- package/dist/esm/utils/handlers.mjs +60 -80
- package/dist/esm/utils/handlers.mjs.map +1 -1
- package/dist/esm/utils/index.mjs +10 -0
- package/dist/esm/utils/llm.mjs +19 -25
- package/dist/esm/utils/llm.mjs.map +1 -1
- package/dist/esm/utils/misc.mjs +30 -44
- package/dist/esm/utils/misc.mjs.map +1 -1
- package/dist/esm/utils/run.mjs +50 -64
- package/dist/esm/utils/run.mjs.map +1 -1
- package/dist/esm/utils/schema.mjs +11 -17
- package/dist/esm/utils/schema.mjs.map +1 -1
- package/dist/esm/utils/title.mjs +71 -104
- package/dist/esm/utils/title.mjs.map +1 -1
- package/dist/esm/utils/tokens.mjs +186 -281
- package/dist/esm/utils/tokens.mjs.map +1 -1
- package/dist/esm/utils/truncation.mjs +95 -112
- package/dist/esm/utils/truncation.mjs.map +1 -1
- package/dist/types/tools/search/tool.d.ts +17 -0
- package/dist/types/tools/search/types.d.ts +4 -0
- package/package.json +4 -10
- package/src/tools/search/highlights.ts +9 -1
- package/src/tools/search/search.ts +41 -3
- package/src/tools/search/source-processing.test.ts +373 -0
- package/src/tools/search/tool.ts +22 -2
- package/src/tools/search/types.ts +4 -0
- package/dist/cjs/langchain/google-common.cjs.map +0 -1
- package/dist/cjs/langchain/index.cjs.map +0 -1
- package/dist/cjs/langchain/language_models/chat_models.cjs.map +0 -1
- package/dist/cjs/langchain/messages/tool.cjs.map +0 -1
- package/dist/cjs/langchain/messages.cjs.map +0 -1
- package/dist/cjs/langchain/openai.cjs.map +0 -1
- package/dist/cjs/langchain/prompts.cjs.map +0 -1
- package/dist/cjs/langchain/runnables.cjs.map +0 -1
- package/dist/cjs/langchain/tools.cjs.map +0 -1
- package/dist/cjs/langchain/utils/env.cjs.map +0 -1
- package/dist/cjs/main.cjs.map +0 -1
- package/dist/esm/langchain/google-common.mjs.map +0 -1
- package/dist/esm/langchain/index.mjs.map +0 -1
- package/dist/esm/langchain/language_models/chat_models.mjs.map +0 -1
- package/dist/esm/langchain/messages/tool.mjs.map +0 -1
- package/dist/esm/langchain/messages.mjs.map +0 -1
- package/dist/esm/langchain/openai.mjs.map +0 -1
- package/dist/esm/langchain/prompts.mjs.map +0 -1
- package/dist/esm/langchain/runnables.mjs.map +0 -1
- package/dist/esm/langchain/tools.mjs.map +0 -1
- package/dist/esm/langchain/utils/env.mjs.map +0 -1
- package/dist/esm/main.mjs.map +0 -1
|
@@ -1,895 +1,750 @@
|
|
|
1
|
-
import
|
|
2
|
-
import
|
|
3
|
-
import
|
|
4
|
-
import {
|
|
5
|
-
import {
|
|
6
|
-
import
|
|
7
|
-
import {
|
|
8
|
-
|
|
9
|
-
|
|
1
|
+
import "../common/enum.mjs";
|
|
2
|
+
import "../common/index.mjs";
|
|
3
|
+
import { getCodeBaseURL } from "./CodeExecutor.mjs";
|
|
4
|
+
import { tool } from "@langchain/core/tools";
|
|
5
|
+
import { config } from "dotenv";
|
|
6
|
+
import fetch from "node-fetch";
|
|
7
|
+
import { HttpsProxyAgent } from "https-proxy-agent";
|
|
8
|
+
import * as okapibm25Module from "okapibm25";
|
|
9
|
+
//#region src/tools/ToolSearch.ts
|
|
10
10
|
function getBM25Function() {
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
if (mod.default != null && typeof mod.default.default === 'function')
|
|
17
|
-
return mod.default.default;
|
|
18
|
-
throw new Error('Could not resolve BM25 function from okapibm25 module');
|
|
11
|
+
const mod = okapibm25Module;
|
|
12
|
+
if (typeof mod === "function") return mod;
|
|
13
|
+
if (typeof mod.default === "function") return mod.default;
|
|
14
|
+
if (mod.default != null && typeof mod.default.default === "function") return mod.default.default;
|
|
15
|
+
throw new Error("Could not resolve BM25 function from okapibm25 module");
|
|
19
16
|
}
|
|
20
17
|
const BM25 = getBM25Function();
|
|
21
18
|
config();
|
|
22
19
|
/** Maximum allowed regex pattern length */
|
|
23
20
|
const MAX_PATTERN_LENGTH = 200;
|
|
24
|
-
const ToolSearchToolName =
|
|
25
|
-
const ToolSearchToolDescription =
|
|
26
|
-
const QUERY_DESCRIPTION_LOCAL =
|
|
27
|
-
const QUERY_DESCRIPTION_REGEX =
|
|
28
|
-
const FIELDS_DESCRIPTION =
|
|
29
|
-
const MAX_RESULTS_DESCRIPTION =
|
|
21
|
+
const ToolSearchToolName = "tool_search";
|
|
22
|
+
const ToolSearchToolDescription = "Searches deferred tools using BM25 ranking. Multi-word queries supported. Use mcp_server param to filter by server.";
|
|
23
|
+
const QUERY_DESCRIPTION_LOCAL = "Search term to find in tool names and descriptions. Case-insensitive substring matching. Optional if mcp_server is provided.";
|
|
24
|
+
const QUERY_DESCRIPTION_REGEX = "Regex pattern to search tool names and descriptions. Optional if mcp_server is provided.";
|
|
25
|
+
const FIELDS_DESCRIPTION = "Which fields to search. Default: name and description";
|
|
26
|
+
const MAX_RESULTS_DESCRIPTION = "Maximum number of matching tools to return";
|
|
30
27
|
const DEFAULT_MAX_RESULTS = 5;
|
|
31
|
-
const MCP_SERVER_DESCRIPTION =
|
|
28
|
+
const MCP_SERVER_DESCRIPTION = "Filter to tools from specific MCP server(s). Can be a single server name or array of names. If provided without a query, lists all tools from those servers.";
|
|
32
29
|
const ToolSearchToolSchema = {
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
30
|
+
type: "object",
|
|
31
|
+
properties: {
|
|
32
|
+
query: {
|
|
33
|
+
type: "string",
|
|
34
|
+
maxLength: MAX_PATTERN_LENGTH,
|
|
35
|
+
default: "",
|
|
36
|
+
description: QUERY_DESCRIPTION_LOCAL
|
|
37
|
+
},
|
|
38
|
+
fields: {
|
|
39
|
+
type: "array",
|
|
40
|
+
items: {
|
|
41
|
+
type: "string",
|
|
42
|
+
enum: [
|
|
43
|
+
"name",
|
|
44
|
+
"description",
|
|
45
|
+
"parameters"
|
|
46
|
+
]
|
|
47
|
+
},
|
|
48
|
+
default: ["name", "description"],
|
|
49
|
+
description: FIELDS_DESCRIPTION
|
|
50
|
+
},
|
|
51
|
+
max_results: {
|
|
52
|
+
type: "integer",
|
|
53
|
+
minimum: 1,
|
|
54
|
+
maximum: 50,
|
|
55
|
+
default: DEFAULT_MAX_RESULTS,
|
|
56
|
+
description: MAX_RESULTS_DESCRIPTION
|
|
57
|
+
},
|
|
58
|
+
mcp_server: {
|
|
59
|
+
oneOf: [{ type: "string" }, {
|
|
60
|
+
type: "array",
|
|
61
|
+
items: { type: "string" }
|
|
62
|
+
}],
|
|
63
|
+
description: MCP_SERVER_DESCRIPTION
|
|
64
|
+
}
|
|
65
|
+
},
|
|
66
|
+
required: []
|
|
60
67
|
};
|
|
61
68
|
const ToolSearchToolDefinition = {
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
69
|
+
name: ToolSearchToolName,
|
|
70
|
+
description: ToolSearchToolDescription,
|
|
71
|
+
schema: ToolSearchToolSchema
|
|
65
72
|
};
|
|
66
73
|
/** Maximum allowed regex nesting depth */
|
|
67
74
|
const MAX_REGEX_COMPLEXITY = 5;
|
|
68
75
|
/** Default search timeout in milliseconds */
|
|
69
|
-
const SEARCH_TIMEOUT =
|
|
76
|
+
const SEARCH_TIMEOUT = 5e3;
|
|
70
77
|
/**
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
78
|
+
* Creates the JSON schema with dynamic query description based on mode.
|
|
79
|
+
* @param mode - The search mode determining query interpretation
|
|
80
|
+
* @returns JSON schema for tool search parameters
|
|
81
|
+
*/
|
|
75
82
|
function createToolSearchSchema(mode) {
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
83
|
+
return {
|
|
84
|
+
type: "object",
|
|
85
|
+
properties: {
|
|
86
|
+
query: {
|
|
87
|
+
type: "string",
|
|
88
|
+
maxLength: MAX_PATTERN_LENGTH,
|
|
89
|
+
default: "",
|
|
90
|
+
description: mode === "local" ? QUERY_DESCRIPTION_LOCAL : QUERY_DESCRIPTION_REGEX
|
|
91
|
+
},
|
|
92
|
+
fields: {
|
|
93
|
+
type: "array",
|
|
94
|
+
items: {
|
|
95
|
+
type: "string",
|
|
96
|
+
enum: [
|
|
97
|
+
"name",
|
|
98
|
+
"description",
|
|
99
|
+
"parameters"
|
|
100
|
+
]
|
|
101
|
+
},
|
|
102
|
+
default: ["name", "description"],
|
|
103
|
+
description: FIELDS_DESCRIPTION
|
|
104
|
+
},
|
|
105
|
+
max_results: {
|
|
106
|
+
type: "integer",
|
|
107
|
+
minimum: 1,
|
|
108
|
+
maximum: 50,
|
|
109
|
+
default: DEFAULT_MAX_RESULTS,
|
|
110
|
+
description: MAX_RESULTS_DESCRIPTION
|
|
111
|
+
},
|
|
112
|
+
mcp_server: {
|
|
113
|
+
oneOf: [{ type: "string" }, {
|
|
114
|
+
type: "array",
|
|
115
|
+
items: { type: "string" }
|
|
116
|
+
}],
|
|
117
|
+
description: MCP_SERVER_DESCRIPTION
|
|
118
|
+
}
|
|
119
|
+
},
|
|
120
|
+
required: []
|
|
121
|
+
};
|
|
109
122
|
}
|
|
110
123
|
/**
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
124
|
+
* Extracts the MCP server name from a tool name.
|
|
125
|
+
* MCP tools follow the pattern: toolName_mcp_serverName
|
|
126
|
+
* @param toolName - The full tool name
|
|
127
|
+
* @returns The server name if it's an MCP tool, undefined otherwise
|
|
128
|
+
*/
|
|
116
129
|
function extractMcpServerName(toolName) {
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
}
|
|
121
|
-
return toolName.substring(delimiterIndex + Constants.MCP_DELIMITER.length);
|
|
130
|
+
const delimiterIndex = toolName.indexOf("_mcp_");
|
|
131
|
+
if (delimiterIndex === -1) return;
|
|
132
|
+
return toolName.substring(delimiterIndex + 5);
|
|
122
133
|
}
|
|
123
134
|
/**
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
135
|
+
* Checks if a tool belongs to a specific MCP server.
|
|
136
|
+
* @param toolName - The full tool name
|
|
137
|
+
* @param serverName - The server name to match
|
|
138
|
+
* @returns True if the tool belongs to the specified server
|
|
139
|
+
*/
|
|
129
140
|
function isFromMcpServer(toolName, serverName) {
|
|
130
|
-
|
|
131
|
-
return toolServer === serverName;
|
|
141
|
+
return extractMcpServerName(toolName) === serverName;
|
|
132
142
|
}
|
|
133
143
|
/**
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
144
|
+
* Checks if a tool belongs to any of the specified MCP servers.
|
|
145
|
+
* @param toolName - The full tool name
|
|
146
|
+
* @param serverNames - Array of server names to match
|
|
147
|
+
* @returns True if the tool belongs to any of the specified servers
|
|
148
|
+
*/
|
|
139
149
|
function isFromAnyMcpServer(toolName, serverNames) {
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
}
|
|
144
|
-
return serverNames.includes(toolServer);
|
|
150
|
+
const toolServer = extractMcpServerName(toolName);
|
|
151
|
+
if (toolServer === void 0) return false;
|
|
152
|
+
return serverNames.includes(toolServer);
|
|
145
153
|
}
|
|
146
154
|
/**
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
155
|
+
* Normalizes server filter input to always be an array.
|
|
156
|
+
* @param serverFilter - String, array of strings, or undefined
|
|
157
|
+
* @returns Array of server names (empty if none specified)
|
|
158
|
+
*/
|
|
151
159
|
function normalizeServerFilter(serverFilter) {
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
if (typeof serverFilter === 'string') {
|
|
156
|
-
return serverFilter === '' ? [] : [serverFilter];
|
|
157
|
-
}
|
|
158
|
-
return serverFilter.filter((s) => s !== '');
|
|
160
|
+
if (serverFilter === void 0) return [];
|
|
161
|
+
if (typeof serverFilter === "string") return serverFilter === "" ? [] : [serverFilter];
|
|
162
|
+
return serverFilter.filter((s) => s !== "");
|
|
159
163
|
}
|
|
160
164
|
/**
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
165
|
+
* Extracts all unique MCP server names from a tool registry.
|
|
166
|
+
* @param toolRegistry - The tool registry to scan
|
|
167
|
+
* @param onlyDeferred - If true, only considers deferred tools
|
|
168
|
+
* @returns Array of unique server names, sorted alphabetically
|
|
169
|
+
*/
|
|
166
170
|
function getAvailableMcpServers(toolRegistry, onlyDeferred = true) {
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
const server = extractMcpServerName(toolDef.name);
|
|
176
|
-
if (server !== undefined && server !== '') {
|
|
177
|
-
servers.add(server);
|
|
178
|
-
}
|
|
179
|
-
}
|
|
180
|
-
return Array.from(servers).sort();
|
|
171
|
+
if (!toolRegistry) return [];
|
|
172
|
+
const servers = /* @__PURE__ */ new Set();
|
|
173
|
+
for (const [, toolDef] of toolRegistry) {
|
|
174
|
+
if (onlyDeferred && toolDef.defer_loading !== true) continue;
|
|
175
|
+
const server = extractMcpServerName(toolDef.name);
|
|
176
|
+
if (server !== void 0 && server !== "") servers.add(server);
|
|
177
|
+
}
|
|
178
|
+
return Array.from(servers).sort();
|
|
181
179
|
}
|
|
182
180
|
/**
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
181
|
+
* Escapes special regex characters in a string to use as a literal pattern.
|
|
182
|
+
* @param pattern - The string to escape
|
|
183
|
+
* @returns The escaped string safe for use in a RegExp
|
|
184
|
+
*/
|
|
187
185
|
function escapeRegexSpecialChars(pattern) {
|
|
188
|
-
|
|
186
|
+
return pattern.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
|
|
189
187
|
}
|
|
190
188
|
/**
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
189
|
+
* Counts the maximum nesting depth of groups in a regex pattern.
|
|
190
|
+
* @param pattern - The regex pattern to analyze
|
|
191
|
+
* @returns The maximum nesting depth
|
|
192
|
+
*/
|
|
195
193
|
function countNestedGroups(pattern) {
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
else if (pattern[i] === ')' && (i === 0 || pattern[i - 1] !== '\\')) {
|
|
204
|
-
currentDepth = Math.max(0, currentDepth - 1);
|
|
205
|
-
}
|
|
206
|
-
}
|
|
207
|
-
return maxDepth;
|
|
194
|
+
let maxDepth = 0;
|
|
195
|
+
let currentDepth = 0;
|
|
196
|
+
for (let i = 0; i < pattern.length; i++) if (pattern[i] === "(" && (i === 0 || pattern[i - 1] !== "\\")) {
|
|
197
|
+
currentDepth++;
|
|
198
|
+
maxDepth = Math.max(maxDepth, currentDepth);
|
|
199
|
+
} else if (pattern[i] === ")" && (i === 0 || pattern[i - 1] !== "\\")) currentDepth = Math.max(0, currentDepth - 1);
|
|
200
|
+
return maxDepth;
|
|
208
201
|
}
|
|
209
202
|
/**
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
203
|
+
* Detects nested quantifiers that can cause catastrophic backtracking.
|
|
204
|
+
* Patterns like (a+)+, (a*)*, (a+)*, etc.
|
|
205
|
+
* @param pattern - The regex pattern to check
|
|
206
|
+
* @returns True if nested quantifiers are detected
|
|
207
|
+
*/
|
|
215
208
|
function hasNestedQuantifiers(pattern) {
|
|
216
|
-
|
|
217
|
-
return nestedQuantifierPattern.test(pattern);
|
|
209
|
+
return /\([^)]*[+*][^)]*\)[+*?]/.test(pattern);
|
|
218
210
|
}
|
|
219
211
|
/**
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
212
|
+
* Checks if a regex pattern contains potentially dangerous constructs.
|
|
213
|
+
* @param pattern - The regex pattern to validate
|
|
214
|
+
* @returns True if the pattern is dangerous
|
|
215
|
+
*/
|
|
224
216
|
function isDangerousPattern(pattern) {
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
/\(\.\*\)\*/, // (.*)*
|
|
238
|
-
/\(\.\+\)\*/, // (.+)*
|
|
239
|
-
];
|
|
240
|
-
for (const dangerous of dangerousPatterns) {
|
|
241
|
-
if (dangerous.test(pattern)) {
|
|
242
|
-
return true;
|
|
243
|
-
}
|
|
244
|
-
}
|
|
245
|
-
return false;
|
|
217
|
+
if (hasNestedQuantifiers(pattern)) return true;
|
|
218
|
+
if (countNestedGroups(pattern) > MAX_REGEX_COMPLEXITY) return true;
|
|
219
|
+
for (const dangerous of [
|
|
220
|
+
/\.\{1000,\}/,
|
|
221
|
+
/\(\?=\.\{100,\}\)/,
|
|
222
|
+
/\([^)]*\|\s*\){20,}/,
|
|
223
|
+
/\(\.\*\)\+/,
|
|
224
|
+
/\(\.\+\)\+/,
|
|
225
|
+
/\(\.\*\)\*/,
|
|
226
|
+
/\(\.\+\)\*/
|
|
227
|
+
]) if (dangerous.test(pattern)) return true;
|
|
228
|
+
return false;
|
|
246
229
|
}
|
|
247
230
|
/**
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
231
|
+
* Sanitizes a regex pattern for safe execution.
|
|
232
|
+
* If the pattern is dangerous, it will be escaped to a literal string search.
|
|
233
|
+
* @param pattern - The regex pattern to sanitize
|
|
234
|
+
* @returns Object containing the safe pattern and whether it was escaped
|
|
235
|
+
*/
|
|
253
236
|
function sanitizeRegex(pattern) {
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
237
|
+
if (isDangerousPattern(pattern)) return {
|
|
238
|
+
safe: escapeRegexSpecialChars(pattern),
|
|
239
|
+
wasEscaped: true
|
|
240
|
+
};
|
|
241
|
+
try {
|
|
242
|
+
new RegExp(pattern);
|
|
243
|
+
return {
|
|
244
|
+
safe: pattern,
|
|
245
|
+
wasEscaped: false
|
|
246
|
+
};
|
|
247
|
+
} catch {
|
|
248
|
+
return {
|
|
249
|
+
safe: escapeRegexSpecialChars(pattern),
|
|
250
|
+
wasEscaped: true
|
|
251
|
+
};
|
|
252
|
+
}
|
|
270
253
|
}
|
|
271
254
|
/**
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
255
|
+
* Simplifies tool parameters for search purposes.
|
|
256
|
+
* Extracts only the essential structure needed for parameter name searching.
|
|
257
|
+
* @param parameters - The tool's JSON schema parameters
|
|
258
|
+
* @returns Simplified parameters object
|
|
259
|
+
*/
|
|
277
260
|
function simplifyParametersForSearch(parameters) {
|
|
278
|
-
|
|
279
|
-
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
properties: Object.fromEntries(Object.entries(parameters.properties).map(([key, value]) => [
|
|
285
|
-
key,
|
|
286
|
-
{ type: value.type },
|
|
287
|
-
])),
|
|
288
|
-
};
|
|
289
|
-
}
|
|
290
|
-
return { type: parameters.type };
|
|
261
|
+
if (!parameters) return;
|
|
262
|
+
if (parameters.properties) return {
|
|
263
|
+
type: parameters.type,
|
|
264
|
+
properties: Object.fromEntries(Object.entries(parameters.properties).map(([key, value]) => [key, { type: value.type }]))
|
|
265
|
+
};
|
|
266
|
+
return { type: parameters.type };
|
|
291
267
|
}
|
|
292
268
|
/**
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
269
|
+
* Tokenizes a string into lowercase words for BM25.
|
|
270
|
+
* Splits on underscores and non-alphanumeric characters for consistent matching.
|
|
271
|
+
* @param text - The text to tokenize
|
|
272
|
+
* @returns Array of lowercase tokens
|
|
273
|
+
*/
|
|
298
274
|
function tokenize(text) {
|
|
299
|
-
|
|
300
|
-
.toLowerCase()
|
|
301
|
-
.replace(/[^a-z0-9]/g, ' ')
|
|
302
|
-
.split(/\s+/)
|
|
303
|
-
.filter((token) => token.length > 0);
|
|
275
|
+
return text.toLowerCase().replace(/[^a-z0-9]/g, " ").split(/\s+/).filter((token) => token.length > 0);
|
|
304
276
|
}
|
|
305
277
|
/**
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
278
|
+
* Creates a searchable document string from tool metadata.
|
|
279
|
+
* @param tool - The tool metadata
|
|
280
|
+
* @param fields - Which fields to include
|
|
281
|
+
* @returns Combined document string for BM25
|
|
282
|
+
*/
|
|
311
283
|
function createToolDocument(tool, fields) {
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
}
|
|
324
|
-
return parts.join(' ');
|
|
284
|
+
const parts = [];
|
|
285
|
+
if (fields.includes("name")) {
|
|
286
|
+
const baseName = tool.name.replace(/_/g, " ");
|
|
287
|
+
parts.push(baseName, baseName);
|
|
288
|
+
}
|
|
289
|
+
if (fields.includes("description") && tool.description) parts.push(tool.description);
|
|
290
|
+
if (fields.includes("parameters") && tool.parameters?.properties) {
|
|
291
|
+
const paramNames = Object.keys(tool.parameters.properties).join(" ");
|
|
292
|
+
parts.push(paramNames);
|
|
293
|
+
}
|
|
294
|
+
return parts.join(" ");
|
|
325
295
|
}
|
|
326
296
|
/**
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
|
|
297
|
+
* Determines which field had the best match for a query.
|
|
298
|
+
* @param tool - The tool to check
|
|
299
|
+
* @param queryTokens - Tokenized query
|
|
300
|
+
* @param fields - Fields to check
|
|
301
|
+
* @returns The matched field and a snippet
|
|
302
|
+
*/
|
|
333
303
|
function findMatchedField(tool, queryTokens, fields) {
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
345
|
-
|
|
346
|
-
|
|
347
|
-
|
|
348
|
-
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
}
|
|
361
|
-
}
|
|
362
|
-
const fallbackSnippet = tool.description
|
|
363
|
-
? tool.description.substring(0, 100)
|
|
364
|
-
: tool.name;
|
|
365
|
-
return { field: 'unknown', snippet: fallbackSnippet };
|
|
304
|
+
if (fields.includes("name")) {
|
|
305
|
+
const nameLower = tool.name.toLowerCase();
|
|
306
|
+
for (const token of queryTokens) if (nameLower.includes(token)) return {
|
|
307
|
+
field: "name",
|
|
308
|
+
snippet: tool.name
|
|
309
|
+
};
|
|
310
|
+
}
|
|
311
|
+
if (fields.includes("description") && tool.description) {
|
|
312
|
+
const descLower = tool.description.toLowerCase();
|
|
313
|
+
for (const token of queryTokens) if (descLower.includes(token)) return {
|
|
314
|
+
field: "description",
|
|
315
|
+
snippet: tool.description.substring(0, 100)
|
|
316
|
+
};
|
|
317
|
+
}
|
|
318
|
+
if (fields.includes("parameters") && tool.parameters?.properties) {
|
|
319
|
+
const paramNames = Object.keys(tool.parameters.properties);
|
|
320
|
+
const paramLower = paramNames.join(" ").toLowerCase();
|
|
321
|
+
for (const token of queryTokens) if (paramLower.includes(token)) return {
|
|
322
|
+
field: "parameters",
|
|
323
|
+
snippet: paramNames.join(", ")
|
|
324
|
+
};
|
|
325
|
+
}
|
|
326
|
+
return {
|
|
327
|
+
field: "unknown",
|
|
328
|
+
snippet: tool.description ? tool.description.substring(0, 100) : tool.name
|
|
329
|
+
};
|
|
366
330
|
}
|
|
367
331
|
/**
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
332
|
+
* Performs BM25-based search for better relevance ranking.
|
|
333
|
+
* Uses Okapi BM25 algorithm for term frequency and document length normalization.
|
|
334
|
+
* If query is empty, returns all tools (up to maxResults) sorted alphabetically.
|
|
335
|
+
* @param tools - Array of tool metadata to search
|
|
336
|
+
* @param query - The search query (empty returns all tools)
|
|
337
|
+
* @param fields - Which fields to search
|
|
338
|
+
* @param maxResults - Maximum results to return
|
|
339
|
+
* @returns Search response with matching tools ranked by BM25 score
|
|
340
|
+
*/
|
|
377
341
|
function performLocalSearch(tools, query, fields, maxResults) {
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
|
|
398
|
-
|
|
399
|
-
|
|
400
|
-
|
|
401
|
-
|
|
402
|
-
|
|
403
|
-
|
|
404
|
-
|
|
405
|
-
|
|
406
|
-
|
|
407
|
-
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
|
|
411
|
-
|
|
412
|
-
|
|
413
|
-
|
|
414
|
-
|
|
415
|
-
|
|
416
|
-
|
|
417
|
-
|
|
418
|
-
|
|
419
|
-
|
|
420
|
-
tool_name: tools[i].name,
|
|
421
|
-
match_score: normalizedScore,
|
|
422
|
-
matched_field: field,
|
|
423
|
-
snippet,
|
|
424
|
-
});
|
|
425
|
-
}
|
|
426
|
-
}
|
|
427
|
-
results.sort((a, b) => b.match_score - a.match_score);
|
|
428
|
-
const topResults = results.slice(0, maxResults);
|
|
429
|
-
return {
|
|
430
|
-
tool_references: topResults,
|
|
431
|
-
total_tools_searched: tools.length,
|
|
432
|
-
pattern_used: query,
|
|
433
|
-
};
|
|
342
|
+
if (tools.length === 0) return {
|
|
343
|
+
tool_references: [],
|
|
344
|
+
total_tools_searched: 0,
|
|
345
|
+
pattern_used: query
|
|
346
|
+
};
|
|
347
|
+
const queryTokens = tokenize(query);
|
|
348
|
+
if (queryTokens.length === 0) return {
|
|
349
|
+
tool_references: tools.slice().sort((a, b) => a.name.localeCompare(b.name)).slice(0, maxResults).map((tool) => ({
|
|
350
|
+
tool_name: tool.name,
|
|
351
|
+
match_score: 1,
|
|
352
|
+
matched_field: "name",
|
|
353
|
+
snippet: tool.description.substring(0, 100) || tool.name
|
|
354
|
+
})),
|
|
355
|
+
total_tools_searched: tools.length,
|
|
356
|
+
pattern_used: query
|
|
357
|
+
};
|
|
358
|
+
const scores = BM25(tools.map((tool) => createToolDocument(tool, fields)), queryTokens, {
|
|
359
|
+
k1: 1.5,
|
|
360
|
+
b: .75
|
|
361
|
+
});
|
|
362
|
+
const maxScore = Math.max(...scores.filter((s) => s > 0), 1);
|
|
363
|
+
const queryLower = query.toLowerCase().trim();
|
|
364
|
+
const results = [];
|
|
365
|
+
for (let i = 0; i < tools.length; i++) if (scores[i] > 0) {
|
|
366
|
+
const { field, snippet } = findMatchedField(tools[i], queryTokens, fields);
|
|
367
|
+
let normalizedScore = Math.min(scores[i] / maxScore, 1);
|
|
368
|
+
const baseName = getBaseToolName(tools[i].name).toLowerCase();
|
|
369
|
+
if (baseName === queryLower) normalizedScore = 1;
|
|
370
|
+
else if (baseName.startsWith(queryLower)) normalizedScore = Math.max(normalizedScore, .95);
|
|
371
|
+
results.push({
|
|
372
|
+
tool_name: tools[i].name,
|
|
373
|
+
match_score: normalizedScore,
|
|
374
|
+
matched_field: field,
|
|
375
|
+
snippet
|
|
376
|
+
});
|
|
377
|
+
}
|
|
378
|
+
results.sort((a, b) => b.match_score - a.match_score);
|
|
379
|
+
return {
|
|
380
|
+
tool_references: results.slice(0, maxResults),
|
|
381
|
+
total_tools_searched: tools.length,
|
|
382
|
+
pattern_used: query
|
|
383
|
+
};
|
|
434
384
|
}
|
|
435
385
|
/**
|
|
436
|
-
|
|
437
|
-
|
|
438
|
-
|
|
439
|
-
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
386
|
+
* Generates the JavaScript search script to be executed in the sandbox.
|
|
387
|
+
* Uses plain JavaScript for maximum compatibility with the Code API.
|
|
388
|
+
* @param deferredTools - Array of tool metadata to search through
|
|
389
|
+
* @param fields - Which fields to search
|
|
390
|
+
* @param maxResults - Maximum number of results to return
|
|
391
|
+
* @param sanitizedPattern - The sanitized regex pattern
|
|
392
|
+
* @returns The JavaScript code string
|
|
393
|
+
*/
|
|
444
394
|
function generateSearchScript(deferredTools, fields, maxResults, sanitizedPattern) {
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
449
|
-
|
|
450
|
-
|
|
451
|
-
|
|
452
|
-
|
|
453
|
-
|
|
454
|
-
|
|
455
|
-
|
|
456
|
-
|
|
457
|
-
|
|
458
|
-
|
|
459
|
-
|
|
460
|
-
|
|
461
|
-
|
|
462
|
-
|
|
463
|
-
|
|
464
|
-
|
|
465
|
-
|
|
466
|
-
|
|
467
|
-
|
|
468
|
-
|
|
469
|
-
|
|
470
|
-
|
|
471
|
-
|
|
472
|
-
|
|
473
|
-
|
|
474
|
-
|
|
475
|
-
|
|
476
|
-
|
|
477
|
-
|
|
478
|
-
|
|
479
|
-
|
|
480
|
-
|
|
481
|
-
|
|
482
|
-
|
|
483
|
-
|
|
484
|
-
|
|
485
|
-
|
|
486
|
-
|
|
487
|
-
|
|
488
|
-
|
|
489
|
-
|
|
490
|
-
|
|
491
|
-
|
|
492
|
-
|
|
493
|
-
|
|
494
|
-
|
|
495
|
-
|
|
496
|
-
|
|
497
|
-
|
|
498
|
-
|
|
499
|
-
|
|
500
|
-
|
|
501
|
-
|
|
502
|
-
|
|
503
|
-
|
|
504
|
-
|
|
505
|
-
|
|
506
|
-
|
|
507
|
-
|
|
508
|
-
|
|
509
|
-
|
|
510
|
-
|
|
511
|
-
|
|
512
|
-
|
|
513
|
-
|
|
514
|
-
|
|
515
|
-
|
|
516
|
-
|
|
517
|
-
|
|
518
|
-
|
|
519
|
-
|
|
520
|
-
|
|
521
|
-
|
|
522
|
-
|
|
523
|
-
|
|
524
|
-
|
|
525
|
-
return lines.join('\n');
|
|
395
|
+
return [
|
|
396
|
+
"// Tool definitions (injected)",
|
|
397
|
+
"var tools = " + JSON.stringify(deferredTools) + ";",
|
|
398
|
+
"var searchFields = " + JSON.stringify(fields) + ";",
|
|
399
|
+
"var maxResults = " + maxResults + ";",
|
|
400
|
+
"var pattern = " + JSON.stringify(sanitizedPattern) + ";",
|
|
401
|
+
"",
|
|
402
|
+
"// Compile regex (pattern is sanitized client-side)",
|
|
403
|
+
"var regex;",
|
|
404
|
+
"try {",
|
|
405
|
+
" regex = new RegExp(pattern, 'i');",
|
|
406
|
+
"} catch (e) {",
|
|
407
|
+
" regex = new RegExp(pattern.replace(/[.*+?^${}()[\\]\\\\|]/g, \"\\\\$&\"), \"i\");",
|
|
408
|
+
"}",
|
|
409
|
+
"",
|
|
410
|
+
"// Search logic",
|
|
411
|
+
"var results = [];",
|
|
412
|
+
"",
|
|
413
|
+
"for (var j = 0; j < tools.length; j++) {",
|
|
414
|
+
" var tool = tools[j];",
|
|
415
|
+
" var bestScore = 0;",
|
|
416
|
+
" var matchedField = '';",
|
|
417
|
+
" var snippet = '';",
|
|
418
|
+
"",
|
|
419
|
+
" // Search name (highest priority)",
|
|
420
|
+
" if (searchFields.indexOf('name') >= 0 && regex.test(tool.name)) {",
|
|
421
|
+
" bestScore = 0.95;",
|
|
422
|
+
" matchedField = 'name';",
|
|
423
|
+
" snippet = tool.name;",
|
|
424
|
+
" }",
|
|
425
|
+
"",
|
|
426
|
+
" // Search description (medium priority)",
|
|
427
|
+
" if (searchFields.indexOf('description') >= 0 && tool.description && regex.test(tool.description)) {",
|
|
428
|
+
" if (bestScore === 0) {",
|
|
429
|
+
" bestScore = 0.75;",
|
|
430
|
+
" matchedField = 'description';",
|
|
431
|
+
" snippet = tool.description.substring(0, 100);",
|
|
432
|
+
" }",
|
|
433
|
+
" }",
|
|
434
|
+
"",
|
|
435
|
+
" // Search parameter names (lower priority)",
|
|
436
|
+
" if (searchFields.indexOf('parameters') >= 0 && tool.parameters && tool.parameters.properties) {",
|
|
437
|
+
" var paramNames = Object.keys(tool.parameters.properties).join(' ');",
|
|
438
|
+
" if (regex.test(paramNames)) {",
|
|
439
|
+
" if (bestScore === 0) {",
|
|
440
|
+
" bestScore = 0.60;",
|
|
441
|
+
" matchedField = 'parameters';",
|
|
442
|
+
" snippet = paramNames;",
|
|
443
|
+
" }",
|
|
444
|
+
" }",
|
|
445
|
+
" }",
|
|
446
|
+
"",
|
|
447
|
+
" if (bestScore > 0) {",
|
|
448
|
+
" results.push({",
|
|
449
|
+
" tool_name: tool.name,",
|
|
450
|
+
" match_score: bestScore,",
|
|
451
|
+
" matched_field: matchedField,",
|
|
452
|
+
" snippet: snippet",
|
|
453
|
+
" });",
|
|
454
|
+
" }",
|
|
455
|
+
"}",
|
|
456
|
+
"",
|
|
457
|
+
"// Sort by score (descending) and limit results",
|
|
458
|
+
"results.sort(function(a, b) { return b.match_score - a.match_score; });",
|
|
459
|
+
"var topResults = results.slice(0, maxResults);",
|
|
460
|
+
"",
|
|
461
|
+
"// Output as JSON",
|
|
462
|
+
"console.log(JSON.stringify({",
|
|
463
|
+
" tool_references: topResults.map(function(r) {",
|
|
464
|
+
" return {",
|
|
465
|
+
" tool_name: r.tool_name,",
|
|
466
|
+
" match_score: r.match_score,",
|
|
467
|
+
" matched_field: r.matched_field,",
|
|
468
|
+
" snippet: r.snippet",
|
|
469
|
+
" };",
|
|
470
|
+
" }),",
|
|
471
|
+
" total_tools_searched: tools.length,",
|
|
472
|
+
" pattern_used: pattern",
|
|
473
|
+
"}));"
|
|
474
|
+
].join("\n");
|
|
526
475
|
}
|
|
527
476
|
/**
|
|
528
|
-
|
|
529
|
-
|
|
530
|
-
|
|
531
|
-
|
|
477
|
+
* Parses the search results from stdout JSON.
|
|
478
|
+
* @param stdout - The stdout string containing JSON results
|
|
479
|
+
* @returns Parsed search response
|
|
480
|
+
*/
|
|
532
481
|
function parseSearchResults(stdout) {
|
|
533
|
-
|
|
534
|
-
|
|
535
|
-
return parsed;
|
|
482
|
+
const jsonMatch = stdout.trim();
|
|
483
|
+
return JSON.parse(jsonMatch);
|
|
536
484
|
}
|
|
537
485
|
/**
|
|
538
|
-
|
|
539
|
-
|
|
540
|
-
|
|
541
|
-
|
|
542
|
-
|
|
543
|
-
function formatSearchResults(searchResponse, nameFormat =
|
|
544
|
-
|
|
545
|
-
|
|
546
|
-
|
|
547
|
-
|
|
548
|
-
|
|
549
|
-
|
|
550
|
-
|
|
551
|
-
|
|
552
|
-
|
|
553
|
-
|
|
554
|
-
|
|
555
|
-
|
|
556
|
-
|
|
557
|
-
|
|
486
|
+
* Formats search results as structured JSON for efficient parsing.
|
|
487
|
+
* @param searchResponse - The parsed search response
|
|
488
|
+
* @param nameFormat - Whether to show 'full' names (tool_mcp_server) or 'base' names (tool only)
|
|
489
|
+
* @returns JSON string with search results
|
|
490
|
+
*/
|
|
491
|
+
function formatSearchResults(searchResponse, nameFormat = "full") {
|
|
492
|
+
const { tool_references, total_tools_searched, pattern_used } = searchResponse;
|
|
493
|
+
const useFullName = nameFormat === "full";
|
|
494
|
+
const output = {
|
|
495
|
+
found: tool_references.length,
|
|
496
|
+
tools: tool_references.map((ref) => ({
|
|
497
|
+
name: useFullName ? ref.tool_name : getBaseToolName(ref.tool_name),
|
|
498
|
+
score: Number(ref.match_score.toFixed(2)),
|
|
499
|
+
matched_in: ref.matched_field,
|
|
500
|
+
snippet: ref.snippet
|
|
501
|
+
})),
|
|
502
|
+
total_searched: total_tools_searched,
|
|
503
|
+
query: pattern_used
|
|
504
|
+
};
|
|
505
|
+
return JSON.stringify(output, null, 2);
|
|
558
506
|
}
|
|
559
507
|
/**
|
|
560
|
-
|
|
561
|
-
|
|
562
|
-
|
|
563
|
-
|
|
508
|
+
* Extracts the base tool name (without MCP server suffix) from a full tool name.
|
|
509
|
+
* @param toolName - The full tool name
|
|
510
|
+
* @returns The base tool name without server suffix
|
|
511
|
+
*/
|
|
564
512
|
function getBaseToolName(toolName) {
|
|
565
|
-
|
|
566
|
-
|
|
567
|
-
|
|
568
|
-
}
|
|
569
|
-
return toolName.substring(0, delimiterIndex);
|
|
513
|
+
const delimiterIndex = toolName.indexOf("_mcp_");
|
|
514
|
+
if (delimiterIndex === -1) return toolName;
|
|
515
|
+
return toolName.substring(0, delimiterIndex);
|
|
570
516
|
}
|
|
571
517
|
/**
|
|
572
|
-
|
|
573
|
-
|
|
574
|
-
|
|
575
|
-
|
|
518
|
+
* Checks whether a tool has any defined parameters in its JSON schema.
|
|
519
|
+
* @param parameters - The tool's JSON schema parameters
|
|
520
|
+
* @returns true if the tool has at least one parameter property
|
|
521
|
+
*/
|
|
576
522
|
function hasParams(parameters) {
|
|
577
|
-
|
|
578
|
-
Object.keys(parameters.properties).length > 0);
|
|
523
|
+
return parameters?.properties != null && Object.keys(parameters.properties).length > 0;
|
|
579
524
|
}
|
|
580
525
|
/**
|
|
581
|
-
|
|
582
|
-
|
|
583
|
-
|
|
584
|
-
|
|
585
|
-
|
|
586
|
-
|
|
587
|
-
|
|
588
|
-
|
|
589
|
-
|
|
526
|
+
* Generates a compact listing of deferred tools grouped by server.
|
|
527
|
+
* Format: "server: tool1, tool2(\u2026), tool3"
|
|
528
|
+
* Tools with parameters are annotated with (\u2026) to signal
|
|
529
|
+
* that the LLM should discover the schema via tool_search before calling.
|
|
530
|
+
* Non-MCP tools are grouped under "other".
|
|
531
|
+
* @param toolRegistry - The tool registry
|
|
532
|
+
* @param onlyDeferred - Whether to only include deferred tools
|
|
533
|
+
* @returns Formatted string with tools grouped by server
|
|
534
|
+
*/
|
|
590
535
|
function getDeferredToolsListing(toolRegistry, onlyDeferred) {
|
|
591
|
-
|
|
592
|
-
|
|
593
|
-
|
|
594
|
-
|
|
595
|
-
|
|
596
|
-
|
|
597
|
-
|
|
598
|
-
|
|
599
|
-
|
|
600
|
-
|
|
601
|
-
|
|
602
|
-
|
|
603
|
-
|
|
604
|
-
|
|
605
|
-
|
|
606
|
-
|
|
607
|
-
|
|
608
|
-
|
|
609
|
-
}
|
|
610
|
-
const serverNames = Object.keys(toolsByServer).sort((a, b) => {
|
|
611
|
-
if (a === 'other')
|
|
612
|
-
return 1;
|
|
613
|
-
if (b === 'other')
|
|
614
|
-
return -1;
|
|
615
|
-
return a.localeCompare(b);
|
|
616
|
-
});
|
|
617
|
-
if (serverNames.length === 0) {
|
|
618
|
-
return '';
|
|
619
|
-
}
|
|
620
|
-
const lines = serverNames.map((server) => `${server}: ${toolsByServer[server].join(', ')}`);
|
|
621
|
-
return lines.join('\n');
|
|
536
|
+
if (!toolRegistry) return "";
|
|
537
|
+
const toolsByServer = {};
|
|
538
|
+
for (const lcTool of toolRegistry.values()) {
|
|
539
|
+
if (onlyDeferred && lcTool.defer_loading !== true) continue;
|
|
540
|
+
const toolName = lcTool.name;
|
|
541
|
+
const serverName = extractMcpServerName(toolName) ?? "other";
|
|
542
|
+
const baseName = getBaseToolName(toolName);
|
|
543
|
+
const displayName = hasParams(lcTool.parameters) ? `${baseName}(\u2026)` : baseName;
|
|
544
|
+
if (!(serverName in toolsByServer)) toolsByServer[serverName] = [];
|
|
545
|
+
toolsByServer[serverName].push(displayName);
|
|
546
|
+
}
|
|
547
|
+
const serverNames = Object.keys(toolsByServer).sort((a, b) => {
|
|
548
|
+
if (a === "other") return 1;
|
|
549
|
+
if (b === "other") return -1;
|
|
550
|
+
return a.localeCompare(b);
|
|
551
|
+
});
|
|
552
|
+
if (serverNames.length === 0) return "";
|
|
553
|
+
return serverNames.map((server) => `${server}: ${toolsByServer[server].join(", ")}`).join("\n");
|
|
622
554
|
}
|
|
623
555
|
/**
|
|
624
|
-
|
|
625
|
-
|
|
626
|
-
|
|
627
|
-
|
|
628
|
-
|
|
629
|
-
|
|
630
|
-
|
|
631
|
-
function formatServerListing(tools, serverNames, nameFormat =
|
|
632
|
-
|
|
633
|
-
|
|
634
|
-
|
|
635
|
-
|
|
636
|
-
|
|
637
|
-
|
|
638
|
-
|
|
639
|
-
|
|
640
|
-
|
|
641
|
-
|
|
642
|
-
|
|
643
|
-
|
|
644
|
-
|
|
645
|
-
|
|
646
|
-
|
|
647
|
-
|
|
648
|
-
|
|
649
|
-
|
|
650
|
-
|
|
651
|
-
|
|
652
|
-
|
|
653
|
-
|
|
654
|
-
|
|
655
|
-
|
|
656
|
-
|
|
657
|
-
|
|
658
|
-
|
|
659
|
-
const output = {
|
|
660
|
-
listing_mode: true,
|
|
661
|
-
servers,
|
|
662
|
-
total_tools: tools.length,
|
|
663
|
-
tools_by_server: toolsByServer,
|
|
664
|
-
hint: `To use a tool, search for it by name (e.g., query: "${exampleToolName}") to load it.`,
|
|
665
|
-
};
|
|
666
|
-
return JSON.stringify(output, null, 2);
|
|
556
|
+
* Formats a server listing response as structured JSON.
|
|
557
|
+
* NOTE: This is a PREVIEW only - tools are NOT discovered/loaded.
|
|
558
|
+
* @param tools - Array of tool metadata from the server(s)
|
|
559
|
+
* @param serverNames - The MCP server name(s)
|
|
560
|
+
* @param nameFormat - Whether to show 'full' names (tool_mcp_server) or 'base' names (tool only)
|
|
561
|
+
* @returns JSON string showing all tools grouped by server
|
|
562
|
+
*/
|
|
563
|
+
function formatServerListing(tools, serverNames, nameFormat = "full") {
|
|
564
|
+
const servers = Array.isArray(serverNames) ? serverNames : [serverNames];
|
|
565
|
+
const useFullName = nameFormat === "full";
|
|
566
|
+
if (tools.length === 0) return JSON.stringify({
|
|
567
|
+
listing_mode: true,
|
|
568
|
+
servers,
|
|
569
|
+
total_tools: 0,
|
|
570
|
+
tools_by_server: {},
|
|
571
|
+
hint: "No tools found from the specified MCP server(s)."
|
|
572
|
+
}, null, 2);
|
|
573
|
+
const toolsByServer = {};
|
|
574
|
+
for (const tool of tools) {
|
|
575
|
+
const server = extractMcpServerName(tool.name) ?? "unknown";
|
|
576
|
+
if (!(server in toolsByServer)) toolsByServer[server] = [];
|
|
577
|
+
toolsByServer[server].push({
|
|
578
|
+
name: useFullName ? tool.name : getBaseToolName(tool.name),
|
|
579
|
+
description: tool.description.length > 100 ? tool.description.substring(0, 97) + "..." : tool.description
|
|
580
|
+
});
|
|
581
|
+
}
|
|
582
|
+
const exampleToolName = useFullName ? tools[0]?.name ?? "tool_name" : getBaseToolName(tools[0]?.name ?? "tool_name");
|
|
583
|
+
const output = {
|
|
584
|
+
listing_mode: true,
|
|
585
|
+
servers,
|
|
586
|
+
total_tools: tools.length,
|
|
587
|
+
tools_by_server: toolsByServer,
|
|
588
|
+
hint: `To use a tool, search for it by name (e.g., query: "${exampleToolName}") to load it.`
|
|
589
|
+
};
|
|
590
|
+
return JSON.stringify(output, null, 2);
|
|
667
591
|
}
|
|
668
592
|
/**
|
|
669
|
-
|
|
670
|
-
|
|
671
|
-
|
|
672
|
-
|
|
673
|
-
|
|
674
|
-
|
|
675
|
-
|
|
676
|
-
|
|
677
|
-
|
|
678
|
-
|
|
679
|
-
|
|
680
|
-
|
|
681
|
-
|
|
682
|
-
|
|
683
|
-
|
|
684
|
-
|
|
685
|
-
|
|
686
|
-
|
|
687
|
-
|
|
688
|
-
|
|
689
|
-
|
|
690
|
-
|
|
691
|
-
|
|
692
|
-
|
|
693
|
-
|
|
694
|
-
|
|
695
|
-
|
|
593
|
+
* Creates a Tool Search tool for discovering tools from a large registry.
|
|
594
|
+
*
|
|
595
|
+
* This tool enables AI agents to dynamically discover tools from a large library
|
|
596
|
+
* without loading all tool definitions into the LLM context window. The agent
|
|
597
|
+
* can search for relevant tools on-demand.
|
|
598
|
+
*
|
|
599
|
+
* **Modes:**
|
|
600
|
+
* - `code_interpreter` (default): Uses external sandbox for regex search. Safer for complex patterns.
|
|
601
|
+
* - `local`: Uses safe substring matching locally. No network call, faster, completely safe from ReDoS.
|
|
602
|
+
*
|
|
603
|
+
* The tool registry can be provided either:
|
|
604
|
+
* 1. At initialization time via params.toolRegistry
|
|
605
|
+
* 2. At runtime via config.configurable.toolRegistry when invoking
|
|
606
|
+
*
|
|
607
|
+
* @param params - Configuration parameters for the tool (toolRegistry is optional)
|
|
608
|
+
* @returns A LangChain DynamicStructuredTool for tool searching
|
|
609
|
+
*
|
|
610
|
+
* @example
|
|
611
|
+
* // Option 1: Code interpreter mode (regex via sandbox)
|
|
612
|
+
* const tool = createToolSearch({ toolRegistry });
|
|
613
|
+
* await tool.invoke({ query: 'expense.*report' });
|
|
614
|
+
*
|
|
615
|
+
* @example
|
|
616
|
+
* // Option 2: Local mode (safe substring search)
|
|
617
|
+
* const tool = createToolSearch({ mode: 'local', toolRegistry });
|
|
618
|
+
* await tool.invoke({ query: 'expense' });
|
|
619
|
+
*/
|
|
696
620
|
function createToolSearch(initParams = {}) {
|
|
697
|
-
|
|
698
|
-
|
|
699
|
-
|
|
700
|
-
|
|
701
|
-
|
|
702
|
-
|
|
703
|
-
|
|
704
|
-
const toolsListSection = deferredToolsListing.length > 0
|
|
705
|
-
? `
|
|
621
|
+
const mode = initParams.mode ?? "code_interpreter";
|
|
622
|
+
const defaultOnlyDeferred = initParams.onlyDeferred ?? true;
|
|
623
|
+
const mcpNameFormat = initParams.mcpNameFormat ?? "full";
|
|
624
|
+
const schema = createToolSearchSchema(mode);
|
|
625
|
+
const EXEC_ENDPOINT = `${initParams.baseUrl ?? getCodeBaseURL()}/exec`;
|
|
626
|
+
const deferredToolsListing = getDeferredToolsListing(initParams.toolRegistry, defaultOnlyDeferred);
|
|
627
|
+
const toolsListSection = deferredToolsListing.length > 0 ? `
|
|
706
628
|
|
|
707
629
|
Deferred tools (search to load; \u2026 = has params, search first):
|
|
708
|
-
${deferredToolsListing}`
|
|
709
|
-
|
|
710
|
-
|
|
711
|
-
|
|
712
|
-
|
|
713
|
-
|
|
714
|
-
|
|
715
|
-
|
|
716
|
-
|
|
717
|
-
|
|
630
|
+
${deferredToolsListing}` : "";
|
|
631
|
+
const mcpNote = deferredToolsListing.includes("_mcp_") || deferredToolsListing.split("\n").some((line) => !line.startsWith("other:")) ? `
|
|
632
|
+
- MCP tools use format: toolName_mcp_serverName
|
|
633
|
+
- Use mcp_server param to filter by server` : "";
|
|
634
|
+
return tool(async (rawParams, config) => {
|
|
635
|
+
const { query = "", fields = ["name", "description"], max_results = DEFAULT_MAX_RESULTS, mcp_server } = rawParams;
|
|
636
|
+
const { toolRegistry: paramToolRegistry, onlyDeferred: paramOnlyDeferred, mcpServer: paramMcpServer } = config.toolCall ?? {};
|
|
637
|
+
const toolRegistry = paramToolRegistry ?? initParams.toolRegistry;
|
|
638
|
+
const onlyDeferred = paramOnlyDeferred !== void 0 ? paramOnlyDeferred : defaultOnlyDeferred;
|
|
639
|
+
const serverFilters = normalizeServerFilter(mcp_server ?? paramMcpServer ?? initParams.mcpServer);
|
|
640
|
+
const hasServerFilter = serverFilters.length > 0;
|
|
641
|
+
if (toolRegistry == null) return ["Error: No tool registry provided. Configure toolRegistry at agent level or initialization.", {
|
|
642
|
+
tool_references: [],
|
|
643
|
+
metadata: {
|
|
644
|
+
total_searched: 0,
|
|
645
|
+
pattern: query,
|
|
646
|
+
error: "No tool registry provided"
|
|
647
|
+
}
|
|
648
|
+
}];
|
|
649
|
+
const deferredTools = Array.from(toolRegistry.values()).filter((lcTool) => {
|
|
650
|
+
if (onlyDeferred === true && lcTool.defer_loading !== true) return false;
|
|
651
|
+
if (hasServerFilter && !isFromAnyMcpServer(lcTool.name, serverFilters)) return false;
|
|
652
|
+
return true;
|
|
653
|
+
}).map((lcTool) => ({
|
|
654
|
+
name: lcTool.name,
|
|
655
|
+
description: lcTool.description ?? "",
|
|
656
|
+
parameters: simplifyParametersForSearch(lcTool.parameters)
|
|
657
|
+
}));
|
|
658
|
+
if (deferredTools.length === 0) return [`No tools available to search${hasServerFilter ? ` from MCP server(s): ${serverFilters.join(", ")}` : ""}. The tool registry is empty or no matching deferred tools are registered.`, {
|
|
659
|
+
tool_references: [],
|
|
660
|
+
metadata: {
|
|
661
|
+
total_searched: 0,
|
|
662
|
+
pattern: query,
|
|
663
|
+
mcp_server: serverFilters
|
|
664
|
+
}
|
|
665
|
+
}];
|
|
666
|
+
if (hasServerFilter && query === "") return [formatServerListing(deferredTools, serverFilters, mcpNameFormat), {
|
|
667
|
+
tool_references: [],
|
|
668
|
+
metadata: {
|
|
669
|
+
total_available: deferredTools.length,
|
|
670
|
+
mcp_server: serverFilters,
|
|
671
|
+
listing_mode: true
|
|
672
|
+
}
|
|
673
|
+
}];
|
|
674
|
+
if (mode === "local") {
|
|
675
|
+
const searchResponse = performLocalSearch(deferredTools, query, fields, max_results);
|
|
676
|
+
return [formatSearchResults(searchResponse, mcpNameFormat), {
|
|
677
|
+
tool_references: searchResponse.tool_references,
|
|
678
|
+
metadata: {
|
|
679
|
+
total_searched: searchResponse.total_tools_searched,
|
|
680
|
+
pattern: searchResponse.pattern_used,
|
|
681
|
+
mcp_server: serverFilters.length > 0 ? serverFilters : void 0
|
|
682
|
+
}
|
|
683
|
+
}];
|
|
684
|
+
}
|
|
685
|
+
const { safe: sanitizedPattern, wasEscaped } = sanitizeRegex(query);
|
|
686
|
+
let warningMessage = "";
|
|
687
|
+
if (wasEscaped) warningMessage = "Note: The provided pattern was converted to a literal search for safety.\n\n";
|
|
688
|
+
const postData = {
|
|
689
|
+
lang: "js",
|
|
690
|
+
code: generateSearchScript(deferredTools, fields, max_results, sanitizedPattern),
|
|
691
|
+
timeout: SEARCH_TIMEOUT
|
|
692
|
+
};
|
|
693
|
+
try {
|
|
694
|
+
const fetchOptions = {
|
|
695
|
+
method: "POST",
|
|
696
|
+
headers: {
|
|
697
|
+
"Content-Type": "application/json",
|
|
698
|
+
"User-Agent": "LibreChat/1.0"
|
|
699
|
+
},
|
|
700
|
+
body: JSON.stringify(postData)
|
|
701
|
+
};
|
|
702
|
+
if (process.env.PROXY != null && process.env.PROXY !== "") fetchOptions.agent = new HttpsProxyAgent(process.env.PROXY);
|
|
703
|
+
const response = await fetch(EXEC_ENDPOINT, fetchOptions);
|
|
704
|
+
if (!response.ok) throw new Error(`HTTP error! status: ${response.status}`);
|
|
705
|
+
const result = await response.json();
|
|
706
|
+
if (result.stderr && result.stderr.trim()) console.warn("[ToolSearch] stderr:", result.stderr);
|
|
707
|
+
if (!result.stdout || !result.stdout.trim()) return [`${warningMessage}No tools matched the pattern "${sanitizedPattern}".\nTotal tools searched: ${deferredTools.length}`, {
|
|
708
|
+
tool_references: [],
|
|
709
|
+
metadata: {
|
|
710
|
+
total_searched: deferredTools.length,
|
|
711
|
+
pattern: sanitizedPattern
|
|
712
|
+
}
|
|
713
|
+
}];
|
|
714
|
+
const searchResponse = parseSearchResults(result.stdout);
|
|
715
|
+
return [`${warningMessage}${formatSearchResults(searchResponse, mcpNameFormat)}`, {
|
|
716
|
+
tool_references: searchResponse.tool_references,
|
|
717
|
+
metadata: {
|
|
718
|
+
total_searched: searchResponse.total_tools_searched,
|
|
719
|
+
pattern: searchResponse.pattern_used
|
|
720
|
+
}
|
|
721
|
+
}];
|
|
722
|
+
} catch (error) {
|
|
723
|
+
console.error("[ToolSearch] Error:", error);
|
|
724
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
725
|
+
return [`Tool search failed: ${errorMessage}\n\nSuggestion: Try a simpler search pattern or search for specific tool names.`, {
|
|
726
|
+
tool_references: [],
|
|
727
|
+
metadata: {
|
|
728
|
+
total_searched: 0,
|
|
729
|
+
pattern: sanitizedPattern,
|
|
730
|
+
error: errorMessage
|
|
731
|
+
}
|
|
732
|
+
}];
|
|
733
|
+
}
|
|
734
|
+
}, {
|
|
735
|
+
name: "tool_search",
|
|
736
|
+
description: mode === "local" ? `
|
|
718
737
|
Searches deferred tools using BM25 ranking. Multi-word queries supported.
|
|
719
738
|
${mcpNote}${toolsListSection}
|
|
720
|
-
`.trim()
|
|
721
|
-
: `
|
|
739
|
+
`.trim() : `
|
|
722
740
|
Searches deferred tools by regex pattern.
|
|
723
741
|
${mcpNote}${toolsListSection}
|
|
724
|
-
`.trim()
|
|
725
|
-
|
|
726
|
-
|
|
727
|
-
|
|
728
|
-
const { toolRegistry: paramToolRegistry, onlyDeferred: paramOnlyDeferred, mcpServer: paramMcpServer, } = config.toolCall ?? {};
|
|
729
|
-
const toolRegistry = paramToolRegistry ?? initParams.toolRegistry;
|
|
730
|
-
const onlyDeferred = paramOnlyDeferred !== undefined
|
|
731
|
-
? paramOnlyDeferred
|
|
732
|
-
: defaultOnlyDeferred;
|
|
733
|
-
const rawServerFilter = mcp_server ?? paramMcpServer ?? initParams.mcpServer;
|
|
734
|
-
const serverFilters = normalizeServerFilter(rawServerFilter);
|
|
735
|
-
const hasServerFilter = serverFilters.length > 0;
|
|
736
|
-
if (toolRegistry == null) {
|
|
737
|
-
return [
|
|
738
|
-
'Error: No tool registry provided. Configure toolRegistry at agent level or initialization.',
|
|
739
|
-
{
|
|
740
|
-
tool_references: [],
|
|
741
|
-
metadata: {
|
|
742
|
-
total_searched: 0,
|
|
743
|
-
pattern: query,
|
|
744
|
-
error: 'No tool registry provided',
|
|
745
|
-
},
|
|
746
|
-
},
|
|
747
|
-
];
|
|
748
|
-
}
|
|
749
|
-
const toolsArray = Array.from(toolRegistry.values());
|
|
750
|
-
const deferredTools = toolsArray
|
|
751
|
-
.filter((lcTool) => {
|
|
752
|
-
if (onlyDeferred === true && lcTool.defer_loading !== true) {
|
|
753
|
-
return false;
|
|
754
|
-
}
|
|
755
|
-
if (hasServerFilter &&
|
|
756
|
-
!isFromAnyMcpServer(lcTool.name, serverFilters)) {
|
|
757
|
-
return false;
|
|
758
|
-
}
|
|
759
|
-
return true;
|
|
760
|
-
})
|
|
761
|
-
.map((lcTool) => ({
|
|
762
|
-
name: lcTool.name,
|
|
763
|
-
description: lcTool.description ?? '',
|
|
764
|
-
parameters: simplifyParametersForSearch(lcTool.parameters),
|
|
765
|
-
}));
|
|
766
|
-
if (deferredTools.length === 0) {
|
|
767
|
-
const serverMsg = hasServerFilter
|
|
768
|
-
? ` from MCP server(s): ${serverFilters.join(', ')}`
|
|
769
|
-
: '';
|
|
770
|
-
return [
|
|
771
|
-
`No tools available to search${serverMsg}. The tool registry is empty or no matching deferred tools are registered.`,
|
|
772
|
-
{
|
|
773
|
-
tool_references: [],
|
|
774
|
-
metadata: {
|
|
775
|
-
total_searched: 0,
|
|
776
|
-
pattern: query,
|
|
777
|
-
mcp_server: serverFilters,
|
|
778
|
-
},
|
|
779
|
-
},
|
|
780
|
-
];
|
|
781
|
-
}
|
|
782
|
-
const isServerListing = hasServerFilter && query === '';
|
|
783
|
-
if (isServerListing) {
|
|
784
|
-
const formattedOutput = formatServerListing(deferredTools, serverFilters, mcpNameFormat);
|
|
785
|
-
return [
|
|
786
|
-
formattedOutput,
|
|
787
|
-
{
|
|
788
|
-
tool_references: [],
|
|
789
|
-
metadata: {
|
|
790
|
-
total_available: deferredTools.length,
|
|
791
|
-
mcp_server: serverFilters,
|
|
792
|
-
listing_mode: true,
|
|
793
|
-
},
|
|
794
|
-
},
|
|
795
|
-
];
|
|
796
|
-
}
|
|
797
|
-
if (mode === 'local') {
|
|
798
|
-
const searchResponse = performLocalSearch(deferredTools, query, fields, max_results);
|
|
799
|
-
const formattedOutput = formatSearchResults(searchResponse, mcpNameFormat);
|
|
800
|
-
return [
|
|
801
|
-
formattedOutput,
|
|
802
|
-
{
|
|
803
|
-
tool_references: searchResponse.tool_references,
|
|
804
|
-
metadata: {
|
|
805
|
-
total_searched: searchResponse.total_tools_searched,
|
|
806
|
-
pattern: searchResponse.pattern_used,
|
|
807
|
-
mcp_server: serverFilters.length > 0 ? serverFilters : undefined,
|
|
808
|
-
},
|
|
809
|
-
},
|
|
810
|
-
];
|
|
811
|
-
}
|
|
812
|
-
const { safe: sanitizedPattern, wasEscaped } = sanitizeRegex(query);
|
|
813
|
-
let warningMessage = '';
|
|
814
|
-
if (wasEscaped) {
|
|
815
|
-
warningMessage =
|
|
816
|
-
'Note: The provided pattern was converted to a literal search for safety.\n\n';
|
|
817
|
-
}
|
|
818
|
-
const searchScript = generateSearchScript(deferredTools, fields, max_results, sanitizedPattern);
|
|
819
|
-
const postData = {
|
|
820
|
-
lang: 'js',
|
|
821
|
-
code: searchScript,
|
|
822
|
-
timeout: SEARCH_TIMEOUT,
|
|
823
|
-
};
|
|
824
|
-
try {
|
|
825
|
-
const fetchOptions = {
|
|
826
|
-
method: 'POST',
|
|
827
|
-
headers: {
|
|
828
|
-
'Content-Type': 'application/json',
|
|
829
|
-
'User-Agent': 'LibreChat/1.0',
|
|
830
|
-
},
|
|
831
|
-
body: JSON.stringify(postData),
|
|
832
|
-
};
|
|
833
|
-
if (process.env.PROXY != null && process.env.PROXY !== '') {
|
|
834
|
-
fetchOptions.agent = new HttpsProxyAgent(process.env.PROXY);
|
|
835
|
-
}
|
|
836
|
-
const response = await fetch(EXEC_ENDPOINT, fetchOptions);
|
|
837
|
-
if (!response.ok) {
|
|
838
|
-
throw new Error(`HTTP error! status: ${response.status}`);
|
|
839
|
-
}
|
|
840
|
-
const result = await response.json();
|
|
841
|
-
if (result.stderr && result.stderr.trim()) {
|
|
842
|
-
// eslint-disable-next-line no-console
|
|
843
|
-
console.warn('[ToolSearch] stderr:', result.stderr);
|
|
844
|
-
}
|
|
845
|
-
if (!result.stdout || !result.stdout.trim()) {
|
|
846
|
-
return [
|
|
847
|
-
`${warningMessage}No tools matched the pattern "${sanitizedPattern}".\nTotal tools searched: ${deferredTools.length}`,
|
|
848
|
-
{
|
|
849
|
-
tool_references: [],
|
|
850
|
-
metadata: {
|
|
851
|
-
total_searched: deferredTools.length,
|
|
852
|
-
pattern: sanitizedPattern,
|
|
853
|
-
},
|
|
854
|
-
},
|
|
855
|
-
];
|
|
856
|
-
}
|
|
857
|
-
const searchResponse = parseSearchResults(result.stdout);
|
|
858
|
-
const formattedOutput = `${warningMessage}${formatSearchResults(searchResponse, mcpNameFormat)}`;
|
|
859
|
-
return [
|
|
860
|
-
formattedOutput,
|
|
861
|
-
{
|
|
862
|
-
tool_references: searchResponse.tool_references,
|
|
863
|
-
metadata: {
|
|
864
|
-
total_searched: searchResponse.total_tools_searched,
|
|
865
|
-
pattern: searchResponse.pattern_used,
|
|
866
|
-
},
|
|
867
|
-
},
|
|
868
|
-
];
|
|
869
|
-
}
|
|
870
|
-
catch (error) {
|
|
871
|
-
// eslint-disable-next-line no-console
|
|
872
|
-
console.error('[ToolSearch] Error:', error);
|
|
873
|
-
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
874
|
-
return [
|
|
875
|
-
`Tool search failed: ${errorMessage}\n\nSuggestion: Try a simpler search pattern or search for specific tool names.`,
|
|
876
|
-
{
|
|
877
|
-
tool_references: [],
|
|
878
|
-
metadata: {
|
|
879
|
-
total_searched: 0,
|
|
880
|
-
pattern: sanitizedPattern,
|
|
881
|
-
error: errorMessage,
|
|
882
|
-
},
|
|
883
|
-
},
|
|
884
|
-
];
|
|
885
|
-
}
|
|
886
|
-
}, {
|
|
887
|
-
name: Constants.TOOL_SEARCH,
|
|
888
|
-
description,
|
|
889
|
-
schema,
|
|
890
|
-
responseFormat: Constants.CONTENT_AND_ARTIFACT,
|
|
891
|
-
});
|
|
742
|
+
`.trim(),
|
|
743
|
+
schema,
|
|
744
|
+
responseFormat: "content_and_artifact"
|
|
745
|
+
});
|
|
892
746
|
}
|
|
893
|
-
|
|
747
|
+
//#endregion
|
|
894
748
|
export { ToolSearchToolDefinition, ToolSearchToolDescription, ToolSearchToolName, ToolSearchToolSchema, countNestedGroups, createToolSearch, escapeRegexSpecialChars, extractMcpServerName, formatServerListing, getAvailableMcpServers, getBaseToolName, getDeferredToolsListing, hasNestedQuantifiers, isDangerousPattern, isFromAnyMcpServer, isFromMcpServer, normalizeServerFilter, performLocalSearch, sanitizeRegex };
|
|
895
|
-
|
|
749
|
+
|
|
750
|
+
//# sourceMappingURL=ToolSearch.mjs.map
|