@adhd/agent-engine-orchestrator 2.1.0

package/package.json

@@ -0,0 +1,64 @@
+{
+  "name": "@adhd/agent-engine-orchestrator",
+  "version": "2.1.0",
+  "description": "Agent engine orchestrator — providers, clients, tools, plugins, and validation extracted from agent-mcp",
+  "type": "module",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "src"
+  ],
+  "exports": {
+    ".": {
+      "types": "./src/index.d.ts",
+      "default": "./src/index.js"
+    }
+  },
+  "scripts": {
+    "build": "nx build agent-engine-orchestrator",
+    "typecheck": "tsc -p tsconfig.lib.json --noEmit"
+  },
+  "dependencies": {
+    "@adhd/agent-base-types": "^2.1.0",
+    "@adhd/agent-store-prompts": "^2.1.0",
+    "@adhd/agent-store-runtime": "^2.1.0",
+    "@anthropic-ai/sdk": "^0.26.0",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "better-sqlite3": "12.10.0",
+    "drizzle-orm": "0.45.2",
+    "openai": "^4.29.0",
+    "p-queue": "^8.0.0",
+    "p-retry": "^6.0.0",
+    "uuid": "14.0.0",
+    "zod": "4.4.3"
+  },
+  "peerDependencies": {
+    "@adhd/agent-engine-compiler": "^2.1.0"
+  },
+  "peerDependenciesMeta": {
+    "@adhd/agent-engine-compiler": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.13",
+    "@vitest/coverage-v8": "^1.0.4",
+    "tslib": "^2.6.0",
+    "typescript": "^5.5.0",
+    "vite": "~5.0.13",
+    "vitest": "^1.6.0"
+  },
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/PseudoSky/adhd.git",
+    "directory": "packages/agent/agent-engine-orchestrator"
+  },
+  "homepage": "https://github.com/PseudoSky/adhd/tree/main/packages/agent/agent-engine-orchestrator#readme",
+  "bugs": {
+    "url": "https://github.com/PseudoSky/adhd/issues"
+  },
+  "module": "./src/index.js",
+  "main": "./src/index.js"
+}

package/src/clients/http-client.d.ts

@@ -0,0 +1,24 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import type { ToolDefinition } from "../providers/types.js";
+import type { McpHttpConfig, McpSseConfig } from "../validation/index.js";
+import type { IMcpClient } from "./types.js";
+declare abstract class BaseHttpMcpClient implements IMcpClient {
+    protected readonly serverName: string;
+    protected client: Client | null;
+    constructor(serverName: string);
+    listTools(): Promise<ToolDefinition[]>;
+    callTool(toolName: string, args: unknown, signal?: AbortSignal): Promise<unknown>;
+    close(): Promise<void>;
+}
+export declare class HttpMcpClient extends BaseHttpMcpClient {
+    private readonly config;
+    constructor(serverName: string, config: McpHttpConfig);
+    connect(): Promise<void>;
+}
+export declare class SseMcpClient extends BaseHttpMcpClient {
+    private readonly config;
+    constructor(serverName: string, config: McpSseConfig);
+    connect(): Promise<void>;
+}
+export {};
+//# sourceMappingURL=http-client.d.ts.map

package/src/clients/http-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-client.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/http-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAInE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAE1E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,uBAAe,iBAAkB,YAAW,UAAU;IAGtC,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM;IAFjD,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAQ;gBAER,UAAU,EAAE,MAAM;IAE3C,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IAgBtC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IAwBjF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAQ/B;AAED,qBAAa,aAAc,SAAQ,iBAAiB;IAG5C,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADvB,UAAU,EAAE,MAAM,EACD,MAAM,EAAE,aAAa;IAKpC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAmBjC;AAED,qBAAa,YAAa,SAAQ,iBAAiB;IAG3C,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADvB,UAAU,EAAE,MAAM,EACD,MAAM,EAAE,YAAY;IAKnC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CA0BjC"}

package/src/clients/http-client.js

@@ -0,0 +1,96 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { ToolError } from "../validation/errors.js";
+class BaseHttpMcpClient {
+    serverName;
+    client = null;
+    constructor(serverName) {
+        this.serverName = serverName;
+    }
+    async listTools() {
+        if (!this.client) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Client '${this.serverName}' not connected`);
+        }
+        const response = await this.client.listTools();
+        return response.tools.map(tool => ({
+            name: tool.name,
+            description: tool.description ?? "",
+            inputSchema: tool.inputSchema ?? {
+                type: "object",
+                properties: {},
+            },
+        }));
+    }
+    async callTool(toolName, args, signal) {
+        if (!this.client) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Client '${this.serverName}' not connected`);
+        }
+        const result = await this.client.callTool({
+            name: toolName,
+            arguments: args,
+        }, undefined, signal ? { signal } : undefined);
+        if (result.isError) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Tool '${toolName}' on server '${this.serverName}' returned an error: ${JSON.stringify(result.content)}`);
+        }
+        return result.content;
+    }
+    async close() {
+        try {
+            await this.client?.close();
+        }
+        catch {
+            // ignore close errors
+        }
+        this.client = null;
+    }
+}
+export class HttpMcpClient extends BaseHttpMcpClient {
+    config;
+    constructor(serverName, config) {
+        super(serverName);
+        this.config = config;
+    }
+    async connect() {
+        const url = new URL(this.config.url);
+        const transport = new StreamableHTTPClientTransport(url, {
+            requestInit: {
+                headers: this.config.headers,
+                signal: this.config.timeoutMs
+                    ? AbortSignal.timeout(this.config.timeoutMs)
+                    : undefined,
+            },
+        });
+        const client = new Client({ name: "agent-mcp-client", version: "1.0.0" }, { capabilities: {} });
+        this.client = client;
+        await client.connect(transport);
+    }
+}
+export class SseMcpClient extends BaseHttpMcpClient {
+    config;
+    constructor(serverName, config) {
+        super(serverName);
+        this.config = config;
+    }
+    async connect() {
+        const url = new URL(this.config.url);
+        const transport = new SSEClientTransport(url, {
+            eventSourceInit: {
+                fetch: (input, init) => fetch(input, {
+                    ...init,
+                    headers: {
+                        ...init?.headers,
+                        ...this.config.headers,
+                    },
+                    signal: this.config.timeoutMs
+                        ? AbortSignal.timeout(this.config.timeoutMs)
+                        : undefined,
+                }),
+            },
+        });
+        const client = new Client({ name: "agent-mcp-client", version: "1.0.0" }, { capabilities: {} });
+        this.client = client;
+        await client.connect(transport);
+    }
+}
+//# sourceMappingURL=http-client.js.map

package/src/clients/http-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-client.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/http-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAInG,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAGpD,MAAe,iBAAiB;IAGG;IAFrB,MAAM,GAAkB,IAAI,CAAC;IAEvC,YAA+B,UAAkB;QAAlB,eAAU,GAAV,UAAU,CAAQ;IAAG,CAAC;IAErD,KAAK,CAAC,SAAS;QACX,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,WAAW,IAAI,CAAC,UAAU,iBAAiB,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAC/C,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,WAAW,EAAG,IAAI,CAAC,WAAuC,IAAI;gBAC1D,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACjB;SACJ,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAa,EAAE,MAAoB;QAChE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,WAAW,IAAI,CAAC,UAAU,iBAAiB,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CACrC;YACI,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,IAA+B;SAC7C,EACD,SAAS,EACT,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAClC,CAAC;QAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,SAAS,CACf,kBAAkB,EAClB,SAAS,QAAQ,gBAAgB,IAAI,CAAC,UAAU,wBAAwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAC3G,CAAC;QACN,CAAC;QAED,OAAO,MAAM,CAAC,OAAO,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACL,sBAAsB;QAC1B,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACvB,CAAC;CACJ;AAED,MAAM,OAAO,aAAc,SAAQ,iBAAiB;IAG3B;IAFrB,YACI,UAAkB,EACD,MAAqB;QAEtC,KAAK,CAAC,UAAU,CAAC,CAAC;QAFD,WAAM,GAAN,MAAM,CAAe;IAG1C,CAAC;IAED,KAAK,CAAC,OAAO;QACT,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC,GAAG,EAAE;YACrD,WAAW,EAAE;gBACT,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAA6C;gBAClE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;oBACzB,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;oBAC5C,CAAC,CAAC,SAAS;aAClB;SACJ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,EAAE,EAC9C,EAAE,YAAY,EAAE,EAAE,EAAE,CACvB,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;CACJ;AAED,MAAM,OAAO,YAAa,SAAQ,iBAAiB;IAG1B;IAFrB,YACI,UAAkB,EACD,MAAoB;QAErC,KAAK,CAAC,UAAU,CAAC,CAAC;QAFD,WAAM,GAAN,MAAM,CAAc;IAGzC,CAAC;IAED,KAAK,CAAC,OAAO;QACT,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,GAAG,EAAE;YAC1C,eAAe,EAAE;gBACb,KAAK,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CACnB,KAAK,CAAC,KAAK,EAAE;oBACT,GAAG,IAAI;oBACP,OAAO,EAAE;wBACL,GAAI,IAAI,EAAE,OAAkC;wBAC5C,GAAI,IAAI,CAAC,MAAM,CAAC,OAAkC;qBAC3B;oBAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;wBACzB,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;wBAC5C,CAAC,CAAC,SAAS;iBAClB,CAAC;aACT;SACJ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,EAAE,EAC9C,EAAE,YAAY,EAAE,EAAE,EAAE,CACvB,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;CACJ"}

package/src/clients/in-process.d.ts

@@ -0,0 +1,19 @@
+import type { ToolDefinition } from "../providers/types.js";
+import type { ExecutionContext } from "../validation/index.js";
+import type { IMcpClient } from "./types.js";
+export type InProcessToolHandler = (toolName: string, args: unknown, ctx: ExecutionContext) => Promise<unknown>;
+export type InProcessToolDescriptor = {
+    name: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+};
+export declare class InProcessMcpClient implements IMcpClient {
+    private readonly tools;
+    private readonly handler;
+    private readonly context;
+    constructor(tools: InProcessToolDescriptor[], handler: InProcessToolHandler, context: ExecutionContext);
+    listTools(): Promise<ToolDefinition[]>;
+    callTool(toolName: string, args: unknown, signal?: AbortSignal): Promise<unknown>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=in-process.d.ts.map

package/src/clients/in-process.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-process.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/in-process.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,MAAM,MAAM,oBAAoB,GAAG,CAC/B,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,GAAG,EAAE,gBAAgB,KACpB,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,MAAM,MAAM,uBAAuB,GAAG;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACxC,CAAC;AAEF,qBAAa,kBAAmB,YAAW,UAAU;IAE7C,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO;gBAFP,KAAK,EAAE,uBAAuB,EAAE,EAChC,OAAO,EAAE,oBAAoB,EAC7B,OAAO,EAAE,gBAAgB;IAGxC,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IAQtC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IAKjF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B"}

package/src/clients/in-process.js

@@ -0,0 +1,25 @@
+export class InProcessMcpClient {
+    tools;
+    handler;
+    context;
+    constructor(tools, handler, context) {
+        this.tools = tools;
+        this.handler = handler;
+        this.context = context;
+    }
+    async listTools() {
+        return this.tools.map(t => ({
+            name: t.name,
+            description: t.description,
+            inputSchema: t.inputSchema,
+        }));
+    }
+    async callTool(toolName, args, signal) {
+        signal?.throwIfAborted();
+        return this.handler(toolName, args, this.context);
+    }
+    async close() {
+        // no-op
+    }
+}
+//# sourceMappingURL=in-process.js.map

package/src/clients/in-process.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-process.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/in-process.ts"],"names":[],"mappings":"AAgBA,MAAM,OAAO,kBAAkB;IAEN;IACA;IACA;IAHrB,YACqB,KAAgC,EAChC,OAA6B,EAC7B,OAAyB;QAFzB,UAAK,GAAL,KAAK,CAA2B;QAChC,YAAO,GAAP,OAAO,CAAsB;QAC7B,YAAO,GAAP,OAAO,CAAkB;IAC3C,CAAC;IAEJ,KAAK,CAAC,SAAS;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,WAAW,EAAE,CAAC,CAAC,WAAW;SAC7B,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAa,EAAE,MAAoB;QAChE,MAAM,EAAE,cAAc,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK;QACP,QAAQ;IACZ,CAAC;CACJ"}

package/src/clients/index.d.ts

@@ -0,0 +1,9 @@
+export type { IMcpClient } from "./types.js";
+export { InProcessMcpClient } from "./in-process.js";
+export type { InProcessToolHandler, InProcessToolDescriptor } from "./in-process.js";
+export { StdioMcpClient } from "./stdio-client.js";
+export { HttpMcpClient, SseMcpClient } from "./http-client.js";
+export { McpClientRegistry } from "./registry.js";
+export { TOOL_NAME_SEPARATOR, normalizeToolName, resolveToolCallName } from "./tool-naming.js";
+export type { ResolvedToolName } from "./tool-naming.js";
+//# sourceMappingURL=index.d.ts.map

package/src/clients/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,YAAY,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AACrF,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAC/F,YAAY,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC"}

package/src/clients/index.js

@@ -0,0 +1,6 @@
+export { InProcessMcpClient } from "./in-process.js";
+export { StdioMcpClient } from "./stdio-client.js";
+export { HttpMcpClient, SseMcpClient } from "./http-client.js";
+export { McpClientRegistry } from "./registry.js";
+export { TOOL_NAME_SEPARATOR, normalizeToolName, resolveToolCallName } from "./tool-naming.js";
+//# sourceMappingURL=index.js.map

package/src/clients/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC"}

package/src/clients/registry.d.ts

@@ -0,0 +1,30 @@
+import type { ToolDefinition } from "../providers/types.js";
+import type { ExecutionContext, McpServerConfig } from "../validation/index.js";
+import type { InProcessToolDescriptor, InProcessToolHandler } from "./in-process.js";
+import type { IMcpClient } from "./types.js";
+import type { EngineLogger } from "../interfaces.js";
+export declare class McpClientRegistry {
+    private readonly mcpServers;
+    private readonly selfUrl;
+    private readonly inProcessDescriptors;
+    private readonly inProcessHandler;
+    private readonly context;
+    private readonly clients;
+    private readonly connectPromises;
+    private readonly toolTargets;
+    private readonly logger;
+    constructor(mcpServers: Record<string, McpServerConfig>, selfUrl: string | undefined, inProcessDescriptors: InProcessToolDescriptor[], inProcessHandler: InProcessToolHandler, context: ExecutionContext, logger?: EngineLogger);
+    private isSelfReferential;
+    private resolveServerName;
+    private getOrCreateClient;
+    getClient(name: string): Promise<IMcpClient>;
+    private isToolHidden;
+    assertToolAllowed(serverName: string, toolName: string): void;
+    listAllTools(): Promise<ToolDefinition[]>;
+    resolveToolName(advertised: string): {
+        server: string;
+        tool: string;
+    };
+    closeAll(): Promise<void>;
+}
+//# sourceMappingURL=registry.d.ts.map

package/src/clients/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAChF,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAIrF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD,qBAAa,iBAAiB;IAOtB,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IACrC,OAAO,CAAC,QAAQ,CAAC,gBAAgB;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAV5B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoC;IACpE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAuD;IACnF,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;gBAGjB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,EAC3C,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,oBAAoB,EAAE,uBAAuB,EAAE,EAC/C,gBAAgB,EAAE,oBAAoB,EACtC,OAAO,EAAE,gBAAgB,EAC1C,MAAM,CAAC,EAAE,YAAY;IAKzB,OAAO,CAAC,iBAAiB;IAczB,OAAO,CAAC,iBAAiB;YAMX,iBAAiB;IAuEzB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAIlD,OAAO,CAAC,YAAY;IAQpB,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAOvD,YAAY,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IA6B/C,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE;IAc/D,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAgBlC"}

package/src/clients/registry.js

@@ -0,0 +1,173 @@
+import { InProcessMcpClient } from "./in-process.js";
+import { HttpMcpClient, SseMcpClient } from "./http-client.js";
+import { StdioMcpClient } from "./stdio-client.js";
+import { TOOL_NAME_SEPARATOR, normalizeToolName } from "./tool-naming.js";
+export class McpClientRegistry {
+    mcpServers;
+    selfUrl;
+    inProcessDescriptors;
+    inProcessHandler;
+    context;
+    clients = new Map();
+    connectPromises = new Map();
+    toolTargets = new Map();
+    logger;
+    constructor(mcpServers, selfUrl, inProcessDescriptors, inProcessHandler, context, logger) {
+        this.mcpServers = mcpServers;
+        this.selfUrl = selfUrl;
+        this.inProcessDescriptors = inProcessDescriptors;
+        this.inProcessHandler = inProcessHandler;
+        this.context = context;
+        this.logger = logger ?? { info: () => { }, warn: () => { }, error: () => { }, debug: () => { } };
+    }
+    isSelfReferential(name, config) {
+        if (name === "agent-mcp")
+            return true;
+        if (this.selfUrl &&
+            (config.transport === "http" || config.transport === "sse") &&
+            config.url === this.selfUrl) {
+            return true;
+        }
+        return false;
+    }
+    resolveServerName(name) {
+        if (this.mcpServers[name])
+            return name;
+        const norm = normalizeToolName(name);
+        return Object.keys(this.mcpServers).find((k) => normalizeToolName(k) === norm) ?? name;
+    }
+    async getOrCreateClient(rawName) {
+        const name = this.resolveServerName(rawName);
+        const existing = this.clients.get(name);
+        if (existing)
+            return existing;
+        const config = this.mcpServers[name];
+        if (!config) {
+            throw new Error(`No MCP server config found for server: '${rawName}'`);
+        }
+        if (this.isSelfReferential(name, config)) {
+            const client = new InProcessMcpClient(this.inProcessDescriptors, this.inProcessHandler, this.context);
+            this.clients.set(name, client);
+            return client;
+        }
+        let connectPromise = this.connectPromises.get(name);
+        if (connectPromise) {
+            await connectPromise;
+            const client = this.clients.get(name);
+            if (!client)
+                throw new Error(`MCP client not found after connect: ${name}`);
+            return client;
+        }
+        let client;
+        if (config.transport === "stdio") {
+            client = new StdioMcpClient(name, config);
+        }
+        else if (config.transport === "http") {
+            client = new HttpMcpClient(name, config);
+        }
+        else if (config.transport === "sse") {
+            client = new SseMcpClient(name, config);
+        }
+        else {
+            const exhaustive = config;
+            throw new Error(`Unknown MCP transport: ${exhaustive.transport}`);
+        }
+        // A REJECTED connect promise must not stay cached. `connectPromises` was only
+        // ever cleared wholesale in close(), so a single failed connect (server not up
+        // yet, transient spawn failure) poisoned the entry forever: every later
+        // getOrCreateClient(name) re-awaited the same rejected promise and rethrew the
+        // original error, and that server could never reconnect for the process's life.
+        // Evict on failure so the next call makes a fresh attempt.
+        const attempt = client.connect().then(() => {
+            this.clients.set(name, client);
+            this.logger.debug({ server: name }, "MCP client connected");
+        });
+        connectPromise = attempt.catch((err) => {
+            // Only evict OUR entry — a newer attempt may already have replaced it.
+            if (this.connectPromises.get(name) === connectPromise) {
+                this.connectPromises.delete(name);
+            }
+            throw err;
+        });
+        this.connectPromises.set(name, connectPromise);
+        await connectPromise;
+        // NB: distinct binding — `client` is already declared in this block scope
+        // (the `let client: Stdio|Http|Sse` above). Re-fetch from the map rather
+        // than returning `client` directly so a concurrent connect that replaced
+        // the entry wins, matching the pre-existing `this.clients.get(name)!`.
+        const connected = this.clients.get(name);
+        if (!connected)
+            throw new Error(`MCP client not found after connect: ${name}`);
+        return connected;
+    }
+    async getClient(name) {
+        return this.getOrCreateClient(name);
+    }
+    isToolHidden(serverName, toolName) {
+        const config = this.mcpServers[serverName];
+        if (!config)
+            return false;
+        if (config.allowedTools && !config.allowedTools.includes(toolName))
+            return true;
+        if (config.disallowedTools?.includes(toolName))
+            return true;
+        return false;
+    }
+    assertToolAllowed(serverName, toolName) {
+        if (this.isToolHidden(serverName, toolName)) {
+            const quoted = `${serverName}__${toolName}`;
+            throw new Error(`Tool "${quoted}" is disallowed by server config and cannot be called`);
+        }
+    }
+    async listAllTools() {
+        const allTools = [];
+        for (const serverName of Object.keys(this.mcpServers)) {
+            try {
+                const client = await this.getOrCreateClient(serverName);
+                const tools = await client.listTools();
+                for (const tool of tools) {
+                    if (this.isToolHidden(serverName, tool.name)) {
+                        continue;
+                    }
+                    const advertised = `${serverName}${TOOL_NAME_SEPARATOR}${tool.name}`;
+                    const target = { server: serverName, tool: tool.name };
+                    this.toolTargets.set(advertised, target);
+                    this.toolTargets.set(normalizeToolName(advertised), target);
+                    allTools.push({ ...tool, name: advertised });
+                }
+            }
+            catch (error) {
+                this.logger.warn({ server: serverName, error }, "Failed to list tools from MCP server");
+            }
+        }
+        return allTools;
+    }
+    resolveToolName(advertised) {
+        const hit = this.toolTargets.get(advertised) ??
+            this.toolTargets.get(normalizeToolName(advertised));
+        if (hit)
+            return hit;
+        const i = advertised.indexOf(TOOL_NAME_SEPARATOR);
+        return i === -1
+            ? { server: advertised, tool: advertised }
+            : {
+                server: advertised.slice(0, i),
+                tool: advertised.slice(i + TOOL_NAME_SEPARATOR.length),
+            };
+    }
+    async closeAll() {
+        const closePromises = Array.from(this.clients.entries()).map(async ([name, client]) => {
+            try {
+                await client.close();
+                this.logger.debug({ server: name }, "MCP client closed");
+            }
+            catch (error) {
+                this.logger.warn({ server: name, error }, "Error closing MCP client");
+            }
+        });
+        await Promise.allSettled(closePromises);
+        this.clients.clear();
+        this.connectPromises.clear();
+    }
+}
+//# sourceMappingURL=registry.js.map

package/src/clients/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/registry.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAG1E,MAAM,OAAO,iBAAiB;IAOL;IACA;IACA;IACA;IACA;IAVJ,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAC;IACxC,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAC;IACnD,WAAW,GAAG,IAAI,GAAG,EAA4C,CAAC;IAClE,MAAM,CAAe;IAEtC,YACqB,UAA2C,EAC3C,OAA2B,EAC3B,oBAA+C,EAC/C,gBAAsC,EACtC,OAAyB,EAC1C,MAAqB;QALJ,eAAU,GAAV,UAAU,CAAiC;QAC3C,YAAO,GAAP,OAAO,CAAoB;QAC3B,yBAAoB,GAApB,oBAAoB,CAA2B;QAC/C,qBAAgB,GAAhB,gBAAgB,CAAsB;QACtC,YAAO,GAAP,OAAO,CAAkB;QAG1C,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC,EAAkB,CAAC;IACjH,CAAC;IAEO,iBAAiB,CAAC,IAAY,EAAE,MAAuB;QAC3D,IAAI,IAAI,KAAK,WAAW;YAAE,OAAO,IAAI,CAAC;QAEtC,IACI,IAAI,CAAC,OAAO;YACZ,CAAC,MAAM,CAAC,SAAS,KAAK,MAAM,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,CAAC;YAC3D,MAAM,CAAC,GAAG,KAAK,IAAI,CAAC,OAAO,EAC7B,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEO,iBAAiB,CAAC,IAAY;QAClC,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QACvC,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,IAAI,IAAI,CAAC;IAC3F,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,OAAe;QAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAE9B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,GAAG,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,IAAI,kBAAkB,CACjC,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,OAAO,CACf,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC/B,OAAO,MAAM,CAAC;QAClB,CAAC;QAED,IAAI,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,cAAc,CAAC;YACrB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,IAAI,EAAE,CAAC,CAAC;YAC5E,OAAO,MAAM,CAAC;QAClB,CAAC;QAED,IAAI,MAAqD,CAAC;QAE1D,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;YAC/B,MAAM,GAAG,IAAI,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,MAAM,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;YACrC,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;YACpC,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACJ,MAAM,UAAU,GAAU,MAAM,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,0BAA2B,UAAoC,CAAC,SAAS,EAAE,CAAC,CAAC;QACjG,CAAC;QAED,8EAA8E;QAC9E,+EAA+E;QAC/E,wEAAwE;QACxE,+EAA+E;QAC/E,gFAAgF;QAChF,2DAA2D;QAC3D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;YACvC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YAC5C,uEAAuE;YACvE,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,cAAc,EAAE,CAAC;gBACpD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,GAAG,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QAC/C,MAAM,cAAc,CAAC;QACrB,0EAA0E;QAC1E,yEAAyE;QACzE,yEAAyE;QACzE,uEAAuE;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,IAAI,EAAE,CAAC,CAAC;QAC/E,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAEO,YAAY,CAAC,UAAkB,EAAE,QAAgB;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC1B,IAAI,MAAM,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QAChF,IAAI,MAAM,CAAC,eAAe,EAAE,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QAC5D,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,iBAAiB,CAAC,UAAkB,EAAE,QAAgB;QAClD,IAAI,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,GAAG,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,SAAS,MAAM,uDAAuD,CAAC,CAAC;QAC5F,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY;QACd,MAAM,QAAQ,GAAqB,EAAE,CAAC;QAEtC,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACpD,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBACxD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBAEvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC3C,SAAS;oBACb,CAAC;oBACD,MAAM,UAAU,GAAG,GAAG,UAAU,GAAG,mBAAmB,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;oBACrE,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;oBACvD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;oBACzC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC;oBAC5D,QAAQ,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;gBACjD,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CACZ,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,EAC7B,sCAAsC,CACzC,CAAC;YACN,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,eAAe,CAAC,UAAkB;QAC9B,MAAM,GAAG,GACL,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC;YAChC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,CAAC;QACxD,IAAI,GAAG;YAAE,OAAO,GAAG,CAAC;QACpB,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAClD,OAAO,CAAC,KAAK,CAAC,CAAC;YACX,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE;YAC1C,CAAC,CAAC;gBACI,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC9B,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,GAAG,mBAAmB,CAAC,MAAM,CAAC;aACzD,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CACxD,KAAK,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE;YACrB,IAAI,CAAC;gBACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,mBAAmB,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,0BAA0B,CAAC,CAAC;YAC1E,CAAC;QACL,CAAC,CACJ,CAAC;QAEF,MAAM,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;CACJ"}

package/src/clients/stdio-client.d.ts

@@ -0,0 +1,19 @@
+import type { ToolDefinition } from "../providers/types.js";
+import type { McpStdioConfig } from "../validation/index.js";
+import type { IMcpClient } from "./types.js";
+import type { EngineLogger, EngineConfig } from "../interfaces.js";
+export declare class StdioMcpClient implements IMcpClient {
+    private readonly serverName;
+    private readonly serverConfig;
+    private client;
+    private transport;
+    private childProcess;
+    private readonly logger;
+    private readonly config;
+    constructor(serverName: string, serverConfig: McpStdioConfig, config?: EngineConfig, logger?: EngineLogger);
+    connect(): Promise<void>;
+    listTools(): Promise<ToolDefinition[]>;
+    callTool(toolName: string, args: unknown, callerSignal?: AbortSignal): Promise<unknown>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=stdio-client.d.ts.map

package/src/clients/stdio-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio-client.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/stdio-client.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAE7D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEnE,qBAAa,cAAe,YAAW,UAAU;IAQzC,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,YAAY;IARjC,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;gBAGjB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,cAAc,EAC7C,MAAM,CAAC,EAAE,YAAY,EACrB,MAAM,CAAC,EAAE,YAAY;IAMnB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAoCxB,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IAgBtC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IA6BvF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAmB/B"}

package/src/clients/stdio-client.js

@@ -0,0 +1,91 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { ToolError } from "../validation/errors.js";
+export class StdioMcpClient {
+    serverName;
+    serverConfig;
+    client = null;
+    transport = null;
+    childProcess = null;
+    logger;
+    config;
+    constructor(serverName, serverConfig, config, logger) {
+        this.serverName = serverName;
+        this.serverConfig = serverConfig;
+        this.config = config ?? { subprocessEnv: () => ({}) };
+        this.logger = logger ?? { info: () => { }, warn: () => { }, error: () => { }, debug: () => { } };
+    }
+    async connect() {
+        const transport = new StdioClientTransport({
+            command: this.serverConfig.command,
+            args: this.serverConfig.args ?? [],
+            env: {
+                ...this.config.subprocessEnv(),
+                ...this.serverConfig.env,
+            },
+        });
+        this.transport = transport;
+        const client = new Client({ name: "agent-mcp-client", version: "1.0.0" }, { capabilities: {} });
+        this.client = client;
+        await client.connect(transport);
+        const childProc = transport.process;
+        if (childProc) {
+            this.childProcess = childProc;
+            childProc.on("exit", (code, signal) => {
+                this.logger.warn({
+                    server: this.serverName,
+                    exitCode: code,
+                    signal,
+                }, "MCP child process exited unexpectedly");
+            });
+        }
+    }
+    async listTools() {
+        if (!this.client) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Client '${this.serverName}' not connected`);
+        }
+        const response = await this.client.listTools();
+        return response.tools.map(tool => ({
+            name: tool.name,
+            description: tool.description ?? "",
+            inputSchema: tool.inputSchema ?? {
+                type: "object",
+                properties: {},
+            },
+        }));
+    }
+    async callTool(toolName, args, callerSignal) {
+        if (!this.client) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Client '${this.serverName}' not connected`);
+        }
+        const timeoutSignal = this.serverConfig.timeoutMs
+            ? AbortSignal.timeout(this.serverConfig.timeoutMs)
+            : undefined;
+        const signal = callerSignal && timeoutSignal
+            ? AbortSignal.any([callerSignal, timeoutSignal])
+            : (callerSignal ?? timeoutSignal);
+        const result = await this.client.callTool({ name: toolName, arguments: args }, undefined, signal ? { signal } : undefined);
+        if (result.isError) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Tool '${toolName}' on server '${this.serverName}' returned an error: ${JSON.stringify(result.content)}`);
+        }
+        return result.content;
+    }
+    async close() {
+        try {
+            await this.client?.close();
+        }
+        catch {
+            // ignore close errors
+        }
+        if (this.childProcess && !this.childProcess.killed) {
+            this.childProcess.kill("SIGTERM");
+            const killTimer = setTimeout(() => {
+                if (this.childProcess && !this.childProcess.killed) {
+                    this.childProcess.kill("SIGKILL");
+                }
+            }, 5_000);
+            killTimer.unref();
+        }
+    }
+}
+//# sourceMappingURL=stdio-client.js.map

package/src/clients/stdio-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio-client.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/stdio-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAKjF,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAIpD,MAAM,OAAO,cAAc;IAQF;IACA;IARb,MAAM,GAAkB,IAAI,CAAC;IAC7B,SAAS,GAAgC,IAAI,CAAC;IAC9C,YAAY,GAAwB,IAAI,CAAC;IAChC,MAAM,CAAe;IACrB,MAAM,CAAe;IAEtC,YACqB,UAAkB,EAClB,YAA4B,EAC7C,MAAqB,EACrB,MAAqB;QAHJ,eAAU,GAAV,UAAU,CAAQ;QAClB,iBAAY,GAAZ,YAAY,CAAgB;QAI7C,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAkB,CAAC;QACtE,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC,EAAkB,CAAC;IACjH,CAAC;IAED,KAAK,CAAC,OAAO;QACT,MAAM,SAAS,GAAG,IAAI,oBAAoB,CAAC;YACvC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;YAClC,GAAG,EAAE;gBACD,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE;gBAC9B,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG;aACD;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,EAAE,EAC9C,EAAE,YAAY,EAAE,EAAE,EAAE,CACvB,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhC,MAAM,SAAS,GAAI,SAAmD,CAAC,OAAO,CAAC;QAC/E,IAAI,SAAS,EAAE,CAAC;YACZ,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;YAC9B,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CACZ;oBACI,MAAM,EAAE,IAAI,CAAC,UAAU;oBACvB,QAAQ,EAAE,IAAI;oBACd,MAAM;iBACT,EACD,uCAAuC,CAC1C,CAAC;YACN,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS;QACX,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,WAAW,IAAI,CAAC,UAAU,iBAAiB,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAC/C,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,WAAW,EAAG,IAAI,CAAC,WAAuC,IAAI;gBAC1D,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACjB;SACJ,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAa,EAAE,YAA0B;QACtE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,WAAW,IAAI,CAAC,UAAU,iBAAiB,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS;YAC7C,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC;YAClD,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,MAAM,GACR,YAAY,IAAI,aAAa;YACzB,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC,YAAY,IAAI,aAAa,CAAC,CAAC;QAE1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CACrC,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAA+B,EAAE,EAC9D,SAAS,EACT,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAClC,CAAC;QAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,SAAS,CACf,kBAAkB,EAClB,SAAS,QAAQ,gBAAgB,IAAI,CAAC,UAAU,wBAAwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAC3G,CAAC;QACN,CAAC;QAED,OAAO,MAAM,CAAC,OAAO,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACL,sBAAsB;QAC1B,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YACjD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;oBACjD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtC,CAAC;YACL,CAAC,EAAE,KAAK,CAAC,CAAC;YAEV,SAAS,CAAC,KAAK,EAAE,CAAC;QACtB,CAAC;IACL,CAAC;CACJ"}

package/src/clients/tool-naming.d.ts

@@ -0,0 +1,8 @@
+export declare const TOOL_NAME_SEPARATOR = "__";
+export declare function normalizeToolName(name: string): string;
+export interface ResolvedToolName {
+    server: string;
+    tool: string;
+}
+export declare function resolveToolCallName(rawName: string, advertised?: readonly string[]): ResolvedToolName;
+//# sourceMappingURL=tool-naming.d.ts.map

package/src/clients/tool-naming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-naming.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-engine-orchestrator/src/clients/tool-naming.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,mBAAmB,OAAO,CAAC;AAExC,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEtD;AAED,MAAM,WAAW,gBAAgB;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;CAChB;AASD,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,GAAE,SAAS,MAAM,EAAO,GAAG,gBAAgB,CAgBzG"}