@probelabs/probe 0.6.0-rc116 → 0.6.0-rc117

package/cjs/index.cjs

@@ -30339,14 +30339,53 @@ var init_probeTool = __esm({
         if (!targetDir.startsWith(secureBaseDir + import_path7.default.sep) && targetDir !== secureBaseDir) {
           throw new Error("Path traversal attempt detected. Access denied.");
         }
+        const debug = process.env.DEBUG === "1";
+        if (debug) {
+          console.log(`[DEBUG] Listing files in directory: ${targetDir}`);
+        }
         try {
-          const files = await listFilesByLevel({
-            directory: targetDir,
-            maxFiles: 100,
-            respectGitignore: !process.env.PROBE_NO_GITIGNORE || process.env.PROBE_NO_GITIGNORE === "",
-            cwd: secureBaseDir
+          const files = await import_fs4.promises.readdir(targetDir, { withFileTypes: true });
+          const formatSize = (size) => {
+            if (size < 1024) return `${size}B`;
+            if (size < 1024 * 1024) return `${(size / 1024).toFixed(1)}K`;
+            if (size < 1024 * 1024 * 1024) return `${(size / (1024 * 1024)).toFixed(1)}M`;
+            return `${(size / (1024 * 1024 * 1024)).toFixed(1)}G`;
+          };
+          const entries = await Promise.all(files.map(async (file) => {
+            const isDirectory = file.isDirectory();
+            const fullPath = import_path7.default.join(targetDir, file.name);
+            let size = 0;
+            try {
+              const stats = await import_fs4.promises.stat(fullPath);
+              size = stats.size;
+            } catch (statError) {
+              if (debug) {
+                console.log(`[DEBUG] Could not stat file ${file.name}:`, statError.message);
+              }
+            }
+            return {
+              name: file.name,
+              isDirectory,
+              size
+            };
+          }));
+          entries.sort((a3, b3) => {
+            if (a3.isDirectory && !b3.isDirectory) return -1;
+            if (!a3.isDirectory && b3.isDirectory) return 1;
+            return a3.name.localeCompare(b3.name);
           });
-          return files;
+          const formatted = entries.map((entry) => {
+            const type = entry.isDirectory ? "dir " : "file";
+            const sizeStr = formatSize(entry.size).padStart(8);
+            return `${type} ${sizeStr}  ${entry.name}`;
+          });
+          if (debug) {
+            console.log(`[DEBUG] Found ${entries.length} files/directories in ${targetDir}`);
+          }
+          const header = `${targetDir}:
+`;
+          const output = header + formatted.join("\n");
+          return output;
         } catch (error2) {
           throw new Error(`Failed to list files: ${error2.message}`);
         }
@@ -30364,6 +30403,9 @@ var init_probeTool = __esm({
         if (!targetDir.startsWith(secureBaseDir + import_path7.default.sep) && targetDir !== secureBaseDir) {
           throw new Error("Path traversal attempt detected. Access denied.");
         }
+        if (pattern.includes("**/**") || pattern.split("*").length > 10) {
+          throw new Error("Pattern too complex. Please use a simpler glob pattern.");
+        }
         try {
           const options = {
             cwd: targetDir,
@@ -30373,7 +30415,17 @@ var init_probeTool = __esm({
           if (!recursive) {
             options.deep = 1;
           }
-          const files = await (0, import_glob.glob)(pattern, options);
+          const timeoutPromise = new Promise((_2, reject2) => {
+            setTimeout(() => reject2(new Error("Search operation timed out after 10 seconds")), 1e4);
+          });
+          const files = await Promise.race([
+            (0, import_glob.glob)(pattern, options),
+            timeoutPromise
+          ]);
+          const maxResults = 1e3;
+          if (files.length > maxResults) {
+            return files.slice(0, maxResults);
+          }
           return files;
         } catch (error2) {
           throw new Error(`Failed to search files: ${error2.message}`);
@@ -57777,8 +57829,8 @@ When presented with a broken Mermaid diagram, analyze it thoroughly and provide
             debug: this.options.debug,
             tracer: this.options.tracer,
             allowEdit: this.options.allowEdit,
-            maxIterations: 2,
-            // Limit mermaid fixing to 2 iterations to prevent long loops
+            maxIterations: 10,
+            // Allow more iterations for mermaid fixing to handle complex diagrams
             disableMermaidValidation: true
             // CRITICAL: Disable mermaid validation in nested agent to prevent infinite recursion
           });
@@ -58678,6 +58730,18 @@ var init_ProbeAgent = __esm({
           this.toolImplementations.bash = wrappedTools.bashToolInstance;
         }
         this.wrappedTools = wrappedTools;
+        if (this.debug) {
+          console.error("\n[DEBUG] ========================================");
+          console.error("[DEBUG] ProbeAgent Tools Initialized");
+          console.error("[DEBUG] Session ID:", this.sessionId);
+          console.error("[DEBUG] Available tools:");
+          for (const toolName of Object.keys(this.toolImplementations)) {
+            console.error(`[DEBUG]   - ${toolName}`);
+          }
+          console.error("[DEBUG] Allowed folders:", this.allowedFolders);
+          console.error("[DEBUG] Outline mode:", this.outline);
+          console.error("[DEBUG] ========================================\n");
+        }
       }
       /**
        * Initialize the AI model based on available API keys and forced provider setting
@@ -59448,12 +59512,28 @@ You are working with a repository located at: ${searchDirectory}
                 const { type } = parsedTool;
                 if (type === "mcp" && this.mcpBridge && this.mcpBridge.isMcpTool(toolName)) {
                   try {
-                    if (this.debug) console.log(`[DEBUG] Executing MCP tool '${toolName}' with params:`, params);
+                    if (this.debug) {
+                      console.error(`
+[DEBUG] ========================================`);
+                      console.error(`[DEBUG] Executing MCP tool: ${toolName}`);
+                      console.error(`[DEBUG] Arguments:`);
+                      for (const [key, value] of Object.entries(params)) {
+                        const displayValue = typeof value === "string" && value.length > 100 ? value.substring(0, 100) + "..." : value;
+                        console.error(`[DEBUG]   ${key}: ${JSON.stringify(displayValue)}`);
+                      }
+                      console.error(`[DEBUG] ========================================
+`);
+                    }
                     const executionResult = await this.mcpBridge.mcpTools[toolName].execute(params);
                     const toolResultContent = typeof executionResult === "string" ? executionResult : JSON.stringify(executionResult, null, 2);
-                    const preview = createMessagePreview(toolResultContent);
                     if (this.debug) {
-                      console.log(`[DEBUG] MCP tool '${toolName}' executed successfully. Result preview: ${preview}`);
+                      const preview = toolResultContent.length > 500 ? toolResultContent.substring(0, 500) + "..." : toolResultContent;
+                      console.error(`[DEBUG] ========================================`);
+                      console.error(`[DEBUG] MCP tool '${toolName}' completed successfully`);
+                      console.error(`[DEBUG] Result preview:`);
+                      console.error(preview);
+                      console.error(`[DEBUG] ========================================
+`);
                     }
                     currentMessages.push({ role: "user", content: `<tool_result>
 ${toolResultContent}
@@ -59461,7 +59541,13 @@ ${toolResultContent}
                   } catch (error2) {
                     console.error(`Error executing MCP tool ${toolName}:`, error2);
                     const toolResultContent = `Error executing MCP tool ${toolName}: ${error2.message}`;
-                    if (this.debug) console.log(`[DEBUG] MCP tool '${toolName}' execution FAILED.`);
+                    if (this.debug) {
+                      console.error(`[DEBUG] ========================================`);
+                      console.error(`[DEBUG] MCP tool '${toolName}' failed with error:`);
+                      console.error(`[DEBUG] ${error2.message}`);
+                      console.error(`[DEBUG] ========================================
+`);
+                    }
                     currentMessages.push({ role: "user", content: `<tool_result>
 ${toolResultContent}
 </tool_result>` });
@@ -59473,6 +59559,18 @@ ${toolResultContent}
                       sessionId: this.sessionId,
                       workingDirectory: this.allowedFolders && this.allowedFolders[0] || process.cwd()
                     };
+                    if (this.debug) {
+                      console.error(`
+[DEBUG] ========================================`);
+                      console.error(`[DEBUG] Executing tool: ${toolName}`);
+                      console.error(`[DEBUG] Arguments:`);
+                      for (const [key, value] of Object.entries(params)) {
+                        const displayValue = typeof value === "string" && value.length > 100 ? value.substring(0, 100) + "..." : value;
+                        console.error(`[DEBUG]   ${key}: ${JSON.stringify(displayValue)}`);
+                      }
+                      console.error(`[DEBUG] ========================================
+`);
+                    }
                     this.events.emit("toolCall", {
                       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
                       name: toolName,
@@ -59514,6 +59612,15 @@ ${toolResultContent}
                       } else {
                         toolResult = await executeToolCall();
                       }
+                      if (this.debug) {
+                        const resultPreview = typeof toolResult === "string" ? toolResult.length > 500 ? toolResult.substring(0, 500) + "..." : toolResult : toolResult ? JSON.stringify(toolResult, null, 2).substring(0, 500) + "..." : "No Result";
+                        console.error(`[DEBUG] ========================================`);
+                        console.error(`[DEBUG] Tool '${toolName}' completed successfully`);
+                        console.error(`[DEBUG] Result preview:`);
+                        console.error(resultPreview);
+                        console.error(`[DEBUG] ========================================
+`);
+                      }
                       this.events.emit("toolCall", {
                         timestamp: (/* @__PURE__ */ new Date()).toISOString(),
                         name: toolName,
@@ -59522,6 +59629,13 @@ ${toolResultContent}
                         status: "completed"
                       });
                     } catch (toolError) {
+                      if (this.debug) {
+                        console.error(`[DEBUG] ========================================`);
+                        console.error(`[DEBUG] Tool '${toolName}' failed with error:`);
+                        console.error(`[DEBUG] ${toolError.message}`);
+                        console.error(`[DEBUG] ========================================
+`);
+                      }
                       this.events.emit("toolCall", {
                         timestamp: (/* @__PURE__ */ new Date()).toISOString(),
                         name: toolName,
@@ -59571,6 +59685,16 @@ Error: Unknown tool '${toolName}'. Available tools: ${allAvailableTools.join(",
                 }
               }
             } else {
+              const hasMermaidCodeBlock = /```mermaid\s*\n[\s\S]*?\n```/.test(assistantResponseContent);
+              const hasNoSchemaOrTools = !options.schema && validTools.length === 0;
+              if (hasMermaidCodeBlock && hasNoSchemaOrTools) {
+                finalResult = assistantResponseContent;
+                completionAttempted = true;
+                if (this.debug) {
+                  console.error(`[DEBUG] Accepting mermaid code block as valid completion (no schema, no tools)`);
+                }
+                break;
+              }
               currentMessages.push({ role: "assistant", content: assistantResponseContent });
               let reminderContent;
               if (options.schema) {
@@ -60284,12 +60408,14 @@ __export(index_exports, {
   getBinaryPath: () => getBinaryPath,
   initializeSimpleTelemetryFromOptions: () => initializeSimpleTelemetryFromOptions,
   listFilesByLevel: () => listFilesByLevel,
+  listFilesToolInstance: () => listFilesToolInstance,
   parseXmlToolCall: () => parseXmlToolCall,
   query: () => query,
   querySchema: () => querySchema,
   queryTool: () => queryTool,
   queryToolDefinition: () => queryToolDefinition,
   search: () => search,
+  searchFilesToolInstance: () => searchFilesToolInstance,
   searchSchema: () => searchSchema,
   searchTool: () => searchTool,
   searchToolDefinition: () => searchToolDefinition,
@@ -60312,6 +60438,7 @@ var init_index = __esm({
     init_bash();
     init_ProbeAgent();
     init_simpleTelemetry();
+    init_probeTool();
   }
 });
 init_index();
@@ -60337,12 +60464,14 @@ init_index();
   getBinaryPath,
   initializeSimpleTelemetryFromOptions,
   listFilesByLevel,
+  listFilesToolInstance,
   parseXmlToolCall,
   query,
   querySchema,
   queryTool,
   queryToolDefinition,
   search,
+  searchFilesToolInstance,
   searchSchema,
   searchTool,
   searchToolDefinition,

package/package.json

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

package/src/agent/ProbeAgent.js

@@ -188,9 +188,23 @@ export class ProbeAgent {
     if (this.enableBash && wrappedTools.bashToolInstance) {
       this.toolImplementations.bash = wrappedTools.bashToolInstance;
     }
-    
+
     // Store wrapped tools for ACP system
     this.wrappedTools = wrappedTools;
+
+    // Log available tools in debug mode
+    if (this.debug) {
+      console.error('\n[DEBUG] ========================================');
+      console.error('[DEBUG] ProbeAgent Tools Initialized');
+      console.error('[DEBUG] Session ID:', this.sessionId);
+      console.error('[DEBUG] Available tools:');
+      for (const toolName of Object.keys(this.toolImplementations)) {
+        console.error(`[DEBUG]   - ${toolName}`);
+      }
+      console.error('[DEBUG] Allowed folders:', this.allowedFolders);
+      console.error('[DEBUG] Outline mode:', this.outline);
+      console.error('[DEBUG] ========================================\n');
+    }
   }
 
   /**
@@ -1117,34 +1131,74 @@ When troubleshooting:
             if (type === 'mcp' && this.mcpBridge && this.mcpBridge.isMcpTool(toolName)) {
               // Execute MCP tool
               try {
-                if (this.debug) console.log(`[DEBUG] Executing MCP tool '${toolName}' with params:`, params);
+                // Log MCP tool execution in debug mode
+                if (this.debug) {
+                  console.error(`\n[DEBUG] ========================================`);
+                  console.error(`[DEBUG] Executing MCP tool: ${toolName}`);
+                  console.error(`[DEBUG] Arguments:`);
+                  for (const [key, value] of Object.entries(params)) {
+                    const displayValue = typeof value === 'string' && value.length > 100
+                      ? value.substring(0, 100) + '...'
+                      : value;
+                    console.error(`[DEBUG]   ${key}: ${JSON.stringify(displayValue)}`);
+                  }
+                  console.error(`[DEBUG] ========================================\n`);
+                }
 
                 // Execute MCP tool through the bridge
                 const executionResult = await this.mcpBridge.mcpTools[toolName].execute(params);
 
                 const toolResultContent = typeof executionResult === 'string' ? executionResult : JSON.stringify(executionResult, null, 2);
-                const preview = createMessagePreview(toolResultContent);
+
+                // Log MCP tool result in debug mode
                 if (this.debug) {
-                  console.log(`[DEBUG] MCP tool '${toolName}' executed successfully. Result preview: ${preview}`);
+                  const preview = toolResultContent.length > 500 ? toolResultContent.substring(0, 500) + '...' : toolResultContent;
+                  console.error(`[DEBUG] ========================================`);
+                  console.error(`[DEBUG] MCP tool '${toolName}' completed successfully`);
+                  console.error(`[DEBUG] Result preview:`);
+                  console.error(preview);
+                  console.error(`[DEBUG] ========================================\n`);
                 }
 
                 currentMessages.push({ role: 'user', content: `<tool_result>\n${toolResultContent}\n</tool_result>` });
               } catch (error) {
                 console.error(`Error executing MCP tool ${toolName}:`, error);
                 const toolResultContent = `Error executing MCP tool ${toolName}: ${error.message}`;
-                if (this.debug) console.log(`[DEBUG] MCP tool '${toolName}' execution FAILED.`);
+
+                // Log MCP tool error in debug mode
+                if (this.debug) {
+                  console.error(`[DEBUG] ========================================`);
+                  console.error(`[DEBUG] MCP tool '${toolName}' failed with error:`);
+                  console.error(`[DEBUG] ${error.message}`);
+                  console.error(`[DEBUG] ========================================\n`);
+                }
+
                 currentMessages.push({ role: 'user', content: `<tool_result>\n${toolResultContent}\n</tool_result>` });
               }
             } else if (this.toolImplementations[toolName]) {
               // Execute native tool
               try {
                 // Add sessionId and workingDirectory to params for tool execution
-                const toolParams = { 
-                  ...params, 
+                const toolParams = {
+                  ...params,
                   sessionId: this.sessionId,
                   workingDirectory: (this.allowedFolders && this.allowedFolders[0]) || process.cwd()
                 };
-                
+
+                // Log tool execution in debug mode
+                if (this.debug) {
+                  console.error(`\n[DEBUG] ========================================`);
+                  console.error(`[DEBUG] Executing tool: ${toolName}`);
+                  console.error(`[DEBUG] Arguments:`);
+                  for (const [key, value] of Object.entries(params)) {
+                    const displayValue = typeof value === 'string' && value.length > 100
+                      ? value.substring(0, 100) + '...'
+                      : value;
+                    console.error(`[DEBUG]   ${key}: ${JSON.stringify(displayValue)}`);
+                  }
+                  console.error(`[DEBUG] ========================================\n`);
+                }
+
                 // Emit tool start event
                 this.events.emit('toolCall', {
                   timestamp: new Date().toISOString(),
@@ -1196,6 +1250,18 @@ When troubleshooting:
                     toolResult = await executeToolCall();
                   }
                   
+                  // Log tool result in debug mode
+                  if (this.debug) {
+                    const resultPreview = typeof toolResult === 'string'
+                      ? (toolResult.length > 500 ? toolResult.substring(0, 500) + '...' : toolResult)
+                      : (toolResult ? JSON.stringify(toolResult, null, 2).substring(0, 500) + '...' : 'No Result');
+                    console.error(`[DEBUG] ========================================`);
+                    console.error(`[DEBUG] Tool '${toolName}' completed successfully`);
+                    console.error(`[DEBUG] Result preview:`);
+                    console.error(resultPreview);
+                    console.error(`[DEBUG] ========================================\n`);
+                  }
+
                   // Emit tool success event
                   this.events.emit('toolCall', {
                     timestamp: new Date().toISOString(),
@@ -1206,8 +1272,16 @@ When troubleshooting:
                       : (toolResult ? JSON.stringify(toolResult).substring(0, 200) + '...' : 'No Result'),
                     status: 'completed'
                   });
-                  
+
                 } catch (toolError) {
+                  // Log tool error in debug mode
+                  if (this.debug) {
+                    console.error(`[DEBUG] ========================================`);
+                    console.error(`[DEBUG] Tool '${toolName}' failed with error:`);
+                    console.error(`[DEBUG] ${toolError.message}`);
+                    console.error(`[DEBUG] ========================================\n`);
+                  }
+
                   // Emit tool error event
                   this.events.emit('toolCall', {
                     timestamp: new Date().toISOString(),
@@ -1262,7 +1336,23 @@ When troubleshooting:
             }
           }
         } else {
-          // No tool call found, add assistant response and ask for tool usage
+          // No tool call found
+          // Special case: If response contains a mermaid code block and no schema was provided,
+          // treat it as a valid completion (for mermaid diagram fixing workflow)
+          const hasMermaidCodeBlock = /```mermaid\s*\n[\s\S]*?\n```/.test(assistantResponseContent);
+          const hasNoSchemaOrTools = !options.schema && validTools.length === 0;
+
+          if (hasMermaidCodeBlock && hasNoSchemaOrTools) {
+            // Accept mermaid code block as final answer for diagram fixing
+            finalResult = assistantResponseContent;
+            completionAttempted = true;
+            if (this.debug) {
+              console.error(`[DEBUG] Accepting mermaid code block as valid completion (no schema, no tools)`);
+            }
+            break;
+          }
+
+          // Add assistant response and ask for tool usage
           currentMessages.push({ role: 'assistant', content: assistantResponseContent });
 
           // Build appropriate reminder message based on whether schema is provided

package/src/agent/probeTool.js

@@ -213,45 +213,99 @@ export function createWrappedTools(baseTools) {
   return wrappedTools;
 }
 
-// Simple file listing tool
+// Simple file listing tool with ls-like formatted output
 export const listFilesTool = {
   execute: async (params) => {
     const { directory = '.', workingDirectory } = params;
-    
+
     // Use the provided working directory, or fall back to process.cwd()
     const baseCwd = workingDirectory || process.cwd();
-    
+
     // Security: Validate path to prevent traversal attacks
     const secureBaseDir = path.resolve(baseCwd);
     const targetDir = path.resolve(secureBaseDir, directory);
     if (!targetDir.startsWith(secureBaseDir + path.sep) && targetDir !== secureBaseDir) {
       throw new Error('Path traversal attempt detected. Access denied.');
     }
-    
+
+    const debug = process.env.DEBUG === '1';
+
+    if (debug) {
+      console.log(`[DEBUG] Listing files in directory: ${targetDir}`);
+    }
+
     try {
-      const files = await listFilesByLevel({
-        directory: targetDir,
-        maxFiles: 100,
-        respectGitignore: !process.env.PROBE_NO_GITIGNORE || process.env.PROBE_NO_GITIGNORE === '',
-        cwd: secureBaseDir
+      // Read the directory contents
+      const files = await fsPromises.readdir(targetDir, { withFileTypes: true });
+
+      // Format size for human readability
+      const formatSize = (size) => {
+        if (size < 1024) return `${size}B`;
+        if (size < 1024 * 1024) return `${(size / 1024).toFixed(1)}K`;
+        if (size < 1024 * 1024 * 1024) return `${(size / (1024 * 1024)).toFixed(1)}M`;
+        return `${(size / (1024 * 1024 * 1024)).toFixed(1)}G`;
+      };
+
+      // Format the results as ls-style output
+      const entries = await Promise.all(files.map(async (file) => {
+        const isDirectory = file.isDirectory();
+        const fullPath = path.join(targetDir, file.name);
+
+        let size = 0;
+        try {
+          const stats = await fsPromises.stat(fullPath);
+          size = stats.size;
+        } catch (statError) {
+          if (debug) {
+            console.log(`[DEBUG] Could not stat file ${file.name}:`, statError.message);
+          }
+        }
+
+        return {
+          name: file.name,
+          isDirectory,
+          size
+        };
+      }));
+
+      // Sort: directories first, then files, both alphabetically
+      entries.sort((a, b) => {
+        if (a.isDirectory && !b.isDirectory) return -1;
+        if (!a.isDirectory && b.isDirectory) return 1;
+        return a.name.localeCompare(b.name);
       });
-      
-      return files;
+
+      // Format entries
+      const formatted = entries.map(entry => {
+        const type = entry.isDirectory ? 'dir ' : 'file';
+        const sizeStr = formatSize(entry.size).padStart(8);
+        return `${type} ${sizeStr}  ${entry.name}`;
+      });
+
+      if (debug) {
+        console.log(`[DEBUG] Found ${entries.length} files/directories in ${targetDir}`);
+      }
+
+      // Return as formatted text output
+      const header = `${targetDir}:\n`;
+      const output = header + formatted.join('\n');
+
+      return output;
     } catch (error) {
       throw new Error(`Failed to list files: ${error.message}`);
     }
   }
 };
 
-// Simple file search tool
+// Simple file search tool with timeout protection
 export const searchFilesTool = {
   execute: async (params) => {
     const { pattern, directory = '.', recursive = true, workingDirectory } = params;
-    
+
     if (!pattern) {
       throw new Error('Pattern is required for file search');
     }
-    
+
     // Security: Validate path to prevent traversal attacks
     const baseCwd = workingDirectory || process.cwd();
     const secureBaseDir = path.resolve(baseCwd);
@@ -259,19 +313,40 @@ export const searchFilesTool = {
     if (!targetDir.startsWith(secureBaseDir + path.sep) && targetDir !== secureBaseDir) {
       throw new Error('Path traversal attempt detected. Access denied.');
     }
-    
+
+    // Validate pattern complexity to prevent DoS
+    if (pattern.includes('**/**') || pattern.split('*').length > 10) {
+      throw new Error('Pattern too complex. Please use a simpler glob pattern.');
+    }
+
     try {
       const options = {
         cwd: targetDir,
         ignore: ['node_modules/**', '.git/**'],
         absolute: false
       };
-      
+
       if (!recursive) {
         options.deep = 1;
       }
-      
-      const files = await glob(pattern, options);
+
+      // Create a timeout promise (10 seconds)
+      const timeoutPromise = new Promise((_, reject) => {
+        setTimeout(() => reject(new Error('Search operation timed out after 10 seconds')), 10000);
+      });
+
+      // Race glob against timeout
+      const files = await Promise.race([
+        glob(pattern, options),
+        timeoutPromise
+      ]);
+
+      // Limit results to prevent memory issues
+      const maxResults = 1000;
+      if (files.length > maxResults) {
+        return files.slice(0, maxResults);
+      }
+
       return files;
     } catch (error) {
       throw new Error(`Failed to search files: ${error.message}`);

package/src/agent/schemaUtils.js

@@ -813,7 +813,7 @@ When presented with a broken Mermaid diagram, analyze it thoroughly and provide
         debug: this.options.debug,
         tracer: this.options.tracer,
         allowEdit: this.options.allowEdit,
-        maxIterations: 2,  // Limit mermaid fixing to 2 iterations to prevent long loops
+        maxIterations: 10,  // Allow more iterations for mermaid fixing to handle complex diagrams
         disableMermaidValidation: true  // CRITICAL: Disable mermaid validation in nested agent to prevent infinite recursion
       });
     }

package/src/index.js

@@ -34,6 +34,7 @@ import { searchTool, queryTool, extractTool, delegateTool } from './tools/vercel
 import { bashTool } from './tools/bash.js';
 import { ProbeAgent } from './agent/ProbeAgent.js';
 import { SimpleTelemetry, SimpleAppTracer, initializeSimpleTelemetryFromOptions } from './agent/simpleTelemetry.js';
+import { listFilesToolInstance, searchFilesToolInstance } from './agent/probeTool.js';
 
 export {
 	search,
@@ -57,6 +58,9 @@ export {
 	extractTool,
 	delegateTool,
 	bashTool,
+	// Export tool instances
+	listFilesToolInstance,
+	searchFilesToolInstance,
 	// Export schemas
 	searchSchema,
 	querySchema,