@probelabs/probe 0.6.0-rc336 → 0.6.0-rc339

package/build/agent/ProbeAgent.js

@@ -2488,7 +2488,6 @@ export class ProbeAgent {
         this.engine = await createCodexEngine({
           agent: this, // Pass reference to ProbeAgent for tool access
           systemPrompt: systemPrompt,
-          customPrompt: this.customPrompt,
           sessionId: this.options?.sessionId,
           debug: this.debug,
           allowedTools: this.allowedTools,  // Pass tool filtering configuration
@@ -3337,7 +3336,7 @@ ${extractGuidance2}
     const { prompt } = this._prepareGovernedAnswerPrompt(message, options);
     const systemPrompt = await this._getCachedCodexNativeSystemPrompt();
     const { previewGovernedCodexInitialDispatch } = await import('./engines/codex.js');
-    return previewGovernedCodexInitialDispatch({ systemPrompt, customPrompt: this.customPrompt, prompt });
+    return previewGovernedCodexInitialDispatch({ systemPrompt, prompt });
   }
 
   /**

package/build/agent/mcp/client.js

@@ -7,6 +7,7 @@ import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
 import { WebSocketClientTransport } from '@modelcontextprotocol/sdk/client/websocket.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { loadMCPConfiguration, parseEnabledServers, DEFAULT_TIMEOUT } from './config.js';
 
 /**
@@ -89,71 +90,24 @@ export function createTransport(serverConfig) {
       }
 
     case 'http':
-    case 'streamable':
-      // For HTTP, we'll use a custom implementation since the SDK
-      // doesn't provide a direct HTTP transport yet
+    case 'streamable': {
       if (!url) {
         throw new Error('HTTP transport requires a URL');
       }
-      // Return a custom HTTP transport wrapper
-      return createHttpTransport(url);
+      const requestInit = {};
+      if (serverConfig.headers && typeof serverConfig.headers === 'object') {
+        requestInit.headers = { ...serverConfig.headers };
+      }
+      return new StreamableHTTPClientTransport(new URL(url), {
+        requestInit: Object.keys(requestInit).length > 0 ? requestInit : undefined,
+      });
+    }
 
     default:
       throw new Error(`Unknown transport type: ${transport}`);
   }
 }
 
-/**
- * Create a custom HTTP transport wrapper
- * This simulates MCP over HTTP REST endpoints
- */
-function createHttpTransport(url) {
-  // This is a simplified HTTP transport
-  // In practice, you'd implement the full MCP protocol over HTTP
-  return {
-    async start() {
-      // Initialize HTTP connection
-      const response = await fetch(`${url}/initialize`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({
-          protocolVersion: '2024-11-05',
-          capabilities: {}
-        })
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP initialization failed: ${response.statusText}`);
-      }
-
-      return response.json();
-    },
-
-    async send(message) {
-      const response = await fetch(`${url}/message`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify(message)
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP request failed: ${response.statusText}`);
-      }
-
-      return response.json();
-    },
-
-    async close() {
-      // Close HTTP connection
-      await fetch(`${url}/close`, {
-        method: 'POST'
-      }).catch(() => {
-        // Ignore close errors
-      });
-    }
-  };
-}
-
 /**
  * MCP Client Manager - manages multiple MCP server connections
  */

package/cjs/agent/ProbeAgent.cjs

@@ -96596,51 +96596,23 @@ function createTransport(serverConfig) {
         throw new Error(`Invalid WebSocket URL: ${url2}`);
       }
     case "http":
-    case "streamable":
+    case "streamable": {
       if (!url2) {
         throw new Error("HTTP transport requires a URL");
       }
-      return createHttpTransport(url2);
-    default:
-      throw new Error(`Unknown transport type: ${transport}`);
-  }
-}
-function createHttpTransport(url2) {
-  return {
-    async start() {
-      const response = await fetch(`${url2}/initialize`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          protocolVersion: "2024-11-05",
-          capabilities: {}
-        })
-      });
-      if (!response.ok) {
-        throw new Error(`HTTP initialization failed: ${response.statusText}`);
+      const requestInit = {};
+      if (serverConfig.headers && typeof serverConfig.headers === "object") {
+        requestInit.headers = { ...serverConfig.headers };
       }
-      return response.json();
-    },
-    async send(message) {
-      const response = await fetch(`${url2}/message`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify(message)
-      });
-      if (!response.ok) {
-        throw new Error(`HTTP request failed: ${response.statusText}`);
-      }
-      return response.json();
-    },
-    async close() {
-      await fetch(`${url2}/close`, {
-        method: "POST"
-      }).catch(() => {
+      return new import_streamableHttp.StreamableHTTPClientTransport(new URL(url2), {
+        requestInit: Object.keys(requestInit).length > 0 ? requestInit : void 0
       });
     }
-  };
+    default:
+      throw new Error(`Unknown transport type: ${transport}`);
+  }
 }
-var import_client, import_stdio, import_sse, import_websocket, MCPClientManager;
+var import_client, import_stdio, import_sse, import_websocket, import_streamableHttp, MCPClientManager;
 var init_client = __esm({
   "src/agent/mcp/client.js"() {
     "use strict";
@@ -96648,6 +96620,7 @@ var init_client = __esm({
     import_stdio = require("@modelcontextprotocol/sdk/client/stdio.js");
     import_sse = require("@modelcontextprotocol/sdk/client/sse.js");
     import_websocket = require("@modelcontextprotocol/sdk/client/websocket.js");
+    import_streamableHttp = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
     init_config2();
     MCPClientManager = class {
       constructor(options = {}) {
@@ -97349,7 +97322,7 @@ function abortable(promise2, signal) {
     });
   });
 }
-var import_http, import_events2, import_crypto6, import_server, import_sse2, import_streamableHttp, import_ai5, import_types5, TOOL_DESCRIPTORS, ARGUMENT_DOMAIN, GOVERNED_CODEX_PROFILE_V2, GOVERNED_CALL_LIMIT, InMemoryEventStore, BuiltInMCPServer;
+var import_http, import_events2, import_crypto6, import_server, import_sse2, import_streamableHttp2, import_ai5, import_types5, TOOL_DESCRIPTORS, ARGUMENT_DOMAIN, GOVERNED_CODEX_PROFILE_V2, GOVERNED_CALL_LIMIT, InMemoryEventStore, BuiltInMCPServer;
 var init_built_in_server = __esm({
   "src/agent/mcp/built-in-server.js"() {
     "use strict";
@@ -97358,7 +97331,7 @@ var init_built_in_server = __esm({
     import_crypto6 = require("crypto");
     import_server = require("@modelcontextprotocol/sdk/server/index.js");
     import_sse2 = require("@modelcontextprotocol/sdk/server/sse.js");
-    import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+    import_streamableHttp2 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
     import_ai5 = require("ai");
     init_common();
     import_types5 = require("@modelcontextprotocol/sdk/types.js");
@@ -97612,7 +97585,7 @@ var init_built_in_server = __esm({
               console.log("[MCP] Creating new Streamable HTTP transport for initialization");
             }
             const eventStore = new InMemoryEventStore();
-            transport = new import_streamableHttp.StreamableHTTPServerTransport({
+            transport = new import_streamableHttp2.StreamableHTTPServerTransport({
               sessionIdGenerator: () => (0, import_crypto6.randomUUID)(),
               eventStore,
               // Enable resumability
@@ -110322,7 +110295,6 @@ var init_ProbeAgent = __esm({
               agent: this,
               // Pass reference to ProbeAgent for tool access
               systemPrompt,
-              customPrompt: this.customPrompt,
               sessionId: this.options?.sessionId,
               debug: this.debug,
               allowedTools: this.allowedTools,
@@ -111031,7 +111003,7 @@ Workspace: ${this.allowedFolders.join(", ")}`;
         const { prompt } = this._prepareGovernedAnswerPrompt(message, options);
         const systemPrompt = await this._getCachedCodexNativeSystemPrompt();
         const { previewGovernedCodexInitialDispatch: previewGovernedCodexInitialDispatch2 } = await Promise.resolve().then(() => (init_codex(), codex_exports));
-        return previewGovernedCodexInitialDispatch2({ systemPrompt, customPrompt: this.customPrompt, prompt });
+        return previewGovernedCodexInitialDispatch2({ systemPrompt, prompt });
       }
       /**
        * Get the system message with instructions for the AI (XML Tool Format)

package/cjs/index.cjs

@@ -96542,51 +96542,23 @@ function createTransport(serverConfig) {
         throw new Error(`Invalid WebSocket URL: ${url2}`);
       }
     case "http":
-    case "streamable":
+    case "streamable": {
       if (!url2) {
         throw new Error("HTTP transport requires a URL");
       }
-      return createHttpTransport(url2);
-    default:
-      throw new Error(`Unknown transport type: ${transport}`);
-  }
-}
-function createHttpTransport(url2) {
-  return {
-    async start() {
-      const response = await fetch(`${url2}/initialize`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          protocolVersion: "2024-11-05",
-          capabilities: {}
-        })
-      });
-      if (!response.ok) {
-        throw new Error(`HTTP initialization failed: ${response.statusText}`);
+      const requestInit = {};
+      if (serverConfig.headers && typeof serverConfig.headers === "object") {
+        requestInit.headers = { ...serverConfig.headers };
       }
-      return response.json();
-    },
-    async send(message) {
-      const response = await fetch(`${url2}/message`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify(message)
-      });
-      if (!response.ok) {
-        throw new Error(`HTTP request failed: ${response.statusText}`);
-      }
-      return response.json();
-    },
-    async close() {
-      await fetch(`${url2}/close`, {
-        method: "POST"
-      }).catch(() => {
+      return new import_streamableHttp.StreamableHTTPClientTransport(new URL(url2), {
+        requestInit: Object.keys(requestInit).length > 0 ? requestInit : void 0
       });
     }
-  };
+    default:
+      throw new Error(`Unknown transport type: ${transport}`);
+  }
 }
-var import_client, import_stdio, import_sse, import_websocket, MCPClientManager;
+var import_client, import_stdio, import_sse, import_websocket, import_streamableHttp, MCPClientManager;
 var init_client = __esm({
   "src/agent/mcp/client.js"() {
     "use strict";
@@ -96594,6 +96566,7 @@ var init_client = __esm({
     import_stdio = require("@modelcontextprotocol/sdk/client/stdio.js");
     import_sse = require("@modelcontextprotocol/sdk/client/sse.js");
     import_websocket = require("@modelcontextprotocol/sdk/client/websocket.js");
+    import_streamableHttp = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
     init_config2();
     MCPClientManager = class {
       constructor(options = {}) {
@@ -97295,7 +97268,7 @@ function abortable(promise2, signal) {
     });
   });
 }
-var import_http, import_events2, import_crypto5, import_server, import_sse2, import_streamableHttp, import_ai5, import_types5, TOOL_DESCRIPTORS, ARGUMENT_DOMAIN, GOVERNED_CODEX_PROFILE_V2, GOVERNED_CALL_LIMIT, InMemoryEventStore, BuiltInMCPServer;
+var import_http, import_events2, import_crypto5, import_server, import_sse2, import_streamableHttp2, import_ai5, import_types5, TOOL_DESCRIPTORS, ARGUMENT_DOMAIN, GOVERNED_CODEX_PROFILE_V2, GOVERNED_CALL_LIMIT, InMemoryEventStore, BuiltInMCPServer;
 var init_built_in_server = __esm({
   "src/agent/mcp/built-in-server.js"() {
     "use strict";
@@ -97304,7 +97277,7 @@ var init_built_in_server = __esm({
     import_crypto5 = require("crypto");
     import_server = require("@modelcontextprotocol/sdk/server/index.js");
     import_sse2 = require("@modelcontextprotocol/sdk/server/sse.js");
-    import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+    import_streamableHttp2 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
     import_ai5 = require("ai");
     init_common();
     import_types5 = require("@modelcontextprotocol/sdk/types.js");
@@ -97558,7 +97531,7 @@ var init_built_in_server = __esm({
               console.log("[MCP] Creating new Streamable HTTP transport for initialization");
             }
             const eventStore = new InMemoryEventStore();
-            transport = new import_streamableHttp.StreamableHTTPServerTransport({
+            transport = new import_streamableHttp2.StreamableHTTPServerTransport({
               sessionIdGenerator: () => (0, import_crypto5.randomUUID)(),
               eventStore,
               // Enable resumability
@@ -110268,7 +110241,6 @@ var init_ProbeAgent = __esm({
               agent: this,
               // Pass reference to ProbeAgent for tool access
               systemPrompt,
-              customPrompt: this.customPrompt,
               sessionId: this.options?.sessionId,
               debug: this.debug,
               allowedTools: this.allowedTools,
@@ -110977,7 +110949,7 @@ Workspace: ${this.allowedFolders.join(", ")}`;
         const { prompt } = this._prepareGovernedAnswerPrompt(message, options);
         const systemPrompt = await this._getCachedCodexNativeSystemPrompt();
         const { previewGovernedCodexInitialDispatch: previewGovernedCodexInitialDispatch2 } = await Promise.resolve().then(() => (init_codex(), codex_exports));
-        return previewGovernedCodexInitialDispatch2({ systemPrompt, customPrompt: this.customPrompt, prompt });
+        return previewGovernedCodexInitialDispatch2({ systemPrompt, prompt });
       }
       /**
        * Get the system message with instructions for the AI (XML Tool Format)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc336",
+  "version": "0.6.0-rc339",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",

package/src/agent/ProbeAgent.js

@@ -2488,7 +2488,6 @@ export class ProbeAgent {
         this.engine = await createCodexEngine({
           agent: this, // Pass reference to ProbeAgent for tool access
           systemPrompt: systemPrompt,
-          customPrompt: this.customPrompt,
           sessionId: this.options?.sessionId,
           debug: this.debug,
           allowedTools: this.allowedTools,  // Pass tool filtering configuration
@@ -3337,7 +3336,7 @@ ${extractGuidance2}
     const { prompt } = this._prepareGovernedAnswerPrompt(message, options);
     const systemPrompt = await this._getCachedCodexNativeSystemPrompt();
     const { previewGovernedCodexInitialDispatch } = await import('./engines/codex.js');
-    return previewGovernedCodexInitialDispatch({ systemPrompt, customPrompt: this.customPrompt, prompt });
+    return previewGovernedCodexInitialDispatch({ systemPrompt, prompt });
   }
 
   /**

package/src/agent/mcp/client.js

@@ -7,6 +7,7 @@ import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
 import { WebSocketClientTransport } from '@modelcontextprotocol/sdk/client/websocket.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { loadMCPConfiguration, parseEnabledServers, DEFAULT_TIMEOUT } from './config.js';
 
 /**
@@ -89,71 +90,24 @@ export function createTransport(serverConfig) {
       }
 
     case 'http':
-    case 'streamable':
-      // For HTTP, we'll use a custom implementation since the SDK
-      // doesn't provide a direct HTTP transport yet
+    case 'streamable': {
       if (!url) {
         throw new Error('HTTP transport requires a URL');
       }
-      // Return a custom HTTP transport wrapper
-      return createHttpTransport(url);
+      const requestInit = {};
+      if (serverConfig.headers && typeof serverConfig.headers === 'object') {
+        requestInit.headers = { ...serverConfig.headers };
+      }
+      return new StreamableHTTPClientTransport(new URL(url), {
+        requestInit: Object.keys(requestInit).length > 0 ? requestInit : undefined,
+      });
+    }
 
     default:
       throw new Error(`Unknown transport type: ${transport}`);
   }
 }
 
-/**
- * Create a custom HTTP transport wrapper
- * This simulates MCP over HTTP REST endpoints
- */
-function createHttpTransport(url) {
-  // This is a simplified HTTP transport
-  // In practice, you'd implement the full MCP protocol over HTTP
-  return {
-    async start() {
-      // Initialize HTTP connection
-      const response = await fetch(`${url}/initialize`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({
-          protocolVersion: '2024-11-05',
-          capabilities: {}
-        })
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP initialization failed: ${response.statusText}`);
-      }
-
-      return response.json();
-    },
-
-    async send(message) {
-      const response = await fetch(`${url}/message`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify(message)
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP request failed: ${response.statusText}`);
-      }
-
-      return response.json();
-    },
-
-    async close() {
-      // Close HTTP connection
-      await fetch(`${url}/close`, {
-        method: 'POST'
-      }).catch(() => {
-        // Ignore close errors
-      });
-    }
-  };
-}
-
 /**
  * MCP Client Manager - manages multiple MCP server connections
  */