@datalyr/wizard 1.0.7 → 1.0.8

package/dist/bin/wizard.js

@@ -2365,7 +2365,7 @@ async function executeAgent(params) {
     { role: "user", content: initialMessage }
   ];
   const filesModified = [];
-  const maxIterations = 20;
+  const maxIterations = 30;
   let iterations = 0;
   while (iterations < maxIterations) {
     iterations++;
@@ -2537,6 +2537,73 @@ async function executeTool(toolUse, cwd, debug) {
         });
         return { content: files.slice(0, 100).join("\n") };
       }
+      case "edit_file": {
+        if (typeof input.path !== "string" || !input.path.trim()) {
+          return { content: "Error: path must be a non-empty string", is_error: true };
+        }
+        if (typeof input.old_string !== "string") {
+          return { content: "Error: old_string must be a string", is_error: true };
+        }
+        if (typeof input.new_string !== "string") {
+          return { content: "Error: new_string must be a string", is_error: true };
+        }
+        const path3 = input.path.trim();
+        const oldString = input.old_string;
+        const newString = input.new_string;
+        if (!isPathSafe(cwd, path3)) {
+          return { content: "Error: path traversal detected - access denied", is_error: true };
+        }
+        const fullPath = (0, import_path3.join)(cwd, path3);
+        const content = await (0, import_promises.readFile)(fullPath, "utf-8");
+        const occurrences = content.split(oldString).length - 1;
+        if (occurrences === 0) {
+          return { content: `Error: old_string not found in ${path3}`, is_error: true };
+        }
+        if (occurrences > 1) {
+          return { content: `Error: old_string found ${occurrences} times in ${path3} - must be unique`, is_error: true };
+        }
+        const newContent = content.replace(oldString, newString);
+        await (0, import_promises.writeFile)(fullPath, newContent, "utf-8");
+        return { content: `Successfully edited ${path3}` };
+      }
+      case "search_files": {
+        if (typeof input.pattern !== "string" || !input.pattern.trim()) {
+          return { content: "Error: pattern must be a non-empty string", is_error: true };
+        }
+        const searchPattern = input.pattern.trim();
+        const searchPath = typeof input.path === "string" ? input.path.trim() : ".";
+        const filePattern = typeof input.file_pattern === "string" ? input.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 = (0, import_path3.join)(cwd, searchPath);
+        const files = await (0, import_glob2.glob)(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 = (0, import_path3.join)(fullPath, file);
+            const content = await (0, import_promises.readFile)(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 };
     }