kimiflare 0.66.0 → 0.67.0

package/dist/index.js

@@ -3150,6 +3150,15 @@ var init_system_prompt = __esm({
 });
 
 // src/agent/loop.ts
+function getSessionWebFetchHistory(sessionId) {
+  const key = sessionId ?? "default";
+  let arr = sessionWebFetchHistory.get(key);
+  if (!arr) {
+    arr = [];
+    sessionWebFetchHistory.set(key, arr);
+  }
+  return arr;
+}
 function isHighSignalMemory(memory) {
   return memory.topicKey === "project_dependencies" || memory.topicKey === "project_tsconfig" || memory.topicKey === "project_entry_point" || memory.category === "instruction" || memory.category === "preference" || memory.category === "event" && memory.importance >= 3;
 }
@@ -3304,7 +3313,8 @@ Use console.log() to return results. Only console.log output will be sent back t
   const recentToolCalls = [];
   const LOOP_WINDOW = 8;
   const LOOP_THRESHOLD = 2;
-  const webFetchHistory = [];
+  const webFetchHistory = getSessionWebFetchHistory(opts2.sessionId);
+  let webFetchesThisTurn = 0;
   const MAX_WEB_FETCH_PER_TURN = 5;
   const WEB_FETCH_DOMAIN_THRESHOLD = 2;
   let cumulativePromptTokens = 0;
@@ -3543,8 +3553,8 @@ Use console.log() to return results. Only console.log output will be sent back t
         try {
           const domain = new URL(url).hostname;
           const domainCount = webFetchHistory.filter((h) => h.domain === domain).length;
-          const totalWebFetches = webFetchHistory.length;
-          if (totalWebFetches >= MAX_WEB_FETCH_PER_TURN) {
+          const totalSessionFetches = webFetchHistory.length;
+          if (webFetchesThisTurn >= MAX_WEB_FETCH_PER_TURN) {
             const warning = `Research budget exceeded: you have already made ${MAX_WEB_FETCH_PER_TURN} web requests this turn. Synthesize what you have learned instead of fetching more pages.`;
             const budgetResult = {
               tool_call_id: tc.id,
@@ -3565,6 +3575,27 @@ Use console.log() to return results. Only console.log output will be sent back t
             blockedCount++;
             continue;
           }
+          if (totalSessionFetches >= SESSION_WEB_FETCH_CAP) {
+            const warning = `Session research budget exceeded: ${totalSessionFetches} web fetches across this session. Synthesize what you have learned from prior fetches instead of starting another page.`;
+            const sessionCapResult = {
+              tool_call_id: tc.id,
+              name: "web_fetch",
+              content: warning,
+              ok: false
+            };
+            toolResults.push(sessionCapResult);
+            opts2.messages.push({
+              role: "tool",
+              tool_call_id: tc.id,
+              content: sanitizeString(warning),
+              name: "web_fetch"
+            });
+            opts2.callbacks.onToolResult?.(sessionCapResult);
+            recentToolCalls.push(loopSignature);
+            if (recentToolCalls.length > LOOP_WINDOW) recentToolCalls.shift();
+            blockedCount++;
+            continue;
+          }
           if (domainCount >= WEB_FETCH_DOMAIN_THRESHOLD) {
             const warning = `Loop detected: you have fetched from ${domain} multiple times. Consider a different approach or synthesize existing findings.`;
             const loopResult = {
@@ -3587,6 +3618,7 @@ Use console.log() to return results. Only console.log output will be sent back t
             continue;
           }
           webFetchHistory.push({ url, domain });
+          webFetchesThisTurn++;
         } catch {
         }
       }
@@ -3622,9 +3654,16 @@ Use console.log() to return results. Only console.log output will be sent back t
 Output:
 ${sandboxResult.output}` : sandboxResult.output;
         if (resultContent.length > MAX_TOOL_CONTENT_CHARS) {
+          const rawBytes = resultContent.length;
           resultContent = resultContent.slice(0, MAX_TOOL_CONTENT_CHARS) + `
 
-[truncated: ${resultContent.length - MAX_TOOL_CONTENT_CHARS} chars omitted]`;
+[truncated: ${rawBytes - MAX_TOOL_CONTENT_CHARS} chars omitted]`;
+          opts2.callbacks.onTruncation?.({
+            tool: "execute_code",
+            toolCallId: tc.id,
+            rawBytes,
+            reducedBytes: resultContent.length
+          });
         }
         const result = {
           tool_call_id: tc.id,
@@ -3653,9 +3692,17 @@ ${sandboxResult.output}` : sandboxResult.output;
         );
         let content2 = result.content;
         if (content2.length > MAX_TOOL_CONTENT_CHARS) {
+          const rawBytes = content2.length;
           content2 = content2.slice(0, MAX_TOOL_CONTENT_CHARS) + `
 
-[truncated: ${content2.length - MAX_TOOL_CONTENT_CHARS} chars omitted]`;
+[truncated: ${rawBytes - MAX_TOOL_CONTENT_CHARS} chars omitted]`;
+          opts2.callbacks.onTruncation?.({
+            tool: tc.function.name,
+            toolCallId: tc.id,
+            rawBytes,
+            reducedBytes: content2.length,
+            artifactId: result.artifactId
+          });
         }
         logger.debug("turn:tool_end", { sessionId: opts2.sessionId, tool: tc.function.name, toolCallId: tc.id, ok: result.ok });
         toolResults.push(result);
@@ -3679,6 +3726,7 @@ ${sandboxResult.output}` : sandboxResult.output;
           );
           const assistantMessage = lastAssistant?.content ?? "";
           const llmOpts = opts2.memoryManager.getExtractionLlmOpts();
+          const turnAtMemoryCommit = turn;
           for (const extractor of EXTRACTORS) {
             if (extractor.match(tc.function.name, filePath)) {
               void (async () => {
@@ -3704,11 +3752,14 @@ ${sandboxResult.output}` : sandboxResult.output;
                     );
                     if (isHighSignalMemory(memory)) {
                       const sid = opts2.sessionId ?? "default";
-                      const current = (driftAccumulator.get(sid) ?? 0) + 1;
-                      driftAccumulator.set(sid, current);
-                      if (current >= DRIFT_THRESHOLD) {
+                      const events2 = driftEvents.get(sid) ?? [];
+                      events2.push(turnAtMemoryCommit);
+                      const cutoff = turnAtMemoryCommit - DRIFT_WINDOW + 1;
+                      const recent = events2.filter((t) => t >= cutoff);
+                      driftEvents.set(sid, recent);
+                      if (recent.length >= DRIFT_THRESHOLD) {
                         opts2.callbacks.onKimiMdStale?.();
-                        driftAccumulator.set(sid, 0);
+                        driftEvents.set(sid, []);
                       }
                     }
                   }
@@ -3740,12 +3791,6 @@ ${sandboxResult.output}` : sandboxResult.output;
     if (blockedCount === toolCalls.length && toolCalls.length > 0) {
       loopExhausted = true;
     }
-    if (opts2.sessionId) {
-      const current = driftAccumulator.get(opts2.sessionId) ?? 0;
-      if (current > 0) {
-        driftAccumulator.set(opts2.sessionId, Math.max(0, current - 1));
-      }
-    }
     if (opts2.onIterationEnd) {
       opts2.messages = await opts2.onIterationEnd(opts2.messages, opts2.signal);
       if (opts2.signal.aborted) throw new DOMException("aborted", "AbortError");
@@ -3804,7 +3849,7 @@ function validateToolArguments(raw) {
     return "{}";
   }
 }
-var BudgetExhaustedError, AgentLoopError, codeModeApiCache, driftAccumulator, DRIFT_THRESHOLD, memoryExtractionErrorCounts, MAX_PROMPT_TOKENS, MAX_TOOL_CONTENT_CHARS;
+var BudgetExhaustedError, AgentLoopError, codeModeApiCache, driftEvents, DRIFT_WINDOW, DRIFT_THRESHOLD, memoryExtractionErrorCounts, sessionWebFetchHistory, SESSION_WEB_FETCH_CAP, MAX_PROMPT_TOKENS, MAX_TOOL_CONTENT_CHARS;
 var init_loop = __esm({
   "src/agent/loop.ts"() {
     "use strict";
@@ -3832,9 +3877,12 @@ var init_loop = __esm({
       }
     };
     codeModeApiCache = /* @__PURE__ */ new Map();
-    driftAccumulator = /* @__PURE__ */ new Map();
-    DRIFT_THRESHOLD = 5;
+    driftEvents = /* @__PURE__ */ new Map();
+    DRIFT_WINDOW = 10;
+    DRIFT_THRESHOLD = 3;
     memoryExtractionErrorCounts = /* @__PURE__ */ new Map();
+    sessionWebFetchHistory = /* @__PURE__ */ new Map();
+    SESSION_WEB_FETCH_CAP = 25;
     MAX_PROMPT_TOKENS = 24e4;
     MAX_TOOL_CONTENT_CHARS = 1e4;
   }