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

package/dist/index.esm.js

@@ -1,4 +1,4 @@
-import { Log, BaseBrowserLabelsAgent, BaseFileAgent } from '@eko-ai/eko';
+import { Log, BaseBrowserLabelsAgent, BaseFileAgent, uuidv4 } from '@eko-ai/eko';
 import { chromium } from 'playwright';
 import * as fs from 'fs/promises';
 import * as path$1 from 'path';
@@ -10,6 +10,7 @@ import { realpath, readlink, readdir, lstat } from 'node:fs/promises';
 import { EventEmitter } from 'node:events';
 import Stream from 'node:stream';
 import { StringDecoder } from 'node:string_decoder';
+import { spawn } from 'child_process';
 
 async function getCdpWsEndpoint(port) {
     // Example => ws://localhost:9222/devtools/browser/{session-id}
@@ -8227,5 +8228,106 @@ class FileAgent extends BaseFileAgent {
     }
 }
 
-export { BrowserAgent, FileAgent, getCdpWsEndpoint };
+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 = spawn(this.command, this.args, this.options);
+        this.process.stdout.on("data", (data) => {
+            const response = data.toString().trim();
+            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) => {
+            Log.error("MCP process error:", this.command, this.args, error);
+        });
+        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 = 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,
+                },
+            });
+            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) {
+            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) {
+            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();
+    }
+}
+
+export { BrowserAgent, FileAgent, SimpleStdioMcpClient, getCdpWsEndpoint };
 //# sourceMappingURL=index.esm.js.map