@strands-agents/sdk 1.0.0-rc.5 → 1.1.0

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 (380) hide show
  1. package/LICENSE +175 -0
  2. package/README.md +340 -0
  3. package/dist/src/__fixtures__/agent-helpers.d.ts +22 -1
  4. package/dist/src/__fixtures__/agent-helpers.d.ts.map +1 -1
  5. package/dist/src/__fixtures__/agent-helpers.js +45 -1
  6. package/dist/src/__fixtures__/agent-helpers.js.map +1 -1
  7. package/dist/src/__fixtures__/mock-plugin.d.ts.map +1 -1
  8. package/dist/src/__fixtures__/mock-plugin.js +3 -1
  9. package/dist/src/__fixtures__/mock-plugin.js.map +1 -1
  10. package/dist/src/__fixtures__/tool-helpers.d.ts +5 -2
  11. package/dist/src/__fixtures__/tool-helpers.d.ts.map +1 -1
  12. package/dist/src/__fixtures__/tool-helpers.js +23 -4
  13. package/dist/src/__fixtures__/tool-helpers.js.map +1 -1
  14. package/dist/src/__tests__/interrupt.test.d.ts +2 -0
  15. package/dist/src/__tests__/interrupt.test.d.ts.map +1 -0
  16. package/dist/src/__tests__/interrupt.test.js +259 -0
  17. package/dist/src/__tests__/interrupt.test.js.map +1 -0
  18. package/dist/src/__tests__/mcp.test.js +448 -2
  19. package/dist/src/__tests__/mcp.test.js.map +1 -1
  20. package/dist/src/a2a/__tests__/events.test.js +2 -0
  21. package/dist/src/a2a/__tests__/events.test.js.map +1 -1
  22. package/dist/src/a2a/__tests__/executor.test.js +16 -5
  23. package/dist/src/a2a/__tests__/executor.test.js.map +1 -1
  24. package/dist/src/a2a/a2a-agent.d.ts +8 -3
  25. package/dist/src/a2a/a2a-agent.d.ts.map +1 -1
  26. package/dist/src/a2a/a2a-agent.js +12 -6
  27. package/dist/src/a2a/a2a-agent.js.map +1 -1
  28. package/dist/src/a2a/executor.d.ts +13 -0
  29. package/dist/src/a2a/executor.d.ts.map +1 -1
  30. package/dist/src/a2a/executor.js +19 -1
  31. package/dist/src/a2a/executor.js.map +1 -1
  32. package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.d.ts +2 -0
  33. package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.d.ts.map +1 -0
  34. package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.js +23 -0
  35. package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.js.map +1 -0
  36. package/dist/src/agent/__tests__/agent.cancel.test.js +1 -1
  37. package/dist/src/agent/__tests__/agent.cancel.test.js.map +1 -1
  38. package/dist/src/agent/__tests__/agent.concurrent.test.d.ts +2 -0
  39. package/dist/src/agent/__tests__/agent.concurrent.test.d.ts.map +1 -0
  40. package/dist/src/agent/__tests__/agent.concurrent.test.js +488 -0
  41. package/dist/src/agent/__tests__/agent.concurrent.test.js.map +1 -0
  42. package/dist/src/agent/__tests__/agent.hook.test.js +724 -12
  43. package/dist/src/agent/__tests__/agent.hook.test.js.map +1 -1
  44. package/dist/src/agent/__tests__/agent.interrupt.test.d.ts +2 -0
  45. package/dist/src/agent/__tests__/agent.interrupt.test.d.ts.map +1 -0
  46. package/dist/src/agent/__tests__/agent.interrupt.test.js +730 -0
  47. package/dist/src/agent/__tests__/agent.interrupt.test.js.map +1 -0
  48. package/dist/src/agent/__tests__/agent.invocation-state.test.d.ts +2 -0
  49. package/dist/src/agent/__tests__/agent.invocation-state.test.d.ts.map +1 -0
  50. package/dist/src/agent/__tests__/agent.invocation-state.test.js +219 -0
  51. package/dist/src/agent/__tests__/agent.invocation-state.test.js.map +1 -0
  52. package/dist/src/agent/__tests__/agent.model-retry.test.d.ts +2 -0
  53. package/dist/src/agent/__tests__/agent.model-retry.test.d.ts.map +1 -0
  54. package/dist/src/agent/__tests__/agent.model-retry.test.js +161 -0
  55. package/dist/src/agent/__tests__/agent.model-retry.test.js.map +1 -0
  56. package/dist/src/agent/__tests__/agent.stateful-model.test.d.ts +2 -0
  57. package/dist/src/agent/__tests__/agent.stateful-model.test.d.ts.map +1 -0
  58. package/dist/src/agent/__tests__/agent.stateful-model.test.js +169 -0
  59. package/dist/src/agent/__tests__/agent.stateful-model.test.js.map +1 -0
  60. package/dist/src/agent/__tests__/agent.test.js +217 -2
  61. package/dist/src/agent/__tests__/agent.test.js.map +1 -1
  62. package/dist/src/agent/__tests__/agent.tracer.test.node.js +39 -0
  63. package/dist/src/agent/__tests__/agent.tracer.test.node.js.map +1 -1
  64. package/dist/src/agent/__tests__/snapshot.test.js +51 -4
  65. package/dist/src/agent/__tests__/snapshot.test.js.map +1 -1
  66. package/dist/src/agent/agent-as-tool.d.ts.map +1 -1
  67. package/dist/src/agent/agent-as-tool.js +4 -2
  68. package/dist/src/agent/agent-as-tool.js.map +1 -1
  69. package/dist/src/agent/agent.d.ts +109 -4
  70. package/dist/src/agent/agent.d.ts.map +1 -1
  71. package/dist/src/agent/agent.js +790 -224
  72. package/dist/src/agent/agent.js.map +1 -1
  73. package/dist/src/agent/snapshot.d.ts +2 -2
  74. package/dist/src/agent/snapshot.d.ts.map +1 -1
  75. package/dist/src/agent/snapshot.js +20 -2
  76. package/dist/src/agent/snapshot.js.map +1 -1
  77. package/dist/src/conversation-manager/__tests__/conversation-manager.test.js +230 -9
  78. package/dist/src/conversation-manager/__tests__/conversation-manager.test.js.map +1 -1
  79. package/dist/src/conversation-manager/__tests__/null-conversation-manager.test.js +19 -6
  80. package/dist/src/conversation-manager/__tests__/null-conversation-manager.test.js.map +1 -1
  81. package/dist/src/conversation-manager/__tests__/sliding-window-conversation-manager.test.js +58 -4
  82. package/dist/src/conversation-manager/__tests__/sliding-window-conversation-manager.test.js.map +1 -1
  83. package/dist/src/conversation-manager/__tests__/summarizing-conversation-manager.test.js +76 -1
  84. package/dist/src/conversation-manager/__tests__/summarizing-conversation-manager.test.js.map +1 -1
  85. package/dist/src/conversation-manager/conversation-manager.d.ts +67 -22
  86. package/dist/src/conversation-manager/conversation-manager.d.ts.map +1 -1
  87. package/dist/src/conversation-manager/conversation-manager.js +65 -13
  88. package/dist/src/conversation-manager/conversation-manager.js.map +1 -1
  89. package/dist/src/conversation-manager/index.d.ts +1 -1
  90. package/dist/src/conversation-manager/index.d.ts.map +1 -1
  91. package/dist/src/conversation-manager/index.js +1 -1
  92. package/dist/src/conversation-manager/index.js.map +1 -1
  93. package/dist/src/conversation-manager/sliding-window-conversation-manager.d.ts +17 -3
  94. package/dist/src/conversation-manager/sliding-window-conversation-manager.d.ts.map +1 -1
  95. package/dist/src/conversation-manager/sliding-window-conversation-manager.js +10 -4
  96. package/dist/src/conversation-manager/sliding-window-conversation-manager.js.map +1 -1
  97. package/dist/src/conversation-manager/summarizing-conversation-manager.d.ts +23 -1
  98. package/dist/src/conversation-manager/summarizing-conversation-manager.d.ts.map +1 -1
  99. package/dist/src/conversation-manager/summarizing-conversation-manager.js +39 -17
  100. package/dist/src/conversation-manager/summarizing-conversation-manager.js.map +1 -1
  101. package/dist/src/errors.d.ts +11 -0
  102. package/dist/src/errors.d.ts.map +1 -1
  103. package/dist/src/errors.js +12 -0
  104. package/dist/src/errors.js.map +1 -1
  105. package/dist/src/hooks/__tests__/events.test.js +267 -73
  106. package/dist/src/hooks/__tests__/events.test.js.map +1 -1
  107. package/dist/src/hooks/__tests__/registry.test.js +182 -18
  108. package/dist/src/hooks/__tests__/registry.test.js.map +1 -1
  109. package/dist/src/hooks/events.d.ts +193 -51
  110. package/dist/src/hooks/events.d.ts.map +1 -1
  111. package/dist/src/hooks/events.js +182 -26
  112. package/dist/src/hooks/events.js.map +1 -1
  113. package/dist/src/hooks/index.d.ts +3 -2
  114. package/dist/src/hooks/index.d.ts.map +1 -1
  115. package/dist/src/hooks/index.js +1 -0
  116. package/dist/src/hooks/index.js.map +1 -1
  117. package/dist/src/hooks/registry.d.ts +12 -12
  118. package/dist/src/hooks/registry.d.ts.map +1 -1
  119. package/dist/src/hooks/registry.js +55 -15
  120. package/dist/src/hooks/registry.js.map +1 -1
  121. package/dist/src/hooks/types.d.ts +23 -0
  122. package/dist/src/hooks/types.d.ts.map +1 -1
  123. package/dist/src/hooks/types.js +17 -1
  124. package/dist/src/hooks/types.js.map +1 -1
  125. package/dist/src/index.d.ts +12 -7
  126. package/dist/src/index.d.ts.map +1 -1
  127. package/dist/src/index.js +4 -1
  128. package/dist/src/index.js.map +1 -1
  129. package/dist/src/interrupt.d.ts +220 -0
  130. package/dist/src/interrupt.d.ts.map +1 -0
  131. package/dist/src/interrupt.js +274 -0
  132. package/dist/src/interrupt.js.map +1 -0
  133. package/dist/src/logging/__tests__/warn-once.test.d.ts +2 -0
  134. package/dist/src/logging/__tests__/warn-once.test.d.ts.map +1 -0
  135. package/dist/src/logging/__tests__/warn-once.test.js +30 -0
  136. package/dist/src/logging/__tests__/warn-once.test.js.map +1 -0
  137. package/dist/src/logging/warn-once.d.ts +13 -0
  138. package/dist/src/logging/warn-once.d.ts.map +1 -0
  139. package/dist/src/logging/warn-once.js +18 -0
  140. package/dist/src/logging/warn-once.js.map +1 -0
  141. package/dist/src/mcp.d.ts +43 -3
  142. package/dist/src/mcp.d.ts.map +1 -1
  143. package/dist/src/mcp.js +85 -17
  144. package/dist/src/mcp.js.map +1 -1
  145. package/dist/src/mime.d.ts +2 -1
  146. package/dist/src/mime.d.ts.map +1 -1
  147. package/dist/src/mime.js +1 -0
  148. package/dist/src/mime.js.map +1 -1
  149. package/dist/src/models/__tests__/anthropic.test.js +147 -3
  150. package/dist/src/models/__tests__/anthropic.test.js.map +1 -1
  151. package/dist/src/models/__tests__/bedrock.test.js +228 -2
  152. package/dist/src/models/__tests__/bedrock.test.js.map +1 -1
  153. package/dist/src/models/__tests__/defaults.test.d.ts +2 -0
  154. package/dist/src/models/__tests__/defaults.test.d.ts.map +1 -0
  155. package/dist/src/models/__tests__/defaults.test.js +36 -0
  156. package/dist/src/models/__tests__/defaults.test.js.map +1 -0
  157. package/dist/src/models/__tests__/google.test.js +135 -0
  158. package/dist/src/models/__tests__/google.test.js.map +1 -1
  159. package/dist/src/models/__tests__/model.test.js +149 -1
  160. package/dist/src/models/__tests__/model.test.js.map +1 -1
  161. package/dist/src/models/anthropic.d.ts +20 -1
  162. package/dist/src/models/anthropic.d.ts.map +1 -1
  163. package/dist/src/models/anthropic.js +42 -8
  164. package/dist/src/models/anthropic.js.map +1 -1
  165. package/dist/src/models/bedrock.d.ts +27 -1
  166. package/dist/src/models/bedrock.d.ts.map +1 -1
  167. package/dist/src/models/bedrock.js +100 -12
  168. package/dist/src/models/bedrock.js.map +1 -1
  169. package/dist/src/models/defaults.d.ts +47 -0
  170. package/dist/src/models/defaults.d.ts.map +1 -0
  171. package/dist/src/models/defaults.js +170 -0
  172. package/dist/src/models/defaults.js.map +1 -0
  173. package/dist/src/models/google/model.d.ts +14 -1
  174. package/dist/src/models/google/model.d.ts.map +1 -1
  175. package/dist/src/models/google/model.js +54 -8
  176. package/dist/src/models/google/model.js.map +1 -1
  177. package/dist/src/models/google/types.d.ts +8 -0
  178. package/dist/src/models/google/types.d.ts.map +1 -1
  179. package/dist/src/models/model.d.ts +65 -0
  180. package/dist/src/models/model.d.ts.map +1 -1
  181. package/dist/src/models/model.js +138 -0
  182. package/dist/src/models/model.js.map +1 -1
  183. package/dist/src/models/openai/__tests__/chat.test.d.ts +2 -0
  184. package/dist/src/models/openai/__tests__/chat.test.d.ts.map +1 -0
  185. package/dist/src/models/{__tests__/openai.test.js → openai/__tests__/chat.test.js} +117 -7
  186. package/dist/src/models/openai/__tests__/chat.test.js.map +1 -0
  187. package/dist/src/models/openai/__tests__/responses.test.d.ts +2 -0
  188. package/dist/src/models/openai/__tests__/responses.test.d.ts.map +1 -0
  189. package/dist/src/models/openai/__tests__/responses.test.js +668 -0
  190. package/dist/src/models/openai/__tests__/responses.test.js.map +1 -0
  191. package/dist/src/models/openai/chat-adapter.d.ts +33 -0
  192. package/dist/src/models/openai/chat-adapter.d.ts.map +1 -0
  193. package/dist/src/models/openai/chat-adapter.js +383 -0
  194. package/dist/src/models/openai/chat-adapter.js.map +1 -0
  195. package/dist/src/models/openai/errors.d.ts +16 -0
  196. package/dist/src/models/openai/errors.d.ts.map +1 -0
  197. package/dist/src/models/openai/errors.js +40 -0
  198. package/dist/src/models/openai/errors.js.map +1 -0
  199. package/dist/src/models/openai/formatting.d.ts +18 -0
  200. package/dist/src/models/openai/formatting.d.ts.map +1 -0
  201. package/dist/src/models/openai/formatting.js +38 -0
  202. package/dist/src/models/openai/formatting.js.map +1 -0
  203. package/dist/src/models/openai/index.d.ts +19 -0
  204. package/dist/src/models/openai/index.d.ts.map +1 -0
  205. package/dist/src/models/openai/index.js +18 -0
  206. package/dist/src/models/openai/index.js.map +1 -0
  207. package/dist/src/models/openai/model.d.ts +77 -0
  208. package/dist/src/models/openai/model.d.ts.map +1 -0
  209. package/dist/src/models/openai/model.js +211 -0
  210. package/dist/src/models/openai/model.js.map +1 -0
  211. package/dist/src/models/openai/responses-adapter.d.ts +78 -0
  212. package/dist/src/models/openai/responses-adapter.d.ts.map +1 -0
  213. package/dist/src/models/openai/responses-adapter.js +467 -0
  214. package/dist/src/models/openai/responses-adapter.js.map +1 -0
  215. package/dist/src/models/openai/types.d.ts +131 -0
  216. package/dist/src/models/openai/types.d.ts.map +1 -0
  217. package/dist/src/models/openai/types.js +5 -0
  218. package/dist/src/models/openai/types.js.map +1 -0
  219. package/dist/src/multiagent/__tests__/events.test.js +122 -28
  220. package/dist/src/multiagent/__tests__/events.test.js.map +1 -1
  221. package/dist/src/multiagent/__tests__/graph.invocation-state.test.d.ts +2 -0
  222. package/dist/src/multiagent/__tests__/graph.invocation-state.test.d.ts.map +1 -0
  223. package/dist/src/multiagent/__tests__/graph.invocation-state.test.js +95 -0
  224. package/dist/src/multiagent/__tests__/graph.invocation-state.test.js.map +1 -0
  225. package/dist/src/multiagent/__tests__/graph.test.js +69 -0
  226. package/dist/src/multiagent/__tests__/graph.test.js.map +1 -1
  227. package/dist/src/multiagent/__tests__/nodes.test.js +18 -2
  228. package/dist/src/multiagent/__tests__/nodes.test.js.map +1 -1
  229. package/dist/src/multiagent/__tests__/swarm.invocation-state.test.d.ts +2 -0
  230. package/dist/src/multiagent/__tests__/swarm.invocation-state.test.d.ts.map +1 -0
  231. package/dist/src/multiagent/__tests__/swarm.invocation-state.test.js +56 -0
  232. package/dist/src/multiagent/__tests__/swarm.invocation-state.test.js.map +1 -0
  233. package/dist/src/multiagent/__tests__/swarm.test.js +77 -0
  234. package/dist/src/multiagent/__tests__/swarm.test.js.map +1 -1
  235. package/dist/src/multiagent/events.d.ts +19 -1
  236. package/dist/src/multiagent/events.d.ts.map +1 -1
  237. package/dist/src/multiagent/events.js +18 -0
  238. package/dist/src/multiagent/events.js.map +1 -1
  239. package/dist/src/multiagent/graph.d.ts +27 -5
  240. package/dist/src/multiagent/graph.d.ts.map +1 -1
  241. package/dist/src/multiagent/graph.js +61 -15
  242. package/dist/src/multiagent/graph.js.map +1 -1
  243. package/dist/src/multiagent/index.d.ts +1 -1
  244. package/dist/src/multiagent/index.d.ts.map +1 -1
  245. package/dist/src/multiagent/multiagent.d.ts +21 -6
  246. package/dist/src/multiagent/multiagent.d.ts.map +1 -1
  247. package/dist/src/multiagent/nodes.d.ts +28 -3
  248. package/dist/src/multiagent/nodes.d.ts.map +1 -1
  249. package/dist/src/multiagent/nodes.js +42 -7
  250. package/dist/src/multiagent/nodes.js.map +1 -1
  251. package/dist/src/multiagent/swarm.d.ts +20 -4
  252. package/dist/src/multiagent/swarm.d.ts.map +1 -1
  253. package/dist/src/multiagent/swarm.js +65 -16
  254. package/dist/src/multiagent/swarm.js.map +1 -1
  255. package/dist/src/plugins/__tests__/registry.test.js +1 -1
  256. package/dist/src/plugins/__tests__/registry.test.js.map +1 -1
  257. package/dist/src/plugins/model-plugin.d.ts +20 -0
  258. package/dist/src/plugins/model-plugin.d.ts.map +1 -0
  259. package/dist/src/plugins/model-plugin.js +29 -0
  260. package/dist/src/plugins/model-plugin.js.map +1 -0
  261. package/dist/src/registry/__tests__/tool-registry.test.js +11 -0
  262. package/dist/src/registry/__tests__/tool-registry.test.js.map +1 -1
  263. package/dist/src/registry/tool-registry.d.ts +4 -0
  264. package/dist/src/registry/tool-registry.d.ts.map +1 -1
  265. package/dist/src/registry/tool-registry.js +6 -0
  266. package/dist/src/registry/tool-registry.js.map +1 -1
  267. package/dist/src/retry/__tests__/backoff-strategy.test.d.ts +2 -0
  268. package/dist/src/retry/__tests__/backoff-strategy.test.d.ts.map +1 -0
  269. package/dist/src/retry/__tests__/backoff-strategy.test.js +116 -0
  270. package/dist/src/retry/__tests__/backoff-strategy.test.js.map +1 -0
  271. package/dist/src/retry/__tests__/default-model-retry-strategy.test.d.ts +2 -0
  272. package/dist/src/retry/__tests__/default-model-retry-strategy.test.d.ts.map +1 -0
  273. package/dist/src/retry/__tests__/default-model-retry-strategy.test.js +225 -0
  274. package/dist/src/retry/__tests__/default-model-retry-strategy.test.js.map +1 -0
  275. package/dist/src/retry/backoff-strategy.d.ts +108 -0
  276. package/dist/src/retry/backoff-strategy.d.ts.map +1 -0
  277. package/dist/src/retry/backoff-strategy.js +86 -0
  278. package/dist/src/retry/backoff-strategy.js.map +1 -0
  279. package/dist/src/retry/default-model-retry-strategy.d.ts +76 -0
  280. package/dist/src/retry/default-model-retry-strategy.d.ts.map +1 -0
  281. package/dist/src/retry/default-model-retry-strategy.js +104 -0
  282. package/dist/src/retry/default-model-retry-strategy.js.map +1 -0
  283. package/dist/src/retry/index.d.ts +8 -0
  284. package/dist/src/retry/index.d.ts.map +1 -0
  285. package/dist/src/retry/index.js +7 -0
  286. package/dist/src/retry/index.js.map +1 -0
  287. package/dist/src/retry/model-retry-strategy.d.ts +80 -0
  288. package/dist/src/retry/model-retry-strategy.d.ts.map +1 -0
  289. package/dist/src/retry/model-retry-strategy.js +85 -0
  290. package/dist/src/retry/model-retry-strategy.js.map +1 -0
  291. package/dist/src/retry/retry-strategy.d.ts +34 -0
  292. package/dist/src/retry/retry-strategy.d.ts.map +1 -0
  293. package/dist/src/retry/retry-strategy.js +25 -0
  294. package/dist/src/retry/retry-strategy.js.map +1 -0
  295. package/dist/src/session/__tests__/session-manager.test.js +52 -11
  296. package/dist/src/session/__tests__/session-manager.test.js.map +1 -1
  297. package/dist/src/session/session-manager.d.ts +6 -0
  298. package/dist/src/session/session-manager.d.ts.map +1 -1
  299. package/dist/src/session/session-manager.js +17 -0
  300. package/dist/src/session/session-manager.js.map +1 -1
  301. package/dist/src/telemetry/__tests__/meter.test.js +23 -0
  302. package/dist/src/telemetry/__tests__/meter.test.js.map +1 -1
  303. package/dist/src/telemetry/meter.d.ts +15 -0
  304. package/dist/src/telemetry/meter.d.ts.map +1 -1
  305. package/dist/src/telemetry/meter.js +14 -0
  306. package/dist/src/telemetry/meter.js.map +1 -1
  307. package/dist/src/tools/__tests__/tool.test.js +24 -1
  308. package/dist/src/tools/__tests__/tool.test.js.map +1 -1
  309. package/dist/src/tools/function-tool.d.ts.map +1 -1
  310. package/dist/src/tools/function-tool.js +6 -1
  311. package/dist/src/tools/function-tool.js.map +1 -1
  312. package/dist/src/tools/mcp-tool.d.ts +24 -3
  313. package/dist/src/tools/mcp-tool.d.ts.map +1 -1
  314. package/dist/src/tools/mcp-tool.js +103 -31
  315. package/dist/src/tools/mcp-tool.js.map +1 -1
  316. package/dist/src/tools/tool.d.ts +21 -2
  317. package/dist/src/tools/tool.d.ts.map +1 -1
  318. package/dist/src/tools/tool.js +12 -0
  319. package/dist/src/tools/tool.js.map +1 -1
  320. package/dist/src/tsconfig.tsbuildinfo +1 -1
  321. package/dist/src/types/__tests__/agent.test.js +48 -0
  322. package/dist/src/types/__tests__/agent.test.js.map +1 -1
  323. package/dist/src/types/agent.d.ts +77 -9
  324. package/dist/src/types/agent.d.ts.map +1 -1
  325. package/dist/src/types/agent.js +30 -6
  326. package/dist/src/types/agent.js.map +1 -1
  327. package/dist/src/types/elicitation.d.ts +15 -0
  328. package/dist/src/types/elicitation.d.ts.map +1 -0
  329. package/dist/src/types/elicitation.js +2 -0
  330. package/dist/src/types/elicitation.js.map +1 -0
  331. package/dist/src/types/interrupt.d.ts +103 -0
  332. package/dist/src/types/interrupt.d.ts.map +1 -0
  333. package/dist/src/types/interrupt.js +63 -0
  334. package/dist/src/types/interrupt.js.map +1 -0
  335. package/dist/src/types/messages.d.ts +2 -1
  336. package/dist/src/types/messages.d.ts.map +1 -1
  337. package/dist/src/types/messages.js.map +1 -1
  338. package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.d.ts +2 -0
  339. package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.d.ts.map +1 -0
  340. package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.js +292 -0
  341. package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.js.map +1 -0
  342. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.d.ts +2 -0
  343. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.d.ts.map +1 -0
  344. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.js +148 -0
  345. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.js.map +1 -0
  346. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.d.ts +2 -0
  347. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.d.ts.map +1 -0
  348. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.js +78 -0
  349. package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.js.map +1 -0
  350. package/dist/src/vended-plugins/context-offloader/index.d.ts +23 -0
  351. package/dist/src/vended-plugins/context-offloader/index.d.ts.map +1 -0
  352. package/dist/src/vended-plugins/context-offloader/index.js +21 -0
  353. package/dist/src/vended-plugins/context-offloader/index.js.map +1 -0
  354. package/dist/src/vended-plugins/context-offloader/plugin.d.ts +48 -0
  355. package/dist/src/vended-plugins/context-offloader/plugin.d.ts.map +1 -0
  356. package/dist/src/vended-plugins/context-offloader/plugin.js +244 -0
  357. package/dist/src/vended-plugins/context-offloader/plugin.js.map +1 -0
  358. package/dist/src/vended-plugins/context-offloader/storage.d.ts +114 -0
  359. package/dist/src/vended-plugins/context-offloader/storage.d.ts.map +1 -0
  360. package/dist/src/vended-plugins/context-offloader/storage.js +204 -0
  361. package/dist/src/vended-plugins/context-offloader/storage.js.map +1 -0
  362. package/dist/src/vended-plugins/skills/__tests__/agent-skills.test.node.js +21 -5
  363. package/dist/src/vended-plugins/skills/__tests__/agent-skills.test.node.js.map +1 -1
  364. package/dist/src/vended-tools/bash/__tests__/bash.test.node.js +4 -0
  365. package/dist/src/vended-tools/bash/__tests__/bash.test.node.js.map +1 -1
  366. package/dist/src/vended-tools/bash/bash.d.ts.map +1 -1
  367. package/dist/src/vended-tools/bash/bash.js +0 -3
  368. package/dist/src/vended-tools/bash/bash.js.map +1 -1
  369. package/dist/src/vended-tools/file-editor/__tests__/file-editor.test.node.js +4 -0
  370. package/dist/src/vended-tools/file-editor/__tests__/file-editor.test.node.js.map +1 -1
  371. package/dist/src/vended-tools/notebook/__tests__/notebook.test.js +4 -0
  372. package/dist/src/vended-tools/notebook/__tests__/notebook.test.js.map +1 -1
  373. package/package.json +17 -9
  374. package/dist/src/models/__tests__/openai.test.d.ts +0 -2
  375. package/dist/src/models/__tests__/openai.test.d.ts.map +0 -1
  376. package/dist/src/models/__tests__/openai.test.js.map +0 -1
  377. package/dist/src/models/openai.d.ts +0 -312
  378. package/dist/src/models/openai.d.ts.map +0 -1
  379. package/dist/src/models/openai.js +0 -789
  380. package/dist/src/models/openai.js.map +0 -1
@@ -0,0 +1,274 @@
1
+ /**
2
+ * Human-in-the-loop interrupt system for agent workflows.
3
+ *
4
+ * Interrupt Flow:
5
+ * 1. Hook or tool calls `event.interrupt()` or `context.interrupt()`
6
+ * 2. If resuming (response exists), the response is returned
7
+ * 3. Otherwise, agent execution halts with `stopReason: 'interrupt'`
8
+ * 4. User resumes by invoking agent with `interruptResponse` content blocks
9
+ * 5. On resume, `interrupt()` returns the user's response
10
+ */
11
+ import { InterruptResponseContent } from './types/interrupt.js';
12
+ import { Message, ToolResultBlock } from './types/messages.js';
13
+ /**
14
+ * Represents an interrupt that can pause agent execution for human-in-the-loop workflows.
15
+ */
16
+ export class Interrupt {
17
+ /**
18
+ * Unique identifier for this interrupt.
19
+ */
20
+ id;
21
+ /**
22
+ * User-defined name for the interrupt.
23
+ */
24
+ name;
25
+ /**
26
+ * User-provided reason for raising the interrupt.
27
+ */
28
+ reason;
29
+ /**
30
+ * Human response provided when resuming the agent after an interrupt.
31
+ */
32
+ response;
33
+ constructor(data) {
34
+ this.id = data.id;
35
+ this.name = data.name;
36
+ if (data.reason !== undefined) {
37
+ this.reason = data.reason;
38
+ }
39
+ if (data.response !== undefined) {
40
+ this.response = data.response;
41
+ }
42
+ }
43
+ /**
44
+ * Serializes the interrupt to a JSON-compatible object.
45
+ */
46
+ toJSON() {
47
+ return {
48
+ id: this.id,
49
+ name: this.name,
50
+ ...(this.reason !== undefined && { reason: this.reason }),
51
+ ...(this.response !== undefined && { response: this.response }),
52
+ };
53
+ }
54
+ /**
55
+ * Creates an Interrupt instance from a JSON object.
56
+ *
57
+ * @param data - JSON data to deserialize
58
+ * @returns Interrupt instance
59
+ */
60
+ static fromJSON(data) {
61
+ return new Interrupt(data);
62
+ }
63
+ }
64
+ /**
65
+ * Error thrown when human input is required to continue agent execution.
66
+ * Caught by the agent loop to trigger an interrupt stop.
67
+ */
68
+ export class InterruptError extends Error {
69
+ /**
70
+ * The interrupts that caused this error.
71
+ */
72
+ interrupts;
73
+ constructor(interrupt) {
74
+ const all = Array.isArray(interrupt) ? interrupt : [interrupt];
75
+ const message = all.length === 1
76
+ ? `Interrupt raised: ${all[0].name}`
77
+ : `${all.length} interrupts raised: ${all.map((i) => i.name).join(', ')}`;
78
+ super(message);
79
+ this.name = 'InterruptError';
80
+ this.interrupts = all;
81
+ }
82
+ }
83
+ /**
84
+ * Tracks the state of interrupt events raised during agent execution.
85
+ *
86
+ * Interrupt state is cleared after resuming.
87
+ */
88
+ export class InterruptState {
89
+ /** Record of interrupt IDs to Interrupt instances. */
90
+ interrupts;
91
+ /** Resume responses provided when resuming from an interrupt. */
92
+ resumeResponses;
93
+ /** Whether the agent is in an interrupted state. */
94
+ activated;
95
+ /** Pending tool execution state for resume. */
96
+ pendingToolExecution;
97
+ constructor() {
98
+ this.interrupts = {};
99
+ this.resumeResponses = undefined;
100
+ this.activated = false;
101
+ this.pendingToolExecution = undefined;
102
+ }
103
+ /**
104
+ * Gets the pending tool execution state with reconstructed Message and ToolResultBlock objects.
105
+ * Returns undefined if there is no pending execution.
106
+ */
107
+ getPendingExecution() {
108
+ if (!this.pendingToolExecution) {
109
+ return undefined;
110
+ }
111
+ const assistantMessage = Message.fromMessageData(this.pendingToolExecution.assistantMessageData);
112
+ const completedToolResults = new Map();
113
+ for (const [toolUseId, resultData] of Object.entries(this.pendingToolExecution.completedToolResults)) {
114
+ completedToolResults.set(toolUseId, ToolResultBlock.fromJSON(resultData));
115
+ }
116
+ return { assistantMessage, completedToolResults };
117
+ }
118
+ /**
119
+ * Sets the pending tool execution state.
120
+ */
121
+ setPendingToolExecution(pending) {
122
+ this.pendingToolExecution = pending;
123
+ }
124
+ /**
125
+ * Clears the pending tool execution state.
126
+ */
127
+ clearPendingToolExecution() {
128
+ this.pendingToolExecution = undefined;
129
+ }
130
+ /**
131
+ * Returns the list of interrupts as an array.
132
+ */
133
+ getInterruptsList() {
134
+ return Object.values(this.interrupts);
135
+ }
136
+ /**
137
+ * Returns all interrupts that have no response (i.e., were raised but not yet answered).
138
+ */
139
+ getUnansweredInterrupts() {
140
+ return Object.values(this.interrupts).filter((interrupt) => interrupt.response === undefined);
141
+ }
142
+ /**
143
+ * Returns the first interrupt that has no response (i.e., was raised but not yet answered).
144
+ */
145
+ getUnansweredInterrupt() {
146
+ for (const interrupt of Object.values(this.interrupts)) {
147
+ if (interrupt.response === undefined) {
148
+ return interrupt;
149
+ }
150
+ }
151
+ return undefined;
152
+ }
153
+ /**
154
+ * Activates the interrupt state.
155
+ */
156
+ activate() {
157
+ this.activated = true;
158
+ }
159
+ /**
160
+ * Deactivates the interrupt state and clears all interrupts and context.
161
+ */
162
+ deactivate() {
163
+ this.interrupts = {};
164
+ this.resumeResponses = undefined;
165
+ this.activated = false;
166
+ this.pendingToolExecution = undefined;
167
+ }
168
+ /**
169
+ * Configures the interrupt state for resuming from an interrupt.
170
+ * Populates interrupt responses from the provided content blocks.
171
+ *
172
+ * @param responses - Array of interrupt response content blocks
173
+ * @throws Error if an interrupt ID is not found
174
+ */
175
+ resume(responses) {
176
+ if (!this.activated) {
177
+ return;
178
+ }
179
+ for (const content of responses) {
180
+ const interruptId = content.interruptResponse.interruptId;
181
+ const response = content.interruptResponse.response;
182
+ const interrupt = this.interrupts[interruptId];
183
+ if (!interrupt) {
184
+ throw new Error(`interrupt_id=<${interruptId}> | no interrupt found`);
185
+ }
186
+ interrupt.response = response;
187
+ }
188
+ this.resumeResponses = responses;
189
+ }
190
+ /**
191
+ * Gets or creates an interrupt with the given ID.
192
+ * If the interrupt already exists, returns it (potentially with a response).
193
+ * If a preemptive response is provided and the interrupt is new, the response
194
+ * is stored on the interrupt so it returns immediately without halting execution.
195
+ *
196
+ * @param id - Unique identifier for the interrupt
197
+ * @param name - User-defined name for the interrupt
198
+ * @param reason - Optional reason for the interrupt
199
+ * @param response - Optional preemptive response to skip the interrupt
200
+ * @returns The interrupt (may have a response if resuming or preemptive)
201
+ */
202
+ getOrCreateInterrupt(id, name, reason, response) {
203
+ const existing = this.interrupts[id];
204
+ if (existing) {
205
+ return existing;
206
+ }
207
+ const interrupt = new Interrupt({
208
+ id,
209
+ name,
210
+ ...(reason !== undefined && { reason }),
211
+ ...(response !== undefined && { response }),
212
+ });
213
+ this.interrupts[id] = interrupt;
214
+ return interrupt;
215
+ }
216
+ /**
217
+ * Serializes the interrupt state to a JSON-compatible object.
218
+ */
219
+ toJSON() {
220
+ const interrupts = {};
221
+ for (const [id, interrupt] of Object.entries(this.interrupts)) {
222
+ interrupts[id] = interrupt.toJSON();
223
+ }
224
+ return {
225
+ interrupts,
226
+ ...(this.resumeResponses && { resumeResponses: this.resumeResponses }),
227
+ activated: this.activated,
228
+ ...(this.pendingToolExecution && { pendingToolExecution: this.pendingToolExecution }),
229
+ };
230
+ }
231
+ /**
232
+ * Creates an InterruptState instance from a JSON object.
233
+ *
234
+ * @param data - JSON data to deserialize
235
+ * @returns InterruptState instance
236
+ */
237
+ static fromJSON(data) {
238
+ const state = new InterruptState();
239
+ state.activated = data.activated;
240
+ for (const [id, interruptData] of Object.entries(data.interrupts)) {
241
+ state.interrupts[id] = Interrupt.fromJSON(interruptData);
242
+ }
243
+ if (data.resumeResponses) {
244
+ state.resumeResponses = data.resumeResponses.map((r) => InterruptResponseContent.fromJSON(r));
245
+ }
246
+ if (data.pendingToolExecution) {
247
+ state.pendingToolExecution = data.pendingToolExecution;
248
+ }
249
+ return state;
250
+ }
251
+ }
252
+ /**
253
+ * Shared interrupt logic that accesses the agent's interrupt state to register or resume an interrupt.
254
+ *
255
+ * @param agent - The agent whose interrupt state to access
256
+ * @param interruptId - Unique identifier for this interrupt instance
257
+ * @param params - Interrupt parameters including name and optional reason
258
+ * @returns The user's response when resuming from an interrupt
259
+ * @throws InterruptError when no response is available (first invocation)
260
+ *
261
+ * @internal
262
+ */
263
+ export function interruptFromAgent(agent, interruptId, params) {
264
+ const interruptState = agent._interruptState;
265
+ if (!interruptState) {
266
+ throw new Error('Interrupt state not available');
267
+ }
268
+ const interrupt = interruptState.getOrCreateInterrupt(interruptId, params.name, params.reason, params.response);
269
+ if (interrupt.response !== undefined) {
270
+ return interrupt.response;
271
+ }
272
+ throw new InterruptError(interrupt);
273
+ }
274
+ //# sourceMappingURL=interrupt.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"interrupt.js","sourceRoot":"","sources":["../../src/interrupt.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,wBAAwB,EAA2D,MAAM,sBAAsB,CAAA;AAGxH,OAAO,EAAE,OAAO,EAAE,eAAe,EAA8C,MAAM,qBAAqB,CAAA;AAE1G;;GAEG;AACH,MAAM,OAAO,SAAS;IACpB;;OAEG;IACM,EAAE,CAAQ;IAEnB;;OAEG;IACM,IAAI,CAAQ;IAErB;;OAEG;IACM,MAAM,CAAY;IAE3B;;OAEG;IACH,QAAQ,CAAY;IAEpB,YAAY,IAA4E;QACtF,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAA;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;QACrB,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QAC3B,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;YACzD,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;SAChE,CAAA;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,QAAQ,CAAC,IAA4E;QAC1F,OAAO,IAAI,SAAS,CAAC,IAAI,CAAC,CAAA;IAC5B,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,KAAK;IACvC;;OAEG;IACM,UAAU,CAAa;IAEhC,YAAY,SAAkC;QAC5C,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;QAC9D,MAAM,OAAO,GACX,GAAG,CAAC,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,qBAAqB,GAAG,CAAC,CAAC,CAAE,CAAC,IAAI,EAAE;YACrC,CAAC,CAAC,GAAG,GAAG,CAAC,MAAM,uBAAuB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;QAC7E,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAA;QAC5B,IAAI,CAAC,UAAU,GAAG,GAAG,CAAA;IACvB,CAAC;CACF;AA4CD;;;;GAIG;AACH,MAAM,OAAO,cAAc;IACzB,sDAAsD;IACtD,UAAU,CAA2B;IAErC,iEAAiE;IACjE,eAAe,CAAyC;IAExD,oDAAoD;IACpD,SAAS,CAAS;IAElB,+CAA+C;IAC/C,oBAAoB,CAAmC;IAEvD;QACE,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,eAAe,GAAG,SAAS,CAAA;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAA;QACtB,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAA;IACvC,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,SAAS,CAAA;QAClB,CAAC;QAED,MAAM,gBAAgB,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAA;QAEhG,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAA2B,CAAA;QAC/D,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACrG,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE,eAAe,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAA;QAC3E,CAAC;QAED,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,CAAA;IACnD,CAAC;IAED;;OAEG;IACH,uBAAuB,CAAC,OAA6B;QACnD,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAA;IACrC,CAAC;IAED;;OAEG;IACH,yBAAyB;QACvB,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAA;IACvC,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IACvC,CAAC;IAED;;OAEG;IACH,uBAAuB;QACrB,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAA;IAC/F,CAAC;IAED;;OAEG;IACH,sBAAsB;QACpB,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACvD,IAAI,SAAS,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO,SAAS,CAAA;YAClB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAA;IAClB,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;IACvB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,eAAe,GAAG,SAAS,CAAA;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAA;QACtB,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAA;IACvC,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,SAAqC;QAC1C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAM;QACR,CAAC;QAED,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;YAChC,MAAM,WAAW,GAAG,OAAO,CAAC,iBAAiB,CAAC,WAAW,CAAA;YACzD,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAA;YAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;YAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,iBAAiB,WAAW,wBAAwB,CAAC,CAAA;YACvE,CAAC;YAED,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAA;QAC/B,CAAC;QAED,IAAI,CAAC,eAAe,GAAG,SAAS,CAAA;IAClC,CAAC;IAED;;;;;;;;;;;OAWG;IACH,oBAAoB,CAAC,EAAU,EAAE,IAAY,EAAE,MAAkB,EAAE,QAAoB;QACrF,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAA;QACpC,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAA;QACjB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;YAC9B,EAAE;YACF,IAAI;YACJ,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,CAAC;YACvC,GAAG,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,CAAC;SAC5C,CAAC,CAAA;QACF,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG,SAAS,CAAA;QAC/B,OAAO,SAAS,CAAA;IAClB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,MAAM,UAAU,GAA2F,EAAE,CAAA;QAC7G,KAAK,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9D,UAAU,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAA;QACrC,CAAC;QAED,OAAO;YACL,UAAU;YACV,GAAG,CAAC,IAAI,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC;YACtE,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,CAAC,IAAI,CAAC,oBAAoB,IAAI,EAAE,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;SACtF,CAAA;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAwB;QACtC,MAAM,KAAK,GAAG,IAAI,cAAc,EAAE,CAAA;QAClC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAEhC,KAAK,MAAM,CAAC,EAAE,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QAC1D,CAAC;QAED,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAA;QAC/F,CAAC;QAED,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,KAAK,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,CAAA;QACxD,CAAC;QAED,OAAO,KAAK,CAAA;IACd,CAAC;CACF;AAUD;;;;;;;;;;GAUG;AACH,MAAM,UAAU,kBAAkB,CAAI,KAAiB,EAAE,WAAmB,EAAE,MAAuB;IACnG,MAAM,cAAc,GAAI,KAAyD,CAAC,eAAe,CAAA;IACjG,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;IAClD,CAAC;IAED,MAAM,SAAS,GAAG,cAAc,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;IAE/G,IAAI,SAAS,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QACrC,OAAO,SAAS,CAAC,QAAa,CAAA;IAChC,CAAC;IAED,MAAM,IAAI,cAAc,CAAC,SAAS,CAAC,CAAA;AACrC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=warn-once.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"warn-once.test.d.ts","sourceRoot":"","sources":["../../../../src/logging/__tests__/warn-once.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,30 @@
1
+ import { describe, it, expect, vi } from 'vitest';
2
+ import { warnOnce } from '../warn-once.js';
3
+ function createLogger() {
4
+ return { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() };
5
+ }
6
+ describe('warnOnce', () => {
7
+ it('emits a warning the first time a message is seen', () => {
8
+ const logger = createLogger();
9
+ warnOnce(logger, 'first-seen-msg');
10
+ expect(logger.warn).toHaveBeenCalledTimes(1);
11
+ expect(logger.warn).toHaveBeenCalledWith('first-seen-msg');
12
+ });
13
+ it('does not emit repeated warnings for the same message', () => {
14
+ const logger = createLogger();
15
+ warnOnce(logger, 'repeated-msg');
16
+ warnOnce(logger, 'repeated-msg');
17
+ warnOnce(logger, 'repeated-msg');
18
+ expect(logger.warn).toHaveBeenCalledTimes(1);
19
+ });
20
+ it('emits distinct messages independently', () => {
21
+ const logger = createLogger();
22
+ warnOnce(logger, 'distinct-alpha-msg');
23
+ warnOnce(logger, 'distinct-beta-msg');
24
+ warnOnce(logger, 'distinct-alpha-msg');
25
+ expect(logger.warn).toHaveBeenCalledTimes(2);
26
+ expect(logger.warn).toHaveBeenNthCalledWith(1, 'distinct-alpha-msg');
27
+ expect(logger.warn).toHaveBeenNthCalledWith(2, 'distinct-beta-msg');
28
+ });
29
+ });
30
+ //# sourceMappingURL=warn-once.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"warn-once.test.js","sourceRoot":"","sources":["../../../../src/logging/__tests__/warn-once.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAE1C,SAAS,YAAY;IACnB,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAA;AACzE,CAAC;AAED,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,MAAM,GAAG,YAAY,EAAE,CAAA;QAC7B,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAA;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC5C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAA;IAC5D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,YAAY,EAAE,CAAA;QAC7B,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAChC,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAChC,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAChC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,MAAM,GAAG,YAAY,EAAE,CAAA;QAC7B,QAAQ,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAA;QACtC,QAAQ,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAA;QACrC,QAAQ,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAA;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC5C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAA;QACpE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,mBAAmB,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -0,0 +1,13 @@
1
+ import type { Logger } from './types.js';
2
+ /**
3
+ * Emits a warning log at most once per unique message per process.
4
+ *
5
+ * Subsequent calls with the same message are no-ops, which prevents
6
+ * repeated nudges (e.g. "using default modelId") from flooding logs
7
+ * when many instances are constructed.
8
+ *
9
+ * @param logger - Logger to emit the warning on
10
+ * @param msg - Warning message; also used as the dedupe key
11
+ */
12
+ export declare function warnOnce(logger: Logger, msg: string): void;
13
+ //# sourceMappingURL=warn-once.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"warn-once.d.ts","sourceRoot":"","sources":["../../../src/logging/warn-once.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAIxC;;;;;;;;;GASG;AACH,wBAAgB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI,CAI1D"}
@@ -0,0 +1,18 @@
1
+ const warned = new Set();
2
+ /**
3
+ * Emits a warning log at most once per unique message per process.
4
+ *
5
+ * Subsequent calls with the same message are no-ops, which prevents
6
+ * repeated nudges (e.g. "using default modelId") from flooding logs
7
+ * when many instances are constructed.
8
+ *
9
+ * @param logger - Logger to emit the warning on
10
+ * @param msg - Warning message; also used as the dedupe key
11
+ */
12
+ export function warnOnce(logger, msg) {
13
+ if (warned.has(msg))
14
+ return;
15
+ logger.warn(msg);
16
+ warned.add(msg);
17
+ }
18
+ //# sourceMappingURL=warn-once.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"warn-once.js","sourceRoot":"","sources":["../../../src/logging/warn-once.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAA;AAEhC;;;;;;;;;GASG;AACH,MAAM,UAAU,QAAQ,CAAC,MAAc,EAAE,GAAW;IAClD,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;QAAE,OAAM;IAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAChB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;AACjB,CAAC"}
package/dist/src/mcp.d.ts CHANGED
@@ -1,7 +1,20 @@
1
1
  import { Client } from '@modelcontextprotocol/sdk/client/index.js';
2
2
  import type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
3
+ import { type ServerCapabilities, type Implementation, type LoggingMessageNotificationParams } from '@modelcontextprotocol/sdk/types.js';
3
4
  import type { JSONValue } from './types/json.js';
5
+ import type { ElicitationCallback } from './types/elicitation.js';
4
6
  import { McpTool } from './tools/mcp-tool.js';
7
+ /**
8
+ * Widened transport type that accepts MCP transport implementations without requiring explicit casts.
9
+ *
10
+ * Under `exactOptionalPropertyTypes`, `StreamableHTTPClientTransport` is not directly assignable
11
+ * to `Transport` because its `sessionId` getter returns `string | undefined`, while `Transport`
12
+ * declares `sessionId?: string` (absent or string, but not explicitly undefined).
13
+ * This type relaxes that constraint so users can pass any MCP transport without `as Transport`.
14
+ */
15
+ export type McpTransport = Omit<Transport, 'sessionId'> & {
16
+ sessionId?: string | undefined;
17
+ };
5
18
  /** Temporary placeholder for RuntimeConfig */
6
19
  export interface RuntimeConfig {
7
20
  applicationName?: string;
@@ -29,9 +42,11 @@ export interface TasksConfig {
29
42
  */
30
43
  pollTimeout?: number;
31
44
  }
45
+ /** Connection state of an MCP client. */
46
+ export type McpConnectionState = 'disconnected' | 'connected' | 'failed';
32
47
  /** Arguments for configuring an MCP Client. */
33
48
  export type McpClientConfig = RuntimeConfig & {
34
- transport: Transport;
49
+ transport: McpTransport;
35
50
  /** Disable OpenTelemetry MCP instrumentation. */
36
51
  disableMcpInstrumentation?: boolean;
37
52
  /**
@@ -40,6 +55,16 @@ export type McpClientConfig = RuntimeConfig & {
40
55
  * When undefined, tools are called directly without task management.
41
56
  */
42
57
  tasksConfig?: TasksConfig;
58
+ /**
59
+ * Callback to handle server-initiated elicitation requests.
60
+ * When provided, the client advertises elicitation support (form + url modes)
61
+ * and routes incoming elicitation requests to this callback.
62
+ */
63
+ elicitationCallback?: ElicitationCallback;
64
+ /** When true, connection failures are logged as warnings instead of throwing. */
65
+ failOpen?: boolean;
66
+ /** Called when the server emits a log message. Defaults to routing through the Strands logger. */
67
+ logHandler?: (params: LoggingMessageNotificationParams) => void;
43
68
  };
44
69
  /** MCP Client for interacting with Model Context Protocol servers. */
45
70
  export declare class McpClient {
@@ -50,17 +75,27 @@ export declare class McpClient {
50
75
  private _clientName;
51
76
  private _clientVersion;
52
77
  private _transport;
53
- private _connected;
78
+ private _state;
54
79
  private _client;
80
+ private _failOpen;
81
+ private _logHandler;
55
82
  private _disableMcpInstrumentation;
56
83
  private _tasksConfig;
84
+ private _elicitationCallback;
57
85
  constructor(args: McpClientConfig);
58
86
  get client(): Client;
87
+ get serverCapabilities(): ServerCapabilities | undefined;
88
+ get serverVersion(): Implementation | undefined;
89
+ get serverInstructions(): string | undefined;
90
+ get connectionState(): McpConnectionState;
59
91
  /**
60
92
  * Connects the MCP client to the server.
61
93
  *
62
- * This function is exposed to allow consumers to connect manually, but will be called lazily before any operations that require a connection.
94
+ * Called lazily before any operation that requires a connection. When `failOpen` is true,
95
+ * connection failures are swallowed and the client enters a `'failed'` state — subsequent
96
+ * calls are no-ops until `connect(true)` is called explicitly to retry.
63
97
  *
98
+ * @param reconnect - When true, forces a reconnect even if already connected or failed.
64
99
  * @returns A promise that resolves when the connection is established.
65
100
  */
66
101
  connect(reconnect?: boolean): Promise<void>;
@@ -70,6 +105,11 @@ export declare class McpClient {
70
105
  * @returns A promise that resolves when the disconnection is complete.
71
106
  */
72
107
  disconnect(): Promise<void>;
108
+ /**
109
+ * Enables the `await using` pattern for automatic resource cleanup.
110
+ * Delegates to {@link McpClient.disconnect}.
111
+ */
112
+ [Symbol.asyncDispose](): Promise<void>;
73
113
  /**
74
114
  * Lists the tools available on the server and returns them as executable McpTool instances.
75
115
  *
@@ -1 +1 @@
1
- {"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../../src/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAA;AAClE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAA;AAG9E,OAAO,KAAK,EAAc,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAC5D,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAA;AAG7C,8CAA8C;AAC9C,MAAM,WAAW,aAAa;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,kBAAkB,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,WAAW;IAC1B;;;OAGG;IACH,GAAG,CAAC,EAAE,MAAM,CAAA;IAEZ;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAA;CACrB;AAED,+CAA+C;AAC/C,MAAM,MAAM,eAAe,GAAG,aAAa,GAAG;IAC5C,SAAS,EAAE,SAAS,CAAA;IAEpB,iDAAiD;IACjD,yBAAyB,CAAC,EAAE,OAAO,CAAA;IAEnC;;;;OAIG;IACH,WAAW,CAAC,EAAE,WAAW,CAAA;CAC1B,CAAA;AAED,sEAAsE;AACtE,qBAAa,SAAS;IACpB,iEAAiE;IACjE,gBAAuB,WAAW,SAAQ;IAE1C,4EAA4E;IAC5E,gBAAuB,oBAAoB,UAAS;IAEpD,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,cAAc,CAAQ;IAC9B,OAAO,CAAC,UAAU,CAAW;IAC7B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,OAAO,CAAQ;IACvB,OAAO,CAAC,0BAA0B,CAAS;IAC3C,OAAO,CAAC,YAAY,CAAyB;gBAEjC,IAAI,EAAE,eAAe;IAcjC,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED;;;;;;OAMG;IACU,OAAO,CAAC,SAAS,GAAE,OAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAe/D;;;;OAIG;IACU,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOxC;;;;OAIG;IACU,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAgB5C;;;;;;;;;;OAUG;IACU,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;CAqC1E"}
1
+ {"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../../src/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAA;AAClE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAA;AAE9E,OAAO,EAGL,KAAK,kBAAkB,EACvB,KAAK,cAAc,EACnB,KAAK,gCAAgC,EACtC,MAAM,oCAAoC,CAAA;AAE3C,OAAO,KAAK,EAAc,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAC5D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAA;AACjE,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAA;AAG7C;;;;;;;GAOG;AACH,MAAM,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAA;AAE5F,8CAA8C;AAC9C,MAAM,WAAW,aAAa;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,kBAAkB,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,WAAW;IAC1B;;;OAGG;IACH,GAAG,CAAC,EAAE,MAAM,CAAA;IAEZ;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAA;CACrB;AAED,yCAAyC;AACzC,MAAM,MAAM,kBAAkB,GAAG,cAAc,GAAG,WAAW,GAAG,QAAQ,CAAA;AAExE,+CAA+C;AAC/C,MAAM,MAAM,eAAe,GAAG,aAAa,GAAG;IAC5C,SAAS,EAAE,YAAY,CAAA;IAEvB,iDAAiD;IACjD,yBAAyB,CAAC,EAAE,OAAO,CAAA;IAEnC;;;;OAIG;IACH,WAAW,CAAC,EAAE,WAAW,CAAA;IAEzB;;;;OAIG;IACH,mBAAmB,CAAC,EAAE,mBAAmB,CAAA;IAEzC,iFAAiF;IACjF,QAAQ,CAAC,EAAE,OAAO,CAAA;IAElB,kGAAkG;IAClG,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,gCAAgC,KAAK,IAAI,CAAA;CAChE,CAAA;AAED,sEAAsE;AACtE,qBAAa,SAAS;IACpB,iEAAiE;IACjE,gBAAuB,WAAW,SAAQ;IAE1C,4EAA4E;IAC5E,gBAAuB,oBAAoB,UAAS;IAEpD,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,cAAc,CAAQ;IAC9B,OAAO,CAAC,UAAU,CAAW;IAC7B,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,OAAO,CAAQ;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAoD;IACvE,OAAO,CAAC,0BAA0B,CAAS;IAC3C,OAAO,CAAC,YAAY,CAAyB;IAC7C,OAAO,CAAC,oBAAoB,CAAiC;gBAEjD,IAAI,EAAE,eAAe;IAwBjC,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED,IAAI,kBAAkB,IAAI,kBAAkB,GAAG,SAAS,CAEvD;IAED,IAAI,aAAa,IAAI,cAAc,GAAG,SAAS,CAE9C;IAED,IAAI,kBAAkB,IAAI,MAAM,GAAG,SAAS,CAE3C;IAED,IAAI,eAAe,IAAI,kBAAkB,CAExC;IAED;;;;;;;;;OASG;IACU,OAAO,CAAC,SAAS,GAAE,OAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IA2B/D;;;;OAIG;IACU,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOxC;;;OAGG;IACG,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5C;;;;OAIG;IACU,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IA4B5C;;;;;;;;;;OAUG;IACU,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;CAsC1E"}
package/dist/src/mcp.js CHANGED
@@ -1,5 +1,6 @@
1
1
  import { Client } from '@modelcontextprotocol/sdk/client/index.js';
2
2
  import { takeResult } from '@modelcontextprotocol/sdk/shared/responseMessage.js';
3
+ import { ElicitRequestSchema, LoggingMessageNotificationSchema, } from '@modelcontextprotocol/sdk/types.js';
3
4
  import { context, propagation, trace } from '@opentelemetry/api';
4
5
  import { McpTool } from './tools/mcp-tool.js';
5
6
  import { logger } from './logging/index.js';
@@ -12,42 +13,79 @@ export class McpClient {
12
13
  _clientName;
13
14
  _clientVersion;
14
15
  _transport;
15
- _connected;
16
+ _state;
16
17
  _client;
18
+ _failOpen;
19
+ _logHandler;
17
20
  _disableMcpInstrumentation;
18
21
  _tasksConfig;
22
+ _elicitationCallback;
19
23
  constructor(args) {
20
24
  this._clientName = args.applicationName || 'strands-agents-ts-sdk';
21
25
  this._clientVersion = args.applicationVersion || '0.0.1';
22
26
  this._transport = args.transport;
23
- this._connected = false;
27
+ this._state = 'disconnected';
28
+ this._failOpen = args.failOpen ?? false;
29
+ this._logHandler = args.logHandler ?? defaultLogHandler;
24
30
  this._tasksConfig = args.tasksConfig;
31
+ this._elicitationCallback = args.elicitationCallback;
25
32
  this._client = new Client({
26
33
  name: this._clientName,
27
34
  version: this._clientVersion,
35
+ }, this._elicitationCallback ? { capabilities: { elicitation: { form: {}, url: {} } } } : undefined);
36
+ this._client.setNotificationHandler(LoggingMessageNotificationSchema, (notification) => {
37
+ this._logHandler(notification.params);
28
38
  });
29
39
  this._disableMcpInstrumentation = args.disableMcpInstrumentation ?? false;
30
40
  }
31
41
  get client() {
32
42
  return this._client;
33
43
  }
44
+ get serverCapabilities() {
45
+ return this._client.getServerCapabilities();
46
+ }
47
+ get serverVersion() {
48
+ return this._client.getServerVersion();
49
+ }
50
+ get serverInstructions() {
51
+ return this._client.getInstructions();
52
+ }
53
+ get connectionState() {
54
+ return this._state;
55
+ }
34
56
  /**
35
57
  * Connects the MCP client to the server.
36
58
  *
37
- * This function is exposed to allow consumers to connect manually, but will be called lazily before any operations that require a connection.
59
+ * Called lazily before any operation that requires a connection. When `failOpen` is true,
60
+ * connection failures are swallowed and the client enters a `'failed'` state — subsequent
61
+ * calls are no-ops until `connect(true)` is called explicitly to retry.
38
62
  *
63
+ * @param reconnect - When true, forces a reconnect even if already connected or failed.
39
64
  * @returns A promise that resolves when the connection is established.
40
65
  */
41
66
  async connect(reconnect = false) {
42
- if (this._connected && !reconnect) {
67
+ if (this._state !== 'disconnected' && !reconnect)
43
68
  return;
44
- }
45
- if (this._connected && reconnect) {
69
+ if (this._state === 'connected' && reconnect) {
46
70
  await this._client.close();
47
- this._connected = false;
71
+ this._state = 'disconnected';
72
+ }
73
+ if (this._elicitationCallback) {
74
+ const callback = this._elicitationCallback;
75
+ this._client.setRequestHandler(ElicitRequestSchema, async (request, extra) => {
76
+ return await callback(extra, request.params);
77
+ });
78
+ }
79
+ try {
80
+ await this._client.connect(this._transport);
81
+ this._state = 'connected';
82
+ }
83
+ catch (error) {
84
+ if (!this._failOpen)
85
+ throw error;
86
+ this._state = 'failed';
87
+ logger.warn(`client=<${this._clientName}>, error=<${error}> | MCP server failed to connect, continuing with failOpen`);
48
88
  }
49
- await this._client.connect(this._transport);
50
- this._connected = true;
51
89
  }
52
90
  /**
53
91
  * Disconnects the MCP client from the server and cleans up resources.
@@ -58,7 +96,14 @@ export class McpClient {
58
96
  // Must be done sequentially
59
97
  await this._client.close();
60
98
  await this._transport.close();
61
- this._connected = false;
99
+ this._state = 'disconnected';
100
+ }
101
+ /**
102
+ * Enables the `await using` pattern for automatic resource cleanup.
103
+ * Delegates to {@link McpClient.disconnect}.
104
+ */
105
+ async [Symbol.asyncDispose]() {
106
+ await this.disconnect();
62
107
  }
63
108
  /**
64
109
  * Lists the tools available on the server and returns them as executable McpTool instances.
@@ -67,16 +112,21 @@ export class McpClient {
67
112
  */
68
113
  async listTools() {
69
114
  await this.connect();
70
- const result = await this._client.listTools();
71
- // Map the tool specifications to fully functional McpTool instances
72
- return result.tools.map((toolSpec) => {
73
- return new McpTool({
115
+ if (this._state === 'failed')
116
+ return [];
117
+ const tools = [];
118
+ let cursor;
119
+ do {
120
+ const result = await this._client.listTools(cursor ? { cursor } : undefined);
121
+ tools.push(...result.tools.map((toolSpec) => new McpTool({
74
122
  name: toolSpec.name,
75
- description: toolSpec.description ?? '',
123
+ description: toolSpec.description || `Tool which performs ${toolSpec.name}`,
76
124
  inputSchema: toolSpec.inputSchema,
77
125
  client: this,
78
- });
79
- });
126
+ })));
127
+ cursor = result.nextCursor;
128
+ } while (cursor);
129
+ return tools;
80
130
  }
81
131
  /**
82
132
  * Invoke a tool on the connected MCP server using an McpTool instance.
@@ -91,6 +141,8 @@ export class McpClient {
91
141
  */
92
142
  async callTool(tool, args) {
93
143
  await this.connect();
144
+ if (this._state === 'failed')
145
+ throw new Error('MCP server failed to connect. Call connect(true) to retry.');
94
146
  if (args === null || args === undefined) {
95
147
  return await this.callTool(tool, {});
96
148
  }
@@ -116,6 +168,22 @@ export class McpClient {
116
168
  return result;
117
169
  }
118
170
  }
171
+ function defaultLogHandler(params) {
172
+ const { level, logger: serverLogger, data } = params;
173
+ const message = `logger=<${serverLogger ?? 'mcp'}>, data=<${JSON.stringify(data)}> | MCP server log`;
174
+ if (level === 'debug') {
175
+ logger.debug(message);
176
+ }
177
+ else if (level === 'info' || level === 'notice') {
178
+ logger.info(message);
179
+ }
180
+ else if (level === 'warning') {
181
+ logger.warn(message);
182
+ }
183
+ else {
184
+ logger.error(message);
185
+ }
186
+ }
119
187
  /**
120
188
  * Injects OpenTelemetry trace context into MCP tool call arguments.
121
189
  * Returns the args with a `_meta` field containing W3C traceparent headers.