agent-bundle 0.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 +21 -0
- package/README.md +161 -0
- package/dist/agent/agent.d.ts +32 -0
- package/dist/agent/agent.js +241 -0
- package/dist/agent/agent.js.map +1 -0
- package/dist/agent/agent.test-helpers.d.ts +63 -0
- package/dist/agent/agent.test-helpers.js +170 -0
- package/dist/agent/agent.test-helpers.js.map +1 -0
- package/dist/agent/define-agent.d.ts +2 -0
- package/dist/agent/define-agent.js +18 -0
- package/dist/agent/define-agent.js.map +1 -0
- package/dist/agent/dependencies.d.ts +15 -0
- package/dist/agent/dependencies.js +14 -0
- package/dist/agent/dependencies.js.map +1 -0
- package/dist/agent/index.d.ts +4 -0
- package/dist/agent/index.js +5 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/internals.d.ts +28 -0
- package/dist/agent/internals.js +160 -0
- package/dist/agent/internals.js.map +1 -0
- package/dist/agent/session.d.ts +4 -0
- package/dist/agent/session.js +2 -0
- package/dist/agent/session.js.map +1 -0
- package/dist/agent/types.d.ts +52 -0
- package/dist/agent/types.js +2 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/agent-loop/agent-loop.d.ts +27 -0
- package/dist/agent-loop/agent-loop.js +2 -0
- package/dist/agent-loop/agent-loop.js.map +1 -0
- package/dist/agent-loop/index.d.ts +4 -0
- package/dist/agent-loop/index.js +5 -0
- package/dist/agent-loop/index.js.map +1 -0
- package/dist/agent-loop/pi-mono/events.d.ts +3 -0
- package/dist/agent-loop/pi-mono/events.js +73 -0
- package/dist/agent-loop/pi-mono/events.js.map +1 -0
- package/dist/agent-loop/pi-mono/index.d.ts +1 -0
- package/dist/agent-loop/pi-mono/index.js +2 -0
- package/dist/agent-loop/pi-mono/index.js.map +1 -0
- package/dist/agent-loop/pi-mono/input.d.ts +6 -0
- package/dist/agent-loop/pi-mono/input.js +120 -0
- package/dist/agent-loop/pi-mono/input.js.map +1 -0
- package/dist/agent-loop/pi-mono/model.d.ts +7 -0
- package/dist/agent-loop/pi-mono/model.js +72 -0
- package/dist/agent-loop/pi-mono/model.js.map +1 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.d.ts +11 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.js +167 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.js.map +1 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.d.ts +28 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.js +76 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.js.map +1 -0
- package/dist/agent-loop/pi-mono/queue.d.ts +9 -0
- package/dist/agent-loop/pi-mono/queue.js +37 -0
- package/dist/agent-loop/pi-mono/queue.js.map +1 -0
- package/dist/agent-loop/pi-mono/tools.d.ts +4 -0
- package/dist/agent-loop/pi-mono/tools.js +164 -0
- package/dist/agent-loop/pi-mono/tools.js.map +1 -0
- package/dist/agent-loop/pi-mono/utils.d.ts +14 -0
- package/dist/agent-loop/pi-mono/utils.js +83 -0
- package/dist/agent-loop/pi-mono/utils.js.map +1 -0
- package/dist/agent-loop/system-prompt/fill.d.ts +1 -0
- package/dist/agent-loop/system-prompt/fill.js +19 -0
- package/dist/agent-loop/system-prompt/fill.js.map +1 -0
- package/dist/agent-loop/system-prompt/generate.d.ts +11 -0
- package/dist/agent-loop/system-prompt/generate.js +21 -0
- package/dist/agent-loop/system-prompt/generate.js.map +1 -0
- package/dist/agent-loop/system-prompt/index.d.ts +2 -0
- package/dist/agent-loop/system-prompt/index.js +3 -0
- package/dist/agent-loop/system-prompt/index.js.map +1 -0
- package/dist/agent-loop/types.d.ts +77 -0
- package/dist/agent-loop/types.js +2 -0
- package/dist/agent-loop/types.js.map +1 -0
- package/dist/cli/build/build.d.ts +35 -0
- package/dist/cli/build/build.js +174 -0
- package/dist/cli/build/build.js.map +1 -0
- package/dist/cli/build/build.test-helpers.d.ts +9 -0
- package/dist/cli/build/build.test-helpers.js +50 -0
- package/dist/cli/build/build.test-helpers.js.map +1 -0
- package/dist/cli/build/codegen-commands.d.ts +16 -0
- package/dist/cli/build/codegen-commands.js +81 -0
- package/dist/cli/build/codegen-commands.js.map +1 -0
- package/dist/cli/build/codegen.d.ts +55 -0
- package/dist/cli/build/codegen.js +212 -0
- package/dist/cli/build/codegen.js.map +1 -0
- package/dist/cli/build/e2b-template.d.ts +38 -0
- package/dist/cli/build/e2b-template.js +210 -0
- package/dist/cli/build/e2b-template.js.map +1 -0
- package/dist/cli/build/e2b-template.test-helpers.d.ts +19 -0
- package/dist/cli/build/e2b-template.test-helpers.js +80 -0
- package/dist/cli/build/e2b-template.test-helpers.js.map +1 -0
- package/dist/cli/build/sandbox-image.d.ts +35 -0
- package/dist/cli/build/sandbox-image.js +57 -0
- package/dist/cli/build/sandbox-image.js.map +1 -0
- package/dist/cli/build-codegen.d.ts +41 -0
- package/dist/cli/build-codegen.js +168 -0
- package/dist/cli/build-codegen.js.map +1 -0
- package/dist/cli/build-e2b-template.d.ts +38 -0
- package/dist/cli/build-e2b-template.js +223 -0
- package/dist/cli/build-e2b-template.js.map +1 -0
- package/dist/cli/build-e2b-template.test-helpers.d.ts +19 -0
- package/dist/cli/build-e2b-template.test-helpers.js +80 -0
- package/dist/cli/build-e2b-template.test-helpers.js.map +1 -0
- package/dist/cli/build-sandbox-image.d.ts +35 -0
- package/dist/cli/build-sandbox-image.js +57 -0
- package/dist/cli/build-sandbox-image.js.map +1 -0
- package/dist/cli/build.d.ts +30 -0
- package/dist/cli/build.js +151 -0
- package/dist/cli/build.js.map +1 -0
- package/dist/cli/build.test-helpers.d.ts +9 -0
- package/dist/cli/build.test-helpers.js +50 -0
- package/dist/cli/build.test-helpers.js.map +1 -0
- package/dist/cli/config/load-bundle-config.d.ts +2 -0
- package/dist/cli/config/load-bundle-config.js +20 -0
- package/dist/cli/config/load-bundle-config.js.map +1 -0
- package/dist/cli/config/resolve-project-root.d.ts +1 -0
- package/dist/cli/config/resolve-project-root.js +19 -0
- package/dist/cli/config/resolve-project-root.js.map +1 -0
- package/dist/cli/deploy/aws-cli.d.ts +42 -0
- package/dist/cli/deploy/aws-cli.js +95 -0
- package/dist/cli/deploy/aws-cli.js.map +1 -0
- package/dist/cli/deploy/aws-ecr.d.ts +20 -0
- package/dist/cli/deploy/aws-ecr.js +97 -0
- package/dist/cli/deploy/aws-ecr.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-infra.d.ts +39 -0
- package/dist/cli/deploy/aws-ecs-infra.js +169 -0
- package/dist/cli/deploy/aws-ecs-infra.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-role.d.ts +9 -0
- package/dist/cli/deploy/aws-ecs-role.js +114 -0
- package/dist/cli/deploy/aws-ecs-role.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-service.d.ts +37 -0
- package/dist/cli/deploy/aws-ecs-service.js +207 -0
- package/dist/cli/deploy/aws-ecs-service.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-shared.d.ts +59 -0
- package/dist/cli/deploy/aws-ecs-shared.js +30 -0
- package/dist/cli/deploy/aws-ecs-shared.js.map +1 -0
- package/dist/cli/deploy/aws-ecs.d.ts +3 -0
- package/dist/cli/deploy/aws-ecs.js +141 -0
- package/dist/cli/deploy/aws-ecs.js.map +1 -0
- package/dist/cli/deploy/aws-prerequisites.d.ts +16 -0
- package/dist/cli/deploy/aws-prerequisites.js +28 -0
- package/dist/cli/deploy/aws-prerequisites.js.map +1 -0
- package/dist/cli/deploy/aws-teardown.d.ts +13 -0
- package/dist/cli/deploy/aws-teardown.js +204 -0
- package/dist/cli/deploy/aws-teardown.js.map +1 -0
- package/dist/cli/deploy/deploy.d.ts +36 -0
- package/dist/cli/deploy/deploy.js +209 -0
- package/dist/cli/deploy/deploy.js.map +1 -0
- package/dist/cli/error.d.ts +1 -0
- package/dist/cli/error.js +4 -0
- package/dist/cli/error.js.map +1 -0
- package/dist/cli/generate/generate.d.ts +37 -0
- package/dist/cli/generate/generate.js +121 -0
- package/dist/cli/generate/generate.js.map +1 -0
- package/dist/cli/generate.d.ts +32 -0
- package/dist/cli/generate.js +102 -0
- package/dist/cli/generate.js.map +1 -0
- package/dist/cli/index.d.ts +2 -0
- package/dist/cli/index.js +182 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/load-bundle-config.d.ts +2 -0
- package/dist/cli/load-bundle-config.js +20 -0
- package/dist/cli/load-bundle-config.js.map +1 -0
- package/dist/cli/resolve-project-root.d.ts +1 -0
- package/dist/cli/resolve-project-root.js +19 -0
- package/dist/cli/resolve-project-root.js.map +1 -0
- package/dist/cli/serve/dev.d.ts +6 -0
- package/dist/cli/serve/dev.js +73 -0
- package/dist/cli/serve/dev.js.map +1 -0
- package/dist/cli/serve/http.d.ts +12 -0
- package/dist/cli/serve/http.js +123 -0
- package/dist/cli/serve/http.js.map +1 -0
- package/dist/cli/serve/init.d.ts +42 -0
- package/dist/cli/serve/init.js +110 -0
- package/dist/cli/serve/init.js.map +1 -0
- package/dist/cli/serve/runtime.d.ts +25 -0
- package/dist/cli/serve/runtime.js +142 -0
- package/dist/cli/serve/runtime.js.map +1 -0
- package/dist/cli/serve/serve.d.ts +7 -0
- package/dist/cli/serve/serve.js +63 -0
- package/dist/cli/serve/serve.js.map +1 -0
- package/dist/cli/serve/serve.test-helpers.d.ts +36 -0
- package/dist/cli/serve/serve.test-helpers.js +205 -0
- package/dist/cli/serve/serve.test-helpers.js.map +1 -0
- package/dist/cli/serve/worktree-port.d.ts +36 -0
- package/dist/cli/serve/worktree-port.js +124 -0
- package/dist/cli/serve/worktree-port.js.map +1 -0
- package/dist/cli/serve-http.d.ts +12 -0
- package/dist/cli/serve-http.js +112 -0
- package/dist/cli/serve-http.js.map +1 -0
- package/dist/cli/serve-runtime.d.ts +16 -0
- package/dist/cli/serve-runtime.js +122 -0
- package/dist/cli/serve-runtime.js.map +1 -0
- package/dist/cli/serve.d.ts +33 -0
- package/dist/cli/serve.js +150 -0
- package/dist/cli/serve.js.map +1 -0
- package/dist/cli/serve.test-helpers.d.ts +30 -0
- package/dist/cli/serve.test-helpers.js +145 -0
- package/dist/cli/serve.test-helpers.js.map +1 -0
- package/dist/code-formatter/bundle.json +41 -0
- package/dist/code-formatter/index.ts +26 -0
- package/dist/code-formatter/package.json +10 -0
- package/dist/code-formatter/types.ts +2 -0
- package/dist/code-formatter-e2b/bundle.json +40 -0
- package/dist/code-formatter-e2b/index.ts +25 -0
- package/dist/code-formatter-e2b/package.json +10 -0
- package/dist/code-formatter-e2b/types.ts +2 -0
- package/dist/coding-assistant-ollama/bundle.json +43 -0
- package/dist/coding-assistant-ollama/index.ts +28 -0
- package/dist/coding-assistant-ollama/package.json +10 -0
- package/dist/coding-assistant-ollama/types.ts +2 -0
- package/dist/commands/find.d.ts +8 -0
- package/dist/commands/find.js +11 -0
- package/dist/commands/find.js.map +1 -0
- package/dist/commands/loader.d.ts +13 -0
- package/dist/commands/loader.js +163 -0
- package/dist/commands/loader.js.map +1 -0
- package/dist/commands/types.d.ts +7 -0
- package/dist/commands/types.js +2 -0
- package/dist/commands/types.js.map +1 -0
- package/dist/commands/with-commands.d.ts +6 -0
- package/dist/commands/with-commands.js +19 -0
- package/dist/commands/with-commands.js.map +1 -0
- package/dist/data-analyst/bundle.json +40 -0
- package/dist/data-analyst/index.ts +25 -0
- package/dist/data-analyst/package.json +10 -0
- package/dist/data-analyst/types.ts +2 -0
- package/dist/financial-analyst/bundle.json +40 -0
- package/dist/financial-analyst/index.ts +25 -0
- package/dist/financial-analyst/package.json +10 -0
- package/dist/financial-analyst/types.ts +2 -0
- package/dist/mcp/client-manager.d.ts +26 -0
- package/dist/mcp/client-manager.js +101 -0
- package/dist/mcp/client-manager.js.map +1 -0
- package/dist/mcp/connect-server.d.ts +21 -0
- package/dist/mcp/connect-server.js +170 -0
- package/dist/mcp/connect-server.js.map +1 -0
- package/dist/mcp/index.d.ts +2 -0
- package/dist/mcp/index.js +3 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/sandbox-stdio-transport.d.ts +32 -0
- package/dist/mcp/sandbox-stdio-transport.js +199 -0
- package/dist/mcp/sandbox-stdio-transport.js.map +1 -0
- package/dist/observability/hooks.d.ts +10 -0
- package/dist/observability/hooks.js +97 -0
- package/dist/observability/hooks.js.map +1 -0
- package/dist/observability/index.d.ts +6 -0
- package/dist/observability/index.js +7 -0
- package/dist/observability/index.js.map +1 -0
- package/dist/observability/metrics.d.ts +31 -0
- package/dist/observability/metrics.js +61 -0
- package/dist/observability/metrics.js.map +1 -0
- package/dist/observability/middleware.d.ts +12 -0
- package/dist/observability/middleware.js +45 -0
- package/dist/observability/middleware.js.map +1 -0
- package/dist/observability/otel-harness.test-util.d.ts +37 -0
- package/dist/observability/otel-harness.test-util.js +96 -0
- package/dist/observability/otel-harness.test-util.js.map +1 -0
- package/dist/observability/provider.d.ts +8 -0
- package/dist/observability/provider.js +19 -0
- package/dist/observability/provider.js.map +1 -0
- package/dist/observability/tracing.d.ts +14 -0
- package/dist/observability/tracing.js +40 -0
- package/dist/observability/tracing.js.map +1 -0
- package/dist/observability/types.d.ts +39 -0
- package/dist/observability/types.js +31 -0
- package/dist/observability/types.js.map +1 -0
- package/dist/personalized-recommend/bundle.json +64 -0
- package/dist/personalized-recommend/index.ts +41 -0
- package/dist/personalized-recommend/package.json +10 -0
- package/dist/personalized-recommend/types.ts +2 -0
- package/dist/plugins/loader.d.ts +11 -0
- package/dist/plugins/loader.js +100 -0
- package/dist/plugins/loader.js.map +1 -0
- package/dist/plugins/merge.d.ts +10 -0
- package/dist/plugins/merge.js +22 -0
- package/dist/plugins/merge.js.map +1 -0
- package/dist/plugins/parse.d.ts +10 -0
- package/dist/plugins/parse.js +139 -0
- package/dist/plugins/parse.js.map +1 -0
- package/dist/plugins/types.d.ts +28 -0
- package/dist/plugins/types.js +2 -0
- package/dist/plugins/types.js.map +1 -0
- package/dist/plugins/urls.d.ts +7 -0
- package/dist/plugins/urls.js +40 -0
- package/dist/plugins/urls.js.map +1 -0
- package/dist/runtime.d.ts +5 -0
- package/dist/runtime.js +3 -0
- package/dist/runtime.js.map +1 -0
- package/dist/sandbox/factory.d.ts +2 -0
- package/dist/sandbox/factory.js +27 -0
- package/dist/sandbox/factory.js.map +1 -0
- package/dist/sandbox/index.d.ts +4 -0
- package/dist/sandbox/index.js +5 -0
- package/dist/sandbox/index.js.map +1 -0
- package/dist/sandbox/providers/e2b.d.ts +28 -0
- package/dist/sandbox/providers/e2b.js +294 -0
- package/dist/sandbox/providers/e2b.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-command-run.d.ts +8 -0
- package/dist/sandbox/providers/kubernetes-command-run.js +195 -0
- package/dist/sandbox/providers/kubernetes-command-run.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-helpers.d.ts +31 -0
- package/dist/sandbox/providers/kubernetes-helpers.js +152 -0
- package/dist/sandbox/providers/kubernetes-helpers.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.d.ts +7 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.js +51 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-spawn.d.ts +2 -0
- package/dist/sandbox/providers/kubernetes-spawn.js +42 -0
- package/dist/sandbox/providers/kubernetes-spawn.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.d.ts +32 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.js +249 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.js.map +1 -0
- package/dist/sandbox/providers/kubernetes.constants.d.ts +3 -0
- package/dist/sandbox/providers/kubernetes.constants.js +4 -0
- package/dist/sandbox/providers/kubernetes.constants.js.map +1 -0
- package/dist/sandbox/providers/kubernetes.d.ts +37 -0
- package/dist/sandbox/providers/kubernetes.js +258 -0
- package/dist/sandbox/providers/kubernetes.js.map +1 -0
- package/dist/sandbox/types.d.ts +57 -0
- package/dist/sandbox/types.js +2 -0
- package/dist/sandbox/types.js.map +1 -0
- package/dist/sandbox/utils.d.ts +1 -0
- package/dist/sandbox/utils.js +4 -0
- package/dist/sandbox/utils.js.map +1 -0
- package/dist/schema/bundle.d.ts +143 -0
- package/dist/schema/bundle.js +203 -0
- package/dist/schema/bundle.js.map +1 -0
- package/dist/service/command-routes.d.ts +8 -0
- package/dist/service/command-routes.js +44 -0
- package/dist/service/command-routes.js.map +1 -0
- package/dist/service/create-server.d.ts +9 -0
- package/dist/service/create-server.js +113 -0
- package/dist/service/create-server.js.map +1 -0
- package/dist/service/index.d.ts +1 -0
- package/dist/service/index.js +2 -0
- package/dist/service/index.js.map +1 -0
- package/dist/skills/index.d.ts +1 -0
- package/dist/skills/index.js +2 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/loader.d.ts +18 -0
- package/dist/skills/loader.js +142 -0
- package/dist/skills/loader.js.map +1 -0
- package/dist/skills/summaries.d.ts +8 -0
- package/dist/skills/summaries.js +9 -0
- package/dist/skills/summaries.js.map +1 -0
- package/dist/test-helpers/env.d.ts +2 -0
- package/dist/test-helpers/env.js +20 -0
- package/dist/test-helpers/env.js.map +1 -0
- package/dist/test-helpers/mock-agent.d.ts +56 -0
- package/dist/test-helpers/mock-agent.js +180 -0
- package/dist/test-helpers/mock-agent.js.map +1 -0
- package/dist/test-helpers/mock-sandbox.d.ts +33 -0
- package/dist/test-helpers/mock-sandbox.js +200 -0
- package/dist/test-helpers/mock-sandbox.js.map +1 -0
- package/dist/tui/index.d.ts +2 -0
- package/dist/tui/index.js +3 -0
- package/dist/tui/index.js.map +1 -0
- package/dist/tui/render.d.ts +8 -0
- package/dist/tui/render.js +40 -0
- package/dist/tui/render.js.map +1 -0
- package/dist/tui/tui.d.ts +19 -0
- package/dist/tui/tui.js +133 -0
- package/dist/tui/tui.js.map +1 -0
- package/dist/webui/create-webui-server.d.ts +22 -0
- package/dist/webui/create-webui-server.js +293 -0
- package/dist/webui/create-webui-server.js.map +1 -0
- package/dist/webui/event-bus.d.ts +16 -0
- package/dist/webui/event-bus.js +24 -0
- package/dist/webui/event-bus.js.map +1 -0
- package/dist/webui/index.d.ts +2 -0
- package/dist/webui/index.js +3 -0
- package/dist/webui/index.js.map +1 -0
- package/dist/webui/public/app.js +847 -0
- package/dist/webui/public/file-transfer.js +114 -0
- package/dist/webui/public/index.html +166 -0
- package/dist/webui/public/styles.css +1678 -0
- package/package.json +99 -0
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
import { Client } from "@modelcontextprotocol/sdk/client";
|
|
2
|
+
import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
|
|
3
|
+
import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
|
|
4
|
+
import { SandboxStdioTransport } from "./sandbox-stdio-transport.js";
|
|
5
|
+
const CLIENT_INFO = {
|
|
6
|
+
name: "agent-bundle",
|
|
7
|
+
version: "1.0.0",
|
|
8
|
+
};
|
|
9
|
+
function toErrorMessage(error) {
|
|
10
|
+
if (error instanceof Error) {
|
|
11
|
+
return error.message;
|
|
12
|
+
}
|
|
13
|
+
return String(error);
|
|
14
|
+
}
|
|
15
|
+
function isRecord(value) {
|
|
16
|
+
return typeof value === "object" && value !== null && !Array.isArray(value);
|
|
17
|
+
}
|
|
18
|
+
function toDescription(description) {
|
|
19
|
+
if (typeof description === "string" && description.trim().length > 0) {
|
|
20
|
+
return description;
|
|
21
|
+
}
|
|
22
|
+
return "MCP tool";
|
|
23
|
+
}
|
|
24
|
+
function normalizeInputSchema(schema) {
|
|
25
|
+
if (!isRecord(schema) || schema.type !== "object") {
|
|
26
|
+
return { type: "object" };
|
|
27
|
+
}
|
|
28
|
+
const normalizedSchema = {
|
|
29
|
+
...schema,
|
|
30
|
+
type: "object",
|
|
31
|
+
};
|
|
32
|
+
const properties = isRecord(schema.properties) ? schema.properties : undefined;
|
|
33
|
+
if (properties) {
|
|
34
|
+
normalizedSchema.properties = properties;
|
|
35
|
+
}
|
|
36
|
+
else {
|
|
37
|
+
delete normalizedSchema.properties;
|
|
38
|
+
}
|
|
39
|
+
const required = Array.isArray(schema.required)
|
|
40
|
+
&& schema.required.every((entry) => typeof entry === "string")
|
|
41
|
+
? schema.required
|
|
42
|
+
: undefined;
|
|
43
|
+
if (required) {
|
|
44
|
+
normalizedSchema.required = required;
|
|
45
|
+
}
|
|
46
|
+
else {
|
|
47
|
+
delete normalizedSchema.required;
|
|
48
|
+
}
|
|
49
|
+
return normalizedSchema;
|
|
50
|
+
}
|
|
51
|
+
function extractTextContent(result) {
|
|
52
|
+
if ("toolResult" in result) {
|
|
53
|
+
return null;
|
|
54
|
+
}
|
|
55
|
+
const textBlocks = result.content
|
|
56
|
+
.flatMap((item) => (item.type === "text" ? [item.text] : []))
|
|
57
|
+
.map((text) => text.trim())
|
|
58
|
+
.filter((text) => text.length > 0);
|
|
59
|
+
if (textBlocks.length === 0) {
|
|
60
|
+
return null;
|
|
61
|
+
}
|
|
62
|
+
return textBlocks.join("\n\n");
|
|
63
|
+
}
|
|
64
|
+
function toToolOutput(result) {
|
|
65
|
+
if ("toolResult" in result) {
|
|
66
|
+
return result.toolResult;
|
|
67
|
+
}
|
|
68
|
+
const textOutput = extractTextContent(result);
|
|
69
|
+
if (textOutput !== null) {
|
|
70
|
+
return textOutput;
|
|
71
|
+
}
|
|
72
|
+
if (result.structuredContent) {
|
|
73
|
+
return result.structuredContent;
|
|
74
|
+
}
|
|
75
|
+
return result.content;
|
|
76
|
+
}
|
|
77
|
+
function isToolCallError(result) {
|
|
78
|
+
return !("toolResult" in result) && result.isError === true;
|
|
79
|
+
}
|
|
80
|
+
async function listAllTools(client, cursor) {
|
|
81
|
+
const page = await client.listTools(cursor ? { cursor } : undefined);
|
|
82
|
+
const next = page.nextCursor ? await listAllTools(client, page.nextCursor) : [];
|
|
83
|
+
return [...page.tools, ...next];
|
|
84
|
+
}
|
|
85
|
+
function createAuthHeaders(token) {
|
|
86
|
+
if (!token) {
|
|
87
|
+
return undefined;
|
|
88
|
+
}
|
|
89
|
+
return {
|
|
90
|
+
Authorization: `Bearer ${token}`,
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
function toCloseErrorMessage(results) {
|
|
94
|
+
const failures = results
|
|
95
|
+
.flatMap((result) => {
|
|
96
|
+
if (result.status === "fulfilled") {
|
|
97
|
+
return [];
|
|
98
|
+
}
|
|
99
|
+
return [toErrorMessage(result.reason)];
|
|
100
|
+
});
|
|
101
|
+
if (failures.length === 0) {
|
|
102
|
+
return null;
|
|
103
|
+
}
|
|
104
|
+
return failures.join(" | ");
|
|
105
|
+
}
|
|
106
|
+
async function createConnectionFromTransport(serverName, transport) {
|
|
107
|
+
const client = new Client(CLIENT_INFO, { capabilities: {} });
|
|
108
|
+
await client.connect(transport);
|
|
109
|
+
const tools = await listAllTools(client);
|
|
110
|
+
const discoveredTools = tools.map((tool) => ({
|
|
111
|
+
name: tool.name,
|
|
112
|
+
description: toDescription(tool.description),
|
|
113
|
+
inputSchema: normalizeInputSchema(tool.inputSchema),
|
|
114
|
+
}));
|
|
115
|
+
return {
|
|
116
|
+
serverName,
|
|
117
|
+
tools: discoveredTools,
|
|
118
|
+
callTool: async (toolName, input) => {
|
|
119
|
+
const result = await client.callTool({
|
|
120
|
+
name: toolName,
|
|
121
|
+
arguments: input,
|
|
122
|
+
});
|
|
123
|
+
return {
|
|
124
|
+
output: toToolOutput(result),
|
|
125
|
+
isError: isToolCallError(result),
|
|
126
|
+
};
|
|
127
|
+
},
|
|
128
|
+
close: async () => {
|
|
129
|
+
const closeResults = await Promise.allSettled([
|
|
130
|
+
client.close(),
|
|
131
|
+
transport.close(),
|
|
132
|
+
]);
|
|
133
|
+
const closeErrorMessage = toCloseErrorMessage(closeResults);
|
|
134
|
+
if (closeErrorMessage) {
|
|
135
|
+
throw new Error(closeErrorMessage);
|
|
136
|
+
}
|
|
137
|
+
},
|
|
138
|
+
};
|
|
139
|
+
}
|
|
140
|
+
export function toServerLocation(server) {
|
|
141
|
+
if (server.transport === "stdio") {
|
|
142
|
+
return `${server.command} ${server.args?.join(" ") ?? ""}`.trim();
|
|
143
|
+
}
|
|
144
|
+
return server.url;
|
|
145
|
+
}
|
|
146
|
+
export async function defaultConnectServer(server, token, sandbox) {
|
|
147
|
+
if (server.transport === "http") {
|
|
148
|
+
const headers = createAuthHeaders(token);
|
|
149
|
+
const requestInit = headers ? { headers } : undefined;
|
|
150
|
+
const transport = new StreamableHTTPClientTransport(new URL(server.url), requestInit ? { requestInit } : undefined);
|
|
151
|
+
return await createConnectionFromTransport(server.name, transport);
|
|
152
|
+
}
|
|
153
|
+
if (server.transport === "stdio") {
|
|
154
|
+
if (!sandbox) {
|
|
155
|
+
throw new Error(`stdio MCP server "${server.name}" requires a sandbox`);
|
|
156
|
+
}
|
|
157
|
+
const transport = new SandboxStdioTransport({
|
|
158
|
+
sandbox,
|
|
159
|
+
command: server.command,
|
|
160
|
+
args: server.args,
|
|
161
|
+
env: server.env,
|
|
162
|
+
});
|
|
163
|
+
return await createConnectionFromTransport(server.name, transport);
|
|
164
|
+
}
|
|
165
|
+
const headers = server.auth === "bearer" ? createAuthHeaders(token) : undefined;
|
|
166
|
+
const requestInit = headers ? { headers } : undefined;
|
|
167
|
+
const transport = new SSEClientTransport(new URL(server.url), requestInit ? { requestInit } : undefined);
|
|
168
|
+
return await createConnectionFromTransport(server.name, transport);
|
|
169
|
+
}
|
|
170
|
+
//# sourceMappingURL=connect-server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"connect-server.js","sourceRoot":"","sources":["../../src/mcp/connect-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAMnG,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAErE,MAAM,WAAW,GAAG;IAClB,IAAI,EAAE,cAAc;IACpB,OAAO,EAAE,OAAO;CACjB,CAAC;AA6BF,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IAED,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,aAAa,CAAC,WAAoB;IACzC,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrE,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAe;IAC3C,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAClD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,gBAAgB,GAAiC;QACrD,GAAG,MAAM;QACT,IAAI,EAAE,QAAQ;KACf,CAAC;IAEF,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC/E,IAAI,UAAU,EAAE,CAAC;QACf,gBAAgB,CAAC,UAAU,GAAG,UAAU,CAAC;IAC3C,CAAC;SAAM,CAAC;QACN,OAAO,gBAAgB,CAAC,UAAU,CAAC;IACrC,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;WAC1C,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC;QAC9D,CAAC,CAAC,MAAM,CAAC,QAAQ;QACjB,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,QAAQ,EAAE,CAAC;QACb,gBAAgB,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,OAAO,gBAAgB,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA4B;IACtD,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO;SAC9B,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SAC5D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAErC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,YAAY,CAAC,MAA4B;IAChD,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,UAAU,CAAC;IAC3B,CAAC;IAED,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC9C,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC,iBAAiB,CAAC;IAClC,CAAC;IAED,OAAO,MAAM,CAAC,OAAO,CAAC;AACxB,CAAC;AAED,SAAS,eAAe,CAAC,MAA4B;IACnD,OAAO,CAAC,CAAC,YAAY,IAAI,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,KAAK,IAAI,CAAC;AAC9D,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,MAAc,EAAE,MAAe;IACzD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrE,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAChF,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAyB;IAClD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,aAAa,EAAE,UAAU,KAAK,EAAE;KACjC,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,OAA8C;IACzE,MAAM,QAAQ,GAAG,OAAO;SACrB,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAClB,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEL,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,6BAA6B,CAC1C,UAAkB,EAClB,SAAoB;IAEpB,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,WAAW,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;IAC7D,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAEhC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC;QAC5C,WAAW,EAAE,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC;KACpD,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,UAAU;QACV,KAAK,EAAE,eAAe;QACtB,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE;YAClC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;gBACnC,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;YAEH,OAAO;gBACL,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC;gBAC5B,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC;aACjC,CAAC;QACJ,CAAC;QACD,KAAK,EAAE,KAAK,IAAI,EAAE;YAChB,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;gBAC5C,MAAM,CAAC,KAAK,EAAE;gBACd,SAAS,CAAC,KAAK,EAAE;aAClB,CAAC,CAAC;YACH,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC;YAC5D,IAAI,iBAAiB,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAuB;IACtD,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;IACpE,CAAC;IAED,OAAO,MAAM,CAAC,GAAG,CAAC;AACpB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAuB,EACvB,KAAyB,EACzB,OAAyB;IAEzB,IAAI,MAAM,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,6BAA6B,CACjD,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EACnB,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAC1C,CAAC;QACF,OAAO,MAAM,6BAA6B,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,IAAI,sBAAsB,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,qBAAqB,CAAC;YAC1C,OAAO;YACP,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,GAAG,EAAE,MAAM,CAAC,GAAG;SAChB,CAAC,CAAC;QACH,OAAO,MAAM,6BAA6B,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACrE,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChF,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACtD,MAAM,SAAS,GAAG,IAAI,kBAAkB,CACtC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EACnB,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAC1C,CAAC;IACF,OAAO,MAAM,6BAA6B,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;AACrE,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/mcp/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,8BAA8B,CAAC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
|
|
2
|
+
import type { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";
|
|
3
|
+
import type { SandboxIO } from "../sandbox/types.js";
|
|
4
|
+
export type SandboxStdioTransportOptions = {
|
|
5
|
+
sandbox: SandboxIO;
|
|
6
|
+
command: string;
|
|
7
|
+
args?: string[];
|
|
8
|
+
cwd?: string;
|
|
9
|
+
env?: Record<string, string>;
|
|
10
|
+
};
|
|
11
|
+
export declare class SandboxStdioTransport implements Transport {
|
|
12
|
+
private readonly options;
|
|
13
|
+
onclose?: () => void;
|
|
14
|
+
onerror?: (error: Error) => void;
|
|
15
|
+
onmessage?: (message: JSONRPCMessage) => void;
|
|
16
|
+
private readonly encoder;
|
|
17
|
+
private readonly decoder;
|
|
18
|
+
private process;
|
|
19
|
+
private stdinWriter;
|
|
20
|
+
private stdoutReaderTask;
|
|
21
|
+
private stderrReaderTask;
|
|
22
|
+
private isClosed;
|
|
23
|
+
private isClosing;
|
|
24
|
+
constructor(options: SandboxStdioTransportOptions);
|
|
25
|
+
start(): Promise<void>;
|
|
26
|
+
send(message: JSONRPCMessage): Promise<void>;
|
|
27
|
+
close(): Promise<void>;
|
|
28
|
+
private finalizeClose;
|
|
29
|
+
private watchExit;
|
|
30
|
+
private consumeStdout;
|
|
31
|
+
private consumeStderr;
|
|
32
|
+
}
|
|
@@ -0,0 +1,199 @@
|
|
|
1
|
+
import { deserializeMessage, serializeMessage } from "@modelcontextprotocol/sdk/shared/stdio.js";
|
|
2
|
+
function toError(error) {
|
|
3
|
+
if (error instanceof Error) {
|
|
4
|
+
return error;
|
|
5
|
+
}
|
|
6
|
+
return new Error(String(error));
|
|
7
|
+
}
|
|
8
|
+
function readLines(buffer) {
|
|
9
|
+
const lines = [];
|
|
10
|
+
let remaining = buffer;
|
|
11
|
+
for (;;) {
|
|
12
|
+
const newline = remaining.indexOf("\n");
|
|
13
|
+
if (newline < 0) {
|
|
14
|
+
break;
|
|
15
|
+
}
|
|
16
|
+
const line = remaining.slice(0, newline).replace(/\r$/, "");
|
|
17
|
+
remaining = remaining.slice(newline + 1);
|
|
18
|
+
if (line.length > 0) {
|
|
19
|
+
lines.push(line);
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
return { rest: remaining, lines };
|
|
23
|
+
}
|
|
24
|
+
function closeWriter(writer) {
|
|
25
|
+
if (writer === null) {
|
|
26
|
+
return;
|
|
27
|
+
}
|
|
28
|
+
try {
|
|
29
|
+
writer.releaseLock();
|
|
30
|
+
}
|
|
31
|
+
catch {
|
|
32
|
+
// Writer lock might already be released.
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
export class SandboxStdioTransport {
|
|
36
|
+
options;
|
|
37
|
+
onclose;
|
|
38
|
+
onerror;
|
|
39
|
+
onmessage;
|
|
40
|
+
encoder = new TextEncoder();
|
|
41
|
+
decoder = new TextDecoder();
|
|
42
|
+
process = null;
|
|
43
|
+
stdinWriter = null;
|
|
44
|
+
stdoutReaderTask = null;
|
|
45
|
+
stderrReaderTask = null;
|
|
46
|
+
isClosed = false;
|
|
47
|
+
isClosing = false;
|
|
48
|
+
constructor(options) {
|
|
49
|
+
this.options = options;
|
|
50
|
+
}
|
|
51
|
+
async start() {
|
|
52
|
+
if (this.process !== null) {
|
|
53
|
+
throw new Error("SandboxStdioTransport already started.");
|
|
54
|
+
}
|
|
55
|
+
this.process = await this.options.sandbox.spawn(this.options.command, this.options.args ?? [], {
|
|
56
|
+
cwd: this.options.cwd,
|
|
57
|
+
env: this.options.env,
|
|
58
|
+
});
|
|
59
|
+
this.stdinWriter = this.process.stdin.getWriter();
|
|
60
|
+
this.stdoutReaderTask = this.consumeStdout(this.process.stdout);
|
|
61
|
+
this.stderrReaderTask = this.consumeStderr(this.process.stderr);
|
|
62
|
+
void this.watchExit(this.process);
|
|
63
|
+
}
|
|
64
|
+
async send(message) {
|
|
65
|
+
const writer = this.stdinWriter;
|
|
66
|
+
if (!writer) {
|
|
67
|
+
throw new Error("SandboxStdioTransport is not connected.");
|
|
68
|
+
}
|
|
69
|
+
const payload = this.encoder.encode(serializeMessage(message));
|
|
70
|
+
await writer.write(payload);
|
|
71
|
+
}
|
|
72
|
+
async close() {
|
|
73
|
+
if (this.isClosed || this.isClosing) {
|
|
74
|
+
return;
|
|
75
|
+
}
|
|
76
|
+
this.isClosing = true;
|
|
77
|
+
const process = this.process;
|
|
78
|
+
const writer = this.stdinWriter;
|
|
79
|
+
try {
|
|
80
|
+
if (writer) {
|
|
81
|
+
await writer.close();
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
catch {
|
|
85
|
+
// Process may already be closing.
|
|
86
|
+
}
|
|
87
|
+
finally {
|
|
88
|
+
closeWriter(writer);
|
|
89
|
+
this.stdinWriter = null;
|
|
90
|
+
}
|
|
91
|
+
try {
|
|
92
|
+
await process?.kill("SIGTERM");
|
|
93
|
+
}
|
|
94
|
+
catch {
|
|
95
|
+
// Best effort.
|
|
96
|
+
}
|
|
97
|
+
await Promise.allSettled([
|
|
98
|
+
process?.exited ?? Promise.resolve(0),
|
|
99
|
+
this.stdoutReaderTask ?? Promise.resolve(),
|
|
100
|
+
this.stderrReaderTask ?? Promise.resolve(),
|
|
101
|
+
]);
|
|
102
|
+
this.finalizeClose();
|
|
103
|
+
}
|
|
104
|
+
finalizeClose() {
|
|
105
|
+
if (this.isClosed) {
|
|
106
|
+
return;
|
|
107
|
+
}
|
|
108
|
+
closeWriter(this.stdinWriter);
|
|
109
|
+
this.stdinWriter = null;
|
|
110
|
+
this.process = null;
|
|
111
|
+
this.isClosed = true;
|
|
112
|
+
this.onclose?.();
|
|
113
|
+
}
|
|
114
|
+
async watchExit(process) {
|
|
115
|
+
try {
|
|
116
|
+
const exitCode = await process.exited;
|
|
117
|
+
if (exitCode !== 0 && !this.isClosing) {
|
|
118
|
+
this.onerror?.(new Error(`MCP stdio process exited with code ${exitCode}.`));
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
catch (error) {
|
|
122
|
+
if (!this.isClosing) {
|
|
123
|
+
this.onerror?.(toError(error));
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
finally {
|
|
127
|
+
this.finalizeClose();
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
async consumeStdout(stream) {
|
|
131
|
+
const reader = stream.getReader();
|
|
132
|
+
let buffer = "";
|
|
133
|
+
try {
|
|
134
|
+
for (;;) {
|
|
135
|
+
const { done, value } = await reader.read();
|
|
136
|
+
if (done) {
|
|
137
|
+
break;
|
|
138
|
+
}
|
|
139
|
+
buffer += this.decoder.decode(value, { stream: true });
|
|
140
|
+
const parsed = readLines(buffer);
|
|
141
|
+
buffer = parsed.rest;
|
|
142
|
+
parsed.lines.forEach((line) => {
|
|
143
|
+
try {
|
|
144
|
+
this.onmessage?.(deserializeMessage(line));
|
|
145
|
+
}
|
|
146
|
+
catch (error) {
|
|
147
|
+
this.onerror?.(toError(error));
|
|
148
|
+
}
|
|
149
|
+
});
|
|
150
|
+
}
|
|
151
|
+
buffer += this.decoder.decode();
|
|
152
|
+
if (buffer.trim().length > 0) {
|
|
153
|
+
try {
|
|
154
|
+
this.onmessage?.(deserializeMessage(buffer.trim()));
|
|
155
|
+
}
|
|
156
|
+
catch (error) {
|
|
157
|
+
this.onerror?.(toError(error));
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
catch (error) {
|
|
162
|
+
if (!this.isClosing) {
|
|
163
|
+
this.onerror?.(toError(error));
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
finally {
|
|
167
|
+
reader.releaseLock();
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
async consumeStderr(stream) {
|
|
171
|
+
const reader = stream.getReader();
|
|
172
|
+
const decoder = new TextDecoder();
|
|
173
|
+
try {
|
|
174
|
+
for (;;) {
|
|
175
|
+
const { done, value } = await reader.read();
|
|
176
|
+
if (done) {
|
|
177
|
+
break;
|
|
178
|
+
}
|
|
179
|
+
const text = decoder.decode(value, { stream: true }).trim();
|
|
180
|
+
if (text.length > 0) {
|
|
181
|
+
this.onerror?.(new Error(`MCP stdio stderr: ${text}`));
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
const trailing = decoder.decode().trim();
|
|
185
|
+
if (trailing.length > 0) {
|
|
186
|
+
this.onerror?.(new Error(`MCP stdio stderr: ${trailing}`));
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
catch (error) {
|
|
190
|
+
if (!this.isClosing) {
|
|
191
|
+
this.onerror?.(toError(error));
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
finally {
|
|
195
|
+
reader.releaseLock();
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
//# sourceMappingURL=sandbox-stdio-transport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sandbox-stdio-transport.js","sourceRoot":"","sources":["../../src/mcp/sandbox-stdio-transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,2CAA2C,CAAC;AAmBjG,SAAS,OAAO,CAAC,KAAc;IAC7B,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,SAAS,CAAC,MAAc;IAC/B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,SAAS,GAAG,MAAM,CAAC;IAEvB,SAAS,CAAC;QACR,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,MAAM;QACR,CAAC;QAED,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,WAAW,CAAC,MAAsD;IACzE,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;QACpB,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,yCAAyC;IAC3C,CAAC;AACH,CAAC;AAED,MAAM,OAAO,qBAAqB;IAcI;IAb7B,OAAO,CAAc;IACrB,OAAO,CAA0B;IACjC,SAAS,CAAqC;IAEpC,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IACrC,OAAO,GAA0B,IAAI,CAAC;IACtC,WAAW,GAAmD,IAAI,CAAC;IACnE,gBAAgB,GAAyB,IAAI,CAAC;IAC9C,gBAAgB,GAAyB,IAAI,CAAC;IAC9C,QAAQ,GAAG,KAAK,CAAC;IACjB,SAAS,GAAG,KAAK,CAAC;IAE1B,YAAoC,OAAqC;QAArC,YAAO,GAAP,OAAO,CAA8B;IAAG,CAAC;IAEtE,KAAK,CAAC,KAAK;QAChB,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAC7C,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,EACvB;YACE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;YACrB,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;SACtB,CACF,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;QAClD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChE,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,OAAuB;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC;QAChC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/D,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpC,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC;QAEhC,IAAI,CAAC;YACH,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,kCAAkC;QACpC,CAAC;gBAAS,CAAC;YACT,WAAW,CAAC,MAAM,CAAC,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;QAED,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,gBAAgB,IAAI,OAAO,CAAC,OAAO,EAAE;YAC1C,IAAI,CAAC,gBAAgB,IAAI,OAAO,CAAC,OAAO,EAAE;SAC3C,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,OAAuB;QAC7C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC;YACtC,IAAI,QAAQ,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACtC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,sCAAsC,QAAQ,GAAG,CAAC,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAkC;QAC5D,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC;YACH,SAAS,CAAC;gBACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM;gBACR,CAAC;gBAED,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;gBACjC,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;gBACrB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC5B,IAAI,CAAC;wBACH,IAAI,CAAC,SAAS,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC7C,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;oBACjC,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAChC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,IAAI,CAAC,SAAS,EAAE,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACtD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAkC;QAC5D,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAElC,IAAI,CAAC;YACH,SAAS,CAAC;gBACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM;gBACR,CAAC;gBAED,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC5D,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpB,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,qBAAqB,IAAI,EAAE,CAAC,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YACzC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { ToolCall, ToolResult, TokenUsage } from "../agent-loop/types.js";
|
|
2
|
+
import { type ObservabilityProvider } from "./types.js";
|
|
3
|
+
export type AgentObservabilityHooks = {
|
|
4
|
+
onRespondStart: () => number;
|
|
5
|
+
onRespondEnd: (startMs: number, error?: unknown) => void;
|
|
6
|
+
onTokenUsage: (usage: TokenUsage) => void;
|
|
7
|
+
};
|
|
8
|
+
export declare function createAgentHooks(provider: ObservabilityProvider, agentName?: string): AgentObservabilityHooks;
|
|
9
|
+
export declare function createToolCallInstrumenter(provider: ObservabilityProvider): (call: ToolCall, execute: (call: ToolCall) => Promise<ToolResult>) => Promise<ToolResult>;
|
|
10
|
+
export declare function createMcpCallInstrumenter(provider: ObservabilityProvider): (serverName: string, toolName: string, execute: () => Promise<ToolResult>) => Promise<ToolResult>;
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
import { createAgentMetrics, createGenAiMetrics, createMcpMetrics, createToolMetrics, } from "./metrics.js";
|
|
2
|
+
import { elapsed, now, withSpan } from "./tracing.js";
|
|
3
|
+
import { AgentAttributes, McpAttributes, } from "./types.js";
|
|
4
|
+
export function createAgentHooks(provider, agentName) {
|
|
5
|
+
const agentMetrics = createAgentMetrics(provider.meter);
|
|
6
|
+
const genAiMetrics = createGenAiMetrics(provider.meter);
|
|
7
|
+
const baseAttrs = agentName ? { [AgentAttributes.AGENT_NAME]: agentName } : {};
|
|
8
|
+
return {
|
|
9
|
+
onRespondStart() {
|
|
10
|
+
agentMetrics.respondActive.add(1, baseAttrs);
|
|
11
|
+
return now();
|
|
12
|
+
},
|
|
13
|
+
onRespondEnd(startMs, error) {
|
|
14
|
+
const errorFlag = error !== undefined && error !== null;
|
|
15
|
+
agentMetrics.respondActive.add(-1, baseAttrs);
|
|
16
|
+
agentMetrics.respondDuration.record(elapsed(startMs), {
|
|
17
|
+
...baseAttrs,
|
|
18
|
+
[AgentAttributes.AGENT_STATUS]: errorFlag ? "error" : "ok",
|
|
19
|
+
});
|
|
20
|
+
},
|
|
21
|
+
onTokenUsage(usage) {
|
|
22
|
+
genAiMetrics.inputTokens.add(usage.inputTokens, baseAttrs);
|
|
23
|
+
genAiMetrics.outputTokens.add(usage.outputTokens, baseAttrs);
|
|
24
|
+
},
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
/* ------------------------------------------------------------------ */
|
|
28
|
+
/* Tool call instrumentation */
|
|
29
|
+
/* ------------------------------------------------------------------ */
|
|
30
|
+
export function createToolCallInstrumenter(provider) {
|
|
31
|
+
const toolMetrics = createToolMetrics(provider.meter);
|
|
32
|
+
return async function instrumentToolCall(call, execute) {
|
|
33
|
+
const startMs = now();
|
|
34
|
+
let threw = false;
|
|
35
|
+
try {
|
|
36
|
+
const result = await withSpan(provider.tracer, `tool ${call.name}`, { [AgentAttributes.TOOL_NAME]: call.name }, async () => execute(call));
|
|
37
|
+
if (result.isError) {
|
|
38
|
+
toolMetrics.callErrors.add(1, {
|
|
39
|
+
[AgentAttributes.TOOL_NAME]: call.name,
|
|
40
|
+
});
|
|
41
|
+
}
|
|
42
|
+
return result;
|
|
43
|
+
}
|
|
44
|
+
catch (error) {
|
|
45
|
+
threw = true;
|
|
46
|
+
toolMetrics.callErrors.add(1, {
|
|
47
|
+
[AgentAttributes.TOOL_NAME]: call.name,
|
|
48
|
+
});
|
|
49
|
+
throw error;
|
|
50
|
+
}
|
|
51
|
+
finally {
|
|
52
|
+
toolMetrics.callDuration.record(elapsed(startMs), {
|
|
53
|
+
[AgentAttributes.TOOL_NAME]: call.name,
|
|
54
|
+
[AgentAttributes.TOOL_ERROR]: String(threw),
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
/* ------------------------------------------------------------------ */
|
|
60
|
+
/* MCP call instrumentation */
|
|
61
|
+
/* ------------------------------------------------------------------ */
|
|
62
|
+
export function createMcpCallInstrumenter(provider) {
|
|
63
|
+
const mcpMetrics = createMcpMetrics(provider.meter);
|
|
64
|
+
return async function instrumentMcpCall(serverName, toolName, execute) {
|
|
65
|
+
const startMs = now();
|
|
66
|
+
let threw = false;
|
|
67
|
+
try {
|
|
68
|
+
const result = await withSpan(provider.tracer, `mcp ${serverName}/${toolName}`, {
|
|
69
|
+
[McpAttributes.SERVER_NAME]: serverName,
|
|
70
|
+
[McpAttributes.TOOL_NAME]: toolName,
|
|
71
|
+
}, async () => execute());
|
|
72
|
+
if (result.isError) {
|
|
73
|
+
mcpMetrics.callErrors.add(1, {
|
|
74
|
+
[McpAttributes.SERVER_NAME]: serverName,
|
|
75
|
+
[McpAttributes.TOOL_NAME]: toolName,
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
return result;
|
|
79
|
+
}
|
|
80
|
+
catch (error) {
|
|
81
|
+
threw = true;
|
|
82
|
+
mcpMetrics.callErrors.add(1, {
|
|
83
|
+
[McpAttributes.SERVER_NAME]: serverName,
|
|
84
|
+
[McpAttributes.TOOL_NAME]: toolName,
|
|
85
|
+
});
|
|
86
|
+
throw error;
|
|
87
|
+
}
|
|
88
|
+
finally {
|
|
89
|
+
mcpMetrics.callDuration.record(elapsed(startMs), {
|
|
90
|
+
[McpAttributes.SERVER_NAME]: serverName,
|
|
91
|
+
[McpAttributes.TOOL_NAME]: toolName,
|
|
92
|
+
[McpAttributes.TOOL_ERROR]: String(threw),
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
//# sourceMappingURL=hooks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../../src/observability/hooks.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,GAKlB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACtD,OAAO,EACL,eAAe,EACf,aAAa,GAEd,MAAM,YAAY,CAAC;AAYpB,MAAM,UAAU,gBAAgB,CAC9B,QAA+B,EAC/B,SAAkB;IAElB,MAAM,YAAY,GAAiB,kBAAkB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtE,MAAM,YAAY,GAAiB,kBAAkB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtE,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/E,OAAO;QACL,cAAc;YACZ,YAAY,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC7C,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;QAED,YAAY,CAAC,OAAe,EAAE,KAAe;YAC3C,MAAM,SAAS,GAAG,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,CAAC;YACxD,YAAY,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC9C,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBACpD,GAAG,SAAS;gBACZ,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;aAC3D,CAAC,CAAC;QACL,CAAC;QAED,YAAY,CAAC,KAAiB;YAC5B,YAAY,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAC3D,YAAY,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC/D,CAAC;KACF,CAAC;AACJ,CAAC;AAED,wEAAwE;AACxE,yEAAyE;AACzE,wEAAwE;AAExE,MAAM,UAAU,0BAA0B,CAAC,QAA+B;IACxE,MAAM,WAAW,GAAgB,iBAAiB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEnE,OAAO,KAAK,UAAU,kBAAkB,CACtC,IAAc,EACd,OAAgD;QAEhD,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;QACtB,IAAI,KAAK,GAAG,KAAK,CAAC;QAElB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC3B,QAAQ,CAAC,MAAM,EACf,QAAQ,IAAI,CAAC,IAAI,EAAE,EACnB,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,EAC1C,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAC1B,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE;oBAC5B,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;iBACvC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,GAAG,IAAI,CAAC;YACb,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE;gBAC5B,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;aACvC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBAChD,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACtC,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,wEAAwE;AACxE,yEAAyE;AACzE,wEAAwE;AAExE,MAAM,UAAU,yBAAyB,CAAC,QAA+B;IACvE,MAAM,UAAU,GAAe,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,OAAO,KAAK,UAAU,iBAAiB,CACrC,UAAkB,EAClB,QAAgB,EAChB,OAAkC;QAElC,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;QACtB,IAAI,KAAK,GAAG,KAAK,CAAC;QAElB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC3B,QAAQ,CAAC,MAAM,EACf,OAAO,UAAU,IAAI,QAAQ,EAAE,EAC/B;gBACE,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,UAAU;gBACvC,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,QAAQ;aACpC,EACD,KAAK,IAAI,EAAE,CAAC,OAAO,EAAE,CACtB,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE;oBAC3B,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,UAAU;oBACvC,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,QAAQ;iBACpC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,GAAG,IAAI,CAAC;YACb,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE;gBAC3B,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,UAAU;gBACvC,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,QAAQ;aACpC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC/C,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,UAAU;gBACvC,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,QAAQ;gBACnC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC;aAC1C,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/observability/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,iBAAiB,CAAC;AAChC,cAAc,YAAY,CAAC"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { Counter, Histogram, Meter, UpDownCounter } from "@opentelemetry/api";
|
|
2
|
+
/** Lazily-created instruments for the HTTP server layer. */
|
|
3
|
+
export type HttpMetrics = {
|
|
4
|
+
requestDuration: Histogram;
|
|
5
|
+
activeRequests: UpDownCounter;
|
|
6
|
+
};
|
|
7
|
+
/** Lazily-created instruments for agent lifecycle. */
|
|
8
|
+
export type AgentMetrics = {
|
|
9
|
+
respondDuration: Histogram;
|
|
10
|
+
respondActive: UpDownCounter;
|
|
11
|
+
};
|
|
12
|
+
/** Lazily-created instruments for tool calls. */
|
|
13
|
+
export type ToolMetrics = {
|
|
14
|
+
callDuration: Histogram;
|
|
15
|
+
callErrors: Counter;
|
|
16
|
+
};
|
|
17
|
+
/** Lazily-created instruments for GenAI token usage. */
|
|
18
|
+
export type GenAiMetrics = {
|
|
19
|
+
inputTokens: Counter;
|
|
20
|
+
outputTokens: Counter;
|
|
21
|
+
};
|
|
22
|
+
/** Lazily-created instruments for MCP tool calls. */
|
|
23
|
+
export type McpMetrics = {
|
|
24
|
+
callDuration: Histogram;
|
|
25
|
+
callErrors: Counter;
|
|
26
|
+
};
|
|
27
|
+
export declare function createHttpMetrics(meter: Meter): HttpMetrics;
|
|
28
|
+
export declare function createAgentMetrics(meter: Meter): AgentMetrics;
|
|
29
|
+
export declare function createToolMetrics(meter: Meter): ToolMetrics;
|
|
30
|
+
export declare function createGenAiMetrics(meter: Meter): GenAiMetrics;
|
|
31
|
+
export declare function createMcpMetrics(meter: Meter): McpMetrics;
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export function createHttpMetrics(meter) {
|
|
2
|
+
return {
|
|
3
|
+
requestDuration: meter.createHistogram("http.server.request.duration", {
|
|
4
|
+
description: "HTTP request latency",
|
|
5
|
+
unit: "ms",
|
|
6
|
+
}),
|
|
7
|
+
activeRequests: meter.createUpDownCounter("http.server.active_requests", {
|
|
8
|
+
description: "Number of in-flight HTTP requests",
|
|
9
|
+
unit: "{request}",
|
|
10
|
+
}),
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
export function createAgentMetrics(meter) {
|
|
14
|
+
return {
|
|
15
|
+
respondDuration: meter.createHistogram("agent.respond.duration", {
|
|
16
|
+
description: "End-to-end agent respond time",
|
|
17
|
+
unit: "ms",
|
|
18
|
+
}),
|
|
19
|
+
respondActive: meter.createUpDownCounter("agent.respond.active", {
|
|
20
|
+
description: "Number of active respond calls",
|
|
21
|
+
unit: "{request}",
|
|
22
|
+
}),
|
|
23
|
+
};
|
|
24
|
+
}
|
|
25
|
+
export function createToolMetrics(meter) {
|
|
26
|
+
return {
|
|
27
|
+
callDuration: meter.createHistogram("agent.tool_call.duration", {
|
|
28
|
+
description: "Tool call execution time",
|
|
29
|
+
unit: "ms",
|
|
30
|
+
}),
|
|
31
|
+
callErrors: meter.createCounter("agent.tool_call.errors", {
|
|
32
|
+
description: "Number of failed tool calls",
|
|
33
|
+
unit: "{error}",
|
|
34
|
+
}),
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
export function createGenAiMetrics(meter) {
|
|
38
|
+
return {
|
|
39
|
+
inputTokens: meter.createCounter("gen_ai.usage.input_tokens", {
|
|
40
|
+
description: "Input tokens consumed",
|
|
41
|
+
unit: "{token}",
|
|
42
|
+
}),
|
|
43
|
+
outputTokens: meter.createCounter("gen_ai.usage.output_tokens", {
|
|
44
|
+
description: "Output tokens consumed",
|
|
45
|
+
unit: "{token}",
|
|
46
|
+
}),
|
|
47
|
+
};
|
|
48
|
+
}
|
|
49
|
+
export function createMcpMetrics(meter) {
|
|
50
|
+
return {
|
|
51
|
+
callDuration: meter.createHistogram("mcp.tool_call.duration", {
|
|
52
|
+
description: "MCP tool call latency",
|
|
53
|
+
unit: "ms",
|
|
54
|
+
}),
|
|
55
|
+
callErrors: meter.createCounter("mcp.tool_call.errors", {
|
|
56
|
+
description: "Number of failed MCP tool calls",
|
|
57
|
+
unit: "{error}",
|
|
58
|
+
}),
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=metrics.js.map
|