@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.
- package/LICENSE +175 -0
- package/README.md +340 -0
- package/dist/src/__fixtures__/agent-helpers.d.ts +22 -1
- package/dist/src/__fixtures__/agent-helpers.d.ts.map +1 -1
- package/dist/src/__fixtures__/agent-helpers.js +45 -1
- package/dist/src/__fixtures__/agent-helpers.js.map +1 -1
- package/dist/src/__fixtures__/mock-plugin.d.ts.map +1 -1
- package/dist/src/__fixtures__/mock-plugin.js +3 -1
- package/dist/src/__fixtures__/mock-plugin.js.map +1 -1
- package/dist/src/__fixtures__/tool-helpers.d.ts +5 -2
- package/dist/src/__fixtures__/tool-helpers.d.ts.map +1 -1
- package/dist/src/__fixtures__/tool-helpers.js +23 -4
- package/dist/src/__fixtures__/tool-helpers.js.map +1 -1
- package/dist/src/__tests__/interrupt.test.d.ts +2 -0
- package/dist/src/__tests__/interrupt.test.d.ts.map +1 -0
- package/dist/src/__tests__/interrupt.test.js +259 -0
- package/dist/src/__tests__/interrupt.test.js.map +1 -0
- package/dist/src/__tests__/mcp.test.js +448 -2
- package/dist/src/__tests__/mcp.test.js.map +1 -1
- package/dist/src/a2a/__tests__/events.test.js +2 -0
- package/dist/src/a2a/__tests__/events.test.js.map +1 -1
- package/dist/src/a2a/__tests__/executor.test.js +16 -5
- package/dist/src/a2a/__tests__/executor.test.js.map +1 -1
- package/dist/src/a2a/a2a-agent.d.ts +8 -3
- package/dist/src/a2a/a2a-agent.d.ts.map +1 -1
- package/dist/src/a2a/a2a-agent.js +12 -6
- package/dist/src/a2a/a2a-agent.js.map +1 -1
- package/dist/src/a2a/executor.d.ts +13 -0
- package/dist/src/a2a/executor.d.ts.map +1 -1
- package/dist/src/a2a/executor.js +19 -1
- package/dist/src/a2a/executor.js.map +1 -1
- package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.js +23 -0
- package/dist/src/agent/__tests__/agent-as-tool.invocation-state.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.cancel.test.js +1 -1
- package/dist/src/agent/__tests__/agent.cancel.test.js.map +1 -1
- package/dist/src/agent/__tests__/agent.concurrent.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent.concurrent.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent.concurrent.test.js +488 -0
- package/dist/src/agent/__tests__/agent.concurrent.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.hook.test.js +724 -12
- package/dist/src/agent/__tests__/agent.hook.test.js.map +1 -1
- package/dist/src/agent/__tests__/agent.interrupt.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent.interrupt.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent.interrupt.test.js +730 -0
- package/dist/src/agent/__tests__/agent.interrupt.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.invocation-state.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent.invocation-state.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent.invocation-state.test.js +219 -0
- package/dist/src/agent/__tests__/agent.invocation-state.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.model-retry.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent.model-retry.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent.model-retry.test.js +161 -0
- package/dist/src/agent/__tests__/agent.model-retry.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.stateful-model.test.d.ts +2 -0
- package/dist/src/agent/__tests__/agent.stateful-model.test.d.ts.map +1 -0
- package/dist/src/agent/__tests__/agent.stateful-model.test.js +169 -0
- package/dist/src/agent/__tests__/agent.stateful-model.test.js.map +1 -0
- package/dist/src/agent/__tests__/agent.test.js +217 -2
- package/dist/src/agent/__tests__/agent.test.js.map +1 -1
- package/dist/src/agent/__tests__/agent.tracer.test.node.js +39 -0
- package/dist/src/agent/__tests__/agent.tracer.test.node.js.map +1 -1
- package/dist/src/agent/__tests__/snapshot.test.js +51 -4
- package/dist/src/agent/__tests__/snapshot.test.js.map +1 -1
- package/dist/src/agent/agent-as-tool.d.ts.map +1 -1
- package/dist/src/agent/agent-as-tool.js +4 -2
- package/dist/src/agent/agent-as-tool.js.map +1 -1
- package/dist/src/agent/agent.d.ts +109 -4
- package/dist/src/agent/agent.d.ts.map +1 -1
- package/dist/src/agent/agent.js +790 -224
- package/dist/src/agent/agent.js.map +1 -1
- package/dist/src/agent/snapshot.d.ts +2 -2
- package/dist/src/agent/snapshot.d.ts.map +1 -1
- package/dist/src/agent/snapshot.js +20 -2
- package/dist/src/agent/snapshot.js.map +1 -1
- package/dist/src/conversation-manager/__tests__/conversation-manager.test.js +230 -9
- package/dist/src/conversation-manager/__tests__/conversation-manager.test.js.map +1 -1
- package/dist/src/conversation-manager/__tests__/null-conversation-manager.test.js +19 -6
- package/dist/src/conversation-manager/__tests__/null-conversation-manager.test.js.map +1 -1
- package/dist/src/conversation-manager/__tests__/sliding-window-conversation-manager.test.js +58 -4
- package/dist/src/conversation-manager/__tests__/sliding-window-conversation-manager.test.js.map +1 -1
- package/dist/src/conversation-manager/__tests__/summarizing-conversation-manager.test.js +76 -1
- package/dist/src/conversation-manager/__tests__/summarizing-conversation-manager.test.js.map +1 -1
- package/dist/src/conversation-manager/conversation-manager.d.ts +67 -22
- package/dist/src/conversation-manager/conversation-manager.d.ts.map +1 -1
- package/dist/src/conversation-manager/conversation-manager.js +65 -13
- package/dist/src/conversation-manager/conversation-manager.js.map +1 -1
- package/dist/src/conversation-manager/index.d.ts +1 -1
- package/dist/src/conversation-manager/index.d.ts.map +1 -1
- package/dist/src/conversation-manager/index.js +1 -1
- package/dist/src/conversation-manager/index.js.map +1 -1
- package/dist/src/conversation-manager/sliding-window-conversation-manager.d.ts +17 -3
- package/dist/src/conversation-manager/sliding-window-conversation-manager.d.ts.map +1 -1
- package/dist/src/conversation-manager/sliding-window-conversation-manager.js +10 -4
- package/dist/src/conversation-manager/sliding-window-conversation-manager.js.map +1 -1
- package/dist/src/conversation-manager/summarizing-conversation-manager.d.ts +23 -1
- package/dist/src/conversation-manager/summarizing-conversation-manager.d.ts.map +1 -1
- package/dist/src/conversation-manager/summarizing-conversation-manager.js +39 -17
- package/dist/src/conversation-manager/summarizing-conversation-manager.js.map +1 -1
- package/dist/src/errors.d.ts +11 -0
- package/dist/src/errors.d.ts.map +1 -1
- package/dist/src/errors.js +12 -0
- package/dist/src/errors.js.map +1 -1
- package/dist/src/hooks/__tests__/events.test.js +267 -73
- package/dist/src/hooks/__tests__/events.test.js.map +1 -1
- package/dist/src/hooks/__tests__/registry.test.js +182 -18
- package/dist/src/hooks/__tests__/registry.test.js.map +1 -1
- package/dist/src/hooks/events.d.ts +193 -51
- package/dist/src/hooks/events.d.ts.map +1 -1
- package/dist/src/hooks/events.js +182 -26
- package/dist/src/hooks/events.js.map +1 -1
- package/dist/src/hooks/index.d.ts +3 -2
- package/dist/src/hooks/index.d.ts.map +1 -1
- package/dist/src/hooks/index.js +1 -0
- package/dist/src/hooks/index.js.map +1 -1
- package/dist/src/hooks/registry.d.ts +12 -12
- package/dist/src/hooks/registry.d.ts.map +1 -1
- package/dist/src/hooks/registry.js +55 -15
- package/dist/src/hooks/registry.js.map +1 -1
- package/dist/src/hooks/types.d.ts +23 -0
- package/dist/src/hooks/types.d.ts.map +1 -1
- package/dist/src/hooks/types.js +17 -1
- package/dist/src/hooks/types.js.map +1 -1
- package/dist/src/index.d.ts +12 -7
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +4 -1
- package/dist/src/index.js.map +1 -1
- package/dist/src/interrupt.d.ts +220 -0
- package/dist/src/interrupt.d.ts.map +1 -0
- package/dist/src/interrupt.js +274 -0
- package/dist/src/interrupt.js.map +1 -0
- package/dist/src/logging/__tests__/warn-once.test.d.ts +2 -0
- package/dist/src/logging/__tests__/warn-once.test.d.ts.map +1 -0
- package/dist/src/logging/__tests__/warn-once.test.js +30 -0
- package/dist/src/logging/__tests__/warn-once.test.js.map +1 -0
- package/dist/src/logging/warn-once.d.ts +13 -0
- package/dist/src/logging/warn-once.d.ts.map +1 -0
- package/dist/src/logging/warn-once.js +18 -0
- package/dist/src/logging/warn-once.js.map +1 -0
- package/dist/src/mcp.d.ts +43 -3
- package/dist/src/mcp.d.ts.map +1 -1
- package/dist/src/mcp.js +85 -17
- package/dist/src/mcp.js.map +1 -1
- package/dist/src/mime.d.ts +2 -1
- package/dist/src/mime.d.ts.map +1 -1
- package/dist/src/mime.js +1 -0
- package/dist/src/mime.js.map +1 -1
- package/dist/src/models/__tests__/anthropic.test.js +147 -3
- package/dist/src/models/__tests__/anthropic.test.js.map +1 -1
- package/dist/src/models/__tests__/bedrock.test.js +228 -2
- package/dist/src/models/__tests__/bedrock.test.js.map +1 -1
- package/dist/src/models/__tests__/defaults.test.d.ts +2 -0
- package/dist/src/models/__tests__/defaults.test.d.ts.map +1 -0
- package/dist/src/models/__tests__/defaults.test.js +36 -0
- package/dist/src/models/__tests__/defaults.test.js.map +1 -0
- package/dist/src/models/__tests__/google.test.js +135 -0
- package/dist/src/models/__tests__/google.test.js.map +1 -1
- package/dist/src/models/__tests__/model.test.js +149 -1
- package/dist/src/models/__tests__/model.test.js.map +1 -1
- package/dist/src/models/anthropic.d.ts +20 -1
- package/dist/src/models/anthropic.d.ts.map +1 -1
- package/dist/src/models/anthropic.js +42 -8
- package/dist/src/models/anthropic.js.map +1 -1
- package/dist/src/models/bedrock.d.ts +27 -1
- package/dist/src/models/bedrock.d.ts.map +1 -1
- package/dist/src/models/bedrock.js +100 -12
- package/dist/src/models/bedrock.js.map +1 -1
- package/dist/src/models/defaults.d.ts +47 -0
- package/dist/src/models/defaults.d.ts.map +1 -0
- package/dist/src/models/defaults.js +170 -0
- package/dist/src/models/defaults.js.map +1 -0
- package/dist/src/models/google/model.d.ts +14 -1
- package/dist/src/models/google/model.d.ts.map +1 -1
- package/dist/src/models/google/model.js +54 -8
- package/dist/src/models/google/model.js.map +1 -1
- package/dist/src/models/google/types.d.ts +8 -0
- package/dist/src/models/google/types.d.ts.map +1 -1
- package/dist/src/models/model.d.ts +65 -0
- package/dist/src/models/model.d.ts.map +1 -1
- package/dist/src/models/model.js +138 -0
- package/dist/src/models/model.js.map +1 -1
- package/dist/src/models/openai/__tests__/chat.test.d.ts +2 -0
- package/dist/src/models/openai/__tests__/chat.test.d.ts.map +1 -0
- package/dist/src/models/{__tests__/openai.test.js → openai/__tests__/chat.test.js} +117 -7
- package/dist/src/models/openai/__tests__/chat.test.js.map +1 -0
- package/dist/src/models/openai/__tests__/responses.test.d.ts +2 -0
- package/dist/src/models/openai/__tests__/responses.test.d.ts.map +1 -0
- package/dist/src/models/openai/__tests__/responses.test.js +668 -0
- package/dist/src/models/openai/__tests__/responses.test.js.map +1 -0
- package/dist/src/models/openai/chat-adapter.d.ts +33 -0
- package/dist/src/models/openai/chat-adapter.d.ts.map +1 -0
- package/dist/src/models/openai/chat-adapter.js +383 -0
- package/dist/src/models/openai/chat-adapter.js.map +1 -0
- package/dist/src/models/openai/errors.d.ts +16 -0
- package/dist/src/models/openai/errors.d.ts.map +1 -0
- package/dist/src/models/openai/errors.js +40 -0
- package/dist/src/models/openai/errors.js.map +1 -0
- package/dist/src/models/openai/formatting.d.ts +18 -0
- package/dist/src/models/openai/formatting.d.ts.map +1 -0
- package/dist/src/models/openai/formatting.js +38 -0
- package/dist/src/models/openai/formatting.js.map +1 -0
- package/dist/src/models/openai/index.d.ts +19 -0
- package/dist/src/models/openai/index.d.ts.map +1 -0
- package/dist/src/models/openai/index.js +18 -0
- package/dist/src/models/openai/index.js.map +1 -0
- package/dist/src/models/openai/model.d.ts +77 -0
- package/dist/src/models/openai/model.d.ts.map +1 -0
- package/dist/src/models/openai/model.js +211 -0
- package/dist/src/models/openai/model.js.map +1 -0
- package/dist/src/models/openai/responses-adapter.d.ts +78 -0
- package/dist/src/models/openai/responses-adapter.d.ts.map +1 -0
- package/dist/src/models/openai/responses-adapter.js +467 -0
- package/dist/src/models/openai/responses-adapter.js.map +1 -0
- package/dist/src/models/openai/types.d.ts +131 -0
- package/dist/src/models/openai/types.d.ts.map +1 -0
- package/dist/src/models/openai/types.js +5 -0
- package/dist/src/models/openai/types.js.map +1 -0
- package/dist/src/multiagent/__tests__/events.test.js +122 -28
- package/dist/src/multiagent/__tests__/events.test.js.map +1 -1
- package/dist/src/multiagent/__tests__/graph.invocation-state.test.d.ts +2 -0
- package/dist/src/multiagent/__tests__/graph.invocation-state.test.d.ts.map +1 -0
- package/dist/src/multiagent/__tests__/graph.invocation-state.test.js +95 -0
- package/dist/src/multiagent/__tests__/graph.invocation-state.test.js.map +1 -0
- package/dist/src/multiagent/__tests__/graph.test.js +69 -0
- package/dist/src/multiagent/__tests__/graph.test.js.map +1 -1
- package/dist/src/multiagent/__tests__/nodes.test.js +18 -2
- package/dist/src/multiagent/__tests__/nodes.test.js.map +1 -1
- package/dist/src/multiagent/__tests__/swarm.invocation-state.test.d.ts +2 -0
- package/dist/src/multiagent/__tests__/swarm.invocation-state.test.d.ts.map +1 -0
- package/dist/src/multiagent/__tests__/swarm.invocation-state.test.js +56 -0
- package/dist/src/multiagent/__tests__/swarm.invocation-state.test.js.map +1 -0
- package/dist/src/multiagent/__tests__/swarm.test.js +77 -0
- package/dist/src/multiagent/__tests__/swarm.test.js.map +1 -1
- package/dist/src/multiagent/events.d.ts +19 -1
- package/dist/src/multiagent/events.d.ts.map +1 -1
- package/dist/src/multiagent/events.js +18 -0
- package/dist/src/multiagent/events.js.map +1 -1
- package/dist/src/multiagent/graph.d.ts +27 -5
- package/dist/src/multiagent/graph.d.ts.map +1 -1
- package/dist/src/multiagent/graph.js +61 -15
- package/dist/src/multiagent/graph.js.map +1 -1
- package/dist/src/multiagent/index.d.ts +1 -1
- package/dist/src/multiagent/index.d.ts.map +1 -1
- package/dist/src/multiagent/multiagent.d.ts +21 -6
- package/dist/src/multiagent/multiagent.d.ts.map +1 -1
- package/dist/src/multiagent/nodes.d.ts +28 -3
- package/dist/src/multiagent/nodes.d.ts.map +1 -1
- package/dist/src/multiagent/nodes.js +42 -7
- package/dist/src/multiagent/nodes.js.map +1 -1
- package/dist/src/multiagent/swarm.d.ts +20 -4
- package/dist/src/multiagent/swarm.d.ts.map +1 -1
- package/dist/src/multiagent/swarm.js +65 -16
- package/dist/src/multiagent/swarm.js.map +1 -1
- package/dist/src/plugins/__tests__/registry.test.js +1 -1
- package/dist/src/plugins/__tests__/registry.test.js.map +1 -1
- package/dist/src/plugins/model-plugin.d.ts +20 -0
- package/dist/src/plugins/model-plugin.d.ts.map +1 -0
- package/dist/src/plugins/model-plugin.js +29 -0
- package/dist/src/plugins/model-plugin.js.map +1 -0
- package/dist/src/registry/__tests__/tool-registry.test.js +11 -0
- package/dist/src/registry/__tests__/tool-registry.test.js.map +1 -1
- package/dist/src/registry/tool-registry.d.ts +4 -0
- package/dist/src/registry/tool-registry.d.ts.map +1 -1
- package/dist/src/registry/tool-registry.js +6 -0
- package/dist/src/registry/tool-registry.js.map +1 -1
- package/dist/src/retry/__tests__/backoff-strategy.test.d.ts +2 -0
- package/dist/src/retry/__tests__/backoff-strategy.test.d.ts.map +1 -0
- package/dist/src/retry/__tests__/backoff-strategy.test.js +116 -0
- package/dist/src/retry/__tests__/backoff-strategy.test.js.map +1 -0
- package/dist/src/retry/__tests__/default-model-retry-strategy.test.d.ts +2 -0
- package/dist/src/retry/__tests__/default-model-retry-strategy.test.d.ts.map +1 -0
- package/dist/src/retry/__tests__/default-model-retry-strategy.test.js +225 -0
- package/dist/src/retry/__tests__/default-model-retry-strategy.test.js.map +1 -0
- package/dist/src/retry/backoff-strategy.d.ts +108 -0
- package/dist/src/retry/backoff-strategy.d.ts.map +1 -0
- package/dist/src/retry/backoff-strategy.js +86 -0
- package/dist/src/retry/backoff-strategy.js.map +1 -0
- package/dist/src/retry/default-model-retry-strategy.d.ts +76 -0
- package/dist/src/retry/default-model-retry-strategy.d.ts.map +1 -0
- package/dist/src/retry/default-model-retry-strategy.js +104 -0
- package/dist/src/retry/default-model-retry-strategy.js.map +1 -0
- package/dist/src/retry/index.d.ts +8 -0
- package/dist/src/retry/index.d.ts.map +1 -0
- package/dist/src/retry/index.js +7 -0
- package/dist/src/retry/index.js.map +1 -0
- package/dist/src/retry/model-retry-strategy.d.ts +80 -0
- package/dist/src/retry/model-retry-strategy.d.ts.map +1 -0
- package/dist/src/retry/model-retry-strategy.js +85 -0
- package/dist/src/retry/model-retry-strategy.js.map +1 -0
- package/dist/src/retry/retry-strategy.d.ts +34 -0
- package/dist/src/retry/retry-strategy.d.ts.map +1 -0
- package/dist/src/retry/retry-strategy.js +25 -0
- package/dist/src/retry/retry-strategy.js.map +1 -0
- package/dist/src/session/__tests__/session-manager.test.js +52 -11
- package/dist/src/session/__tests__/session-manager.test.js.map +1 -1
- package/dist/src/session/session-manager.d.ts +6 -0
- package/dist/src/session/session-manager.d.ts.map +1 -1
- package/dist/src/session/session-manager.js +17 -0
- package/dist/src/session/session-manager.js.map +1 -1
- package/dist/src/telemetry/__tests__/meter.test.js +23 -0
- package/dist/src/telemetry/__tests__/meter.test.js.map +1 -1
- package/dist/src/telemetry/meter.d.ts +15 -0
- package/dist/src/telemetry/meter.d.ts.map +1 -1
- package/dist/src/telemetry/meter.js +14 -0
- package/dist/src/telemetry/meter.js.map +1 -1
- package/dist/src/tools/__tests__/tool.test.js +24 -1
- package/dist/src/tools/__tests__/tool.test.js.map +1 -1
- package/dist/src/tools/function-tool.d.ts.map +1 -1
- package/dist/src/tools/function-tool.js +6 -1
- package/dist/src/tools/function-tool.js.map +1 -1
- package/dist/src/tools/mcp-tool.d.ts +24 -3
- package/dist/src/tools/mcp-tool.d.ts.map +1 -1
- package/dist/src/tools/mcp-tool.js +103 -31
- package/dist/src/tools/mcp-tool.js.map +1 -1
- package/dist/src/tools/tool.d.ts +21 -2
- package/dist/src/tools/tool.d.ts.map +1 -1
- package/dist/src/tools/tool.js +12 -0
- package/dist/src/tools/tool.js.map +1 -1
- package/dist/src/tsconfig.tsbuildinfo +1 -1
- package/dist/src/types/__tests__/agent.test.js +48 -0
- package/dist/src/types/__tests__/agent.test.js.map +1 -1
- package/dist/src/types/agent.d.ts +77 -9
- package/dist/src/types/agent.d.ts.map +1 -1
- package/dist/src/types/agent.js +30 -6
- package/dist/src/types/agent.js.map +1 -1
- package/dist/src/types/elicitation.d.ts +15 -0
- package/dist/src/types/elicitation.d.ts.map +1 -0
- package/dist/src/types/elicitation.js +2 -0
- package/dist/src/types/elicitation.js.map +1 -0
- package/dist/src/types/interrupt.d.ts +103 -0
- package/dist/src/types/interrupt.d.ts.map +1 -0
- package/dist/src/types/interrupt.js +63 -0
- package/dist/src/types/interrupt.js.map +1 -0
- package/dist/src/types/messages.d.ts +2 -1
- package/dist/src/types/messages.d.ts.map +1 -1
- package/dist/src/types/messages.js.map +1 -1
- package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.d.ts +2 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.js +292 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/plugin.test.js.map +1 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.d.ts +2 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.js +148 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.js.map +1 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.d.ts +2 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.js +78 -0
- package/dist/src/vended-plugins/context-offloader/__tests__/storage.test.node.js.map +1 -0
- package/dist/src/vended-plugins/context-offloader/index.d.ts +23 -0
- package/dist/src/vended-plugins/context-offloader/index.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/index.js +21 -0
- package/dist/src/vended-plugins/context-offloader/index.js.map +1 -0
- package/dist/src/vended-plugins/context-offloader/plugin.d.ts +48 -0
- package/dist/src/vended-plugins/context-offloader/plugin.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/plugin.js +244 -0
- package/dist/src/vended-plugins/context-offloader/plugin.js.map +1 -0
- package/dist/src/vended-plugins/context-offloader/storage.d.ts +114 -0
- package/dist/src/vended-plugins/context-offloader/storage.d.ts.map +1 -0
- package/dist/src/vended-plugins/context-offloader/storage.js +204 -0
- package/dist/src/vended-plugins/context-offloader/storage.js.map +1 -0
- package/dist/src/vended-plugins/skills/__tests__/agent-skills.test.node.js +21 -5
- package/dist/src/vended-plugins/skills/__tests__/agent-skills.test.node.js.map +1 -1
- package/dist/src/vended-tools/bash/__tests__/bash.test.node.js +4 -0
- package/dist/src/vended-tools/bash/__tests__/bash.test.node.js.map +1 -1
- package/dist/src/vended-tools/bash/bash.d.ts.map +1 -1
- package/dist/src/vended-tools/bash/bash.js +0 -3
- package/dist/src/vended-tools/bash/bash.js.map +1 -1
- package/dist/src/vended-tools/file-editor/__tests__/file-editor.test.node.js +4 -0
- package/dist/src/vended-tools/file-editor/__tests__/file-editor.test.node.js.map +1 -1
- package/dist/src/vended-tools/notebook/__tests__/notebook.test.js +4 -0
- package/dist/src/vended-tools/notebook/__tests__/notebook.test.js.map +1 -1
- package/package.json +17 -9
- package/dist/src/models/__tests__/openai.test.d.ts +0 -2
- package/dist/src/models/__tests__/openai.test.d.ts.map +0 -1
- package/dist/src/models/__tests__/openai.test.js.map +0 -1
- package/dist/src/models/openai.d.ts +0 -312
- package/dist/src/models/openai.d.ts.map +0 -1
- package/dist/src/models/openai.js +0 -789
- 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 @@
|
|
|
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:
|
|
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
|
|
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
|
-
*
|
|
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
|
*
|
package/dist/src/mcp.d.ts.map
CHANGED
|
@@ -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;
|
|
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
|
-
|
|
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.
|
|
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
|
-
*
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
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.
|