@probelabs/probe 0.6.0-rc118 → 0.6.0-rc120

package/build/agent/mcp/client.js

@@ -242,11 +242,25 @@ export class MCPClientManager {
         console.error(`[MCP] Calling ${toolName} with args:`, args);
       }
 
-      const result = await clientInfo.client.callTool({
-        name: tool.originalName,
-        arguments: args
+      // Get timeout from config (default 30 seconds)
+      const timeout = this.config?.settings?.timeout || 30000;
+
+      // Create a timeout promise
+      const timeoutPromise = new Promise((_, reject) => {
+        setTimeout(() => {
+          reject(new Error(`MCP tool call timeout after ${timeout}ms`));
+        }, timeout);
       });
 
+      // Race between the actual call and timeout
+      const result = await Promise.race([
+        clientInfo.client.callTool({
+          name: tool.originalName,
+          arguments: args
+        }),
+        timeoutPromise
+      ]);
+
       return result;
     } catch (error) {
       console.error(`[MCP] Error calling tool ${toolName}:`, error);

package/build/agent/schemaUtils.js

@@ -562,15 +562,18 @@ export async function validateMermaidDiagram(diagram) {
     const result = validate(diagram);
 
     // Maid returns { type: string, errors: array }
-    // Valid if errors array is empty
-    if (result.errors && result.errors.length === 0) {
+    // Only count actual errors (severity: 'error'), not warnings
+    const actualErrors = (result.errors || []).filter(err => err.severity === 'error');
+
+    // Valid if no actual errors (warnings are OK)
+    if (actualErrors.length === 0) {
       return {
         isValid: true,
         diagramType: result.type || 'unknown'
       };
     } else {
       // Format maid errors into a readable error message
-      const errorMessages = (result.errors || []).map(err => {
+      const errorMessages = actualErrors.map(err => {
         const location = err.line ? `line ${err.line}${err.column ? `:${err.column}` : ''}` : '';
         return location ? `${location} - ${err.message}` : err.message;
       });
@@ -580,7 +583,7 @@ export async function validateMermaidDiagram(diagram) {
         diagramType: result.type || 'unknown',
         error: errorMessages[0] || 'Validation failed',
         detailedError: errorMessages.join('\n'),
-        errors: result.errors || [] // Include raw maid errors for AI fixing
+        errors: actualErrors // Include only actual errors for AI fixing
       };
     }