mcp-use 0.0.1

package/dist/src/task_managers/sse.js

@@ -0,0 +1,44 @@
+import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
+import { logger } from '../logging.js';
+import { ConnectionManager } from './base.js';
+export class SseConnectionManager extends ConnectionManager {
+    url;
+    opts;
+    _transport = null;
+    /**
+     * Create an SSE connection manager.
+     *
+     * @param url  The SSE endpoint URL.
+     * @param opts Optional transport options (auth, headers, etc.).
+     */
+    constructor(url, opts) {
+        super();
+        this.url = typeof url === 'string' ? new URL(url) : url;
+        this.opts = opts;
+    }
+    /**
+     * Spawn a new `SSEClientTransport` and start the connection.
+     */
+    async establishConnection() {
+        this._transport = new SSEClientTransport(this.url, this.opts);
+        // await this._transport.start()
+        logger.debug(`${this.constructor.name} connected successfully`);
+        return this._transport;
+    }
+    /**
+     * Close the underlying transport and clean up resources.
+     */
+    async closeConnection(_connection) {
+        if (this._transport) {
+            try {
+                await this._transport.close();
+            }
+            catch (e) {
+                logger.warn(`Error closing SSE transport: ${e}`);
+            }
+            finally {
+                this._transport = null;
+            }
+        }
+    }
+}

package/dist/src/task_managers/stdio.js

@@ -0,0 +1,52 @@
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { logger } from '../logging.js';
+import { ConnectionManager } from './base.js';
+export class StdioConnectionManager extends ConnectionManager {
+    serverParams;
+    errlog;
+    _transport = null;
+    /**
+     * Create a new stdio connection manager.
+     *
+     * @param serverParams Parameters for the stdio server process.
+     * @param errlog       Stream to which the server's stderr should be piped.
+     *                     Defaults to `process.stderr`.
+     */
+    constructor(serverParams, errlog = process.stderr) {
+        super();
+        this.serverParams = serverParams;
+        this.errlog = errlog;
+    }
+    /**
+     * Establish the stdio connection by spawning the server process and starting
+     * the SDK's transport. Returns the live `StdioClientTransport` instance.
+     */
+    async establishConnection() {
+        // Instantiate and start the transport
+        this._transport = new StdioClientTransport(this.serverParams);
+        // await this._transport.start()
+        // If stderr was piped, forward it to `errlog` for visibility
+        if (this._transport.stderr && typeof this._transport.stderr.pipe === 'function') {
+            this._transport.stderr.pipe(this.errlog);
+        }
+        logger.debug(`${this.constructor.name} connected successfully`);
+        return this._transport;
+    }
+    /**
+     * Close the stdio connection, making sure the transport cleans up the child
+     * process and associated resources.
+     */
+    async closeConnection(_connection) {
+        if (this._transport) {
+            try {
+                await this._transport.close();
+            }
+            catch (e) {
+                logger.warn(`Error closing stdio transport: ${e}`);
+            }
+            finally {
+                this._transport = null;
+            }
+        }
+    }
+}

package/dist/src/task_managers/websocket.js

@@ -0,0 +1,67 @@
+import WS from 'ws';
+import { logger } from '../logging.js';
+import { ConnectionManager } from './base.js';
+export class WebSocketConnectionManager extends ConnectionManager {
+    url;
+    headers;
+    _ws = null;
+    /**
+     * @param url     The WebSocket URL to connect to.
+     * @param headers Optional headers to include in the connection handshake.
+     */
+    constructor(url, headers = {}) {
+        super();
+        this.url = url;
+        this.headers = headers;
+    }
+    /** Establish a WebSocket connection and wait until it is open. */
+    async establishConnection() {
+        logger.debug(`Connecting to WebSocket: ${this.url}`);
+        return new Promise((resolve, reject) => {
+            const ws = new WS(this.url, { headers: this.headers });
+            this._ws = ws;
+            const onOpen = () => {
+                // eslint-disable-next-line ts/no-use-before-define
+                cleanup();
+                logger.debug('WebSocket connected successfully');
+                resolve(ws);
+            };
+            const onError = (err) => {
+                // eslint-disable-next-line ts/no-use-before-define
+                cleanup();
+                logger.error(`Failed to connect to WebSocket: ${err}`);
+                reject(err);
+            };
+            const cleanup = () => {
+                ws.off('open', onOpen);
+                ws.off('error', onError);
+            };
+            // Register listeners (browser vs Node API differences handled)
+            ws.on('open', onOpen);
+            ws.on('error', onError);
+        });
+    }
+    /** Cleanly close the WebSocket connection. */
+    async closeConnection(connection) {
+        logger.debug('Closing WebSocket connection');
+        return new Promise((resolve) => {
+            const onClose = () => {
+                connection.off('close', onClose);
+                this._ws = null;
+                resolve();
+            };
+            if (connection.readyState === WS.CLOSED) {
+                onClose();
+                return;
+            }
+            connection.on('close', onClose);
+            try {
+                connection.close();
+            }
+            catch (e) {
+                logger.warn(`Error closing WebSocket connection: ${e}`);
+                onClose();
+            }
+        });
+    }
+}

package/package.json

@@ -0,0 +1,75 @@
+{
+  "name": "mcp-use",
+  "type": "module",
+  "version": "0.0.1",
+  "description": "A utility library for integrating Model Context Protocol (MCP) with LangChain, Zod, and related tools. Provides helpers for schema conversion, event streaming, and SDK usage.",
+  "author": "Zane",
+  "license": "MIT",
+  "homepage": "https://github.com/zandko/mcp-use#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/zandko/mcp-use.git"
+  },
+  "bugs": {
+    "url": "https://github.com/zandko/mcp-use/issues"
+  },
+  "keywords": [
+    "MCP",
+    "Model Context Protocol",
+    "LangChain",
+    "Zod",
+    "schema",
+    "SDK",
+    "eventsource",
+    "AI",
+    "utility",
+    "typescript"
+  ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "registry": "https://registry.npmjs.org"
+  },
+  "scripts": {
+    "build": "tsc && shx chmod +x dist/*.js",
+    "watch": "tsc --watch",
+    "start": "node dist/index.js",
+    "prepublishOnly": "pnpm run build",
+    "lint": "eslint",
+    "lint:fix": "eslint --fix",
+    "release": "npm version patch && git push --follow-tags",
+    "release:minor": "npm version minor && git push --follow-tags",
+    "release:major": "npm version major && git push --follow-tags"
+  },
+  "dependencies": {
+    "@dmitryrechkin/json-schema-to-zod": "^1.0.1",
+    "@langchain/community": "0.3.41",
+    "@langchain/core": "0.3.45",
+    "@modelcontextprotocol/sdk": "1.10.1",
+    "eventsource": "^3.0.6",
+    "langchain": "^0.3.23",
+    "uuid": "^11.1.0",
+    "winston": "^3.17.0",
+    "ws": "^8.18.1",
+    "zod": "^3.24.3",
+    "zod-to-json-schema": "^3.24.5"
+  },
+  "devDependencies": {
+    "@antfu/eslint-config": "^4.12.0",
+    "@types/ws": "^8.18.1",
+    "eslint": "^9.24.0",
+    "eslint-plugin-format": "^1.0.1",
+    "shx": "^0.4.0",
+    "typescript": "^5.8.3"
+  }
+}