@adhd/agent-mcp 0.0.1

package/src/clients/in-process.js

@@ -0,0 +1,32 @@
+/**
+ * InProcessMcpClient routes tool calls directly to runtime handler
+ * functions without any network round-trip.
+ *
+ * This is the client used when an agent definition includes an
+ * "agent-mcp" server entry — it allows recursive delegation while
+ * preserving the ExecutionContext thread.
+ */
+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) {
+        return this.handler(toolName, args, this.context);
+    }
+    async close() {
+        // no-op — in-process, nothing to tear down
+    }
+}
+//# sourceMappingURL=in-process.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"in-process.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/clients/in-process.ts"],"names":[],"mappings":"AAgBA;;;;;;;GAOG;AACH,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;QAC1C,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK;QACP,2CAA2C;IAC/C,CAAC;CACJ"}

package/src/clients/index.d.ts

@@ -0,0 +1,7 @@
+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";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/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"}

package/src/clients/index.js

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

package/src/clients/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/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"}

package/src/clients/registry.d.ts

@@ -0,0 +1,37 @@
+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";
+/**
+ * McpClientRegistry — per-task lifetime.
+ *
+ * Created fresh for each task() call and torn down in the Orchestrator's
+ * finally block via closeAll(). Never reused across tasks.
+ *
+ * Self-referential detection (routes to InProcessMcpClient):
+ *   1. Key === "agent-mcp" (always applies)
+ *   2. URL matches selfUrl (only when selfUrl is defined — http/sse transport)
+ *      When TRANSPORT=stdio, selfUrl is undefined and only condition 1 applies.
+ */
+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;
+    constructor(mcpServers: Record<string, McpServerConfig>, selfUrl: string | undefined, inProcessDescriptors: InProcessToolDescriptor[], inProcessHandler: InProcessToolHandler, context: ExecutionContext);
+    private isSelfReferential;
+    private getOrCreateClient;
+    /** Get a client for a specific server, connecting lazily if needed */
+    getClient(name: string): Promise<IMcpClient>;
+    /**
+     * List all tools across all configured servers.
+     * Prefixes each tool name with "<server>__<tool>" for disambiguation.
+     */
+    listAllTools(): Promise<ToolDefinition[]>;
+    /** Tear down all clients — called from Orchestrator's finally block */
+    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/ai/agent-mcp/src/clients/registry.ts"],"names":[],"mappings":"AACA,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;;;;;;;;;;GAUG;AACH,qBAAa,iBAAiB;IAKtB,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;IAR5B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoC;gBAG/C,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;IAG9C,OAAO,CAAC,iBAAiB;YAcX,iBAAiB;IAiD/B,sEAAsE;IAChE,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAIlD;;;OAGG;IACG,YAAY,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IAyB/C,uEAAuE;IACjE,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAgBlC"}

package/src/clients/registry.js

@@ -0,0 +1,125 @@
+import { logger } from "../logger.js";
+import { InProcessMcpClient } from "./in-process.js";
+import { HttpMcpClient, SseMcpClient } from "./http-client.js";
+import { StdioMcpClient } from "./stdio-client.js";
+/**
+ * McpClientRegistry — per-task lifetime.
+ *
+ * Created fresh for each task() call and torn down in the Orchestrator's
+ * finally block via closeAll(). Never reused across tasks.
+ *
+ * Self-referential detection (routes to InProcessMcpClient):
+ *   1. Key === "agent-mcp" (always applies)
+ *   2. URL matches selfUrl (only when selfUrl is defined — http/sse transport)
+ *      When TRANSPORT=stdio, selfUrl is undefined and only condition 1 applies.
+ */
+export class McpClientRegistry {
+    mcpServers;
+    selfUrl;
+    inProcessDescriptors;
+    inProcessHandler;
+    context;
+    clients = new Map();
+    connectPromises = new Map();
+    constructor(mcpServers, selfUrl, inProcessDescriptors, inProcessHandler, context) {
+        this.mcpServers = mcpServers;
+        this.selfUrl = selfUrl;
+        this.inProcessDescriptors = inProcessDescriptors;
+        this.inProcessHandler = inProcessHandler;
+        this.context = context;
+    }
+    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;
+    }
+    async getOrCreateClient(name) {
+        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: '${name}'`);
+        }
+        if (this.isSelfReferential(name, config)) {
+            const client = new InProcessMcpClient(this.inProcessDescriptors, this.inProcessHandler, this.context);
+            this.clients.set(name, client);
+            return client;
+        }
+        // Create and connect the appropriate transport client
+        let connectPromise = this.connectPromises.get(name);
+        if (connectPromise) {
+            await connectPromise;
+            return this.clients.get(name);
+        }
+        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}`);
+        }
+        connectPromise = client.connect().then(() => {
+            this.clients.set(name, client);
+            logger.debug({ server: name }, "MCP client connected");
+        });
+        this.connectPromises.set(name, connectPromise);
+        await connectPromise;
+        return this.clients.get(name);
+    }
+    /** Get a client for a specific server, connecting lazily if needed */
+    async getClient(name) {
+        return this.getOrCreateClient(name);
+    }
+    /**
+     * List all tools across all configured servers.
+     * Prefixes each tool name with "<server>__<tool>" for disambiguation.
+     */
+    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) {
+                    allTools.push({
+                        ...tool,
+                        name: `${serverName}__${tool.name}`,
+                    });
+                }
+            }
+            catch (error) {
+                logger.warn({ server: serverName, error }, "Failed to list tools from MCP server");
+            }
+        }
+        return allTools;
+    }
+    /** Tear down all clients — called from Orchestrator's finally block */
+    async closeAll() {
+        const closePromises = Array.from(this.clients.entries()).map(async ([name, client]) => {
+            try {
+                await client.close();
+                logger.debug({ server: name }, "MCP client closed");
+            }
+            catch (error) {
+                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/ai/agent-mcp/src/clients/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAItC,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;AAGnD;;;;;;;;;;GAUG;AACH,MAAM,OAAO,iBAAiB;IAKL;IACA;IACA;IACA;IACA;IARJ,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAC;IACxC,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAC;IAEpE,YACqB,UAA2C,EAC3C,OAA2B,EAC3B,oBAA+C,EAC/C,gBAAsC,EACtC,OAAyB;QAJzB,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;IAC3C,CAAC;IAEI,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,KAAK,CAAC,iBAAiB,CAAC,IAAY;QACxC,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,IAAI,GAAG,CAAC,CAAC;QACxE,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,sDAAsD;QACtD,IAAI,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,cAAc,CAAC;YACrB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAE,CAAC;QACnC,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,cAAc,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QAC/C,MAAM,cAAc,CAAC;QACrB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAE,CAAC;IACnC,CAAC;IAED,sEAAsE;IACtE,KAAK,CAAC,SAAS,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED;;;OAGG;IACH,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,QAAQ,CAAC,IAAI,CAAC;wBACV,GAAG,IAAI;wBACP,IAAI,EAAE,GAAG,UAAU,KAAK,IAAI,CAAC,IAAI,EAAE;qBACtC,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,EAC7B,sCAAsC,CACzC,CAAC;YACN,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,uEAAuE;IACvE,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,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,mBAAmB,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,0BAA0B,CAAC,CAAC;YACrE,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,16 @@
+import type { ToolDefinition } from "../providers/types.js";
+import type { McpStdioConfig } from "../validation/index.js";
+import type { IMcpClient } from "./types.js";
+export declare class StdioMcpClient implements IMcpClient {
+    private readonly serverName;
+    private readonly config;
+    private client;
+    private transport;
+    private childProcess;
+    constructor(serverName: string, config: McpStdioConfig);
+    connect(): Promise<void>;
+    listTools(): Promise<ToolDefinition[]>;
+    callTool(toolName: string, args: unknown): 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/ai/agent-mcp/src/clients/stdio-client.ts"],"names":[],"mappings":"AAKA,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;AAE7C,qBAAa,cAAe,YAAW,UAAU;IAMzC,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAN3B,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,YAAY,CAA6B;gBAG5B,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,cAAc;IAGrC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAwCxB,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IAgBtC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAyB3D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAoB/B"}

package/src/clients/stdio-client.js

@@ -0,0 +1,90 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { logger } from "../logger.js";
+import { ToolError } from "../validation/errors.js";
+export class StdioMcpClient {
+    serverName;
+    config;
+    client = null;
+    transport = null;
+    childProcess = null;
+    constructor(serverName, config) {
+        this.serverName = serverName;
+        this.config = config;
+    }
+    async connect() {
+        const transport = new StdioClientTransport({
+            command: this.config.command,
+            args: this.config.args ?? [],
+            env: {
+                ...process.env,
+                ...this.config.env,
+            },
+        });
+        this.transport = transport;
+        // Capture child process reference for lifecycle management.
+        // The StdioClientTransport exposes it after start().
+        const client = new Client({ name: "agent-mcp-client", version: "1.0.0" }, { capabilities: {} });
+        this.client = client;
+        await client.connect(transport);
+        // Attach exit handler to the underlying child process.
+        // StdioClientTransport exposes `process` after connect().
+        const childProc = transport.process;
+        if (childProc) {
+            this.childProcess = childProc;
+            childProc.on("exit", (code, signal) => {
+                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) {
+        if (!this.client) {
+            throw new ToolError("MCP_CLIENT_ERROR", `Client '${this.serverName}' not connected`);
+        }
+        const signal = this.config.timeoutMs
+            ? AbortSignal.timeout(this.config.timeoutMs)
+            : undefined;
+        const result = await this.client.callTool({ name: toolName, arguments: args }, undefined, { signal });
+        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");
+            // Force-kill after 5 seconds if still alive
+            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/ai/agent-mcp/src/clients/stdio-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAGjF,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAGtC,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAGpD,MAAM,OAAO,cAAc;IAMF;IACA;IANb,MAAM,GAAkB,IAAI,CAAC;IAC7B,SAAS,GAAgC,IAAI,CAAC;IAC9C,YAAY,GAAwB,IAAI,CAAC;IAEjD,YACqB,UAAkB,EAClB,MAAsB;QADtB,eAAU,GAAV,UAAU,CAAQ;QAClB,WAAM,GAAN,MAAM,CAAgB;IACxC,CAAC;IAEJ,KAAK,CAAC,OAAO;QACT,MAAM,SAAS,GAAG,IAAI,oBAAoB,CAAC;YACvC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE;YAC5B,GAAG,EAAE;gBACD,GAAG,OAAO,CAAC,GAAG;gBACd,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG;aACK;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,4DAA4D;QAC5D,qDAAqD;QACrD,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,uDAAuD;QACvD,0DAA0D;QAC1D,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,MAAM,CAAC,IAAI,CACP;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;QAC1C,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,IAAI,CAAC,MAAM,CAAC,SAAS;YAChC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YAC5C,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CACrC,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAA+B,EAAE,EAC9D,SAAS,EACT,EAAE,MAAM,EAAE,CACb,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,4CAA4C;YAC5C,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/types.d.ts

@@ -0,0 +1,14 @@
+import type { ToolDefinition } from "../providers/types.js";
+export interface IMcpClient {
+    /** Returns all tools advertised by this MCP server */
+    listTools(): Promise<ToolDefinition[]>;
+    /**
+     * Call a single tool on this server.
+     * @param toolName  the tool name (without server prefix)
+     * @param args      the arguments to pass
+     */
+    callTool(toolName: string, args: unknown): Promise<unknown>;
+    /** Tear down the client (kill child processes, close connections) */
+    close(): Promise<void>;
+}
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/clients/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,MAAM,WAAW,UAAU;IACvB,sDAAsD;IACtD,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAEvC;;;;OAIG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAE5D,qEAAqE;IACrE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1B"}

package/src/clients/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/src/clients/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/clients/types.ts"],"names":[],"mappings":""}

package/src/db/client.d.ts

@@ -0,0 +1,6 @@
+import Database from "better-sqlite3";
+import * as schema from "./schema.js";
+export declare const db: import("drizzle-orm/better-sqlite3").BetterSQLite3Database<typeof schema> & {
+    $client: Database.Database;
+};
+//# sourceMappingURL=client.d.ts.map

package/src/db/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/db/client.ts"],"names":[],"mappings":"AAGA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAGtC,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAwBtC,eAAO,MAAM,EAAE;;CAA8B,CAAC"}

package/src/db/client.js

@@ -0,0 +1,19 @@
+import fs from "node:fs";
+import path from "node:path";
+import Database from "better-sqlite3";
+import { drizzle } from "drizzle-orm/better-sqlite3";
+import * as schema from "./schema.js";
+const databasePath = process.env["DATABASE_PATH"] ||
+    "./data/agents.db";
+const resolvedPath = path.resolve(databasePath);
+const directory = path.dirname(resolvedPath);
+if (!fs.existsSync(directory)) {
+    fs.mkdirSync(directory, {
+        recursive: true
+    });
+}
+const sqlite = new Database(resolvedPath);
+sqlite.pragma("journal_mode = WAL");
+sqlite.pragma("foreign_keys = ON");
+export const db = drizzle(sqlite, { schema });
+//# sourceMappingURL=client.js.map

package/src/db/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/db/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAEtC,MAAM,YAAY,GACd,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC5B,kBAAkB,CAAC;AAEvB,MAAM,YAAY,GACd,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AAE/B,MAAM,SAAS,GACX,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AAE/B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;IAC5B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE;QACpB,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;AACP,CAAC;AAED,MAAM,MAAM,GACR,IAAI,QAAQ,CAAC,YAAY,CAAC,CAAC;AAE/B,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;AACpC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;AAEnC,MAAM,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC"}

package/src/db/migrate.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Runs pending Drizzle migrations synchronously before the server starts.
+ * Migrations are read from the ./drizzle folder (generated by drizzle-kit).
+ */
+export declare function runMigrations(): void;
+//# sourceMappingURL=migrate.d.ts.map

package/src/db/migrate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/db/migrate.ts"],"names":[],"mappings":"AAGA;;;GAGG;AACH,wBAAgB,aAAa,IAAI,IAAI,CAOpC"}

package/src/db/migrate.js

@@ -0,0 +1,12 @@
+import { migrate } from "drizzle-orm/better-sqlite3/migrator";
+import { db } from "./client.js";
+/**
+ * Runs pending Drizzle migrations synchronously before the server starts.
+ * Migrations are read from the ./drizzle folder (generated by drizzle-kit).
+ */
+export function runMigrations() {
+    migrate(db, {
+        migrationsFolder: new URL("../../drizzle", import.meta.url).pathname
+    });
+}
+//# sourceMappingURL=migrate.js.map

package/src/db/migrate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/db/migrate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qCAAqC,CAAC;AAC9D,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAEjC;;;GAGG;AACH,MAAM,UAAU,aAAa;IACzB,OAAO,CAAC,EAAE,EAAE;QACR,gBAAgB,EAAE,IAAI,GAAG,CACrB,eAAe,EACf,MAAM,CAAC,IAAI,CAAC,GAAG,CAClB,CAAC,QAAQ;KACb,CAAC,CAAC;AACP,CAAC"}