opencode-toolbox 0.10.3 → 0.10.4

package/dist/index.js

@@ -36280,21 +36280,21 @@ class LocalMCPClient {
   toolsCache;
   name;
   config;
-  constructor(config3) {
+  transportFactory;
+  constructor(config3, options) {
     this.transport = null;
     this.toolsCache = null;
     this.name = config3.name;
     this.config = config3;
-    this.client = new Client({
-      name: `opencode-toolbox-client-${this.name}`,
-      version: "0.1.0"
-    }, {});
+    const clientFactory = options?.clientFactory ?? ((name) => new Client({ name: `opencode-toolbox-client-${name}`, version: "0.1.0" }, {}));
+    this.transportFactory = options?.transportFactory ?? ((opts) => new StdioClientTransport(opts));
+    this.client = clientFactory(this.name);
   }
   async connect() {
     if (!this.config.command || this.config.command.length === 0) {
       throw new Error(`Local MCP server ${this.name} has no command`);
     }
-    this.transport = new StdioClientTransport({
+    this.transport = this.transportFactory({
       command: this.config.command[0],
       args: this.config.command.slice(1),
       env: {
@@ -38028,20 +38028,45 @@ class RemoteMCPClient {
   name;
   config;
   transportType;
-  constructor(config3) {
+  options;
+  constructor(config3, options) {
     this.transport = null;
     this.toolsCache = null;
     this.name = config3.name;
     this.config = config3;
     this.transportType = null;
+    this.options = options ?? {};
     this.client = this.createClient();
   }
   createClient() {
+    if (this.options.clientFactory) {
+      return this.options.clientFactory(this.name);
+    }
     return new Client({
       name: `opencode-toolbox-client-${this.name}`,
       version: "0.1.0"
     }, {});
   }
+  createStreamableTransport(url3) {
+    if (this.options.streamableTransportFactory) {
+      return this.options.streamableTransportFactory(url3, this.config.headers);
+    }
+    return new StreamableHTTPClientTransport(url3, {
+      requestInit: {
+        headers: this.config.headers
+      }
+    });
+  }
+  createSSETransport(url3, headers) {
+    if (this.options.sseTransportFactory) {
+      return this.options.sseTransportFactory(url3, headers);
+    }
+    return new SSEClientTransport(url3, {
+      requestInit: {
+        headers
+      }
+    });
+  }
   async connect() {
     if (!this.config.url) {
       throw new Error(`Remote MCP server ${this.name} has no URL`);
@@ -38050,11 +38075,7 @@ class RemoteMCPClient {
     this.transportType = null;
     let streamableTransport = null;
     try {
-      streamableTransport = new StreamableHTTPClientTransport(url3, {
-        requestInit: {
-          headers: this.config.headers
-        }
-      });
+      streamableTransport = this.createStreamableTransport(url3);
       await this.client.connect(streamableTransport);
       this.transport = streamableTransport;
       this.transportType = "streamable-http";
@@ -38071,11 +38092,7 @@ class RemoteMCPClient {
         Accept: "text/event-stream",
         ...this.config.headers
       };
-      sseTransport = new SSEClientTransport(url3, {
-        requestInit: {
-          headers: sseHeaders
-        }
-      });
+      sseTransport = this.createSSETransport(url3, sseHeaders);
       await this.client.connect(sseTransport);
       this.transport = sseTransport;
       this.transportType = "sse";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-toolbox",
-  "version": "0.10.3",
+  "version": "0.10.4",
   "description": "Tool Search Tool Plugin for OpenCode - search and execute tools from MCP servers on-demand",
   "main": "dist/index.js",
   "module": "dist/index.js",