@eko-ai/eko-nodejs 2.2.0 → 3.0.0-alpha.1

package/dist/index.cjs.js

@@ -12,6 +12,7 @@ var promises = require('node:fs/promises');
 var node_events = require('node:events');
 var Stream = require('node:stream');
 var node_string_decoder = require('node:string_decoder');
+var child_process = require('child_process');
 
 function _interopNamespaceDefault(e) {
     var n = Object.create(null);
@@ -8250,7 +8251,109 @@ class FileAgent extends eko.BaseFileAgent {
     }
 }
 
+class SimpleStdioMcpClient {
+    constructor(command, args, options) {
+        this.process = null;
+        this.command = command;
+        this.args = args || [];
+        this.options = options || {
+            stdio: ["pipe", "pipe", "pipe"],
+        };
+        this.requestMap = new Map();
+    }
+    async connect(signal) {
+        if (this.process) {
+            try {
+                this.process.kill();
+            }
+            catch (e) { }
+        }
+        this.process = child_process.spawn(this.command, this.args, this.options);
+        this.process.stdout.on("data", (data) => {
+            const response = data.toString().trim();
+            eko.Log.debug("MCP Client, onmessage", this.command, this.args, response);
+            if (!response.startsWith("{")) {
+                return;
+            }
+            const message = JSON.parse(response);
+            if (message.id) {
+                const callback = this.requestMap.get(message.id);
+                if (callback) {
+                    callback(message);
+                }
+            }
+        });
+        this.process.on("error", (error) => {
+            eko.Log.error("MCP process error:", this.command, this.args, error);
+        });
+        eko.Log.info("MCP Client, connection successful:", this.command, this.args);
+    }
+    async sendMessage(method, params = {}, signal) {
+        if (!this.process) {
+            await this.connect();
+            if (!this.process) {
+                throw new Error("Failed to connect to MCP server");
+            }
+        }
+        const id = eko.uuidv4();
+        try {
+            const callback = new Promise((resolve, reject) => {
+                if (signal) {
+                    signal.addEventListener("abort", () => {
+                        reject(new Error("AbortError"));
+                    });
+                }
+                this.requestMap.set(id, resolve);
+            });
+            const message = JSON.stringify({
+                jsonrpc: "2.0",
+                id: id,
+                method: method,
+                params: {
+                    ...params,
+                },
+            });
+            eko.Log.debug(`MCP Client, ${method}`, id, params);
+            const suc = this.process.stdin.write(message + "\n", "utf-8");
+            if (!suc) {
+                throw new Error("SseClient Response Exception: " + message);
+            }
+            return await callback;
+        }
+        finally {
+            this.requestMap.delete(id);
+        }
+    }
+    async listTools(param, signal) {
+        const message = await this.sendMessage("tools/list", {
+            ...param,
+        }, signal);
+        if (message.error) {
+            eko.Log.error("McpClient listTools error: ", param, message);
+            throw new Error("listTools Exception");
+        }
+        return message.result.tools || [];
+    }
+    async callTool(param, signal) {
+        const message = await this.sendMessage("tools/call", {
+            ...param,
+        }, signal);
+        if (message.error) {
+            eko.Log.error("McpClient callTool error: ", param, message);
+            throw new Error("callTool Exception");
+        }
+        return message.result;
+    }
+    isConnected() {
+        return (this.process != null && !this.process.killed && !this.process.exitCode);
+    }
+    async close() {
+        this.process && this.process.kill();
+    }
+}
+
 exports.BrowserAgent = BrowserAgent;
 exports.FileAgent = FileAgent;
+exports.SimpleStdioMcpClient = SimpleStdioMcpClient;
 exports.getCdpWsEndpoint = getCdpWsEndpoint;
 //# sourceMappingURL=index.cjs.js.map