@probelabs/probe 0.6.0-rc268 → 0.6.0-rc269

package/build/agent/ProbeAgent.js

@@ -1879,7 +1879,15 @@ export class ProbeAgent {
     if (this.mcpBridge && !options._disableTools) {
       const mcpTools = this.mcpBridge.getVercelTools(this._filterMcpTools(this.mcpBridge.getToolNames()));
       for (const [name, mcpTool] of Object.entries(mcpTools)) {
-        nativeTools[name] = mcpTool;
+        // MCP tools have raw JSON Schema inputSchema that must be wrapped with jsonSchema()
+        // for the Vercel AI SDK. Without wrapping, asSchema() misidentifies them as Zod schemas.
+        const mcpSchema = mcpTool.inputSchema || mcpTool.parameters;
+        const wrappedSchema = mcpSchema && mcpSchema._def ? mcpSchema : jsonSchema(mcpSchema || { type: 'object', properties: {} });
+        nativeTools[name] = tool({
+          description: mcpTool.description || `MCP tool: ${name}`,
+          inputSchema: wrappedSchema,
+          execute: mcpTool.execute,
+        });
       }
     }
 

package/build/agent/index.js

@@ -83190,7 +83190,13 @@ var init_ProbeAgent = __esm({
         if (this.mcpBridge && !options._disableTools) {
           const mcpTools = this.mcpBridge.getVercelTools(this._filterMcpTools(this.mcpBridge.getToolNames()));
           for (const [name, mcpTool] of Object.entries(mcpTools)) {
-            nativeTools[name] = mcpTool;
+            const mcpSchema = mcpTool.inputSchema || mcpTool.parameters;
+            const wrappedSchema = mcpSchema && mcpSchema._def ? mcpSchema : jsonSchema(mcpSchema || { type: "object", properties: {} });
+            nativeTools[name] = tool5({
+              description: mcpTool.description || `MCP tool: ${name}`,
+              inputSchema: wrappedSchema,
+              execute: mcpTool.execute
+            });
           }
         }
         if (this.apiType === "google" && this._geminiToolsEnabled && !options._disableTools) {

package/cjs/agent/ProbeAgent.cjs

@@ -110138,7 +110138,13 @@ var init_ProbeAgent = __esm({
         if (this.mcpBridge && !options._disableTools) {
           const mcpTools = this.mcpBridge.getVercelTools(this._filterMcpTools(this.mcpBridge.getToolNames()));
           for (const [name14, mcpTool] of Object.entries(mcpTools)) {
-            nativeTools[name14] = mcpTool;
+            const mcpSchema = mcpTool.inputSchema || mcpTool.parameters;
+            const wrappedSchema = mcpSchema && mcpSchema._def ? mcpSchema : (0, import_ai6.jsonSchema)(mcpSchema || { type: "object", properties: {} });
+            nativeTools[name14] = (0, import_ai6.tool)({
+              description: mcpTool.description || `MCP tool: ${name14}`,
+              inputSchema: wrappedSchema,
+              execute: mcpTool.execute
+            });
           }
         }
         if (this.apiType === "google" && this._geminiToolsEnabled && !options._disableTools) {

package/cjs/index.cjs

@@ -107433,7 +107433,13 @@ var init_ProbeAgent = __esm({
         if (this.mcpBridge && !options._disableTools) {
           const mcpTools = this.mcpBridge.getVercelTools(this._filterMcpTools(this.mcpBridge.getToolNames()));
           for (const [name14, mcpTool] of Object.entries(mcpTools)) {
-            nativeTools[name14] = mcpTool;
+            const mcpSchema = mcpTool.inputSchema || mcpTool.parameters;
+            const wrappedSchema = mcpSchema && mcpSchema._def ? mcpSchema : (0, import_ai4.jsonSchema)(mcpSchema || { type: "object", properties: {} });
+            nativeTools[name14] = (0, import_ai4.tool)({
+              description: mcpTool.description || `MCP tool: ${name14}`,
+              inputSchema: wrappedSchema,
+              execute: mcpTool.execute
+            });
           }
         }
         if (this.apiType === "google" && this._geminiToolsEnabled && !options._disableTools) {

package/package.json

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

package/src/agent/ProbeAgent.js

@@ -1879,7 +1879,15 @@ export class ProbeAgent {
     if (this.mcpBridge && !options._disableTools) {
       const mcpTools = this.mcpBridge.getVercelTools(this._filterMcpTools(this.mcpBridge.getToolNames()));
       for (const [name, mcpTool] of Object.entries(mcpTools)) {
-        nativeTools[name] = mcpTool;
+        // MCP tools have raw JSON Schema inputSchema that must be wrapped with jsonSchema()
+        // for the Vercel AI SDK. Without wrapping, asSchema() misidentifies them as Zod schemas.
+        const mcpSchema = mcpTool.inputSchema || mcpTool.parameters;
+        const wrappedSchema = mcpSchema && mcpSchema._def ? mcpSchema : jsonSchema(mcpSchema || { type: 'object', properties: {} });
+        nativeTools[name] = tool({
+          description: mcpTool.description || `MCP tool: ${name}`,
+          inputSchema: wrappedSchema,
+          execute: mcpTool.execute,
+        });
       }
     }