@morphllm/morphsdk 0.2.82 → 0.2.84

package/dist/tools/warp_grep/anthropic.cjs

@@ -649,14 +649,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) => {
@@ -1008,10 +1017,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_path2.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;
@@ -1020,22 +1034,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");
@@ -1076,7 +1097,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);
     }
@@ -1085,6 +1108,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 ?? [];
@@ -1094,7 +1118,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) {
@@ -1102,6 +1127,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,
@@ -1121,13 +1147,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
   };
 }