@morphllm/morphsdk 0.2.57 → 0.2.59

package/dist/tools/warp_grep/agent/runner.cjs

@@ -36,9 +36,13 @@ module.exports = __toCommonJS(runner_exports);
 
 // tools/warp_grep/agent/config.ts
 var AGENT_CONFIG = {
-  // Give the model freedom; failsafe cap to prevent infinite loops
-  MAX_ROUNDS: 10,
-  TIMEOUT_MS: 3e4
+  MAX_TURNS: 4,
+  TIMEOUT_MS: 3e4,
+  MAX_CONTEXT_CHARS: 54e4,
+  MAX_OUTPUT_LINES: 200,
+  MAX_READ_LINES: 800,
+  MAX_LIST_DEPTH: 3,
+  LIST_TIMEOUT_MS: 2e3
 };
 var BUILTIN_EXCLUDES = [
   // Version control
@@ -120,113 +124,191 @@ var BUILTIN_EXCLUDES = [
   ".*"
 ];
 var DEFAULT_EXCLUDES = (process.env.MORPH_WARP_GREP_EXCLUDE || "").split(",").map((s) => s.trim()).filter(Boolean).concat(BUILTIN_EXCLUDES);
-var DEFAULT_MODEL = "morph-warp-grep";
+var DEFAULT_MODEL = "morph-warp-grep-v1";
 
 // tools/warp_grep/agent/prompt.ts
-var SYSTEM_PROMPT = `You are a code search agent. Your task is to find all relevant code for a given query.
+var SYSTEM_PROMPT = `You are a code search agent. Your task is to find all relevant code for a given search_string.
 
-<workflow>
+### workflow
 You have exactly 4 turns. The 4th turn MUST be a \`finish\` call. Each turn allows up to 8 parallel tool calls.
 
-- Turn 1: Map the territory OR dive deep (based on query specificity)
+- Turn 1: Map the territory OR dive deep (based on search_string specificity)
 - Turn 2-3: Refine based on findings
 - Turn 4: MUST call \`finish\` with all relevant code locations
 - You MAY call \`finish\` early if confident\u2014but never before at least 1 search turn.
+- The user strongly prefers if you can call the finish tool early, but you must be correct
 
-Remember, if the task feels easy to you, it is strongly desirable to call \`finish\` early using fewer turns, but quality over speed.
-</workflow>
+Remember, if the task feels easy to you, it is strongly desirable to call 'finish' early using fewer turns, but quality over speed
 
-<tools>
-### \`analyse <path> [pattern]\`
-Directory tree or file search. Shows structure of a path, optionally filtered by regex pattern.
-- \`path\`: Required. Directory or file path (use \`.\` for repo root)
-- \`pattern\`: Optional regex to filter results
+### tools
+Tool calls use nested XML elements:
+\`\`\`xml
+<tool_name>
+  <parameter>value</parameter>
+</tool_name>
+\`\`\`
+
+### \`list_directory\`
+Directory tree view. Shows structure of a path, optionally filtered by regex pattern.
+
+Elements:
+- \`<path>\` (required): Directory path to list (use \`.\` for repo root)
+- \`<pattern>\` (optional): Regex to filter results
 
 Examples:
 \`\`\`
-analyse .
-analyse src/api
-analyse . ".*\\.ts$"
-analyse src "test.*"
+<list_directory>
+  <path>src/services</path>
+</list_directory>
+
+<list_directory>
+  <path>lib/utils</path>
+  <pattern>.*\\.(ts|js)$</pattern>
+</list_directory>
 \`\`\`
 
-### \`read <path>[:start-end]\`
-Read file contents. Line range is 1-based, inclusive.
+### \`read\`
+Read file contents. Supports multiple line ranges.
 - Returns numbered lines for easy reference
-- Omit range to read entire file
+- ALWAYS include import statements (usually lines 1-20). Better to over-include than miss context.
+
+Elements:
+- \`<path>\` (required): File path to read
+- \`<lines>\` (optional): Line ranges like "1-50,75-80,100-120" (omit to read entire file)
 
 Examples:
 \`\`\`
-read src/main.py
-read src/db/conn.py:10-50
-read package.json:1-20
+<read>
+  <path>src/main.py</path>
+</read>
+
+<read>
+  <path>src/auth.py</path>
+  <lines>1-20,45-80,150-200</lines>
+</read>
 \`\`\`
 
-### \`grep '<pattern>' <path>\`
-Ripgrep search. Finds pattern matches across files.
-- \`'<pattern>'\`: Required. Regex pattern wrapped in single quotes
-- \`<path>\`: Required. Directory or file to search (use \`.\` for repo root)
+### \`grep\`
+Search for pattern matches across files. Returns matches with 1 line of context above and below.
+- Match lines use \`:\` separator \u2192 \`filepath:linenum:content\`
+- Context lines use \`-\` separator \u2192 \`filepath-linenum-content\`
+
+Elements:
+- \`<pattern>\` (required): Search pattern (regex). Use \`(a|b)\` for OR patterns.
+- \`<sub_dir>\` (optional): Subdirectory to search in (defaults to \`.\`)
+- \`<glob>\` (optional): File pattern filter like \`*.py\` or \`*.{ts,tsx}\`
 
 Examples:
 \`\`\`
-grep 'class.*Service' src/
-grep 'def authenticate' .
-grep 'import.*from' src/components/
-grep 'TODO' .
+<grep>
+  <pattern>(authenticate|authorize|login)</pattern>
+  <sub_dir>src/auth/</sub_dir>
+</grep>
+
+<grep>
+  <pattern>class.*(Service|Controller)</pattern>
+  <glob>*.{ts,js}</glob>
+</grep>
+
+<grep>
+  <pattern>(DB_HOST|DATABASE_URL|connection)</pattern>
+  <glob>*.{py,yaml,env}</glob>
+  <sub_dir>lib/</sub_dir>
+</grep>
 \`\`\`
 
-### \`finish <file1:ranges> [file2:ranges ...]\`
-Submit final answer with all relevant code locations.
-- Include generous line ranges\u2014don't be stingy with context
-- Ranges are comma-separated: \`file.py:10-30,50-60\`
-- ALWAYS include import statements at the top of files (usually lines 1-20)
-- If code spans multiple files, include ALL of them
-- Small files can be returned in full
+### \`finish\`
+Submit final answer with all relevant code locations. Uses nested \`<file>\` elements.
+
+File elements:
+- \`<path>\` (required): File path
+- \`<lines>\` (optional): Line ranges like "1-50,75-80" (\`*\` for entire file)
+
+ALWAYS include import statements (usually lines 1-20). Better to over-include than miss context.
 
 Examples:
 \`\`\`
-finish src/auth.py:1-15,25-50,75-80 src/models/user.py:1-10,20-45
-finish src/index.ts:1-100
+<finish>
+  <file>
+    <path>src/auth.py</path>
+    <lines>1-15,25-50,75-80</lines>
+  </file>
+  <file>
+    <path>src/models/user.py</path>
+    <lines>*</lines>
+  </file>
+</finish>
 \`\`\`
 </tools>
 
 <strategy>
-**Before your first tool call, classify the query:**
+**Before your first tool call, classify the search_string:**
 
-| Query Type | Turn 1 Strategy | Early Finish? |
-|------------|-----------------|---------------|
-| **Specific** (function name, error string, unique identifier) | 8 parallel greps on likely paths | Often by turn 2 |
-| **Conceptual** (how does X work, where is Y handled) | analyse + 2-3 broad greps | Rarely early |
-| **Exploratory** (find all tests, list API endpoints) | analyse at multiple depths | Usually needs 3 turns |
+| Search_string Type | Round 1 Strategy | Early Finish? |
+|------------|------------------|---------------|
+| **Specific** (function name, error string, unique identifier) | 8 parallel greps on likely paths | Often by round 2 |
+| **Conceptual** (how does X work, where is Y handled) | list_directory + 2-3 broad greps | Rarely early |
+| **Exploratory** (find all tests, list API endpoints) | list_directory at multiple depths | Usually needs 3 rounds |
 
 **Parallel call patterns:**
 - **Shotgun grep**: Same pattern, 8 different directories\u2014fast coverage
 - **Variant grep**: 8 pattern variations (synonyms, naming conventions)\u2014catches inconsistent codebases
-- **Funnel**: 1 analyse + 7 greps\u2014orient and search simultaneously
+- **Funnel**: 1 list_directory + 7 greps\u2014orient and search simultaneously
 - **Deep read**: 8 reads on files you already identified\u2014gather full context fast
+
+**Tool call expectations:**
+- Low quality tool calls are ones that give back sparse information. This either means they are not well thought out and are not educated guesses OR, they are too broad and give back too many results.
+- High quality tool calls strike a balance between complexity in the tool call to exclude results we know we don't want, and how wide the search space is so that we don't miss anything. It is ok to start off with wider search spaces, but is imperative that you use your intuition from there on out and seek high quality tool calls only.
+- You are not starting blind, you have some information about root level repo structure going in, so use that to prevent making trivial repo wide queries.
+- The grep tool shows you which file path and line numbers the pattern was found in, use this information smartly when trying to read the file.
 </strategy>
 
 <output_format>
 EVERY response MUST follow this exact format:
 
 1. First, wrap your reasoning in \`<think>...</think>\` tags containing:
-   - Query classification (specific/conceptual/exploratory)
-   - Confidence estimate (can I finish in 1-2 turns?)
-   - This turn's parallel strategy
+   - Search_string classification (specific/conceptual/exploratory)
+   - Confidence estimate (can I finish in 1-2 rounds?)
+   - This round's parallel strategy
    - What signals would let me finish early?
 
-2. Then, output tool calls wrapped in \`<tool_call>...</tool_call>\` tags, one per line.
+2. Then, output up to 8 tool calls using nested XML elements.
 
 Example:
 \`\`\`
 <think>
-This is a specific query about authentication. I'll grep for auth-related patterns.
-High confidence I can finish in 2 turns if I find the auth module.
+This is a specific search_string about authentication. I'll grep for auth-related patterns.
+High confidence I can finish in 2 rounds if I find the auth module. I have already been shown the repo's structure at root
 Strategy: Shotgun grep across likely directories.
 </think>
-<tool_call>grep 'authenticate' src/</tool_call>
-<tool_call>grep 'login' src/</tool_call>
-<tool_call>analyse src/auth</tool_call>
+<grep>
+  <pattern>(authenticate|login|session)</pattern>
+  <sub_dir>src/auth/</sub_dir>
+</grep>
+<grep>
+  <pattern>(middleware|interceptor)</pattern>
+  <glob>*.{ts,js}</glob>
+</grep>
+<list_directory>
+  <path>src/auth</path>
+</list_directory>
+\`\`\`
+
+Finishing example:
+\`\`\`
+<think>
+I think I have a rough idea, but this is my last turn so I must call the finish tool regardless.
+</think>
+<finish>
+  <file>
+    <path>src/auth/login.py</path>
+    <lines>1-50</lines>
+  </file>
+  <file>
+    <path>src/middleware/session.py</path>
+    <lines>10-80</lines>
+  </file>
+</finish>
 \`\`\`
 
 No commentary outside \`<think>\`. No explanations after tool calls.
@@ -239,17 +321,111 @@ When calling \`finish\`:
 - Include any type definitions, interfaces, or constants used
 - Better to over-include than leave the user missing context
 - If unsure about boundaries, include more rather than less
-</finishing_requirements>
-
-Begin your exploration now to find code relevant to the query.`;
+</finishing_requirements>`;
 function getSystemPrompt() {
   return SYSTEM_PROMPT;
 }
 
 // tools/warp_grep/agent/parser.ts
-var VALID_COMMANDS = ["analyse", "grep", "read", "finish"];
+var VALID_COMMANDS = ["list_directory", "grep", "read", "finish"];
+function isValidCommand(name) {
+  return VALID_COMMANDS.includes(name);
+}
+function getXmlElementText(xml, tagName) {
+  const regex = new RegExp(`<${tagName}>([\\s\\S]*?)</${tagName}>`, "i");
+  const match = xml.match(regex);
+  return match ? match[1].trim() : null;
+}
+function parseNestedXmlTools(text) {
+  const tools = [];
+  const toolRegex = /<([a-z_][a-z0-9_]*)>([\s\S]*?)<\/\1>/gi;
+  let match;
+  while ((match = toolRegex.exec(text)) !== null) {
+    const rawToolName = match[1].toLowerCase();
+    const content = match[2];
+    if (!isValidCommand(rawToolName)) continue;
+    const toolName = rawToolName;
+    if (toolName === "list_directory") {
+      const path3 = getXmlElementText(content, "path");
+      const pattern = getXmlElementText(content, "pattern");
+      if (path3) {
+        tools.push({ name: "list_directory", arguments: { path: path3, pattern } });
+      }
+    } else if (toolName === "grep") {
+      const pattern = getXmlElementText(content, "pattern");
+      const subDir = getXmlElementText(content, "sub_dir");
+      const glob = getXmlElementText(content, "glob");
+      if (pattern) {
+        tools.push({
+          name: "grep",
+          arguments: {
+            pattern,
+            path: subDir || ".",
+            ...glob && { glob }
+          }
+        });
+      }
+    } else if (toolName === "read") {
+      const path3 = getXmlElementText(content, "path");
+      const linesStr = getXmlElementText(content, "lines");
+      if (path3) {
+        const args = { path: path3 };
+        if (linesStr) {
+          const ranges = [];
+          for (const rangeStr of linesStr.split(",")) {
+            const trimmed = rangeStr.trim();
+            if (!trimmed) continue;
+            const [s, e] = trimmed.split("-").map((v) => parseInt(v.trim(), 10));
+            if (Number.isFinite(s) && Number.isFinite(e)) {
+              ranges.push([s, e]);
+            } else if (Number.isFinite(s)) {
+              ranges.push([s, s]);
+            }
+          }
+          if (ranges.length === 1) {
+            args.start = ranges[0][0];
+            args.end = ranges[0][1];
+          } else if (ranges.length > 1) {
+            args.lines = ranges;
+          }
+        }
+        tools.push({ name: "read", arguments: args });
+      }
+    } else if (toolName === "finish") {
+      const fileRegex = /<file>([\s\S]*?)<\/file>/gi;
+      const files = [];
+      let fileMatch;
+      while ((fileMatch = fileRegex.exec(content)) !== null) {
+        const fileContent = fileMatch[1];
+        const filePath = getXmlElementText(fileContent, "path");
+        const linesStr = getXmlElementText(fileContent, "lines");
+        if (filePath && linesStr) {
+          const ranges = [];
+          for (const rangeStr of linesStr.split(",")) {
+            if (rangeStr.trim() === "*") {
+              ranges.push([1, 999999]);
+            } else {
+              const [s, e] = rangeStr.split("-").map((v) => parseInt(v.trim(), 10));
+              if (Number.isFinite(s) && Number.isFinite(e)) {
+                ranges.push([s, e]);
+              }
+            }
+          }
+          if (ranges.length > 0) {
+            files.push({ path: filePath, lines: ranges });
+          }
+        }
+      }
+      if (files.length > 0) {
+        tools.push({ name: "finish", arguments: { files } });
+      }
+    }
+  }
+  return tools;
+}
 function preprocessText(text) {
   let processed = text.replace(/<think>[\s\S]*?<\/think>/gi, "");
+  const nestedTools = parseNestedXmlTools(processed);
   const openingTagRegex = /<tool_call>|<tool>/gi;
   const closingTagRegex = /<\/tool_call>|<\/tool>/gi;
   const openingMatches = processed.match(openingTagRegex) || [];
@@ -286,7 +462,7 @@ function preprocessText(text) {
       }
     }
   }
-  return toolCallLines;
+  return { lines: toolCallLines, nestedTools };
 }
 var LLMResponseParser = class {
   finishSpecSplitRe = /,(?=[^,\s]+:)/;
@@ -294,8 +470,8 @@ var LLMResponseParser = class {
     if (typeof text !== "string") {
       throw new TypeError("Command text must be a string.");
     }
-    const lines = preprocessText(text);
-    const commands = [];
+    const { lines, nestedTools } = preprocessText(text);
+    const commands = [...nestedTools];
     let finishAccumulator = null;
     lines.forEach((line) => {
       if (!line || line.startsWith("#")) return;
@@ -303,8 +479,8 @@ var LLMResponseParser = class {
       if (parts.length === 0) return;
       const cmd = parts[0];
       switch (cmd) {
-        case "analyse":
-          this.handleAnalyse(parts, line, commands);
+        case "list_directory":
+          this.handleListDirectory(parts, line, commands);
           break;
         case "grep":
           this.handleGrep(parts, line, commands);
@@ -322,8 +498,8 @@ var LLMResponseParser = class {
     if (finishAccumulator) {
       const map = finishAccumulator;
       const entries = [...map.entries()];
-      const filesPayload = entries.map(([path2, ranges]) => ({
-        path: path2,
+      const filesPayload = entries.map(([path3, ranges]) => ({
+        path: path3,
         lines: [...ranges].sort((a, b) => a[0] - b[0])
       }));
       commands.push({ name: "finish", arguments: { files: filesPayload } });
@@ -355,18 +531,17 @@ var LLMResponseParser = class {
   skip(message) {
     return { name: "_skip", arguments: { message } };
   }
-  handleAnalyse(parts, rawLine, commands) {
+  handleListDirectory(parts, rawLine, commands) {
     if (parts.length < 2) {
       commands.push(this.skip(
-        `[SKIPPED] Your command "${rawLine}" is missing a path. Correct format: analyse <path> [pattern]. Example: analyse src/`
+        `[SKIPPED] Your command "${rawLine}" is missing a path. Correct format: list_directory <path> [pattern]. Example: list_directory src/`
       ));
       return;
     }
-    const path2 = parts[1];
+    const path3 = parts[1];
     const pattern = parts[2]?.replace(/^"|"$/g, "") ?? null;
-    commands.push({ name: "analyse", arguments: { path: path2, pattern } });
+    commands.push({ name: "list_directory", arguments: { path: path3, pattern } });
   }
-  // no glob tool in MCP
   handleGrep(parts, rawLine, commands) {
     if (parts.length < 3) {
       commands.push(this.skip(
@@ -437,8 +612,30 @@ var LLMResponseParser = class {
   }
 };
 
-// tools/warp_grep/tools/read.ts
+// tools/warp_grep/agent/tools/grep.ts
+async function toolGrep(provider, args) {
+  const res = await provider.grep(args);
+  if (res.error) {
+    return { output: res.error };
+  }
+  if (!res.lines.length) {
+    return { output: "no matches" };
+  }
+  return { output: res.lines.join("\n") };
+}
+
+// tools/warp_grep/agent/tools/read.ts
 async function toolRead(provider, args) {
+  if (args.lines && args.lines.length > 0) {
+    const chunks = [];
+    for (const [start, end] of args.lines) {
+      const res2 = await provider.read({ path: args.path, start, end });
+      if (res2.error) return res2.error;
+      chunks.push(res2.lines.join("\n"));
+    }
+    if (chunks.every((c) => c === "")) return "(empty file)";
+    return chunks.join("\n...\n");
+  }
   const res = await provider.read({ path: args.path, start: args.start, end: args.end });
   if (res.error) {
     return res.error;
@@ -447,16 +644,56 @@ async function toolRead(provider, args) {
   return res.lines.join("\n");
 }
 
-// tools/warp_grep/tools/analyse.ts
-async function toolAnalyse(provider, args) {
-  const list = await provider.analyse({
+// tools/warp_grep/agent/tools/list_directory.ts
+async function toolListDirectory(provider, args) {
+  const list = await provider.listDirectory({
     path: args.path,
     pattern: args.pattern ?? null,
-    maxResults: args.maxResults ?? 100,
-    maxDepth: args.maxDepth ?? 2
+    maxResults: args.maxResults ?? AGENT_CONFIG.MAX_OUTPUT_LINES,
+    maxDepth: args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH
   });
   if (!list.length) return "empty";
-  return list.map((e) => `${"  ".repeat(e.depth)}- ${e.type === "dir" ? "[D]" : "[F]"} ${e.name}`).join("\n");
+  if (list.length >= AGENT_CONFIG.MAX_OUTPUT_LINES) {
+    return "query not specific enough, tool called tried to return too much context and failed";
+  }
+  return list.map((e) => {
+    const indent = "  ".repeat(e.depth);
+    const name = e.type === "dir" ? `${e.name}/` : e.name;
+    return `${indent}${name}`;
+  }).join("\n");
+}
+
+// tools/warp_grep/agent/tools/finish.ts
+async function readFinishFiles(repoRoot, files, reader) {
+  const out = [];
+  for (const f of files) {
+    const ranges = mergeRanges(f.lines);
+    const chunks = [];
+    for (const [s, e] of ranges) {
+      const lines = await reader(f.path, s, e);
+      chunks.push(lines.join("\n"));
+    }
+    out.push({ path: f.path, ranges, content: chunks.join("\n") });
+  }
+  return out;
+}
+function mergeRanges(ranges) {
+  if (!ranges.length) return [];
+  const sorted = [...ranges].sort((a, b) => a[0] - b[0]);
+  const merged = [];
+  let [cs, ce] = sorted[0];
+  for (let i = 1; i < sorted.length; i++) {
+    const [s, e] = sorted[i];
+    if (s <= ce + 1) {
+      ce = Math.max(ce, e);
+    } else {
+      merged.push([cs, ce]);
+      cs = s;
+      ce = e;
+    }
+  }
+  merged.push([cs, ce]);
+  return merged;
 }
 
 // tools/utils/resilience.ts
@@ -549,8 +786,8 @@ var ToolOutputFormatter = class {
     switch (name) {
       case "read":
         return this.formatRead(safeArgs, payload, isError);
-      case "analyse":
-        return this.formatAnalyse(safeArgs, payload, isError);
+      case "list_directory":
+        return this.formatListDirectory(safeArgs, payload, isError);
       case "grep":
         return this.formatGrep(safeArgs, payload, isError);
       default:
@@ -563,39 +800,56 @@ ${payload}
     if (isError) {
       return payload;
     }
-    const path2 = this.asString(args.path) || "...";
-    return `<file path="${path2}">
+    const path3 = this.asString(args.path) || "...";
+    const start = args.start;
+    const end = args.end;
+    const linesArray = args.lines;
+    const attributes = [`path="${path3}"`];
+    if (linesArray && linesArray.length > 0) {
+      const rangeStr = linesArray.map(([s, e]) => `${s}-${e}`).join(",");
+      attributes.push(`lines="${rangeStr}"`);
+    } else if (start !== void 0 && end !== void 0) {
+      attributes.push(`lines="${start}-${end}"`);
+    }
+    return `<read ${attributes.join(" ")}>
 ${payload}
-</file>`;
+</read>`;
   }
-  formatAnalyse(args, payload, isError) {
-    const path2 = this.asString(args.path) || ".";
+  formatListDirectory(args, payload, isError) {
+    const path3 = this.asString(args.path) || ".";
+    const pattern = this.asString(args.pattern);
+    const attributes = [`path="${path3}"`];
+    if (pattern) {
+      attributes.push(`pattern="${pattern}"`);
+    }
     if (isError) {
-      return `<analyse_results path="${path2}" status="error">
-${payload}
-</analyse_results>`;
+      attributes.push('status="error"');
     }
-    return `<analyse_results path="${path2}">
+    return `<list_directory ${attributes.join(" ")}>
 ${payload}
-</analyse_results>`;
+</list_directory>`;
   }
   formatGrep(args, payload, isError) {
     const pattern = this.asString(args.pattern);
-    const path2 = this.asString(args.path);
+    const subDir = this.asString(args.path);
+    const glob = this.asString(args.glob);
     const attributes = [];
     if (pattern !== void 0) {
       attributes.push(`pattern="${pattern}"`);
     }
-    if (path2 !== void 0) {
-      attributes.push(`path="${path2}"`);
+    if (subDir !== void 0) {
+      attributes.push(`sub_dir="${subDir}"`);
+    }
+    if (glob !== void 0) {
+      attributes.push(`glob="${glob}"`);
     }
     if (isError) {
       attributes.push('status="error"');
     }
     const attrText = attributes.length ? ` ${attributes.join(" ")}` : "";
-    return `<grep_output${attrText}>
+    return `<grep${attrText}>
 ${payload}
-</grep_output>`;
+</grep>`;
   }
   asString(value) {
     if (value === null || value === void 0) {
@@ -609,66 +863,100 @@ function formatAgentToolOutput(toolName, args, output, options = {}) {
   return sharedFormatter.format(toolName, args, output, options);
 }
 
-// tools/warp_grep/tools/finish.ts
-async function readFinishFiles(repoRoot, files, reader) {
-  const out = [];
-  for (const f of files) {
-    const ranges = mergeRanges(f.lines);
-    const chunks = [];
-    for (const [s, e] of ranges) {
-      const lines = await reader(f.path, s, e);
-      chunks.push(lines.join("\n"));
-    }
-    out.push({ path: f.path, ranges, content: chunks.join("\n") });
+// tools/warp_grep/agent/helpers.ts
+var import_path = __toESM(require("path"), 1);
+var TRUNCATED_MARKER = "[truncated for context limit]";
+function formatTurnMessage(turnsUsed, maxTurns) {
+  const turnsRemaining = maxTurns - turnsUsed;
+  if (turnsRemaining === 1) {
+    return `
+You have used ${turnsUsed} turns, you only have 1 turn remaining. You have run out of turns to explore the code base and MUST call the finish tool now`;
   }
-  return out;
+  return `
+You have used ${turnsUsed} turn${turnsUsed === 1 ? "" : "s"} and have ${turnsRemaining} remaining`;
 }
-function mergeRanges(ranges) {
-  if (!ranges.length) return [];
-  const sorted = [...ranges].sort((a, b) => a[0] - b[0]);
-  const merged = [];
-  let [cs, ce] = sorted[0];
-  for (let i = 1; i < sorted.length; i++) {
-    const [s, e] = sorted[i];
-    if (s <= ce + 1) {
-      ce = Math.max(ce, e);
-    } else {
-      merged.push([cs, ce]);
-      cs = s;
-      ce = e;
-    }
-  }
-  merged.push([cs, ce]);
-  return merged;
+function calculateContextBudget(messages) {
+  const totalChars = messages.reduce((sum, m) => sum + m.content.length, 0);
+  const maxChars = AGENT_CONFIG.MAX_CONTEXT_CHARS;
+  const percent = Math.round(totalChars / maxChars * 100);
+  const usedK = Math.round(totalChars / 1e3);
+  const maxK = Math.round(maxChars / 1e3);
+  return `<context_budget>${percent}% (${usedK}K/${maxK}K chars)</context_budget>`;
 }
-
-// tools/warp_grep/agent/runner.ts
-var import_path = __toESM(require("path"), 1);
-var parser = new LLMResponseParser();
 async function buildInitialState(repoRoot, query, provider) {
   try {
-    const entries = await provider.analyse({ path: ".", maxResults: 100 });
-    const dirs = entries.filter((e) => e.type === "dir").map((d) => d.name).slice(0, 50);
-    const files = entries.filter((e) => e.type === "file").map((f) => f.name).slice(0, 50);
-    const parts = [
-      `<repo_root>${repoRoot}</repo_root>`,
-      `<top_dirs>${dirs.join(", ")}</top_dirs>`,
-      `<top_files>${files.join(", ")}</top_files>`
-    ];
-    return parts.join("\n");
+    const entries = await provider.listDirectory({
+      path: ".",
+      maxResults: AGENT_CONFIG.MAX_OUTPUT_LINES,
+      maxDepth: 2
+    });
+    const treeLines = entries.map((e) => {
+      const indent = "  ".repeat(e.depth);
+      const name = e.type === "dir" ? `${e.name}/` : e.name;
+      return `${indent}${name}`;
+    });
+    const repoName = import_path.default.basename(repoRoot);
+    const treeOutput = treeLines.length > 0 ? `${repoName}/
+${treeLines.join("\n")}` : `${repoName}/`;
+    return `<repo_structure>
+${treeOutput}
+</repo_structure>
+
+<search_string>
+${query}
+</search_string>`;
   } catch {
-    return `<repo_root>${repoRoot}</repo_root>`;
+    const repoName = import_path.default.basename(repoRoot);
+    return `<repo_structure>
+${repoName}/
+</repo_structure>
+
+<search_string>
+${query}
+</search_string>`;
+  }
+}
+function enforceContextLimit(messages, maxChars = AGENT_CONFIG.MAX_CONTEXT_CHARS) {
+  const getTotalChars = () => messages.reduce((sum, m) => sum + m.content.length, 0);
+  if (getTotalChars() <= maxChars) {
+    return messages;
+  }
+  const userIndices = [];
+  let firstUserSkipped = false;
+  for (let i = 0; i < messages.length; i++) {
+    if (messages[i].role === "user") {
+      if (!firstUserSkipped) {
+        firstUserSkipped = true;
+        continue;
+      }
+      userIndices.push(i);
+    }
   }
+  for (const idx of userIndices) {
+    if (getTotalChars() <= maxChars) {
+      break;
+    }
+    if (messages[idx].content !== TRUNCATED_MARKER) {
+      messages[idx] = { role: "user", content: TRUNCATED_MARKER };
+    }
+  }
+  return messages;
 }
-async function callModel(messages, model, apiKey) {
-  const api = "https://api.morphllm.com/v1/chat/completions";
+
+// tools/warp_grep/agent/runner.ts
+var import_path2 = __toESM(require("path"), 1);
+var parser = new LLMResponseParser();
+var DEFAULT_API_URL = "https://api.morphllm.com";
+async function callModel(messages, model, options = {}) {
+  const baseUrl = DEFAULT_API_URL;
+  const apiKey = options.morphApiKey || process.env.MORPH_API_KEY || "";
   const fetchPromise = fetchWithRetry(
-    api,
+    `${baseUrl}/v1/chat/completions`,
     {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
-        Authorization: `Bearer ${apiKey || process.env.MORPH_API_KEY || ""}`
+        Authorization: `Bearer ${apiKey}`
       },
       body: JSON.stringify({
         model,
@@ -677,10 +965,15 @@ async function callModel(messages, model, apiKey) {
         messages
       })
     },
-    {}
+    options.retryConfig
   );
   const resp = await withTimeout(fetchPromise, AGENT_CONFIG.TIMEOUT_MS, "morph-warp-grep request timed out");
   if (!resp.ok) {
+    if (resp.status === 404) {
+      throw new Error(
+        "The endpoint you are trying to call is likely deprecated. Please update with: npm cache clean --force && npx -y @morphllm/morphmcp@latest or visit: https://morphllm.com/mcp"
+      );
+    }
     const t = await resp.text();
     throw new Error(`morph-warp-grep error ${resp.status}: ${t}`);
   }
@@ -692,22 +985,24 @@ async function callModel(messages, model, apiKey) {
   return content;
 }
 async function runWarpGrep(config) {
-  const repoRoot = import_path.default.resolve(config.repoRoot || process.cwd());
+  const repoRoot = import_path2.default.resolve(config.repoRoot || process.cwd());
   const messages = [];
-  const systemMessage = { role: "system", content: getSystemPrompt() };
-  messages.push(systemMessage);
-  const queryContent = `<query>${config.query}</query>`;
-  messages.push({ role: "user", content: queryContent });
+  messages.push({ role: "system", content: getSystemPrompt() });
   const initialState = await buildInitialState(repoRoot, config.query, config.provider);
   messages.push({ role: "user", content: initialState });
-  const maxRounds = AGENT_CONFIG.MAX_ROUNDS;
+  const maxTurns = AGENT_CONFIG.MAX_TURNS;
   const model = config.model || DEFAULT_MODEL;
   const provider = config.provider;
   const errors = [];
   let finishMeta;
   let terminationReason = "terminated";
-  for (let round = 1; round <= maxRounds; round += 1) {
-    const assistantContent = await callModel(messages, model, config.apiKey).catch((e) => {
+  for (let turn = 1; turn <= maxTurns; turn += 1) {
+    enforceContextLimit(messages);
+    const assistantContent = await callModel(messages, model, {
+      morphApiKey: config.morphApiKey,
+      morphApiUrl: config.morphApiUrl,
+      retryConfig: config.retryConfig
+    }).catch((e) => {
       errors.push({ message: e instanceof Error ? e.message : String(e) });
       return "";
     });
@@ -715,13 +1010,13 @@ async function runWarpGrep(config) {
     messages.push({ role: "assistant", content: assistantContent });
     const toolCalls = parser.parse(assistantContent);
     if (toolCalls.length === 0) {
-      errors.push({ message: "No tool calls produced by the model." });
+      errors.push({ message: "No tool calls produced by the model. Your MCP is likely out of date! Update it by running: npm cache clean --force && npx -y @morphllm/morphmcp@latest" });
       terminationReason = "terminated";
       break;
     }
     const finishCalls = toolCalls.filter((c) => c.name === "finish");
     const grepCalls = toolCalls.filter((c) => c.name === "grep");
-    const analyseCalls = toolCalls.filter((c) => c.name === "analyse");
+    const listDirCalls = toolCalls.filter((c) => c.name === "list_directory");
     const readCalls = toolCalls.filter((c) => c.name === "read");
     const skipCalls = toolCalls.filter((c) => c.name === "_skip");
     const formatted = [];
@@ -733,24 +1028,18 @@ async function runWarpGrep(config) {
     for (const c of grepCalls) {
       const args = c.arguments ?? {};
       allPromises.push(
-        provider.grep({ pattern: args.pattern, path: args.path }).then(
-          (grepRes) => {
-            if (grepRes.error) {
-              return { terminate: true, error: grepRes.error };
-            }
-            const output = grepRes.lines.join("\n") || "no matches";
-            return formatAgentToolOutput("grep", args, output, { isError: false });
-          },
+        toolGrep(provider, args).then(
+          ({ output }) => formatAgentToolOutput("grep", args, output, { isError: false }),
           (err) => formatAgentToolOutput("grep", args, String(err), { isError: true })
         )
       );
     }
-    for (const c of analyseCalls) {
+    for (const c of listDirCalls) {
       const args = c.arguments ?? {};
       allPromises.push(
-        toolAnalyse(provider, args).then(
-          (p) => formatAgentToolOutput("analyse", args, p, { isError: false }),
-          (err) => formatAgentToolOutput("analyse", args, String(err), { isError: true })
+        toolListDirectory(provider, args).then(
+          (p) => formatAgentToolOutput("list_directory", args, p, { isError: false }),
+          (err) => formatAgentToolOutput("list_directory", args, String(err), { isError: true })
         )
       );
     }
@@ -765,34 +1054,12 @@ async function runWarpGrep(config) {
     }
     const allResults = await Promise.all(allPromises);
     for (const result of allResults) {
-      if (typeof result === "object" && "terminate" in result) {
-        errors.push({ message: result.error });
-        return {
-          terminationReason: "terminated",
-          messages,
-          errors
-        };
-      }
       formatted.push(result);
     }
     if (formatted.length > 0) {
-      const turnsUsed = round;
-      const turnsRemaining = 4 - turnsUsed;
-      let turnMessage;
-      if (turnsRemaining === 0) {
-        turnMessage = `
-
-[Turn ${turnsUsed}/4] This is your LAST turn. You MUST call the finish tool now.`;
-      } else if (turnsRemaining === 1) {
-        turnMessage = `
-
-[Turn ${turnsUsed}/4] You have 1 turn remaining. Next turn you MUST call the finish tool.`;
-      } else {
-        turnMessage = `
-
-[Turn ${turnsUsed}/4] You have ${turnsRemaining} turns remaining.`;
-      }
-      messages.push({ role: "user", content: formatted.join("\n") + turnMessage });
+      const turnMessage = formatTurnMessage(turn, maxTurns);
+      const contextBudget = calculateContextBudget(messages);
+      messages.push({ role: "user", content: formatted.join("\n") + turnMessage + "\n" + contextBudget });
     }
     if (finishCalls.length) {
       const fc = finishCalls[0];