@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,533 @@
1
+ // src/types/graph.ts
2
+ import type {
3
+ START,
4
+ StateType,
5
+ UpdateType,
6
+ StateGraph,
7
+ StateGraphArgs,
8
+ StateDefinition,
9
+ CompiledStateGraph,
10
+ BinaryOperatorAggregate,
11
+ } from '@langchain/langgraph';
12
+ import type { BindToolsInput } from '@langchain/core/language_models/chat_models';
13
+ import type {
14
+ BaseMessage,
15
+ AIMessageChunk,
16
+ SystemMessage,
17
+ } from '@langchain/core/messages';
18
+ import type { RunnableConfig, Runnable } from '@langchain/core/runnables';
19
+ import type { ChatGenerationChunk } from '@langchain/core/outputs';
20
+ import type { GoogleAIToolType } from '@langchain/google-common';
21
+ import type { ToolMap, ToolEndEvent, GenericTool, LCTool } from '@/types/tools';
22
+ import type { Providers, Callback, GraphNodeKeys } from '@/common';
23
+ import type { StandardGraph, MultiAgentGraph } from '@/graphs';
24
+ import type { ClientOptions } from '@/types/llm';
25
+ import type {
26
+ RunStep,
27
+ RunStepDeltaEvent,
28
+ MessageDeltaEvent,
29
+ ReasoningDeltaEvent,
30
+ } from '@/types/stream';
31
+ import type { TokenCounter } from '@/types/run';
32
+
33
+ /** Interface for bound model with stream and invoke methods */
34
+ export interface ChatModel {
35
+ stream?: (
36
+ messages: BaseMessage[],
37
+ config?: RunnableConfig
38
+ ) => Promise<AsyncIterable<AIMessageChunk>>;
39
+ invoke: (
40
+ messages: BaseMessage[],
41
+ config?: RunnableConfig
42
+ ) => Promise<AIMessageChunk>;
43
+ }
44
+
45
+ export type GraphNode = GraphNodeKeys | typeof START;
46
+ export type ClientCallback<T extends unknown[]> = (
47
+ graph: StandardGraph,
48
+ ...args: T
49
+ ) => void;
50
+
51
+ export type ClientCallbacks = {
52
+ [Callback.TOOL_ERROR]?: ClientCallback<[Error, string]>;
53
+ [Callback.TOOL_START]?: ClientCallback<unknown[]>;
54
+ [Callback.TOOL_END]?: ClientCallback<unknown[]>;
55
+ };
56
+
57
+ export type SystemCallbacks = {
58
+ [K in keyof ClientCallbacks]: ClientCallbacks[K] extends ClientCallback<
59
+ infer Args
60
+ >
61
+ ? (...args: Args) => void
62
+ : never;
63
+ };
64
+
65
+ export type BaseGraphState = {
66
+ messages: BaseMessage[];
67
+ /**
68
+ * Structured response when using structured output mode.
69
+ * Contains the validated JSON response conforming to the configured schema.
70
+ */
71
+ structuredResponse?: Record<string, unknown>;
72
+ };
73
+
74
+ export type MultiAgentGraphState = BaseGraphState & {
75
+ agentMessages?: BaseMessage[];
76
+ };
77
+
78
+ export type IState = BaseGraphState;
79
+
80
+ export interface EventHandler {
81
+ handle(
82
+ event: string,
83
+ data:
84
+ | StreamEventData
85
+ | ModelEndData
86
+ | RunStep
87
+ | RunStepDeltaEvent
88
+ | MessageDeltaEvent
89
+ | ReasoningDeltaEvent
90
+ | { result: ToolEndEvent },
91
+ metadata?: Record<string, unknown>,
92
+ graph?: StandardGraph | MultiAgentGraph
93
+ ): void | Promise<void>;
94
+ }
95
+
96
+ export type GraphStateChannels<T extends BaseGraphState> =
97
+ StateGraphArgs<T>['channels'];
98
+
99
+ export type Workflow<
100
+ T extends BaseGraphState = BaseGraphState,
101
+ U extends Partial<T> = Partial<T>,
102
+ N extends string = string,
103
+ > = StateGraph<T, U, N>;
104
+
105
+ export type CompiledWorkflow<
106
+ T extends BaseGraphState = BaseGraphState,
107
+ U extends Partial<T> = Partial<T>,
108
+ N extends string = string,
109
+ > = CompiledStateGraph<T, U, N>;
110
+
111
+ export type CompiledStateWorkflow = CompiledStateGraph<
112
+ StateType<{
113
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
114
+ }>,
115
+ UpdateType<{
116
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
117
+ }>,
118
+ string,
119
+ {
120
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
121
+ },
122
+ {
123
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
124
+ },
125
+ StateDefinition
126
+ >;
127
+
128
+ export type CompiledMultiAgentWorkflow = CompiledStateGraph<
129
+ StateType<{
130
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
131
+ agentMessages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
132
+ }>,
133
+ UpdateType<{
134
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
135
+ agentMessages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
136
+ }>,
137
+ string,
138
+ {
139
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
140
+ agentMessages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
141
+ },
142
+ {
143
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
144
+ agentMessages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
145
+ },
146
+ StateDefinition
147
+ >;
148
+
149
+ export type CompiledAgentWorfklow = CompiledStateGraph<
150
+ {
151
+ messages: BaseMessage[];
152
+ },
153
+ {
154
+ messages?: BaseMessage[] | undefined;
155
+ },
156
+ '__start__' | `agent=${string}` | `tools=${string}`,
157
+ {
158
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
159
+ },
160
+ {
161
+ messages: BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
162
+ },
163
+ StateDefinition,
164
+ {
165
+ [x: `agent=${string}`]: Partial<BaseGraphState>;
166
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
167
+ [x: `tools=${string}`]: any;
168
+ }
169
+ >;
170
+
171
+ export type SystemRunnable =
172
+ | Runnable<
173
+ BaseMessage[],
174
+ (BaseMessage | SystemMessage)[],
175
+ RunnableConfig<Record<string, unknown>>
176
+ >
177
+ | undefined;
178
+
179
+ /**
180
+ * Optional compile options passed to workflow.compile().
181
+ * These are intentionally untyped to avoid coupling to library internals.
182
+ */
183
+ export type CompileOptions = {
184
+ // A checkpointer instance (e.g., MemorySaver, SQL saver)
185
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
186
+ checkpointer?: any;
187
+ interruptBefore?: string[];
188
+ interruptAfter?: string[];
189
+ };
190
+
191
+ export type EventStreamCallbackHandlerInput =
192
+ Parameters<CompiledWorkflow['streamEvents']>[2] extends Omit<
193
+ infer T,
194
+ 'autoClose'
195
+ >
196
+ ? T
197
+ : never;
198
+
199
+ export type StreamChunk =
200
+ | (ChatGenerationChunk & {
201
+ message: AIMessageChunk;
202
+ })
203
+ | AIMessageChunk;
204
+
205
+ /**
206
+ * Data associated with a StreamEvent.
207
+ */
208
+ export type StreamEventData = {
209
+ /**
210
+ * The input passed to the runnable that generated the event.
211
+ * Inputs will sometimes be available at the *START* of the runnable, and
212
+ * sometimes at the *END* of the runnable.
213
+ * If a runnable is able to stream its inputs, then its input by definition
214
+ * won't be known until the *END* of the runnable when it has finished streaming
215
+ * its inputs.
216
+ */
217
+ input?: unknown;
218
+ /**
219
+ * The output of the runnable that generated the event.
220
+ * Outputs will only be available at the *END* of the runnable.
221
+ * For most runnables, this field can be inferred from the `chunk` field,
222
+ * though there might be some exceptions for special cased runnables (e.g., like
223
+ * chat models), which may return more information.
224
+ */
225
+ output?: unknown;
226
+ /**
227
+ * A streaming chunk from the output that generated the event.
228
+ * chunks support addition in general, and adding them up should result
229
+ * in the output of the runnable that generated the event.
230
+ */
231
+ chunk?: StreamChunk;
232
+ /**
233
+ * Runnable config for invoking other runnables within handlers.
234
+ */
235
+ config?: RunnableConfig;
236
+ /**
237
+ * Custom result from the runnable that generated the event.
238
+ */
239
+ result?: unknown;
240
+ /**
241
+ * Custom field to indicate the event was manually emitted, and may have been handled already
242
+ */
243
+ emitted?: boolean;
244
+ };
245
+
246
+ /**
247
+ * A streaming event.
248
+ *
249
+ * Schema of a streaming event which is produced from the streamEvents method.
250
+ */
251
+ export type StreamEvent = {
252
+ /**
253
+ * Event names are of the format: on_[runnable_type]_(start|stream|end).
254
+ *
255
+ * Runnable types are one of:
256
+ * - llm - used by non chat models
257
+ * - chat_model - used by chat models
258
+ * - prompt -- e.g., ChatPromptTemplate
259
+ * - tool -- LangChain tools
260
+ * - chain - most Runnables are of this type
261
+ *
262
+ * Further, the events are categorized as one of:
263
+ * - start - when the runnable starts
264
+ * - stream - when the runnable is streaming
265
+ * - end - when the runnable ends
266
+ *
267
+ * start, stream and end are associated with slightly different `data` payload.
268
+ *
269
+ * Please see the documentation for `EventData` for more details.
270
+ */
271
+ event: string;
272
+ /** The name of the runnable that generated the event. */
273
+ name: string;
274
+ /**
275
+ * An randomly generated ID to keep track of the execution of the given runnable.
276
+ *
277
+ * Each child runnable that gets invoked as part of the execution of a parent runnable
278
+ * is assigned its own unique ID.
279
+ */
280
+ run_id: string;
281
+ /**
282
+ * Tags associated with the runnable that generated this event.
283
+ * Tags are always inherited from parent runnables.
284
+ */
285
+ tags?: string[];
286
+ /** Metadata associated with the runnable that generated this event. */
287
+ metadata: Record<string, unknown>;
288
+ /**
289
+ * Event data.
290
+ *
291
+ * The contents of the event data depend on the event type.
292
+ */
293
+ data: StreamEventData;
294
+ };
295
+
296
+ export type GraphConfig = {
297
+ provider: string;
298
+ thread_id?: string;
299
+ run_id?: string;
300
+ };
301
+
302
+ export type PartMetadata = {
303
+ progress?: number;
304
+ asset_pointer?: string;
305
+ status?: string;
306
+ action?: boolean;
307
+ output?: string;
308
+ };
309
+
310
+ export type ModelEndData =
311
+ | (StreamEventData & { output: AIMessageChunk | undefined })
312
+ | undefined;
313
+ export type GraphTools = GenericTool[] | BindToolsInput[] | GoogleAIToolType[];
314
+ export type StandardGraphInput = {
315
+ runId?: string;
316
+ signal?: AbortSignal;
317
+ agents: AgentInputs[];
318
+ tokenCounter?: TokenCounter;
319
+ indexTokenCountMap?: Record<string, number>;
320
+ };
321
+
322
+ export type GraphEdge = {
323
+ /** Agent ID, use a list for multiple sources */
324
+ from: string | string[];
325
+ /** Agent ID, use a list for multiple destinations */
326
+ to: string | string[];
327
+ description?: string;
328
+ /** Can return boolean or specific destination(s) */
329
+ condition?: (state: BaseGraphState) => boolean | string | string[];
330
+ /** 'handoff' creates tools for dynamic routing, 'direct' creates direct edges, which also allow parallel execution */
331
+ edgeType?: 'handoff' | 'direct';
332
+ /**
333
+ * For direct edges: Optional prompt to add when transitioning through this edge.
334
+ * String prompts can include variables like {results} which will be replaced with
335
+ * messages from startIndex onwards. When {results} is used, excludeResults defaults to true.
336
+ *
337
+ * For handoff edges: Description for the input parameter that the handoff tool accepts,
338
+ * allowing the supervisor to pass specific instructions/context to the transferred agent.
339
+ */
340
+ prompt?:
341
+ | string
342
+ | ((
343
+ messages: BaseMessage[],
344
+ runStartIndex: number
345
+ ) => string | Promise<string> | undefined);
346
+ /**
347
+ * When true, excludes messages from startIndex when adding prompt.
348
+ * Automatically set to true when {results} variable is used in prompt.
349
+ */
350
+ excludeResults?: boolean;
351
+ /**
352
+ * For handoff edges: Customizes the parameter name for the handoff input.
353
+ * Defaults to "instructions" if not specified.
354
+ * Only applies when prompt is provided for handoff edges.
355
+ */
356
+ promptKey?: string;
357
+ };
358
+
359
+ export type MultiAgentGraphInput = StandardGraphInput & {
360
+ edges: GraphEdge[];
361
+ };
362
+
363
+ /**
364
+ * Structured output mode determines how the agent returns structured data.
365
+ * - 'tool': Uses tool calling to return structured output (works with all tool-calling models)
366
+ * - 'provider': Uses provider-native structured output via LangChain's jsonMode (OpenAI, Anthropic, etc.)
367
+ * - 'native': Uses provider's constrained decoding API directly for guaranteed schema compliance
368
+ * (Anthropic output_config.format, OpenAI response_format.json_schema). Falls back to 'tool' for unsupported providers.
369
+ * - 'auto': Automatically selects the best strategy — 'native' for supported providers, 'tool' for others
370
+ */
371
+ export type StructuredOutputMode = 'tool' | 'provider' | 'native' | 'auto';
372
+
373
+ /**
374
+ * Resolved method used internally after mode resolution.
375
+ * Maps to LangChain's withStructuredOutput method parameter plus our native path.
376
+ */
377
+ export type ResolvedStructuredOutputMethod = 'functionCalling' | 'jsonMode' | 'jsonSchema' | 'native' | undefined;
378
+
379
+ /**
380
+ * Error thrown when the model refuses to produce structured output due to safety policies.
381
+ */
382
+ export class StructuredOutputRefusalError extends Error {
383
+ constructor(public refusalText: string) {
384
+ super(`Model refused to produce structured output: ${refusalText}`);
385
+ this.name = 'StructuredOutputRefusalError';
386
+ }
387
+ }
388
+
389
+ /**
390
+ * Error thrown when the structured output response was truncated due to max_tokens.
391
+ */
392
+ export class StructuredOutputTruncatedError extends Error {
393
+ constructor(public stopReason: string) {
394
+ super(
395
+ `Structured output was truncated (stop_reason: ${stopReason}). ` +
396
+ `Increase max_tokens to allow the full JSON response to be generated.`
397
+ );
398
+ this.name = 'StructuredOutputTruncatedError';
399
+ }
400
+ }
401
+
402
+ /**
403
+ * Configuration for structured JSON output from agents.
404
+ * When configured, the agent will return a validated JSON response
405
+ * instead of streaming text.
406
+ */
407
+ export interface StructuredOutputConfig {
408
+ /**
409
+ * JSON Schema defining the output structure.
410
+ * The model will be forced to return data conforming to this schema.
411
+ */
412
+ schema: Record<string, unknown>;
413
+ /**
414
+ * Name for the structured output format (used in tool mode).
415
+ * @default 'StructuredResponse'
416
+ */
417
+ name?: string;
418
+ /**
419
+ * Description of what the structured output represents.
420
+ * Helps the model understand the expected format.
421
+ */
422
+ description?: string;
423
+ /**
424
+ * Output mode strategy.
425
+ * @default 'auto'
426
+ */
427
+ mode?: StructuredOutputMode;
428
+ /**
429
+ * Enable strict schema validation.
430
+ * When true, the response must exactly match the schema.
431
+ * @default true
432
+ */
433
+ strict?: boolean;
434
+ /**
435
+ * Error handling configuration.
436
+ * - true: Auto-retry on validation errors (default)
437
+ * - false: Throw error on validation failure
438
+ * - string: Custom error message for retry
439
+ */
440
+ handleErrors?: boolean | string;
441
+ /**
442
+ * Maximum number of retry attempts on validation failure.
443
+ * @default 2
444
+ */
445
+ maxRetries?: number;
446
+ /**
447
+ * Include the raw AI message along with structured response.
448
+ * Useful for debugging.
449
+ * @default false
450
+ */
451
+ includeRaw?: boolean;
452
+ }
453
+
454
+ /**
455
+ * Database/API structured output format (snake_case with enabled flag).
456
+ * This matches the format stored in MongoDB and sent from frontends.
457
+ */
458
+ export interface StructuredOutputInput {
459
+ /** Whether structured output is enabled */
460
+ enabled?: boolean;
461
+ /** JSON Schema defining the expected response structure */
462
+ schema?: Record<string, unknown>;
463
+ /** Name identifier for the structured output */
464
+ name?: string;
465
+ /** Description of what the structured output represents */
466
+ description?: string;
467
+ /** Mode for structured output: 'tool' | 'provider' | 'native' | 'auto' */
468
+ mode?: StructuredOutputMode;
469
+ /** Whether to enforce strict schema validation */
470
+ strict?: boolean;
471
+ }
472
+
473
+ export interface AgentInputs {
474
+ agentId: string;
475
+ /** Human-readable name for the agent (used in handoff context). Defaults to agentId if not provided. */
476
+ name?: string;
477
+ toolEnd?: boolean;
478
+ toolMap?: ToolMap;
479
+ tools?: GraphTools;
480
+ provider: Providers;
481
+ instructions?: string;
482
+ streamBuffer?: number;
483
+ maxContextTokens?: number;
484
+ clientOptions?: ClientOptions;
485
+ additional_instructions?: string;
486
+ reasoningKey?: 'reasoning_content' | 'reasoning';
487
+ /** Format content blocks as strings (for legacy compatibility i.e. Ollama/Azure Serverless) */
488
+ useLegacyContent?: boolean;
489
+ /**
490
+ * Tool definitions for all tools, including deferred and programmatic.
491
+ * Used for tool search and programmatic tool calling.
492
+ * Maps tool name to LCTool definition.
493
+ */
494
+ toolRegistry?: Map<string, LCTool>;
495
+ /**
496
+ * Dynamic context that changes per-request (e.g., current time, user info).
497
+ * This is injected as a user message rather than system prompt to preserve cache.
498
+ * Keeping this separate from instructions ensures the system message stays static
499
+ * and can be cached by Bedrock/Anthropic prompt caching.
500
+ */
501
+ dynamicContext?: string;
502
+ /**
503
+ * Structured output configuration (camelCase).
504
+ * When set, disables streaming and returns a validated JSON response
505
+ * conforming to the specified schema.
506
+ */
507
+ structuredOutput?: StructuredOutputConfig;
508
+ /**
509
+ * Structured output configuration (snake_case - database/API format).
510
+ * Alternative to structuredOutput for compatibility with MongoDB/frontend.
511
+ * Uses an `enabled` flag to control activation.
512
+ * @deprecated Use structuredOutput instead when possible
513
+ */
514
+ structured_output?: StructuredOutputInput;
515
+ /**
516
+ * Serializable tool definitions for event-driven execution.
517
+ * When provided, ToolNode operates in event-driven mode, dispatching
518
+ * ON_TOOL_EXECUTE events instead of invoking tools directly.
519
+ */
520
+ toolDefinitions?: LCTool[];
521
+ /**
522
+ * Tool names discovered from previous conversation history.
523
+ * These tools will be pre-marked as discovered so they're included
524
+ * in tool binding without requiring tool_search.
525
+ */
526
+ discoveredTools?: string[];
527
+ /**
528
+ * Optional callback for summarizing messages that were pruned from context.
529
+ * When provided, discarded messages are summarized by the caller (e.g., Ranger)
530
+ * using a cheap LLM call, and the summary is prepended to the context.
531
+ */
532
+ summarizeCallback?: (messagesToRefine: import('@langchain/core/messages').BaseMessage[]) => Promise<string | undefined>;
533
+ }
@@ -0,0 +1,6 @@
1
+ // src/types/index.ts
2
+ export * from './graph';
3
+ export * from './llm';
4
+ export * from './run';
5
+ export * from './stream';
6
+ export * from './tools';
@@ -0,0 +1,140 @@
1
+ // src/types/llm.ts
2
+ import { ChatMistralAI } from '@langchain/mistralai';
3
+ import type {
4
+ BindToolsInput,
5
+ BaseChatModelParams,
6
+ } from '@langchain/core/language_models/chat_models';
7
+ import type {
8
+ OpenAIChatInput,
9
+ ChatOpenAIFields,
10
+ AzureOpenAIInput,
11
+ ClientOptions as OAIClientOptions,
12
+ } from '@langchain/openai';
13
+ import type { GoogleGenerativeAIChatInput } from '@langchain/google-genai';
14
+ import type { GeminiGenerationConfig } from '@langchain/google-common';
15
+ import type { ChatVertexAIInput } from '@langchain/google-vertexai';
16
+ import type { ChatDeepSeekCallOptions } from '@langchain/deepseek';
17
+ import type { ChatOpenRouterCallOptions } from '@/llm/openrouter';
18
+ import type { ChatBedrockConverseInput } from '@langchain/aws';
19
+ import type { ChatMistralAIInput } from '@langchain/mistralai';
20
+ import type { RequestOptions } from '@google/generative-ai';
21
+ import type { StructuredTool } from '@langchain/core/tools';
22
+ import type { AnthropicInput } from '@langchain/anthropic';
23
+ import type { Runnable } from '@langchain/core/runnables';
24
+ import type { OpenAI as OpenAIClient } from 'openai';
25
+ import type { ChatXAIInput } from '@langchain/xai';
26
+ import {
27
+ AzureChatOpenAI,
28
+ ChatDeepSeek,
29
+ ChatMoonshot,
30
+ ChatOpenAI,
31
+ ChatXAI,
32
+ } from '@/llm/openai';
33
+ import { CustomChatGoogleGenerativeAI } from '@/llm/google';
34
+ import { CustomChatBedrockConverse } from '@/llm/bedrock';
35
+ import { CustomAnthropic } from '@/llm/anthropic';
36
+ import { ChatOpenRouter } from '@/llm/openrouter';
37
+ import { ChatVertexAI } from '@/llm/vertexai';
38
+ import { Providers } from '@/common';
39
+
40
+ export type AzureClientOptions = Partial<OpenAIChatInput> &
41
+ Partial<AzureOpenAIInput> & {
42
+ openAIApiKey?: string;
43
+ openAIApiVersion?: string;
44
+ openAIBasePath?: string;
45
+ deploymentName?: string;
46
+ } & BaseChatModelParams & {
47
+ configuration?: OAIClientOptions;
48
+ };
49
+ export type ThinkingConfig = AnthropicInput['thinking'];
50
+ export type ChatOpenAIToolType =
51
+ | BindToolsInput
52
+ | OpenAIClient.ChatCompletionTool;
53
+ export type CommonToolType = StructuredTool | ChatOpenAIToolType;
54
+ export type AnthropicReasoning = {
55
+ thinking?: ThinkingConfig | boolean;
56
+ thinkingBudget?: number;
57
+ };
58
+ export type OpenAIClientOptions = ChatOpenAIFields;
59
+ export type AnthropicClientOptions = AnthropicInput & {
60
+ promptCache?: boolean;
61
+ };
62
+ export type MistralAIClientOptions = ChatMistralAIInput;
63
+ export type VertexAIClientOptions = ChatVertexAIInput & {
64
+ includeThoughts?: boolean;
65
+ };
66
+ export type BedrockAnthropicInput = ChatBedrockConverseInput & {
67
+ additionalModelRequestFields?: ChatBedrockConverseInput['additionalModelRequestFields'] &
68
+ AnthropicReasoning;
69
+ promptCache?: boolean;
70
+ };
71
+ export type BedrockConverseClientOptions = ChatBedrockConverseInput;
72
+ export type BedrockAnthropicClientOptions = BedrockAnthropicInput;
73
+ export type GoogleClientOptions = GoogleGenerativeAIChatInput & {
74
+ customHeaders?: RequestOptions['customHeaders'];
75
+ thinkingConfig?: GeminiGenerationConfig['thinkingConfig'];
76
+ };
77
+ export type DeepSeekClientOptions = ChatDeepSeekCallOptions;
78
+ export type XAIClientOptions = ChatXAIInput;
79
+
80
+ export type ClientOptions =
81
+ | OpenAIClientOptions
82
+ | AzureClientOptions
83
+ | AnthropicClientOptions
84
+ | MistralAIClientOptions
85
+ | VertexAIClientOptions
86
+ | BedrockConverseClientOptions
87
+ | GoogleClientOptions
88
+ | DeepSeekClientOptions
89
+ | XAIClientOptions;
90
+
91
+ export type SharedLLMConfig = {
92
+ provider: Providers;
93
+ _lc_stream_delay?: number;
94
+ };
95
+
96
+ export type LLMConfig = SharedLLMConfig &
97
+ ClientOptions & {
98
+ /** Optional provider fallbacks in order of attempt */
99
+ fallbacks?: Array<{ provider: Providers; clientOptions?: ClientOptions }>;
100
+ };
101
+
102
+ export type ProviderOptionsMap = {
103
+ [Providers.AZURE]: AzureClientOptions;
104
+ [Providers.OPENAI]: OpenAIClientOptions;
105
+ [Providers.GOOGLE]: GoogleClientOptions;
106
+ [Providers.VERTEXAI]: VertexAIClientOptions;
107
+ [Providers.DEEPSEEK]: DeepSeekClientOptions;
108
+ [Providers.ANTHROPIC]: AnthropicClientOptions;
109
+ [Providers.MISTRALAI]: MistralAIClientOptions;
110
+ [Providers.MISTRAL]: MistralAIClientOptions;
111
+ [Providers.OPENROUTER]: ChatOpenRouterCallOptions;
112
+ [Providers.BEDROCK]: BedrockConverseClientOptions;
113
+ [Providers.XAI]: XAIClientOptions;
114
+ [Providers.MOONSHOT]: OpenAIClientOptions;
115
+ };
116
+
117
+ export type ChatModelMap = {
118
+ [Providers.XAI]: ChatXAI;
119
+ [Providers.OPENAI]: ChatOpenAI;
120
+ [Providers.AZURE]: AzureChatOpenAI;
121
+ [Providers.DEEPSEEK]: ChatDeepSeek;
122
+ [Providers.VERTEXAI]: ChatVertexAI;
123
+ [Providers.ANTHROPIC]: CustomAnthropic;
124
+ [Providers.MISTRALAI]: ChatMistralAI;
125
+ [Providers.MISTRAL]: ChatMistralAI;
126
+ [Providers.OPENROUTER]: ChatOpenRouter;
127
+ [Providers.BEDROCK]: CustomChatBedrockConverse;
128
+ [Providers.GOOGLE]: CustomChatGoogleGenerativeAI;
129
+ [Providers.MOONSHOT]: ChatMoonshot;
130
+ };
131
+
132
+ export type ChatModelConstructorMap = {
133
+ [P in Providers]: new (config: ProviderOptionsMap[P]) => ChatModelMap[P];
134
+ };
135
+
136
+ export type ChatModelInstance = ChatModelMap[Providers];
137
+
138
+ export type ModelWithTools = ChatModelInstance & {
139
+ bindTools(tools: CommonToolType[]): Runnable;
140
+ };
@@ -0,0 +1,4 @@
1
+ import type Anthropic from '@anthropic-ai/sdk';
2
+ import type { BaseMessage } from '@langchain/core/messages';
3
+ export type AnthropicMessages = Array<AnthropicMessage | BaseMessage>;
4
+ export type AnthropicMessage = Anthropic.MessageParam;