@illuma-ai/agents 1.0.81

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