@morphllm/morphsdk 0.2.82 → 0.2.84

package/dist/{chunk-ZLJAODDJ.js → chunk-262WRPS5.js}

@@ -5,7 +5,7 @@ import {
 import {
   executeToolCall,
   formatResult
-} from "./chunk-FJKPMMNQ.js";
+} from "./chunk-D4EQYM6O.js";
 import {
   getSystemPrompt
 } from "./chunk-FMLHRJDF.js";
@@ -50,4 +50,4 @@ export {
   execute,
   createWarpGrepTool
 };
-//# sourceMappingURL=chunk-ZLJAODDJ.js.map
+//# sourceMappingURL=chunk-262WRPS5.js.map

package/dist/{chunk-FJKPMMNQ.js → chunk-D4EQYM6O.js}

@@ -1,6 +1,6 @@
 import {
   runWarpGrep
-} from "./chunk-WIAYUEJK.js";
+} from "./chunk-LF2X6YNP.js";
 import {
   RemoteCommandsProvider
 } from "./chunk-PUGSTXLO.js";
@@ -117,4 +117,4 @@ export {
   executeToolCall,
   formatResult
 };
-//# sourceMappingURL=chunk-FJKPMMNQ.js.map
+//# sourceMappingURL=chunk-D4EQYM6O.js.map

package/dist/{chunk-P2O5JKE5.js → chunk-KRDIR7GG.js}

@@ -51,14 +51,23 @@ async function toolRead(provider, args) {
 
 // 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 ?? AGENT_CONFIG.MAX_OUTPUT_LINES,
-    maxDepth: args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH
-  });
+  const maxResults = args.maxResults ?? AGENT_CONFIG.MAX_OUTPUT_LINES;
+  const initialDepth = args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH;
+  async function getListRecursive(currentDepth) {
+    const entries = await provider.listDirectory({
+      path: args.path,
+      pattern: args.pattern ?? null,
+      maxResults,
+      maxDepth: currentDepth
+    });
+    if (entries.length >= maxResults && currentDepth > 0) {
+      return getListRecursive(currentDepth - 1);
+    }
+    return { entries };
+  }
+  const { entries: list } = await getListRecursive(initialDepth);
   if (!list.length) return "empty";
-  if (list.length >= AGENT_CONFIG.MAX_OUTPUT_LINES) {
+  if (list.length >= maxResults) {
     return "query not specific enough, tool called tried to return too much context and failed";
   }
   return list.map((e) => {
@@ -228,4 +237,4 @@ export {
   formatListDirectoryTree,
   enforceContextLimit
 };
-//# sourceMappingURL=chunk-P2O5JKE5.js.map
+//# sourceMappingURL=chunk-KRDIR7GG.js.map

package/dist/chunk-KRDIR7GG.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../tools/warp_grep/agent/tools/grep.ts","../tools/warp_grep/agent/tools/read.ts","../tools/warp_grep/agent/tools/list_directory.ts","../tools/warp_grep/agent/tools/finish.ts","../tools/warp_grep/agent/helpers.ts"],"sourcesContent":["import type { WarpGrepProvider } from '../../providers/types.js';\n\nexport async function toolGrep(\n  provider: WarpGrepProvider,\n  args: { pattern: string; path: string; glob?: string }\n): Promise<{ output: string }> {\n  const res = await provider.grep(args);\n  \n  // Return errors as output - let the model see and handle them\n  if (res.error) {\n    return { output: res.error };\n  }\n  \n  if (!res.lines.length) {\n    return { output: 'no matches' };\n  }\n  \n  return { output: res.lines.join('\\n') };\n}\n\n","import type { WarpGrepProvider } from '../../providers/types.js';\n\nfunction isValidRange(start: unknown, end: unknown): boolean {\n  return (\n    typeof start === 'number' &&\n    typeof end === 'number' &&\n    Number.isFinite(start) &&\n    Number.isFinite(end) &&\n    start > 0 &&\n    end >= start\n  );\n}\n\nexport async function toolRead(\n  provider: WarpGrepProvider,\n  args: { path: string; start?: number; end?: number; lines?: Array<[number, number]> }\n): Promise<string> {\n  if (args.lines && Array.isArray(args.lines) && args.lines.length > 0) {\n    const validRanges: Array<[number, number]> = [];\n    for (const range of args.lines) {\n      if (Array.isArray(range) && range.length >= 2 && isValidRange(range[0], range[1])) {\n        validRanges.push([range[0], range[1]]);\n      }\n    }\n    \n    if (validRanges.length === 0) {\n      const res = await provider.read({ path: args.path });\n      if (res.error) return res.error;\n      if (!res.lines.length) return '(empty file)';\n      return res.lines.join('\\n');\n    }\n    \n    const chunks: string[] = [];\n    for (const [start, end] of validRanges) {\n      const res = await provider.read({ path: args.path, start, end });\n      if (res.error) return res.error;\n      chunks.push(res.lines.join('\\n'));\n    }\n    if (chunks.every(c => c === '')) return '(empty file)';\n    return chunks.join('\\n...\\n');\n  }\n\n  const res = await provider.read({ path: args.path, start: args.start, end: args.end });\n  if (res.error) {\n    return res.error;\n  }\n  if (!res.lines.length) return '(empty file)';\n  return res.lines.join('\\n');\n}\n\n","import type { WarpGrepProvider } from '../../providers/types.js';\nimport { AGENT_CONFIG } from '../config.js';\n\nexport async function toolListDirectory(\n  provider: WarpGrepProvider,\n  args: { path: string; pattern?: string | null; maxResults?: number; maxDepth?: number }\n): Promise<string> {\n\n  const maxResults = args.maxResults ?? AGENT_CONFIG.MAX_OUTPUT_LINES;\n  const initialDepth = args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH;\n\n  type DirectoryEntries = Awaited<ReturnType<typeof provider.listDirectory>>;\n  async function getListRecursive(currentDepth: number): Promise<{ entries: DirectoryEntries; }> {\n    const entries = await provider.listDirectory({\n      path: args.path,\n      pattern: args.pattern ?? null,\n      maxResults,\n      maxDepth: currentDepth,\n    });\n\n    if (entries.length >= maxResults && currentDepth > 0) {\n      return getListRecursive(currentDepth - 1);\n    }\n\n    return { entries };\n  }\n  const { entries: list } = await getListRecursive(initialDepth);\n\n  if (!list.length) return 'empty';\n  if (list.length >= maxResults) {\n    return 'query not specific enough, tool called tried to return too much context and failed';\n  }\n  return list\n    .map((e) => {\n      const indent = '  '.repeat(e.depth);\n      const name = e.type === 'dir' ? `${e.name}/` : e.name;\n      return `${indent}${name}`;\n    })\n    .join('\\n');\n}","import type { FinishFileSpec } from '../types.js';\n\nfunction isValidRange(range: unknown): range is [number, number] {\n  return (\n    Array.isArray(range) &&\n    range.length >= 2 &&\n    typeof range[0] === 'number' &&\n    typeof range[1] === 'number' &&\n    Number.isFinite(range[0]) &&\n    Number.isFinite(range[1]) &&\n    range[0] > 0 &&\n    range[1] >= range[0]\n  );\n}\n\nfunction extractValidRanges(lines: unknown): Array<[number, number]> | null {\n  if (!Array.isArray(lines)) return null;\n  const valid: Array<[number, number]> = [];\n  for (const range of lines) {\n    if (isValidRange(range)) {\n      valid.push([range[0], range[1]]);\n    }\n  }\n  return valid.length > 0 ? valid : null;\n}\n\nexport function normalizeFinishFiles(files: FinishFileSpec[]): FinishFileSpec[] {\n  return files.map((f) => {\n    if (f.lines === '*') return { path: f.path, lines: '*' };\n    const validRanges = extractValidRanges(f.lines);\n    if (!validRanges) return { path: f.path, lines: '*' };\n    return { path: f.path, lines: mergeRanges(validRanges) };\n  });\n}\n\nexport async function readFinishFiles(\n  repoRoot: string,\n  files: FinishFileSpec[],\n  reader: (path: string, start?: number, end?: number) => Promise<string[]>\n): Promise<{ path: string; ranges: '*' | Array<[number, number]>; content: string }[]> {\n  const out: { path: string; ranges: '*' | Array<[number, number]>; content: string }[] = [];\n  for (const f of files) {\n    const validRanges = f.lines === '*' ? null : extractValidRanges(f.lines);\n    if (f.lines === '*' || !validRanges) {\n      const lines = await reader(f.path);\n      out.push({ path: f.path, ranges: '*', content: lines.join('\\n') });\n    } else {\n      const ranges = mergeRanges(validRanges);\n      const chunks: string[] = [];\n      for (const [s, e] of ranges) {\n        const lines = await reader(f.path, s, e);\n        chunks.push(lines.join('\\n'));\n      }\n      out.push({ path: f.path, ranges, content: chunks.join('\\n') });\n    }\n  }\n  return out;\n}\n\nfunction mergeRanges(ranges: Array<[number, number]>): Array<[number, number]> {\n  if (!ranges.length) return [];\n  const sorted = [...ranges].sort((a, b) => a[0] - b[0]);\n  const merged: Array<[number, number]> = [];\n  let [cs, ce] = sorted[0];\n  for (let i = 1; i < sorted.length; i++) {\n    const [s, e] = sorted[i];\n    if (s <= ce + 1) {\n      ce = Math.max(ce, e);\n    } else {\n      merged.push([cs, ce]);\n      cs = s;\n      ce = e;\n    }\n  }\n  merged.push([cs, ce]);\n  return merged;\n}\n\n","/**\n * Shared helper functions for warp-grep agent loop.\n * Used by both the internal runner and exported for harness users.\n */\n\nimport path from 'path';\nimport { AGENT_CONFIG } from './config.js';\nimport type { ChatMessage } from './types.js';\nimport type { WarpGrepProvider, ListDirectoryEntry } from '../providers/types.js';\n\nconst TRUNCATED_MARKER = '[truncated for context limit]';\n\n/**\n * Format the turn counter message shown to the model.\n */\nexport function formatTurnMessage(turnsUsed: number, maxTurns: number): string {\n  const turnsRemaining = maxTurns - turnsUsed;\n  if (turnsRemaining === 1) {\n    return `\\nYou 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`;\n  }\n  return `\\nYou have used ${turnsUsed} turn${turnsUsed === 1 ? '' : 's'} and have ${turnsRemaining} remaining`;\n}\n\n/**\n * Calculate and format the context budget indicator.\n */\nexport function calculateContextBudget(messages: ChatMessage[]): string {\n  const totalChars = messages.reduce((sum, m) => sum + m.content.length, 0);\n  const maxChars = AGENT_CONFIG.MAX_CONTEXT_CHARS;\n  const percent = Math.round((totalChars / maxChars) * 100);\n  const usedK = Math.round(totalChars / 1000);\n  const maxK = Math.round(maxChars / 1000);\n  return `<context_budget>${percent}% (${usedK}K/${maxK}K chars)</context_budget>`;\n}\n\n/**\n * Build the initial state message containing repo structure and query.\n */\nexport async function buildInitialState(\n  repoRoot: string,\n  query: string,\n  provider: WarpGrepProvider\n): Promise<string> {\n  try {\n    const entries = await provider.listDirectory({\n      path: '.',\n      maxResults: AGENT_CONFIG.MAX_OUTPUT_LINES,\n      maxDepth: 2,\n    });\n    const treeLines = entries.map((e) => {\n      const indent = '  '.repeat(e.depth);\n      const name = e.type === 'dir' ? `${e.name}/` : e.name;\n      return `${indent}${name}`;\n    });\n\n    const repoName = path.basename(repoRoot);\n    const treeOutput =\n      treeLines.length > 0 ? `${repoName}/\\n${treeLines.join('\\n')}` : `${repoName}/`;\n\n    return `<repo_structure>\\n${treeOutput}\\n</repo_structure>\\n\\n<search_string>\\n${query}\\n</search_string>`;\n  } catch {\n    const repoName = path.basename(repoRoot);\n    return `<repo_structure>\\n${repoName}/\\n</repo_structure>\\n\\n<search_string>\\n${query}\\n</search_string>`;\n  }\n}\n\n/**\n * Format directory entries as an indented tree string.\n */\nexport function formatListDirectoryTree(entries: ListDirectoryEntry[]): string {\n  if (!entries.length) return 'empty';\n  return entries\n    .map((e) => {\n      const indent = '  '.repeat(e.depth);\n      const name = e.type === 'dir' ? `${e.name}/` : e.name;\n      return `${indent}${name}`;\n    })\n    .join('\\n');\n}\n\n/**\n * Enforce hard context limit by truncating user messages.\n * Preserves system prompt, first user message (initial query), and all assistant messages.\n * Removes tool result user messages from oldest to newest until under the threshold.\n * \n * @param messages - The conversation messages array (mutated in place)\n * @param maxChars - Maximum total character count (defaults to AGENT_CONFIG.MAX_CONTEXT_CHARS)\n * @returns The same messages array, truncated if necessary\n */\nexport function enforceContextLimit(\n  messages: ChatMessage[],\n  maxChars: number = AGENT_CONFIG.MAX_CONTEXT_CHARS\n): ChatMessage[] {\n  const getTotalChars = () => messages.reduce((sum, m) => sum + m.content.length, 0);\n  \n  if (getTotalChars() <= maxChars) {\n    return messages;\n  }\n\n  // Find indices of user messages, skipping the first one (initial query with repo structure)\n  // We truncate tool result messages from oldest to newest\n  const userIndices: number[] = [];\n  let firstUserSkipped = false;\n  for (let i = 0; i < messages.length; i++) {\n    if (messages[i].role === 'user') {\n      if (!firstUserSkipped) {\n        firstUserSkipped = true;\n        continue; // Skip first user message (contains the search query)\n      }\n      userIndices.push(i);\n    }\n  }\n\n  // Truncate user messages from oldest to newest until under limit\n  for (const idx of userIndices) {\n    if (getTotalChars() <= maxChars) {\n      break;\n    }\n    // Only truncate if not already truncated\n    if (messages[idx].content !== TRUNCATED_MARKER) {\n      messages[idx] = { role: 'user', content: TRUNCATED_MARKER };\n    }\n  }\n\n  return messages;\n}\n\n"],"mappings":";;;;;AAEA,eAAsB,SACpB,UACA,MAC6B;AAC7B,QAAM,MAAM,MAAM,SAAS,KAAK,IAAI;AAGpC,MAAI,IAAI,OAAO;AACb,WAAO,EAAE,QAAQ,IAAI,MAAM;AAAA,EAC7B;AAEA,MAAI,CAAC,IAAI,MAAM,QAAQ;AACrB,WAAO,EAAE,QAAQ,aAAa;AAAA,EAChC;AAEA,SAAO,EAAE,QAAQ,IAAI,MAAM,KAAK,IAAI,EAAE;AACxC;;;AChBA,SAAS,aAAa,OAAgB,KAAuB;AAC3D,SACE,OAAO,UAAU,YACjB,OAAO,QAAQ,YACf,OAAO,SAAS,KAAK,KACrB,OAAO,SAAS,GAAG,KACnB,QAAQ,KACR,OAAO;AAEX;AAEA,eAAsB,SACpB,UACA,MACiB;AACjB,MAAI,KAAK,SAAS,MAAM,QAAQ,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG;AACpE,UAAM,cAAuC,CAAC;AAC9C,eAAW,SAAS,KAAK,OAAO;AAC9B,UAAI,MAAM,QAAQ,KAAK,KAAK,MAAM,UAAU,KAAK,aAAa,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,GAAG;AACjF,oBAAY,KAAK,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;AAAA,MACvC;AAAA,IACF;AAEA,QAAI,YAAY,WAAW,GAAG;AAC5B,YAAMA,OAAM,MAAM,SAAS,KAAK,EAAE,MAAM,KAAK,KAAK,CAAC;AACnD,UAAIA,KAAI,MAAO,QAAOA,KAAI;AAC1B,UAAI,CAACA,KAAI,MAAM,OAAQ,QAAO;AAC9B,aAAOA,KAAI,MAAM,KAAK,IAAI;AAAA,IAC5B;AAEA,UAAM,SAAmB,CAAC;AAC1B,eAAW,CAAC,OAAO,GAAG,KAAK,aAAa;AACtC,YAAMA,OAAM,MAAM,SAAS,KAAK,EAAE,MAAM,KAAK,MAAM,OAAO,IAAI,CAAC;AAC/D,UAAIA,KAAI,MAAO,QAAOA,KAAI;AAC1B,aAAO,KAAKA,KAAI,MAAM,KAAK,IAAI,CAAC;AAAA,IAClC;AACA,QAAI,OAAO,MAAM,OAAK,MAAM,EAAE,EAAG,QAAO;AACxC,WAAO,OAAO,KAAK,SAAS;AAAA,EAC9B;AAEA,QAAM,MAAM,MAAM,SAAS,KAAK,EAAE,MAAM,KAAK,MAAM,OAAO,KAAK,OAAO,KAAK,KAAK,IAAI,CAAC;AACrF,MAAI,IAAI,OAAO;AACb,WAAO,IAAI;AAAA,EACb;AACA,MAAI,CAAC,IAAI,MAAM,OAAQ,QAAO;AAC9B,SAAO,IAAI,MAAM,KAAK,IAAI;AAC5B;;;AC7CA,eAAsB,kBACpB,UACA,MACiB;AAEjB,QAAM,aAAa,KAAK,cAAc,aAAa;AACnD,QAAM,eAAe,KAAK,YAAY,aAAa;AAGnD,iBAAe,iBAAiB,cAA+D;AAC7F,UAAM,UAAU,MAAM,SAAS,cAAc;AAAA,MAC3C,MAAM,KAAK;AAAA,MACX,SAAS,KAAK,WAAW;AAAA,MACzB;AAAA,MACA,UAAU;AAAA,IACZ,CAAC;AAED,QAAI,QAAQ,UAAU,cAAc,eAAe,GAAG;AACpD,aAAO,iBAAiB,eAAe,CAAC;AAAA,IAC1C;AAEA,WAAO,EAAE,QAAQ;AAAA,EACnB;AACA,QAAM,EAAE,SAAS,KAAK,IAAI,MAAM,iBAAiB,YAAY;AAE7D,MAAI,CAAC,KAAK,OAAQ,QAAO;AACzB,MAAI,KAAK,UAAU,YAAY;AAC7B,WAAO;AAAA,EACT;AACA,SAAO,KACJ,IAAI,CAAC,MAAM;AACV,UAAM,SAAS,KAAK,OAAO,EAAE,KAAK;AAClC,UAAM,OAAO,EAAE,SAAS,QAAQ,GAAG,EAAE,IAAI,MAAM,EAAE;AACjD,WAAO,GAAG,MAAM,GAAG,IAAI;AAAA,EACzB,CAAC,EACA,KAAK,IAAI;AACd;;;ACrCA,SAASC,cAAa,OAA2C;AAC/D,SACE,MAAM,QAAQ,KAAK,KACnB,MAAM,UAAU,KAChB,OAAO,MAAM,CAAC,MAAM,YACpB,OAAO,MAAM,CAAC,MAAM,YACpB,OAAO,SAAS,MAAM,CAAC,CAAC,KACxB,OAAO,SAAS,MAAM,CAAC,CAAC,KACxB,MAAM,CAAC,IAAI,KACX,MAAM,CAAC,KAAK,MAAM,CAAC;AAEvB;AAEA,SAAS,mBAAmB,OAAgD;AAC1E,MAAI,CAAC,MAAM,QAAQ,KAAK,EAAG,QAAO;AAClC,QAAM,QAAiC,CAAC;AACxC,aAAW,SAAS,OAAO;AACzB,QAAIA,cAAa,KAAK,GAAG;AACvB,YAAM,KAAK,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;AAAA,IACjC;AAAA,EACF;AACA,SAAO,MAAM,SAAS,IAAI,QAAQ;AACpC;AAEO,SAAS,qBAAqB,OAA2C;AAC9E,SAAO,MAAM,IAAI,CAAC,MAAM;AACtB,QAAI,EAAE,UAAU,IAAK,QAAO,EAAE,MAAM,EAAE,MAAM,OAAO,IAAI;AACvD,UAAM,cAAc,mBAAmB,EAAE,KAAK;AAC9C,QAAI,CAAC,YAAa,QAAO,EAAE,MAAM,EAAE,MAAM,OAAO,IAAI;AACpD,WAAO,EAAE,MAAM,EAAE,MAAM,OAAO,YAAY,WAAW,EAAE;AAAA,EACzD,CAAC;AACH;AAEA,eAAsB,gBACpB,UACA,OACA,QACqF;AACrF,QAAM,MAAkF,CAAC;AACzF,aAAW,KAAK,OAAO;AACrB,UAAM,cAAc,EAAE,UAAU,MAAM,OAAO,mBAAmB,EAAE,KAAK;AACvE,QAAI,EAAE,UAAU,OAAO,CAAC,aAAa;AACnC,YAAM,QAAQ,MAAM,OAAO,EAAE,IAAI;AACjC,UAAI,KAAK,EAAE,MAAM,EAAE,MAAM,QAAQ,KAAK,SAAS,MAAM,KAAK,IAAI,EAAE,CAAC;AAAA,IACnE,OAAO;AACL,YAAM,SAAS,YAAY,WAAW;AACtC,YAAM,SAAmB,CAAC;AAC1B,iBAAW,CAAC,GAAG,CAAC,KAAK,QAAQ;AAC3B,cAAM,QAAQ,MAAM,OAAO,EAAE,MAAM,GAAG,CAAC;AACvC,eAAO,KAAK,MAAM,KAAK,IAAI,CAAC;AAAA,MAC9B;AACA,UAAI,KAAK,EAAE,MAAM,EAAE,MAAM,QAAQ,SAAS,OAAO,KAAK,IAAI,EAAE,CAAC;AAAA,IAC/D;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,YAAY,QAA0D;AAC7E,MAAI,CAAC,OAAO,OAAQ,QAAO,CAAC;AAC5B,QAAM,SAAS,CAAC,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;AACrD,QAAM,SAAkC,CAAC;AACzC,MAAI,CAAC,IAAI,EAAE,IAAI,OAAO,CAAC;AACvB,WAAS,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,UAAM,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC;AACvB,QAAI,KAAK,KAAK,GAAG;AACf,WAAK,KAAK,IAAI,IAAI,CAAC;AAAA,IACrB,OAAO;AACL,aAAO,KAAK,CAAC,IAAI,EAAE,CAAC;AACpB,WAAK;AACL,WAAK;AAAA,IACP;AAAA,EACF;AACA,SAAO,KAAK,CAAC,IAAI,EAAE,CAAC;AACpB,SAAO;AACT;;;ACvEA,OAAO,UAAU;AAKjB,IAAM,mBAAmB;AAKlB,SAAS,kBAAkB,WAAmB,UAA0B;AAC7E,QAAM,iBAAiB,WAAW;AAClC,MAAI,mBAAmB,GAAG;AACxB,WAAO;AAAA,gBAAmB,SAAS;AAAA,EACrC;AACA,SAAO;AAAA,gBAAmB,SAAS,QAAQ,cAAc,IAAI,KAAK,GAAG,aAAa,cAAc;AAClG;AAKO,SAAS,uBAAuB,UAAiC;AACtE,QAAM,aAAa,SAAS,OAAO,CAAC,KAAK,MAAM,MAAM,EAAE,QAAQ,QAAQ,CAAC;AACxE,QAAM,WAAW,aAAa;AAC9B,QAAM,UAAU,KAAK,MAAO,aAAa,WAAY,GAAG;AACxD,QAAM,QAAQ,KAAK,MAAM,aAAa,GAAI;AAC1C,QAAM,OAAO,KAAK,MAAM,WAAW,GAAI;AACvC,SAAO,mBAAmB,OAAO,MAAM,KAAK,KAAK,IAAI;AACvD;AAKA,eAAsB,kBACpB,UACA,OACA,UACiB;AACjB,MAAI;AACF,UAAM,UAAU,MAAM,SAAS,cAAc;AAAA,MAC3C,MAAM;AAAA,MACN,YAAY,aAAa;AAAA,MACzB,UAAU;AAAA,IACZ,CAAC;AACD,UAAM,YAAY,QAAQ,IAAI,CAAC,MAAM;AACnC,YAAM,SAAS,KAAK,OAAO,EAAE,KAAK;AAClC,YAAM,OAAO,EAAE,SAAS,QAAQ,GAAG,EAAE,IAAI,MAAM,EAAE;AACjD,aAAO,GAAG,MAAM,GAAG,IAAI;AAAA,IACzB,CAAC;AAED,UAAM,WAAW,KAAK,SAAS,QAAQ;AACvC,UAAM,aACJ,UAAU,SAAS,IAAI,GAAG,QAAQ;AAAA,EAAM,UAAU,KAAK,IAAI,CAAC,KAAK,GAAG,QAAQ;AAE9E,WAAO;AAAA,EAAqB,UAAU;AAAA;AAAA;AAAA;AAAA,EAA2C,KAAK;AAAA;AAAA,EACxF,QAAQ;AACN,UAAM,WAAW,KAAK,SAAS,QAAQ;AACvC,WAAO;AAAA,EAAqB,QAAQ;AAAA;AAAA;AAAA;AAAA,EAA4C,KAAK;AAAA;AAAA,EACvF;AACF;AAKO,SAAS,wBAAwB,SAAuC;AAC7E,MAAI,CAAC,QAAQ,OAAQ,QAAO;AAC5B,SAAO,QACJ,IAAI,CAAC,MAAM;AACV,UAAM,SAAS,KAAK,OAAO,EAAE,KAAK;AAClC,UAAM,OAAO,EAAE,SAAS,QAAQ,GAAG,EAAE,IAAI,MAAM,EAAE;AACjD,WAAO,GAAG,MAAM,GAAG,IAAI;AAAA,EACzB,CAAC,EACA,KAAK,IAAI;AACd;AAWO,SAAS,oBACd,UACA,WAAmB,aAAa,mBACjB;AACf,QAAM,gBAAgB,MAAM,SAAS,OAAO,CAAC,KAAK,MAAM,MAAM,EAAE,QAAQ,QAAQ,CAAC;AAEjF,MAAI,cAAc,KAAK,UAAU;AAC/B,WAAO;AAAA,EACT;AAIA,QAAM,cAAwB,CAAC;AAC/B,MAAI,mBAAmB;AACvB,WAAS,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;AACxC,QAAI,SAAS,CAAC,EAAE,SAAS,QAAQ;AAC/B,UAAI,CAAC,kBAAkB;AACrB,2BAAmB;AACnB;AAAA,MACF;AACA,kBAAY,KAAK,CAAC;AAAA,IACpB;AAAA,EACF;AAGA,aAAW,OAAO,aAAa;AAC7B,QAAI,cAAc,KAAK,UAAU;AAC/B;AAAA,IACF;AAEA,QAAI,SAAS,GAAG,EAAE,YAAY,kBAAkB;AAC9C,eAAS,GAAG,IAAI,EAAE,MAAM,QAAQ,SAAS,iBAAiB;AAAA,IAC5D;AAAA,EACF;AAEA,SAAO;AACT;","names":["res","isValidRange"]}

package/dist/{chunk-WIAYUEJK.js → chunk-LF2X6YNP.js}

@@ -7,7 +7,7 @@ import {
   toolGrep,
   toolListDirectory,
   toolRead
-} from "./chunk-P2O5JKE5.js";
+} from "./chunk-KRDIR7GG.js";
 import {
   formatAgentToolOutput
 } from "./chunk-APP75CBN.js";
@@ -69,10 +69,15 @@ async function callModel(messages, model, options = {}) {
   return content;
 }
 async function runWarpGrep(config) {
+  const totalStart = Date.now();
+  const timeoutMs = config.timeout ?? AGENT_CONFIG.TIMEOUT_MS;
+  const timings = { turns: [], timeout_ms: timeoutMs };
   const repoRoot = path.resolve(config.repoRoot || process.cwd());
   const messages = [];
   messages.push({ role: "system", content: getSystemPrompt() });
+  const initialStateStart = Date.now();
   const initialState = await buildInitialState(repoRoot, config.query, config.provider);
+  timings.initial_state_ms = Date.now() - initialStateStart;
   messages.push({ role: "user", content: initialState });
   const maxTurns = AGENT_CONFIG.MAX_TURNS;
   const model = config.model || DEFAULT_MODEL;
@@ -81,22 +86,29 @@ async function runWarpGrep(config) {
   let finishMeta;
   let terminationReason = "terminated";
   for (let turn = 1; turn <= maxTurns; turn += 1) {
+    const turnMetrics = { turn, morph_api_ms: 0, local_tools_ms: 0 };
     enforceContextLimit(messages);
+    const modelCallStart = Date.now();
     const assistantContent = await callModel(messages, model, {
       morphApiKey: config.morphApiKey,
       morphApiUrl: config.morphApiUrl,
       retryConfig: config.retryConfig,
-      timeout: config.timeout
+      timeout: timeoutMs
     }).catch((e) => {
       errors.push({ message: e instanceof Error ? e.message : String(e) });
       return "";
     });
-    if (!assistantContent) break;
+    turnMetrics.morph_api_ms = Date.now() - modelCallStart;
+    if (!assistantContent) {
+      timings.turns.push(turnMetrics);
+      break;
+    }
     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. Your MCP is likely out of date! Update it by running: rm -rf ~/.npm/_npx && npm cache clean --force && npx -y @morphllm/morphmcp@latest" });
       terminationReason = "terminated";
+      timings.turns.push(turnMetrics);
       break;
     }
     const finishCalls = toolCalls.filter((c) => c.name === "finish");
@@ -137,7 +149,9 @@ async function runWarpGrep(config) {
         )
       );
     }
+    const toolExecStart = Date.now();
     const allResults = await Promise.all(allPromises);
+    turnMetrics.local_tools_ms = Date.now() - toolExecStart;
     for (const result of allResults) {
       formatted.push(result);
     }
@@ -146,6 +160,7 @@ async function runWarpGrep(config) {
       const contextBudget = calculateContextBudget(messages);
       messages.push({ role: "user", content: formatted.join("\n") + turnMessage + "\n" + contextBudget });
     }
+    timings.turns.push(turnMetrics);
     if (finishCalls.length) {
       const fc = finishCalls[0];
       const files = fc.arguments?.files ?? [];
@@ -155,7 +170,8 @@ async function runWarpGrep(config) {
     }
   }
   if (terminationReason !== "completed" || !finishMeta) {
-    return { terminationReason, messages, errors };
+    timings.total_ms = Date.now() - totalStart;
+    return { terminationReason, messages, errors, timings };
   }
   const parts = ["Relevant context found:"];
   for (const f of finishMeta.files) {
@@ -163,6 +179,7 @@ async function runWarpGrep(config) {
     parts.push(`- ${f.path}: ${ranges}`);
   }
   const payload = parts.join("\n");
+  const finishResolutionStart = Date.now();
   const fileReadErrors = [];
   const resolved = await readFinishFiles(
     repoRoot,
@@ -182,17 +199,20 @@ async function runWarpGrep(config) {
       }
     }
   );
+  timings.finish_resolution_ms = Date.now() - finishResolutionStart;
   if (fileReadErrors.length > 0) {
     errors.push(...fileReadErrors.map((e) => ({ message: `File read error: ${e.path} - ${e.error}` })));
   }
+  timings.total_ms = Date.now() - totalStart;
   return {
     terminationReason: "completed",
     messages,
-    finish: { payload, metadata: finishMeta, resolved }
+    finish: { payload, metadata: finishMeta, resolved },
+    timings
   };
 }
 
 export {
   runWarpGrep
 };
-//# sourceMappingURL=chunk-WIAYUEJK.js.map
+//# sourceMappingURL=chunk-LF2X6YNP.js.map

package/dist/chunk-LF2X6YNP.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../tools/warp_grep/agent/runner.ts"],"sourcesContent":["import { AGENT_CONFIG, DEFAULT_MODEL } from './config.js';\nimport { getSystemPrompt } from './prompt.js';\nimport type { AgentRunResult, ChatMessage, SessionConfig, AgentFinish, WarpGrepExecutionMetrics, WarpGrepTurnMetrics } from './types.js';\nimport { LLMResponseParser } from './parser.js';\nimport type { WarpGrepProvider } from '../providers/types.js';\nimport { toolGrep } from './tools/grep.js';\nimport { toolRead } from './tools/read.js';\nimport { toolListDirectory } from './tools/list_directory.js';\nimport { readFinishFiles } from './tools/finish.js';\nimport { fetchWithRetry, withTimeout, type RetryConfig } from '../../utils/resilience.js';\nimport { formatAgentToolOutput } from './formatter.js';\nimport { formatTurnMessage, calculateContextBudget, buildInitialState, enforceContextLimit } from './helpers.js';\nimport path from 'path';\n\ntype EventName =\n  | 'initial_state'\n  | 'round_start'\n  | 'round_end'\n  | 'finish'\n  | 'error';\n\nexport type EventCallback = (name: EventName, payload: Record<string, unknown>) => void;\n\nconst parser = new LLMResponseParser();\n\nconst DEFAULT_API_URL = 'https://api.morphllm.com';\n\ninterface CallModelOptions {\n  morphApiKey?: string;\n  morphApiUrl?: string;\n  retryConfig?: RetryConfig;\n  timeout?: number;\n}\n\nasync function callModel(\n  messages: ChatMessage[],\n  model: string,\n  options: CallModelOptions = {}\n): Promise<string> {\n  const baseUrl = options.morphApiUrl || DEFAULT_API_URL;\n  const apiKey = options.morphApiKey || process.env.MORPH_API_KEY || '';\n  \n  const fetchPromise = fetchWithRetry(\n    `${baseUrl}/v1/chat/completions`,\n    {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n        Authorization: `Bearer ${apiKey}`,\n      },\n      body: JSON.stringify({\n        model,\n        temperature: 0.0,\n        max_tokens: 1024,\n        messages,\n      }),\n    },\n    options.retryConfig\n  );\n  const timeoutMs = options.timeout ?? AGENT_CONFIG.TIMEOUT_MS;\n  const resp = await withTimeout(fetchPromise, timeoutMs, 'morph-warp-grep request timed out');\n  if (!resp.ok) {\n    if (resp.status === 404) {\n      throw new Error(\n        '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'\n      );\n    }\n    // keeping these cases are real throws, if this happens retry will likely not help, so best we just throw here, notice the error and fix\n    const t = await resp.text();\n    throw new Error(`morph-warp-grep error ${resp.status}: ${t}`);\n  }\n  const data = await resp.json();\n  const content = data?.choices?.[0]?.message?.content;\n  if (!content || typeof content !== 'string') {\n    throw new Error('Invalid response from model');\n  }\n  return content;\n}\n\nexport async function runWarpGrep(config: SessionConfig & { provider: WarpGrepProvider }): Promise<AgentRunResult> {\n  const totalStart = Date.now();\n  const timeoutMs = config.timeout ?? AGENT_CONFIG.TIMEOUT_MS;\n  const timings: Partial<WarpGrepExecutionMetrics> = { turns: [], timeout_ms: timeoutMs };\n  \n  const repoRoot = path.resolve(config.repoRoot || process.cwd());\n  const messages: ChatMessage[] = [];\n\n  messages.push({ role: 'system' as const, content: getSystemPrompt() });\n  \n  const initialStateStart = Date.now();\n  const initialState = await buildInitialState(repoRoot, config.query, config.provider);\n  timings.initial_state_ms = Date.now() - initialStateStart;\n  \n  messages.push({ role: 'user', content: initialState });\n\n  const maxTurns = AGENT_CONFIG.MAX_TURNS;\n  const model = config.model || DEFAULT_MODEL;\n  const provider = config.provider;\n  const errors: Array<{ message: string }> = [];\n\n  let finishMeta: AgentFinish | undefined;\n  let terminationReason: AgentRunResult['terminationReason'] = 'terminated';\n\n  for (let turn = 1; turn <= maxTurns; turn += 1) {\n    const turnMetrics: WarpGrepTurnMetrics = { turn, morph_api_ms: 0, local_tools_ms: 0 };\n    \n    // Enforce hard context limit before calling model\n    enforceContextLimit(messages);\n    \n    // call model\n    const modelCallStart = Date.now();\n    const assistantContent = await callModel(messages, model, {\n      morphApiKey: config.morphApiKey,\n      morphApiUrl: config.morphApiUrl,\n      retryConfig: config.retryConfig,\n      timeout: timeoutMs,\n    }).catch((e: unknown) => {\n      errors.push({ message: e instanceof Error ? e.message : String(e) });\n      return '';\n    });\n    turnMetrics.morph_api_ms = Date.now() - modelCallStart;\n    \n    if (!assistantContent) {\n      timings.turns!.push(turnMetrics);\n      break;\n    }\n    messages.push({ role: 'assistant', content: assistantContent });\n\n    // parse tool calls (no longer throws - returns _skip calls for malformed commands)\n    const toolCalls = parser.parse(assistantContent);\n    if (toolCalls.length === 0) {\n      errors.push({ message: 'No tool calls produced by the model. Your MCP is likely out of date! Update it by running: rm -rf ~/.npm/_npx && npm cache clean --force && npx -y @morphllm/morphmcp@latest' });\n      terminationReason = 'terminated';\n      timings.turns!.push(turnMetrics);\n      break;\n    }\n\n    const finishCalls = toolCalls.filter(c => c.name === 'finish');\n    const grepCalls = toolCalls.filter(c => c.name === 'grep');\n    const listDirCalls = toolCalls.filter(c => c.name === 'list_directory');\n    const readCalls = toolCalls.filter(c => c.name === 'read');\n    const skipCalls = toolCalls.filter(c => c.name === '_skip');\n\n    const formatted: string[] = [];\n\n    // Surface any skipped commands as feedback to the LLM\n    for (const c of skipCalls) {\n      const msg = (c.arguments as { message?: string })?.message || 'Command skipped due to parsing error';\n      formatted.push(msg);\n    }\n\n    const allPromises: Array<Promise<string>> = [];\n    \n    for (const c of grepCalls) {\n      const args = (c.arguments ?? {}) as { pattern: string; path: string; glob?: string };\n      allPromises.push(\n        toolGrep(provider, args).then(\n          ({ output }) => formatAgentToolOutput('grep', args, output, { isError: false }),\n          err => formatAgentToolOutput('grep', args, String(err), { isError: true })\n        )\n      );\n    }\n    \n    for (const c of listDirCalls) {\n      const args = (c.arguments ?? {}) as { path: string; pattern?: string | null };\n      allPromises.push(\n        toolListDirectory(provider, args).then(\n          p => formatAgentToolOutput('list_directory', args, p, { isError: false }),\n          err => formatAgentToolOutput('list_directory', args, String(err), { isError: true })\n        )\n      );\n    }\n    \n    for (const c of readCalls) {\n      const args = (c.arguments ?? {}) as { path: string; start?: number; end?: number; lines?: Array<[number, number]> };\n      allPromises.push(\n        toolRead(provider, args).then(\n          p => formatAgentToolOutput('read', args, p, { isError: false }),\n          err => formatAgentToolOutput('read', args, String(err), { isError: true })\n        )\n      );\n    }\n    \n    const toolExecStart = Date.now();\n    const allResults = await Promise.all(allPromises);\n    turnMetrics.local_tools_ms = Date.now() - toolExecStart;\n    \n    for (const result of allResults) {\n      formatted.push(result);\n    }\n\n    if (formatted.length > 0) {\n      const turnMessage = formatTurnMessage(turn, maxTurns);\n      const contextBudget = calculateContextBudget(messages);\n      messages.push({ role: 'user', content: formatted.join('\\n') + turnMessage + '\\n' + contextBudget });\n    }\n\n    timings.turns!.push(turnMetrics);\n\n    if (finishCalls.length) {\n      const fc = finishCalls[0];\n      const files = ((fc.arguments as any)?.files ?? []) as AgentFinish['files'];\n      finishMeta = { files };\n      terminationReason = 'completed';\n      break;\n    }\n  }\n\n  if (terminationReason !== 'completed' || !finishMeta) {\n    timings.total_ms = Date.now() - totalStart;\n    return { terminationReason, messages, errors, timings: timings as WarpGrepExecutionMetrics };\n  }\n\n  // Build finish payload\n  const parts: string[] = ['Relevant context found:'];\n  for (const f of finishMeta.files) {\n    const ranges = f.lines === '*' ? '*' \n      : Array.isArray(f.lines) ? f.lines.map(([s, e]) => `${s}-${e}`).join(', ') \n      : '*';\n    parts.push(`- ${f.path}: ${ranges}`);\n  }\n  const payload = parts.join('\\n');\n\n  // Resolve file contents for returned ranges\n  // Wrap reader in try-catch to handle non-existent or unreadable files gracefully\n  // Track files that couldn't be read for error reporting\n  const finishResolutionStart = Date.now();\n  const fileReadErrors: Array<{ path: string; error: string }> = [];\n  const resolved = await readFinishFiles(\n    repoRoot,\n    finishMeta.files,\n    async (p: string, s?: number, e?: number) => {\n      try {\n        const rr = await provider.read({ path: p, start: s, end: e });\n        // rr.lines are \"line|content\" → strip the \"line|\" prefix\n        return rr.lines.map(l => {\n          const idx = l.indexOf('|');\n          return idx >= 0 ? l.slice(idx + 1) : l;\n        });\n      } catch (err) {\n        // File doesn't exist or can't be read - log error but don't throw\n        // This handles cases where the agent hallucinated a path or the file was deleted\n        const errorMsg = err instanceof Error ? err.message : String(err);\n        fileReadErrors.push({ path: p, error: errorMsg });\n        console.error(`[warp_grep] Failed to read file: ${p} - ${errorMsg}`);\n        return [`[couldn't find: ${p}]`];\n      }\n    }\n  );\n\n  timings.finish_resolution_ms = Date.now() - finishResolutionStart;\n\n  // Add file read errors to the result so MCP can report them\n  if (fileReadErrors.length > 0) {\n    errors.push(...fileReadErrors.map(e => ({ message: `File read error: ${e.path} - ${e.error}` })));\n  }\n\n  timings.total_ms = Date.now() - totalStart;\n  return {\n    terminationReason: 'completed',\n    messages,\n    finish: { payload, metadata: finishMeta, resolved },\n    timings: timings as WarpGrepExecutionMetrics,\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAYA,OAAO,UAAU;AAWjB,IAAM,SAAS,IAAI,kBAAkB;AAErC,IAAM,kBAAkB;AASxB,eAAe,UACb,UACA,OACA,UAA4B,CAAC,GACZ;AACjB,QAAM,UAAU,QAAQ,eAAe;AACvC,QAAM,SAAS,QAAQ,eAAe,QAAQ,IAAI,iBAAiB;AAEnE,QAAM,eAAe;AAAA,IACnB,GAAG,OAAO;AAAA,IACV;AAAA,MACE,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,gBAAgB;AAAA,QAChB,eAAe,UAAU,MAAM;AAAA,MACjC;AAAA,MACA,MAAM,KAAK,UAAU;AAAA,QACnB;AAAA,QACA,aAAa;AAAA,QACb,YAAY;AAAA,QACZ;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,QAAQ;AAAA,EACV;AACA,QAAM,YAAY,QAAQ,WAAW,aAAa;AAClD,QAAM,OAAO,MAAM,YAAY,cAAc,WAAW,mCAAmC;AAC3F,MAAI,CAAC,KAAK,IAAI;AACZ,QAAI,KAAK,WAAW,KAAK;AACvB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,IAAI,MAAM,KAAK,KAAK;AAC1B,UAAM,IAAI,MAAM,yBAAyB,KAAK,MAAM,KAAK,CAAC,EAAE;AAAA,EAC9D;AACA,QAAM,OAAO,MAAM,KAAK,KAAK;AAC7B,QAAM,UAAU,MAAM,UAAU,CAAC,GAAG,SAAS;AAC7C,MAAI,CAAC,WAAW,OAAO,YAAY,UAAU;AAC3C,UAAM,IAAI,MAAM,6BAA6B;AAAA,EAC/C;AACA,SAAO;AACT;AAEA,eAAsB,YAAY,QAAiF;AACjH,QAAM,aAAa,KAAK,IAAI;AAC5B,QAAM,YAAY,OAAO,WAAW,aAAa;AACjD,QAAM,UAA6C,EAAE,OAAO,CAAC,GAAG,YAAY,UAAU;AAEtF,QAAM,WAAW,KAAK,QAAQ,OAAO,YAAY,QAAQ,IAAI,CAAC;AAC9D,QAAM,WAA0B,CAAC;AAEjC,WAAS,KAAK,EAAE,MAAM,UAAmB,SAAS,gBAAgB,EAAE,CAAC;AAErE,QAAM,oBAAoB,KAAK,IAAI;AACnC,QAAM,eAAe,MAAM,kBAAkB,UAAU,OAAO,OAAO,OAAO,QAAQ;AACpF,UAAQ,mBAAmB,KAAK,IAAI,IAAI;AAExC,WAAS,KAAK,EAAE,MAAM,QAAQ,SAAS,aAAa,CAAC;AAErD,QAAM,WAAW,aAAa;AAC9B,QAAM,QAAQ,OAAO,SAAS;AAC9B,QAAM,WAAW,OAAO;AACxB,QAAM,SAAqC,CAAC;AAE5C,MAAI;AACJ,MAAI,oBAAyD;AAE7D,WAAS,OAAO,GAAG,QAAQ,UAAU,QAAQ,GAAG;AAC9C,UAAM,cAAmC,EAAE,MAAM,cAAc,GAAG,gBAAgB,EAAE;AAGpF,wBAAoB,QAAQ;AAG5B,UAAM,iBAAiB,KAAK,IAAI;AAChC,UAAM,mBAAmB,MAAM,UAAU,UAAU,OAAO;AAAA,MACxD,aAAa,OAAO;AAAA,MACpB,aAAa,OAAO;AAAA,MACpB,aAAa,OAAO;AAAA,MACpB,SAAS;AAAA,IACX,CAAC,EAAE,MAAM,CAAC,MAAe;AACvB,aAAO,KAAK,EAAE,SAAS,aAAa,QAAQ,EAAE,UAAU,OAAO,CAAC,EAAE,CAAC;AACnE,aAAO;AAAA,IACT,CAAC;AACD,gBAAY,eAAe,KAAK,IAAI,IAAI;AAExC,QAAI,CAAC,kBAAkB;AACrB,cAAQ,MAAO,KAAK,WAAW;AAC/B;AAAA,IACF;AACA,aAAS,KAAK,EAAE,MAAM,aAAa,SAAS,iBAAiB,CAAC;AAG9D,UAAM,YAAY,OAAO,MAAM,gBAAgB;AAC/C,QAAI,UAAU,WAAW,GAAG;AAC1B,aAAO,KAAK,EAAE,SAAS,+KAA+K,CAAC;AACvM,0BAAoB;AACpB,cAAQ,MAAO,KAAK,WAAW;AAC/B;AAAA,IACF;AAEA,UAAM,cAAc,UAAU,OAAO,OAAK,EAAE,SAAS,QAAQ;AAC7D,UAAM,YAAY,UAAU,OAAO,OAAK,EAAE,SAAS,MAAM;AACzD,UAAM,eAAe,UAAU,OAAO,OAAK,EAAE,SAAS,gBAAgB;AACtE,UAAM,YAAY,UAAU,OAAO,OAAK,EAAE,SAAS,MAAM;AACzD,UAAM,YAAY,UAAU,OAAO,OAAK,EAAE,SAAS,OAAO;AAE1D,UAAM,YAAsB,CAAC;AAG7B,eAAW,KAAK,WAAW;AACzB,YAAM,MAAO,EAAE,WAAoC,WAAW;AAC9D,gBAAU,KAAK,GAAG;AAAA,IACpB;AAEA,UAAM,cAAsC,CAAC;AAE7C,eAAW,KAAK,WAAW;AACzB,YAAM,OAAQ,EAAE,aAAa,CAAC;AAC9B,kBAAY;AAAA,QACV,SAAS,UAAU,IAAI,EAAE;AAAA,UACvB,CAAC,EAAE,OAAO,MAAM,sBAAsB,QAAQ,MAAM,QAAQ,EAAE,SAAS,MAAM,CAAC;AAAA,UAC9E,SAAO,sBAAsB,QAAQ,MAAM,OAAO,GAAG,GAAG,EAAE,SAAS,KAAK,CAAC;AAAA,QAC3E;AAAA,MACF;AAAA,IACF;AAEA,eAAW,KAAK,cAAc;AAC5B,YAAM,OAAQ,EAAE,aAAa,CAAC;AAC9B,kBAAY;AAAA,QACV,kBAAkB,UAAU,IAAI,EAAE;AAAA,UAChC,OAAK,sBAAsB,kBAAkB,MAAM,GAAG,EAAE,SAAS,MAAM,CAAC;AAAA,UACxE,SAAO,sBAAsB,kBAAkB,MAAM,OAAO,GAAG,GAAG,EAAE,SAAS,KAAK,CAAC;AAAA,QACrF;AAAA,MACF;AAAA,IACF;AAEA,eAAW,KAAK,WAAW;AACzB,YAAM,OAAQ,EAAE,aAAa,CAAC;AAC9B,kBAAY;AAAA,QACV,SAAS,UAAU,IAAI,EAAE;AAAA,UACvB,OAAK,sBAAsB,QAAQ,MAAM,GAAG,EAAE,SAAS,MAAM,CAAC;AAAA,UAC9D,SAAO,sBAAsB,QAAQ,MAAM,OAAO,GAAG,GAAG,EAAE,SAAS,KAAK,CAAC;AAAA,QAC3E;AAAA,MACF;AAAA,IACF;AAEA,UAAM,gBAAgB,KAAK,IAAI;AAC/B,UAAM,aAAa,MAAM,QAAQ,IAAI,WAAW;AAChD,gBAAY,iBAAiB,KAAK,IAAI,IAAI;AAE1C,eAAW,UAAU,YAAY;AAC/B,gBAAU,KAAK,MAAM;AAAA,IACvB;AAEA,QAAI,UAAU,SAAS,GAAG;AACxB,YAAM,cAAc,kBAAkB,MAAM,QAAQ;AACpD,YAAM,gBAAgB,uBAAuB,QAAQ;AACrD,eAAS,KAAK,EAAE,MAAM,QAAQ,SAAS,UAAU,KAAK,IAAI,IAAI,cAAc,OAAO,cAAc,CAAC;AAAA,IACpG;AAEA,YAAQ,MAAO,KAAK,WAAW;AAE/B,QAAI,YAAY,QAAQ;AACtB,YAAM,KAAK,YAAY,CAAC;AACxB,YAAM,QAAU,GAAG,WAAmB,SAAS,CAAC;AAChD,mBAAa,EAAE,MAAM;AACrB,0BAAoB;AACpB;AAAA,IACF;AAAA,EACF;AAEA,MAAI,sBAAsB,eAAe,CAAC,YAAY;AACpD,YAAQ,WAAW,KAAK,IAAI,IAAI;AAChC,WAAO,EAAE,mBAAmB,UAAU,QAAQ,QAA6C;AAAA,EAC7F;AAGA,QAAM,QAAkB,CAAC,yBAAyB;AAClD,aAAW,KAAK,WAAW,OAAO;AAChC,UAAM,SAAS,EAAE,UAAU,MAAM,MAC7B,MAAM,QAAQ,EAAE,KAAK,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,IAAI,IACvE;AACJ,UAAM,KAAK,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE;AAAA,EACrC;AACA,QAAM,UAAU,MAAM,KAAK,IAAI;AAK/B,QAAM,wBAAwB,KAAK,IAAI;AACvC,QAAM,iBAAyD,CAAC;AAChE,QAAM,WAAW,MAAM;AAAA,IACrB;AAAA,IACA,WAAW;AAAA,IACX,OAAO,GAAW,GAAY,MAAe;AAC3C,UAAI;AACF,cAAM,KAAK,MAAM,SAAS,KAAK,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,EAAE,CAAC;AAE5D,eAAO,GAAG,MAAM,IAAI,OAAK;AACvB,gBAAM,MAAM,EAAE,QAAQ,GAAG;AACzB,iBAAO,OAAO,IAAI,EAAE,MAAM,MAAM,CAAC,IAAI;AAAA,QACvC,CAAC;AAAA,MACH,SAAS,KAAK;AAGZ,cAAM,WAAW,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAChE,uBAAe,KAAK,EAAE,MAAM,GAAG,OAAO,SAAS,CAAC;AAChD,gBAAQ,MAAM,oCAAoC,CAAC,MAAM,QAAQ,EAAE;AACnE,eAAO,CAAC,mBAAmB,CAAC,GAAG;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAEA,UAAQ,uBAAuB,KAAK,IAAI,IAAI;AAG5C,MAAI,eAAe,SAAS,GAAG;AAC7B,WAAO,KAAK,GAAG,eAAe,IAAI,QAAM,EAAE,SAAS,oBAAoB,EAAE,IAAI,MAAM,EAAE,KAAK,GAAG,EAAE,CAAC;AAAA,EAClG;AAEA,UAAQ,WAAW,KAAK,IAAI,IAAI;AAChC,SAAO;AAAA,IACL,mBAAmB;AAAA,IACnB;AAAA,IACA,QAAQ,EAAE,SAAS,UAAU,YAAY,SAAS;AAAA,IAClD;AAAA,EACF;AACF;","names":[]}

package/dist/{chunk-24EYSWME.js → chunk-LX2WNS3L.js}

@@ -5,7 +5,7 @@ import {
 import {
   executeToolCall,
   formatResult
-} from "./chunk-FJKPMMNQ.js";
+} from "./chunk-D4EQYM6O.js";
 import {
   getSystemPrompt
 } from "./chunk-FMLHRJDF.js";
@@ -58,4 +58,4 @@ export {
   createWarpGrepTool,
   openai_default
 };
-//# sourceMappingURL=chunk-24EYSWME.js.map
+//# sourceMappingURL=chunk-LX2WNS3L.js.map

package/dist/{chunk-3ONNAQZU.js → chunk-MJ7JODAY.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-KW7OEGZK.js";
 import {
   executeToolCall
-} from "./chunk-FJKPMMNQ.js";
+} from "./chunk-D4EQYM6O.js";
 
 // tools/warp_grep/vercel.ts
 import { tool } from "ai";
@@ -38,4 +38,4 @@ export {
   createWarpGrepTool,
   vercel_default
 };
-//# sourceMappingURL=chunk-3ONNAQZU.js.map
+//# sourceMappingURL=chunk-MJ7JODAY.js.map

package/dist/{chunk-EK5ZEOI3.js → chunk-N3RNM4A4.js}

@@ -1,15 +1,15 @@
 import {
   createWarpGrepTool as createWarpGrepTool2
-} from "./chunk-ZLJAODDJ.js";
+} from "./chunk-262WRPS5.js";
 import {
   createWarpGrepTool
-} from "./chunk-24EYSWME.js";
+} from "./chunk-LX2WNS3L.js";
 import {
   createWarpGrepTool as createWarpGrepTool3
-} from "./chunk-3ONNAQZU.js";
+} from "./chunk-MJ7JODAY.js";
 import {
   WarpGrepClient
-} from "./chunk-FJKPMMNQ.js";
+} from "./chunk-D4EQYM6O.js";
 import {
   createCodebaseSearchTool as createCodebaseSearchTool3
 } from "./chunk-UBX7QYBD.js";
@@ -280,4 +280,4 @@ export {
   VercelToolFactory,
   MorphClient
 };
-//# sourceMappingURL=chunk-EK5ZEOI3.js.map
+//# sourceMappingURL=chunk-N3RNM4A4.js.map

package/dist/client.cjs

@@ -1597,14 +1597,23 @@ async function toolRead(provider, args) {
 
 // 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 ?? AGENT_CONFIG.MAX_OUTPUT_LINES,
-    maxDepth: args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH
-  });
+  const maxResults = args.maxResults ?? AGENT_CONFIG.MAX_OUTPUT_LINES;
+  const initialDepth = args.maxDepth ?? AGENT_CONFIG.MAX_LIST_DEPTH;
+  async function getListRecursive(currentDepth) {
+    const entries = await provider.listDirectory({
+      path: args.path,
+      pattern: args.pattern ?? null,
+      maxResults,
+      maxDepth: currentDepth
+    });
+    if (entries.length >= maxResults && currentDepth > 0) {
+      return getListRecursive(currentDepth - 1);
+    }
+    return { entries };
+  }
+  const { entries: list } = await getListRecursive(initialDepth);
   if (!list.length) return "empty";
-  if (list.length >= AGENT_CONFIG.MAX_OUTPUT_LINES) {
+  if (list.length >= maxResults) {
     return "query not specific enough, tool called tried to return too much context and failed";
   }
   return list.map((e) => {
@@ -1882,10 +1891,15 @@ async function callModel(messages, model, options = {}) {
   return content;
 }
 async function runWarpGrep(config) {
+  const totalStart = Date.now();
+  const timeoutMs = config.timeout ?? AGENT_CONFIG.TIMEOUT_MS;
+  const timings = { turns: [], timeout_ms: timeoutMs };
   const repoRoot = import_path3.default.resolve(config.repoRoot || process.cwd());
   const messages = [];
   messages.push({ role: "system", content: getSystemPrompt() });
+  const initialStateStart = Date.now();
   const initialState = await buildInitialState(repoRoot, config.query, config.provider);
+  timings.initial_state_ms = Date.now() - initialStateStart;
   messages.push({ role: "user", content: initialState });
   const maxTurns = AGENT_CONFIG.MAX_TURNS;
   const model = config.model || DEFAULT_MODEL;
@@ -1894,22 +1908,29 @@ async function runWarpGrep(config) {
   let finishMeta;
   let terminationReason = "terminated";
   for (let turn = 1; turn <= maxTurns; turn += 1) {
+    const turnMetrics = { turn, morph_api_ms: 0, local_tools_ms: 0 };
     enforceContextLimit(messages);
+    const modelCallStart = Date.now();
     const assistantContent = await callModel(messages, model, {
       morphApiKey: config.morphApiKey,
       morphApiUrl: config.morphApiUrl,
       retryConfig: config.retryConfig,
-      timeout: config.timeout
+      timeout: timeoutMs
     }).catch((e) => {
       errors.push({ message: e instanceof Error ? e.message : String(e) });
       return "";
     });
-    if (!assistantContent) break;
+    turnMetrics.morph_api_ms = Date.now() - modelCallStart;
+    if (!assistantContent) {
+      timings.turns.push(turnMetrics);
+      break;
+    }
     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. Your MCP is likely out of date! Update it by running: rm -rf ~/.npm/_npx && npm cache clean --force && npx -y @morphllm/morphmcp@latest" });
       terminationReason = "terminated";
+      timings.turns.push(turnMetrics);
       break;
     }
     const finishCalls = toolCalls.filter((c) => c.name === "finish");
@@ -1950,7 +1971,9 @@ async function runWarpGrep(config) {
         )
       );
     }
+    const toolExecStart = Date.now();
     const allResults = await Promise.all(allPromises);
+    turnMetrics.local_tools_ms = Date.now() - toolExecStart;
     for (const result of allResults) {
       formatted.push(result);
     }
@@ -1959,6 +1982,7 @@ async function runWarpGrep(config) {
       const contextBudget = calculateContextBudget(messages);
       messages.push({ role: "user", content: formatted.join("\n") + turnMessage + "\n" + contextBudget });
     }
+    timings.turns.push(turnMetrics);
     if (finishCalls.length) {
       const fc = finishCalls[0];
       const files = fc.arguments?.files ?? [];
@@ -1968,7 +1992,8 @@ async function runWarpGrep(config) {
     }
   }
   if (terminationReason !== "completed" || !finishMeta) {
-    return { terminationReason, messages, errors };
+    timings.total_ms = Date.now() - totalStart;
+    return { terminationReason, messages, errors, timings };
   }
   const parts = ["Relevant context found:"];
   for (const f of finishMeta.files) {
@@ -1976,6 +2001,7 @@ async function runWarpGrep(config) {
     parts.push(`- ${f.path}: ${ranges}`);
   }
   const payload = parts.join("\n");
+  const finishResolutionStart = Date.now();
   const fileReadErrors = [];
   const resolved = await readFinishFiles(
     repoRoot,
@@ -1995,13 +2021,16 @@ async function runWarpGrep(config) {
       }
     }
   );
+  timings.finish_resolution_ms = Date.now() - finishResolutionStart;
   if (fileReadErrors.length > 0) {
     errors.push(...fileReadErrors.map((e) => ({ message: `File read error: ${e.path} - ${e.error}` })));
   }
+  timings.total_ms = Date.now() - totalStart;
   return {
     terminationReason: "completed",
     messages,
-    finish: { payload, metadata: finishMeta, resolved }
+    finish: { payload, metadata: finishMeta, resolved },
+    timings
   };
 }