@mozaik-ai/core 3.13.1 → 3.14.0

package/dist/index.mjs

@@ -204,6 +204,98 @@ var TokenUsage = class {
   }
 };
 
+// src/infrastructure/mcp/mcp-client.ts
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+var McpClient = class {
+  constructor(config) {
+    this.config = config;
+    this.connected = false;
+    var _a, _b;
+    const headers = { ...(_a = config.headers) != null ? _a : {} };
+    if (config.authToken) headers.Authorization = `Bearer ${config.authToken}`;
+    this.transport = new StreamableHTTPClientTransport(
+      new URL(config.url),
+      Object.keys(headers).length > 0 ? { requestInit: { headers } } : void 0
+    );
+    this.client = new Client({ name: (_b = config.name) != null ? _b : "mozaik", version: "1.0.0" }, { capabilities: {} });
+  }
+  async connect() {
+    if (this.connected) return;
+    await this.client.connect(this.transport);
+    this.connected = true;
+  }
+  /** The tools this server exposes. */
+  async listTools() {
+    var _a;
+    await this.connect();
+    const res = await this.client.listTools();
+    return ((_a = res.tools) != null ? _a : []).map((t) => {
+      var _a2, _b;
+      return {
+        name: t.name,
+        description: (_a2 = t.description) != null ? _a2 : "",
+        inputSchema: (_b = t.inputSchema) != null ? _b : { type: "object", properties: {} }
+      };
+    });
+  }
+  /** Call a tool by name; returns its text output. Throws if the server flags an error. */
+  async callTool(name, args) {
+    await this.connect();
+    const res = await this.client.callTool({ name, arguments: args != null ? args : {} });
+    const text = extractText(res);
+    if (res.isError) {
+      throw new Error(`MCP tool "${name}" failed: ${text || "unknown error"}`);
+    }
+    return text;
+  }
+  async close() {
+    if (!this.connected) return;
+    try {
+      await this.client.close();
+    } catch (e) {
+    }
+    this.connected = false;
+  }
+};
+function extractText(res) {
+  const content = res.content;
+  if (!Array.isArray(content)) return "";
+  return content.filter((c) => (c == null ? void 0 : c.type) === "text" && typeof c.text === "string").map((c) => c.text).join("\n");
+}
+
+// src/infrastructure/mcp/mcp-tool-registry.ts
+var McpToolRegistry = class {
+  constructor(servers, createClient = (config) => new McpClient(config)) {
+    this.servers = servers;
+    this.createClient = createClient;
+    this.clients = [];
+  }
+  /** Discover every tool across the configured servers, as FunctionTools ready to pass to a model. */
+  async discoverTools() {
+    const tools = [];
+    for (const server of this.servers) {
+      const client = this.createClient(server);
+      this.clients.push(client);
+      for (const spec of await client.listTools()) {
+        tools.push({
+          type: "function",
+          name: spec.name,
+          description: spec.description,
+          parameters: spec.inputSchema,
+          strict: false,
+          invoke: (args) => client.callTool(spec.name, args != null ? args : {})
+        });
+      }
+    }
+    return tools;
+  }
+  /** Close all underlying MCP connections. */
+  async close() {
+    await Promise.all(this.clients.map((c) => c.close()));
+  }
+};
+
 // src/infrastructure/repository/in-memory-model-context-repository.ts
 var InMemoryModelContextRepository = class {
   constructor() {
@@ -566,39 +658,39 @@ var FunctionCallRunner = class {
 
 // src/application/participants/participant.ts
 var BaseParticipant = class extends Participant {
-  onFunctionCall(item) {
+  onFunctionCall(_item) {
   }
-  onExternalFunctionCall(source, item) {
+  onExternalFunctionCall(_source, _item) {
   }
-  onFunctionCallOutput(item) {
+  onFunctionCallOutput(_item) {
   }
-  onExternalFunctionCallOutput(source, item) {
+  onExternalFunctionCallOutput(_source, _item) {
   }
-  onReasoning(item) {
+  onReasoning(_item) {
   }
-  onExternalReasoning(source, item) {
+  onExternalReasoning(_source, _item) {
   }
-  onModelMessage(item) {
+  onModelMessage(_item) {
   }
-  onExternalModelMessage(source, item) {
+  onExternalModelMessage(_source, _item) {
   }
-  onMessage(message) {
+  onMessage(_message) {
   }
   onJoined() {
   }
   onLeft() {
   }
-  onParticipantJoined(participant) {
+  onParticipantJoined(_participant) {
   }
-  onParticipantLeft(participant) {
+  onParticipantLeft(_participant) {
   }
-  onInternalEvent(item) {
+  onInternalEvent(_item) {
   }
-  onExternalEvent(source, item) {
+  onExternalEvent(_source, _item) {
   }
-  onError(error) {
+  onError(_error) {
   }
-  onParticipantError(source, error) {
+  onParticipantError(_source, _error) {
   }
 };
 
@@ -1758,6 +1850,8 @@ export {
   InMemoryModelContextRepository,
   InferenceResponse,
   InputTokenDetails,
+  McpClient,
+  McpToolRegistry,
   ModelContext,
   ModelMessageItem,
   OutputTokenDetails,