@datalyr/wizard 1.0.7 → 1.0.8

package/dist/index.mjs

@@ -3309,7 +3309,7 @@ async function executeAgent(params) {
     { role: "user", content: initialMessage }
   ];
   const filesModified = [];
-  const maxIterations = 20;
+  const maxIterations = 30;
   let iterations = 0;
   while (iterations < maxIterations) {
     iterations++;
@@ -3481,6 +3481,73 @@ async function executeTool(toolUse, cwd, debug) {
         });
         return { content: files.slice(0, 100).join("\n") };
       }
+      case "edit_file": {
+        if (typeof input2.path !== "string" || !input2.path.trim()) {
+          return { content: "Error: path must be a non-empty string", is_error: true };
+        }
+        if (typeof input2.old_string !== "string") {
+          return { content: "Error: old_string must be a string", is_error: true };
+        }
+        if (typeof input2.new_string !== "string") {
+          return { content: "Error: new_string must be a string", is_error: true };
+        }
+        const path6 = input2.path.trim();
+        const oldString = input2.old_string;
+        const newString = input2.new_string;
+        if (!isPathSafe(cwd, path6)) {
+          return { content: "Error: path traversal detected - access denied", is_error: true };
+        }
+        const fullPath = join(cwd, path6);
+        const content = await readFile2(fullPath, "utf-8");
+        const occurrences = content.split(oldString).length - 1;
+        if (occurrences === 0) {
+          return { content: `Error: old_string not found in ${path6}`, is_error: true };
+        }
+        if (occurrences > 1) {
+          return { content: `Error: old_string found ${occurrences} times in ${path6} - must be unique`, is_error: true };
+        }
+        const newContent = content.replace(oldString, newString);
+        await writeFile2(fullPath, newContent, "utf-8");
+        return { content: `Successfully edited ${path6}` };
+      }
+      case "search_files": {
+        if (typeof input2.pattern !== "string" || !input2.pattern.trim()) {
+          return { content: "Error: pattern must be a non-empty string", is_error: true };
+        }
+        const searchPattern = input2.pattern.trim();
+        const searchPath = typeof input2.path === "string" ? input2.path.trim() : ".";
+        const filePattern = typeof input2.file_pattern === "string" ? input2.file_pattern : "**/*.{ts,tsx,js,jsx,json,md}";
+        if (!isPathSafe(cwd, searchPath)) {
+          return { content: "Error: path traversal detected - access denied", is_error: true };
+        }
+        const fullPath = join(cwd, searchPath);
+        const files = await glob2(filePattern, {
+          cwd: fullPath,
+          nodir: true,
+          maxDepth: 5
+        });
+        const results = [];
+        const regex = new RegExp(searchPattern, "gi");
+        for (const file of files.slice(0, 50)) {
+          try {
+            const filePath = join(fullPath, file);
+            const content = await readFile2(filePath, "utf-8");
+            const lines = content.split("\n");
+            lines.forEach((line, index) => {
+              if (regex.test(line)) {
+                results.push(`${file}:${index + 1}: ${line.trim().slice(0, 100)}`);
+              }
+              regex.lastIndex = 0;
+            });
+          } catch {
+          }
+          if (results.length >= 50) break;
+        }
+        if (results.length === 0) {
+          return { content: `No matches found for "${searchPattern}"` };
+        }
+        return { content: results.join("\n") };
+      }
       default:
         return { content: `Unknown tool: ${name}`, is_error: true };
     }