@morphllm/morphsdk 0.2.81 → 0.2.83

package/dist/{chunk-4NI25AKV.js → chunk-4KMBU6T3.js}

@@ -14,7 +14,7 @@ import {
 import {
   AGENT_CONFIG,
   DEFAULT_EXCLUDES
-} from "./chunk-XT5ZO6ES.js";
+} from "./chunk-5PNMAWLC.js";
 
 // tools/warp_grep/providers/local.ts
 import fs from "fs/promises";
@@ -289,4 +289,4 @@ Details: ${res.stderr}` : ""}`
 export {
   LocalRipgrepProvider
 };
-//# sourceMappingURL=chunk-4NI25AKV.js.map
+//# sourceMappingURL=chunk-4KMBU6T3.js.map

package/dist/{chunk-XT5ZO6ES.js → chunk-5PNMAWLC.js}

@@ -1,7 +1,13 @@
 // tools/warp_grep/agent/config.ts
+var parseEnvTimeout = (envValue, defaultMs) => {
+  if (!envValue) return defaultMs;
+  const parsed = parseInt(envValue, 10);
+  return isNaN(parsed) || parsed <= 0 ? defaultMs : parsed;
+};
 var AGENT_CONFIG = {
   MAX_TURNS: 4,
-  TIMEOUT_MS: 3e4,
+  /** Default timeout for model calls. Can be overridden via MORPH_WARP_GREP_TIMEOUT env var (in ms) */
+  TIMEOUT_MS: parseEnvTimeout(process.env.MORPH_WARP_GREP_TIMEOUT, 3e4),
   MAX_CONTEXT_CHARS: 54e4,
   MAX_OUTPUT_LINES: 200,
   MAX_READ_LINES: 800,
@@ -95,4 +101,4 @@ export {
   DEFAULT_EXCLUDES,
   DEFAULT_MODEL
 };
-//# sourceMappingURL=chunk-XT5ZO6ES.js.map
+//# sourceMappingURL=chunk-5PNMAWLC.js.map

package/dist/chunk-5PNMAWLC.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../tools/warp_grep/agent/config.ts"],"sourcesContent":["// Parse timeout from env, defaulting to 30 seconds\nconst parseEnvTimeout = (envValue: string | undefined, defaultMs: number): number => {\n  if (!envValue) return defaultMs;\n  const parsed = parseInt(envValue, 10);\n  return isNaN(parsed) || parsed <= 0 ? defaultMs : parsed;\n};\n\nexport const AGENT_CONFIG = {\n  MAX_TURNS: 4,\n  /** Default timeout for model calls. Can be overridden via MORPH_WARP_GREP_TIMEOUT env var (in ms) */\n  TIMEOUT_MS: parseEnvTimeout(process.env.MORPH_WARP_GREP_TIMEOUT, 30_000),\n  MAX_CONTEXT_CHARS: 540_000,\n  MAX_OUTPUT_LINES: 200,\n  MAX_READ_LINES: 800,\n  MAX_LIST_DEPTH: 3,\n  LIST_TIMEOUT_MS: 2_000,\n};\n\n/**\n * Comprehensive exclusion list for directories and files\n * These patterns are used with ripgrep's -g flag\n */\nconst BUILTIN_EXCLUDES = [\n  // Version control\n  '.git', '.svn', '.hg', '.bzr',\n  \n  // Dependencies\n  'node_modules', 'bower_components', '.pnpm', '.yarn',\n  'vendor', 'packages', 'Pods', '.bundle',\n  \n  // Python\n  '__pycache__', '.pytest_cache', '.mypy_cache', '.ruff_cache',\n  '.venv', 'venv', '.tox', '.nox', '.eggs', '*.egg-info',\n  \n  // Build outputs\n  'dist', 'build', 'out', 'output', 'target', '_build',\n  '.next', '.nuxt', '.output', '.vercel', '.netlify',\n  \n  // Cache directories\n  '.cache', '.parcel-cache', '.turbo', '.nx', '.gradle',\n  \n  // IDE/Editor\n  '.idea', '.vscode', '.vs',\n  \n  // Coverage\n  'coverage', '.coverage', 'htmlcov', '.nyc_output',\n  \n  // Temporary\n  'tmp', 'temp', '.tmp', '.temp',\n  \n  // Lock files\n  'package-lock.json', 'yarn.lock', 'pnpm-lock.yaml', 'bun.lockb',\n  'Cargo.lock', 'Gemfile.lock', 'poetry.lock',\n  \n  // Binary/minified\n  '*.min.js', '*.min.css', '*.bundle.js',\n  '*.wasm', '*.so', '*.dll', '*.pyc',\n  '*.map', '*.js.map',\n  \n  // Hidden directories catch-all\n  '.*',\n];\n\nexport const DEFAULT_EXCLUDES = (process.env.MORPH_WARP_GREP_EXCLUDE || '')\n  .split(',')\n  .map(s => s.trim())\n  .filter(Boolean)\n  .concat(BUILTIN_EXCLUDES);\n\nexport const DEFAULT_MODEL = 'morph-warp-grep-v1';\n\n\n"],"mappings":";AACA,IAAM,kBAAkB,CAAC,UAA8B,cAA8B;AACnF,MAAI,CAAC,SAAU,QAAO;AACtB,QAAM,SAAS,SAAS,UAAU,EAAE;AACpC,SAAO,MAAM,MAAM,KAAK,UAAU,IAAI,YAAY;AACpD;AAEO,IAAM,eAAe;AAAA,EAC1B,WAAW;AAAA;AAAA,EAEX,YAAY,gBAAgB,QAAQ,IAAI,yBAAyB,GAAM;AAAA,EACvE,mBAAmB;AAAA,EACnB,kBAAkB;AAAA,EAClB,gBAAgB;AAAA,EAChB,gBAAgB;AAAA,EAChB,iBAAiB;AACnB;AAMA,IAAM,mBAAmB;AAAA;AAAA,EAEvB;AAAA,EAAQ;AAAA,EAAQ;AAAA,EAAO;AAAA;AAAA,EAGvB;AAAA,EAAgB;AAAA,EAAoB;AAAA,EAAS;AAAA,EAC7C;AAAA,EAAU;AAAA,EAAY;AAAA,EAAQ;AAAA;AAAA,EAG9B;AAAA,EAAe;AAAA,EAAiB;AAAA,EAAe;AAAA,EAC/C;AAAA,EAAS;AAAA,EAAQ;AAAA,EAAQ;AAAA,EAAQ;AAAA,EAAS;AAAA;AAAA,EAG1C;AAAA,EAAQ;AAAA,EAAS;AAAA,EAAO;AAAA,EAAU;AAAA,EAAU;AAAA,EAC5C;AAAA,EAAS;AAAA,EAAS;AAAA,EAAW;AAAA,EAAW;AAAA;AAAA,EAGxC;AAAA,EAAU;AAAA,EAAiB;AAAA,EAAU;AAAA,EAAO;AAAA;AAAA,EAG5C;AAAA,EAAS;AAAA,EAAW;AAAA;AAAA,EAGpB;AAAA,EAAY;AAAA,EAAa;AAAA,EAAW;AAAA;AAAA,EAGpC;AAAA,EAAO;AAAA,EAAQ;AAAA,EAAQ;AAAA;AAAA,EAGvB;AAAA,EAAqB;AAAA,EAAa;AAAA,EAAkB;AAAA,EACpD;AAAA,EAAc;AAAA,EAAgB;AAAA;AAAA,EAG9B;AAAA,EAAY;AAAA,EAAa;AAAA,EACzB;AAAA,EAAU;AAAA,EAAQ;AAAA,EAAS;AAAA,EAC3B;AAAA,EAAS;AAAA;AAAA,EAGT;AACF;AAEO,IAAM,oBAAoB,QAAQ,IAAI,2BAA2B,IACrE,MAAM,GAAG,EACT,IAAI,OAAK,EAAE,KAAK,CAAC,EACjB,OAAO,OAAO,EACd,OAAO,gBAAgB;AAEnB,IAAM,gBAAgB;","names":[]}

package/dist/{chunk-X6A64F2F.js → chunk-HPR6BJA7.js}

@@ -1,15 +1,15 @@
 import {
   createWarpGrepTool as createWarpGrepTool2
-} from "./chunk-Z2HDXUH7.js";
+} from "./chunk-RL4JBZ3T.js";
 import {
   createWarpGrepTool
-} from "./chunk-NTTQJM6O.js";
+} from "./chunk-LXG37353.js";
 import {
   createWarpGrepTool as createWarpGrepTool3
-} from "./chunk-OFMDENAQ.js";
+} from "./chunk-S27A4NQM.js";
 import {
   WarpGrepClient
-} from "./chunk-4VMIPD4G.js";
+} from "./chunk-HRK53IKL.js";
 import {
   createCodebaseSearchTool as createCodebaseSearchTool3
 } from "./chunk-UBX7QYBD.js";
@@ -280,4 +280,4 @@ export {
   VercelToolFactory,
   MorphClient
 };
-//# sourceMappingURL=chunk-X6A64F2F.js.map
+//# sourceMappingURL=chunk-HPR6BJA7.js.map

package/dist/{chunk-4VMIPD4G.js → chunk-HRK53IKL.js}

@@ -1,12 +1,12 @@
 import {
   runWarpGrep
-} from "./chunk-B3UC7UVA.js";
+} from "./chunk-QOYI77CN.js";
 import {
   RemoteCommandsProvider
-} from "./chunk-PHJQ4N3T.js";
+} from "./chunk-PUGSTXLO.js";
 import {
   LocalRipgrepProvider
-} from "./chunk-4NI25AKV.js";
+} from "./chunk-4KMBU6T3.js";
 
 // tools/warp_grep/client.ts
 var WarpGrepClient = class {
@@ -53,7 +53,8 @@ var WarpGrepClient = class {
         debug: input.debug ?? this.config.debug,
         morphApiKey: this.config.morphApiKey,
         morphApiUrl: this.config.morphApiUrl,
-        retryConfig: this.config.retryConfig
+        retryConfig: this.config.retryConfig,
+        timeout: this.config.timeout
       }
     );
   }
@@ -74,7 +75,8 @@ async function executeToolCall(input, config) {
     debug: config.debug ?? false,
     morphApiKey: config.morphApiKey,
     morphApiUrl: config.morphApiUrl,
-    retryConfig: config.retryConfig
+    retryConfig: config.retryConfig,
+    timeout: config.timeout
   });
   const finish = result.finish;
   if (result.terminationReason !== "completed" || !finish?.metadata) {
@@ -115,4 +117,4 @@ export {
   executeToolCall,
   formatResult
 };
-//# sourceMappingURL=chunk-4VMIPD4G.js.map
+//# sourceMappingURL=chunk-HRK53IKL.js.map

package/dist/chunk-HRK53IKL.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../tools/warp_grep/client.ts"],"sourcesContent":["/**\n * WarpGrep client for programmatic code search\n */\n\nimport { runWarpGrep } from './agent/runner.js';\nimport { LocalRipgrepProvider } from './providers/local.js';\nimport { RemoteCommandsProvider } from './providers/remote.js';\nimport type { \n  WarpGrepClientConfig, \n  WarpGrepInput, \n  WarpGrepResult,\n  WarpGrepContext,\n  WarpGrepToolConfig,\n} from './types.js';\n\n/**\n * WarpGrep client for programmatic code search\n * \n * @example\n * ```typescript\n * import { WarpGrepClient } from '@morphllm/morphsdk';\n * \n * const client = new WarpGrepClient({ morphApiKey: process.env.MORPH_API_KEY });\n * \n * // Simple usage - defaults to LocalRipgrepProvider\n * const result = await client.execute({\n *   query: 'Find authentication middleware',\n *   repoRoot: '.'\n * });\n * \n * // With custom excludes\n * const result = await client.execute({\n *   query: 'Find database models',\n *   repoRoot: '.',\n *   excludes: ['node_modules', '.git', 'dist']\n * });\n * ```\n */\nexport class WarpGrepClient {\n  private config: WarpGrepClientConfig;\n\n  constructor(config: WarpGrepClientConfig = {}) {\n    this.config = {\n      morphApiKey: config.morphApiKey,\n      morphApiUrl: config.morphApiUrl,\n      debug: config.debug,\n      timeout: config.timeout,\n      retryConfig: config.retryConfig,\n    };\n  }\n\n  /**\n   * Execute a code search query\n   * \n   * @param input - Search parameters including query, repoRoot, and optional provider\n   * @returns Search results with relevant code contexts\n   * \n   * @example\n   * ```typescript\n   * const result = await client.execute({\n   *   query: 'Find authentication middleware',\n   *   repoRoot: '.'\n   * });\n   * \n   * if (result.success) {\n   *   for (const ctx of result.contexts) {\n   *     console.log(`File: ${ctx.file}`);\n   *     console.log(ctx.content);\n   *   }\n   * }\n   * ```\n   */\n  async execute(input: WarpGrepInput): Promise<WarpGrepResult> {\n    return executeToolCall(\n      { query: input.query },\n      {\n        repoRoot: input.repoRoot,\n        remoteCommands: input.remoteCommands,\n        provider: input.provider,\n        excludes: input.excludes,\n        includes: input.includes,\n        debug: input.debug ?? this.config.debug,\n        morphApiKey: this.config.morphApiKey,\n        morphApiUrl: this.config.morphApiUrl,\n        retryConfig: this.config.retryConfig,\n        timeout: this.config.timeout,\n      }\n    );\n  }\n}\n\n/**\n * Execute a warp grep search directly\n * \n * @param input - Search parameters\n * @param config - Optional client configuration\n * @returns Search results\n * \n * @example\n * ```typescript\n * import { executeWarpGrep } from '@morphllm/morphsdk/tools/warp-grep';\n * \n * const result = await executeWarpGrep({\n *   query: 'Find authentication middleware',\n *   repoRoot: '.'\n * });\n * ```\n */\nexport async function executeWarpGrep(\n  input: WarpGrepInput,\n  config?: WarpGrepClientConfig\n): Promise<WarpGrepResult> {\n  const client = new WarpGrepClient(config);\n  return client.execute(input);\n}\n\nexport async function executeToolCall(\n  input: { query: string } | string,\n  config: WarpGrepToolConfig\n): Promise<WarpGrepResult> {\n  const parsed = typeof input === 'string' ? JSON.parse(input) : input;\n  \n  // Priority: remoteCommands > provider > LocalRipgrepProvider\n  const provider = config.remoteCommands\n    ? new RemoteCommandsProvider(config.repoRoot, config.remoteCommands)\n    : config.provider ?? new LocalRipgrepProvider(config.repoRoot, config.excludes);\n  \n  const result = await runWarpGrep({\n    query: parsed.query,\n    repoRoot: config.repoRoot,\n    provider,\n    excludes: config.excludes,\n    includes: config.includes,\n    debug: config.debug ?? false,\n    morphApiKey: config.morphApiKey,\n    morphApiUrl: config.morphApiUrl,\n    retryConfig: config.retryConfig,\n    timeout: config.timeout,\n  });\n\n  const finish = result.finish;\n  if (result.terminationReason !== 'completed' || !finish?.metadata) {\n    return { success: false, error: 'Search did not complete' };\n  }\n\n  const contexts: WarpGrepContext[] = (finish.resolved ?? []).map(r => ({\n    file: r.path,\n    content: r.content,\n  }));\n\n  return { success: true, contexts, summary: finish.payload };\n}\n\n/**\n * Format warp grep results for display or tool responses\n * \n * @param result - The search result\n * @returns Formatted string representation\n */\nexport function formatResult(result: WarpGrepResult): string {\n  if (!result.success) {\n    return `Search failed: ${result.error}`;\n  }\n\n  if (!result.contexts || result.contexts.length === 0) {\n    return 'No relevant code found. Try rephrasing your query.';\n  }\n\n  const lines: string[] = [];\n  lines.push(`Found ${result.contexts.length} relevant code sections:\\n`);\n\n  result.contexts.forEach((ctx, i) => {\n    lines.push(`${i + 1}. ${ctx.file}`);\n    lines.push('```');\n    lines.push(ctx.content);\n    lines.push('```');\n    lines.push('');\n  });\n\n  if (result.summary) {\n    lines.push(`Summary: ${result.summary}`);\n  }\n\n  return lines.join('\\n');\n}\n\n"],"mappings":";;;;;;;;;;;AAsCO,IAAM,iBAAN,MAAqB;AAAA,EAClB;AAAA,EAER,YAAY,SAA+B,CAAC,GAAG;AAC7C,SAAK,SAAS;AAAA,MACZ,aAAa,OAAO;AAAA,MACpB,aAAa,OAAO;AAAA,MACpB,OAAO,OAAO;AAAA,MACd,SAAS,OAAO;AAAA,MAChB,aAAa,OAAO;AAAA,IACtB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAuBA,MAAM,QAAQ,OAA+C;AAC3D,WAAO;AAAA,MACL,EAAE,OAAO,MAAM,MAAM;AAAA,MACrB;AAAA,QACE,UAAU,MAAM;AAAA,QAChB,gBAAgB,MAAM;AAAA,QACtB,UAAU,MAAM;AAAA,QAChB,UAAU,MAAM;AAAA,QAChB,UAAU,MAAM;AAAA,QAChB,OAAO,MAAM,SAAS,KAAK,OAAO;AAAA,QAClC,aAAa,KAAK,OAAO;AAAA,QACzB,aAAa,KAAK,OAAO;AAAA,QACzB,aAAa,KAAK,OAAO;AAAA,QACzB,SAAS,KAAK,OAAO;AAAA,MACvB;AAAA,IACF;AAAA,EACF;AACF;AAmBA,eAAsB,gBACpB,OACA,QACyB;AACzB,QAAM,SAAS,IAAI,eAAe,MAAM;AACxC,SAAO,OAAO,QAAQ,KAAK;AAC7B;AAEA,eAAsB,gBACpB,OACA,QACyB;AACzB,QAAM,SAAS,OAAO,UAAU,WAAW,KAAK,MAAM,KAAK,IAAI;AAG/D,QAAM,WAAW,OAAO,iBACpB,IAAI,uBAAuB,OAAO,UAAU,OAAO,cAAc,IACjE,OAAO,YAAY,IAAI,qBAAqB,OAAO,UAAU,OAAO,QAAQ;AAEhF,QAAM,SAAS,MAAM,YAAY;AAAA,IAC/B,OAAO,OAAO;AAAA,IACd,UAAU,OAAO;AAAA,IACjB;AAAA,IACA,UAAU,OAAO;AAAA,IACjB,UAAU,OAAO;AAAA,IACjB,OAAO,OAAO,SAAS;AAAA,IACvB,aAAa,OAAO;AAAA,IACpB,aAAa,OAAO;AAAA,IACpB,aAAa,OAAO;AAAA,IACpB,SAAS,OAAO;AAAA,EAClB,CAAC;AAED,QAAM,SAAS,OAAO;AACtB,MAAI,OAAO,sBAAsB,eAAe,CAAC,QAAQ,UAAU;AACjE,WAAO,EAAE,SAAS,OAAO,OAAO,0BAA0B;AAAA,EAC5D;AAEA,QAAM,YAA+B,OAAO,YAAY,CAAC,GAAG,IAAI,QAAM;AAAA,IACpE,MAAM,EAAE;AAAA,IACR,SAAS,EAAE;AAAA,EACb,EAAE;AAEF,SAAO,EAAE,SAAS,MAAM,UAAU,SAAS,OAAO,QAAQ;AAC5D;AAQO,SAAS,aAAa,QAAgC;AAC3D,MAAI,CAAC,OAAO,SAAS;AACnB,WAAO,kBAAkB,OAAO,KAAK;AAAA,EACvC;AAEA,MAAI,CAAC,OAAO,YAAY,OAAO,SAAS,WAAW,GAAG;AACpD,WAAO;AAAA,EACT;AAEA,QAAM,QAAkB,CAAC;AACzB,QAAM,KAAK,SAAS,OAAO,SAAS,MAAM;AAAA,CAA4B;AAEtE,SAAO,SAAS,QAAQ,CAAC,KAAK,MAAM;AAClC,UAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,EAAE;AAClC,UAAM,KAAK,KAAK;AAChB,UAAM,KAAK,IAAI,OAAO;AACtB,UAAM,KAAK,KAAK;AAChB,UAAM,KAAK,EAAE;AAAA,EACf,CAAC;AAED,MAAI,OAAO,SAAS;AAClB,UAAM,KAAK,YAAY,OAAO,OAAO,EAAE;AAAA,EACzC;AAEA,SAAO,MAAM,KAAK,IAAI;AACxB;","names":[]}

package/dist/{chunk-NTTQJM6O.js → chunk-LXG37353.js}

@@ -5,7 +5,7 @@ import {
 import {
   executeToolCall,
   formatResult
-} from "./chunk-4VMIPD4G.js";
+} from "./chunk-HRK53IKL.js";
 import {
   getSystemPrompt
 } from "./chunk-FMLHRJDF.js";
@@ -58,4 +58,4 @@ export {
   createWarpGrepTool,
   openai_default
 };
-//# sourceMappingURL=chunk-NTTQJM6O.js.map
+//# sourceMappingURL=chunk-LXG37353.js.map

package/dist/{chunk-467MCW5R.js → chunk-P2O5JKE5.js}

@@ -1,6 +1,6 @@
 import {
   AGENT_CONFIG
-} from "./chunk-XT5ZO6ES.js";
+} from "./chunk-5PNMAWLC.js";
 
 // tools/warp_grep/agent/tools/grep.ts
 async function toolGrep(provider, args) {
@@ -228,4 +228,4 @@ export {
   formatListDirectoryTree,
   enforceContextLimit
 };
-//# sourceMappingURL=chunk-467MCW5R.js.map
+//# sourceMappingURL=chunk-P2O5JKE5.js.map

package/dist/{chunk-PHJQ4N3T.js → chunk-PUGSTXLO.js}

@@ -1,6 +1,6 @@
 import {
   AGENT_CONFIG
-} from "./chunk-XT5ZO6ES.js";
+} from "./chunk-5PNMAWLC.js";
 
 // tools/warp_grep/providers/remote.ts
 var SKIP_NAMES = /* @__PURE__ */ new Set([
@@ -185,4 +185,4 @@ var RemoteCommandsProvider = class {
 export {
   RemoteCommandsProvider
 };
-//# sourceMappingURL=chunk-PHJQ4N3T.js.map
+//# sourceMappingURL=chunk-PUGSTXLO.js.map

package/dist/{chunk-B3UC7UVA.js → chunk-QOYI77CN.js}

@@ -7,7 +7,7 @@ import {
   toolGrep,
   toolListDirectory,
   toolRead
-} from "./chunk-467MCW5R.js";
+} from "./chunk-P2O5JKE5.js";
 import {
   formatAgentToolOutput
 } from "./chunk-APP75CBN.js";
@@ -20,7 +20,7 @@ import {
 import {
   AGENT_CONFIG,
   DEFAULT_MODEL
-} from "./chunk-XT5ZO6ES.js";
+} from "./chunk-5PNMAWLC.js";
 import {
   fetchWithRetry,
   withTimeout
@@ -50,7 +50,8 @@ async function callModel(messages, model, options = {}) {
     },
     options.retryConfig
   );
-  const resp = await withTimeout(fetchPromise, AGENT_CONFIG.TIMEOUT_MS, "morph-warp-grep request timed out");
+  const timeoutMs = options.timeout ?? AGENT_CONFIG.TIMEOUT_MS;
+  const resp = await withTimeout(fetchPromise, timeoutMs, "morph-warp-grep request timed out");
   if (!resp.ok) {
     if (resp.status === 404) {
       throw new Error(
@@ -68,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;
@@ -80,21 +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
+      retryConfig: config.retryConfig,
+      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");
@@ -135,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);
     }
@@ -144,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 ?? [];
@@ -153,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) {
@@ -161,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,
@@ -180,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-B3UC7UVA.js.map
+//# sourceMappingURL=chunk-QOYI77CN.js.map

package/dist/chunk-QOYI77CN.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-Z2HDXUH7.js → chunk-RL4JBZ3T.js}

@@ -5,7 +5,7 @@ import {
 import {
   executeToolCall,
   formatResult
-} from "./chunk-4VMIPD4G.js";
+} from "./chunk-HRK53IKL.js";
 import {
   getSystemPrompt
 } from "./chunk-FMLHRJDF.js";
@@ -50,4 +50,4 @@ export {
   execute,
   createWarpGrepTool
 };
-//# sourceMappingURL=chunk-Z2HDXUH7.js.map
+//# sourceMappingURL=chunk-RL4JBZ3T.js.map

package/dist/{chunk-OFMDENAQ.js → chunk-S27A4NQM.js}

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

package/dist/{client-CSINf0lQ.d.ts → client-CsO9LifG.d.ts}

@@ -7,7 +7,7 @@ import { MorphGit } from './git/client.js';
 import { OpenAIRouter, AnthropicRouter, GeminiRouter, RawRouter } from './modelrouter/core.js';
 import { EditFileConfig, EditFileInput, EditFileResult, EditChanges } from './tools/fastapply/types.js';
 import { CodebaseSearchConfig, CodebaseSearchInput, CodebaseSearchResult } from './tools/codebase_search/types.js';
-import { d as WarpGrepToolConfig, b as WarpGrepResult, c as WarpGrepContext } from './types-CnvVDM63.js';
+import { d as WarpGrepToolConfig, b as WarpGrepResult, c as WarpGrepContext } from './types-Cv4LpqVl.js';
 import * as openai_resources_index_mjs from 'openai/resources/index.mjs';
 import * as _anthropic_ai_sdk_resources_messages_mjs from '@anthropic-ai/sdk/resources/messages.mjs';
 import * as ai from 'ai';

package/dist/client.cjs

@@ -987,9 +987,15 @@ async function checkHealth(config = {}) {
 }
 
 // tools/warp_grep/agent/config.ts
+var parseEnvTimeout = (envValue, defaultMs) => {
+  if (!envValue) return defaultMs;
+  const parsed = parseInt(envValue, 10);
+  return isNaN(parsed) || parsed <= 0 ? defaultMs : parsed;
+};
 var AGENT_CONFIG = {
   MAX_TURNS: 4,
-  TIMEOUT_MS: 3e4,
+  /** Default timeout for model calls. Can be overridden via MORPH_WARP_GREP_TIMEOUT env var (in ms) */
+  TIMEOUT_MS: parseEnvTimeout(process.env.MORPH_WARP_GREP_TIMEOUT, 3e4),
   MAX_CONTEXT_CHARS: 54e4,
   MAX_OUTPUT_LINES: 200,
   MAX_READ_LINES: 800,
@@ -1857,7 +1863,8 @@ async function callModel(messages, model, options = {}) {
     },
     options.retryConfig
   );
-  const resp = await withTimeout(fetchPromise, AGENT_CONFIG.TIMEOUT_MS, "morph-warp-grep request timed out");
+  const timeoutMs = options.timeout ?? AGENT_CONFIG.TIMEOUT_MS;
+  const resp = await withTimeout(fetchPromise, timeoutMs, "morph-warp-grep request timed out");
   if (!resp.ok) {
     if (resp.status === 404) {
       throw new Error(
@@ -1875,10 +1882,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;
@@ -1887,21 +1899,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
+      retryConfig: config.retryConfig,
+      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");
@@ -1942,7 +1962,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);
     }
@@ -1951,6 +1973,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 ?? [];
@@ -1960,7 +1983,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) {
@@ -1968,6 +1992,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,
@@ -1987,13 +2012,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
   };
 }
 
@@ -2622,7 +2650,8 @@ var WarpGrepClient = class {
         debug: input.debug ?? this.config.debug,
         morphApiKey: this.config.morphApiKey,
         morphApiUrl: this.config.morphApiUrl,
-        retryConfig: this.config.retryConfig
+        retryConfig: this.config.retryConfig,
+        timeout: this.config.timeout
       }
     );
   }
@@ -2639,7 +2668,8 @@ async function executeToolCall(input, config) {
     debug: config.debug ?? false,
     morphApiKey: config.morphApiKey,
     morphApiUrl: config.morphApiUrl,
-    retryConfig: config.retryConfig
+    retryConfig: config.retryConfig,
+    timeout: config.timeout
   });
   const finish = result.finish;
   if (result.terminationReason !== "completed" || !finish?.metadata) {