micro-models-agent 2.16.1 → 2.17.2

package/dist/modules/mcp/client.js

@@ -1,445 +0,0 @@
-// MCP-клиент: JSON-RPC поверх stdio/sse/http, listTools/callTool, таймауты.
-import { spawn } from "child_process";
-import { defaultOutputBus } from "../../output";
-import { MCP_HTTP_TIMEOUT_MS as HTTP_TIMEOUT_MS } from "../../config/timeouts";
-/**
- * Accept-заголовок для MCP Streamable HTTP: сервер отвечает 406, если клиент
- * не принимает оба media type.
- */
-const MCP_ACCEPT_HEADER = "application/json, text/event-stream";
-/** MCP-клиент: коннект по транспорту, listTools, callTool. */
-export class MCPClient {
-    serverName;
-    config;
-    process = null;
-    _connected = false;
-    abortController = null;
-    /** In-flight SSE-вызовы по JSON-RPC id (один слот мешал конкурентные
-     *  вызовы и матчил ответы по любому id — теперь слот на вызов). */
-    ssePending = new Map();
-    sseSeq = 0;
-    constructor(config) {
-        this.config = config;
-        this.serverName = config.name;
-    }
-    /** Подключён ли клиент. */
-    isConnected() {
-        return this._connected;
-    }
-    /** Подключиться по транспорту из конфига (http/sse/stdio). */
-    async connect() {
-        if (this._connected)
-            return;
-        if (this.config.transport === "http") {
-            this._connected = true;
-            return;
-        }
-        if (this.config.transport === "sse" || this.config.url) {
-            return this.connectSSE();
-        }
-        return this.connectStdio();
-    }
-    async connectSSE() {
-        const url = this.config.url;
-        if (!url)
-            throw new Error("SSE transport requires a url");
-        this.abortController = new AbortController();
-        const timeout = setTimeout(() => this.abortController?.abort(), this.config.timeout || HTTP_TIMEOUT_MS);
-        try {
-            const response = await fetch(url, {
-                method: "GET",
-                headers: {
-                    Accept: "text/event-stream",
-                    ...this.config.headers,
-                },
-                signal: this.abortController.signal,
-            });
-            clearTimeout(timeout);
-            if (!response.ok) {
-                throw new Error(`SSE connection failed: ${response.status} ${response.statusText}`);
-            }
-            this._connected = true;
-            const reader = response.body.getReader();
-            const decoder = new TextDecoder();
-            const buffer = "";
-            this.readSSE(reader, decoder, buffer);
-        }
-        catch (err) {
-            clearTimeout(timeout);
-            throw err;
-        }
-    }
-    async readSSE(reader, decoder, buffer) {
-        try {
-            while (this._connected) {
-                const { done, value } = await reader.read();
-                if (done)
-                    break;
-                buffer += decoder.decode(value, { stream: true });
-                const lines = buffer.split("\n");
-                buffer = lines.pop() || "";
-                let eventType = "";
-                let data = "";
-                for (const line of lines) {
-                    if (line.startsWith("event: ")) {
-                        eventType = line.slice(7).trim();
-                    }
-                    else if (line.startsWith("data: ")) {
-                        data = line.slice(6);
-                    }
-                    else if (line === "") {
-                        if (eventType === "message" && data) {
-                            this.handleSSEMessage(data);
-                        }
-                        eventType = "";
-                        data = "";
-                    }
-                }
-            }
-        }
-        catch (err) {
-            if (err.name !== "AbortError") {
-                this.handleSSEError(err);
-            }
-        }
-        this._connected = false;
-    }
-    handleSSEMessage(data) {
-        try {
-            const msg = JSON.parse(data);
-            if (msg.id !== undefined) {
-                const pending = this.ssePending.get(msg.id);
-                if (pending) {
-                    this.ssePending.delete(msg.id);
-                    clearTimeout(pending.timer);
-                    if (msg.error)
-                        pending.reject(new Error(msg.error.message));
-                    else
-                        pending.resolve(msg.result);
-                }
-            }
-        }
-        catch (e) {
-            // Хартбиты/комменты — не JSON: ожидаемый шум, но логируем, чтобы
-            // реально битый стрим был виден в диагностике.
-            defaultOutputBus.log("error", "mcp", `[MCPClient] Unparseable SSE message: ${String(data).slice(0, 120)} ${e instanceof Error ? e.message : String(e)}`, { persist: true });
-        }
-    }
-    handleSSEError(err) {
-        for (const [, pending] of this.ssePending) {
-            clearTimeout(pending.timer);
-            pending.reject(err);
-        }
-        this.ssePending.clear();
-    }
-    connectStdio() {
-        return new Promise((resolve, reject) => {
-            const child = spawn(this.config.command, this.config.args || [], {
-                env: { ...process.env, ...this.config.env },
-                stdio: ["pipe", "pipe", "pipe"],
-            });
-            let _resolved = false;
-            child.on("error", (err) => {
-                this._connected = false;
-                if (!_resolved) {
-                    _resolved = true;
-                    reject(err);
-                }
-            });
-            child.on("exit", (code) => {
-                this._connected = false;
-                if (code !== 0 && !_resolved) {
-                    _resolved = true;
-                    reject(new Error(`MCP server exited with code ${code}`));
-                }
-            });
-            child.on("spawn", () => {
-                this._connected = true;
-                this.process = child;
-                // MCP-handshake: настоящие серверы требуют `initialize` перед
-                // tools/list и tools/call.
-                this.stdioRequest("initialize", {
-                    protocolVersion: "2024-11-05",
-                    capabilities: {},
-                    clientInfo: { name: "mma", version: "1.0.0" },
-                }, 10000)
-                    .then(() => {
-                    child.stdin?.write(`${JSON.stringify({ jsonrpc: "2.0", method: "notifications/initialized" })}\n`);
-                })
-                    .then(() => {
-                    _resolved = true;
-                    resolve();
-                })
-                    .catch((err) => {
-                    this._connected = false;
-                    if (!_resolved) {
-                        _resolved = true;
-                        reject(err);
-                    }
-                });
-            });
-        });
-    }
-    /** Неразобранный хвост stdio-ответов (частичная строка). */
-    stdioBuffer = "";
-    /** Монотонный id stdio-запросов (Date.now() коллизил на вызовах в одну мс). */
-    stdioRequestSeq = 0;
-    /**
-     * JSON-RPC запрос поверх stdio: таймаут, обработка exit и построчный парсинг
-     * с учётом буфера (нотификации и чужие ответы между запросом и ответом
-     * матчинг больше не ломают).
-     */
-    stdioRequest(method, params, timeoutMs) {
-        const proc = this.process;
-        if (!proc?.stdout || !proc.stdin) {
-            return Promise.reject(new Error("Not connected to MCP server"));
-        }
-        const stdout = proc.stdout;
-        const timeout = timeoutMs ?? this.config.timeout ?? 30000;
-        return new Promise((resolve, reject) => {
-            const request = {
-                jsonrpc: "2.0",
-                id: ++this.stdioRequestSeq,
-                method,
-                ...(params !== undefined ? { params } : {}),
-            };
-            let timer;
-            let settled = false;
-            const finish = (fn) => {
-                if (settled)
-                    return;
-                settled = true;
-                if (timer !== undefined)
-                    clearTimeout(timer);
-                stdout.off("data", onData);
-                proc.off("exit", onExit);
-                fn();
-            };
-            const onData = (data) => {
-                this.stdioBuffer += data.toString();
-                let idx;
-                while ((idx = this.stdioBuffer.indexOf("\n")) !== -1 && !settled) {
-                    const line = this.stdioBuffer.slice(0, idx).trim();
-                    this.stdioBuffer = this.stdioBuffer.slice(idx + 1);
-                    if (!line)
-                        continue;
-                    let msg;
-                    try {
-                        msg = JSON.parse(line);
-                    }
-                    catch {
-                        continue;
-                    }
-                    if (msg.id !== request.id)
-                        continue; // нотификация или чужой ответ
-                    finish(() => {
-                        if (msg.error) {
-                            reject(new Error(msg.error.message));
-                        }
-                        else {
-                            resolve(msg.result);
-                        }
-                    });
-                    return;
-                }
-            };
-            const onExit = () => {
-                finish(() => reject(new Error(`MCP server exited before responding to ${method}`)));
-            };
-            stdout.on("data", onData);
-            proc.once("exit", onExit);
-            timer = setTimeout(() => {
-                finish(() => reject(new Error(`MCP ${method} timeout`)));
-            }, timeout);
-            proc.stdin.write(`${JSON.stringify(request)}\n`);
-        });
-    }
-    /** Список тулов сервера (автоконнект; транспорт выбирается из конфига). */
-    async listTools() {
-        if (!this._connected) {
-            await this.connect();
-        }
-        if (this.config.transport === "http") {
-            return this.listToolsHTTP();
-        }
-        if (this.config.transport === "sse" || this.config.url) {
-            return this.listToolsSSE();
-        }
-        return this.listToolsStdio();
-    }
-    async listToolsHTTP() {
-        const request = {
-            jsonrpc: "2.0",
-            id: Date.now(),
-            method: "tools/list",
-        };
-        const response = await fetch(this.config.url, {
-            method: "POST",
-            headers: {
-                Accept: MCP_ACCEPT_HEADER,
-                "Content-Type": "application/json",
-                ...this.config.headers,
-            },
-            body: JSON.stringify(request),
-            signal: AbortSignal.timeout(this.config.timeout || HTTP_TIMEOUT_MS),
-        });
-        if (!response.ok) {
-            const text = await response.text().catch(() => "");
-            throw new Error(`MCP listTools failed: ${response.status} ${text}`);
-        }
-        const result = await response.json();
-        if (result.error) {
-            throw new Error(result.error.message);
-        }
-        return result.result?.tools || [];
-    }
-    async listToolsSSE() {
-        const request = {
-            jsonrpc: "2.0",
-            id: Date.now(),
-            method: "tools/list",
-        };
-        const response = await fetch(this.config.url, {
-            method: "POST",
-            headers: {
-                Accept: MCP_ACCEPT_HEADER,
-                "Content-Type": "application/json",
-                ...this.config.headers,
-            },
-            body: JSON.stringify(request),
-            signal: this.abortController?.signal,
-        });
-        if (!response.ok) {
-            const text = await response.text().catch(() => "");
-            throw new Error(`MCP listTools failed: ${response.status} ${text}`);
-        }
-        const result = await response.json();
-        if (result.error) {
-            throw new Error(result.error.message);
-        }
-        return result.result?.tools || [];
-    }
-    async listToolsStdio() {
-        if (!this.process) {
-            throw new Error("Not connected to MCP server");
-        }
-        const result = await this.stdioRequest("tools/list");
-        return result?.tools || [];
-    }
-    /** Отключиться: http — флаг, sse — abort, stdio — kill процесса. */
-    async disconnect() {
-        if (this.config.transport === "http") {
-            this._connected = false;
-            return;
-        }
-        if (this.config.transport === "sse" || this.config.url) {
-            this.abortController?.abort();
-            this.abortController = null;
-        }
-        else if (this.process) {
-            this.process.kill();
-            this.process = null;
-        }
-        this._connected = false;
-    }
-    /** Вызвать тул (без коннекта — ошибка; транспорт из конфига). */
-    async callTool(toolName, args) {
-        if (!this._connected) {
-            throw new Error("Not connected to MCP server");
-        }
-        if (this.config.transport === "http") {
-            return this.callToolHTTP(toolName, args);
-        }
-        if (this.config.transport === "sse" || this.config.url) {
-            return this.callToolSSE(toolName, args);
-        }
-        return this.callToolStdio(toolName, args);
-    }
-    async callToolHTTP(toolName, args) {
-        const request = {
-            jsonrpc: "2.0",
-            id: Date.now(),
-            method: "tools/call",
-            params: {
-                name: toolName,
-                arguments: args,
-            },
-        };
-        const response = await fetch(this.config.url, {
-            method: "POST",
-            headers: {
-                Accept: MCP_ACCEPT_HEADER,
-                "Content-Type": "application/json",
-                ...this.config.headers,
-            },
-            body: JSON.stringify(request),
-            signal: AbortSignal.timeout(this.config.timeout || HTTP_TIMEOUT_MS),
-        });
-        if (!response.ok) {
-            const text = await response.text().catch(() => "");
-            throw new Error(`MCP call failed: ${response.status} ${text}`);
-        }
-        const result = await response.json();
-        if (result.error) {
-            throw new Error(result.error.message);
-        }
-        return result.result || result;
-    }
-    async callToolSSE(toolName, args) {
-        const id = ++this.sseSeq;
-        const request = {
-            jsonrpc: "2.0",
-            id,
-            method: "tools/call",
-            params: {
-                name: toolName,
-                arguments: args,
-            },
-        };
-        return new Promise((resolve, reject) => {
-            const settle = (fn) => {
-                const pending = this.ssePending.get(id);
-                if (!pending)
-                    return;
-                this.ssePending.delete(id);
-                clearTimeout(pending.timer);
-                fn();
-            };
-            const timeout = this.config.timeout || 30000;
-            const timer = setTimeout(() => settle(() => reject(new Error("MCP call timeout"))), timeout);
-            this.ssePending.set(id, { resolve, reject, timer });
-            fetch(this.config.url, {
-                method: "POST",
-                headers: {
-                    Accept: MCP_ACCEPT_HEADER,
-                    "Content-Type": "application/json",
-                    ...this.config.headers,
-                },
-                body: JSON.stringify(request),
-                signal: this.abortController?.signal,
-            })
-                .then(async (response) => {
-                if (!response.ok) {
-                    const text = await response.text().catch(() => "");
-                    settle(() => reject(new Error(`MCP POST failed: ${response.status} ${text}`)));
-                    return;
-                }
-                const result = await response.json();
-                settle(() => result.error ? reject(new Error(result.error.message)) : resolve(result.result || result));
-            })
-                .catch((err) => {
-                if (err.name !== "AbortError")
-                    settle(() => reject(err));
-            });
-        });
-    }
-    callToolStdio(toolName, args) {
-        if (!this.process) {
-            throw new Error("Not connected to MCP server");
-        }
-        return this.stdioRequest("tools/call", {
-            name: toolName,
-            arguments: args,
-        });
-    }
-}

package/dist/modules/mcp/index.js

@@ -1,4 +0,0 @@
-// Публичный фасад модуля MCP: клиент, реестр, модуль.
-export { MCPClient } from "./client";
-export { MCPRegistry } from "./registry";
-export { MCPModule } from "./module";