@cognipeer/agent-sdk 0.4.6 → 0.4.8

package/dist/index.cjs

@@ -1793,14 +1793,6 @@ var DEFAULT_PROFILE_CONFIGS = {
       schemaValidation: "strict",
       retryOnSchemaError: true,
       fallbackPolicy: "keep_structured"
-    },
-    watchdog: {
-      enabled: true,
-      tokenDriftThreshold: 1200,
-      overToolingSpikeThreshold: 0.7,
-      contextRotThreshold: 0.65,
-      autoCompaction: true,
-      autoReplanOnFailure: true
     }
   },
   balanced: {
@@ -1858,14 +1850,6 @@ var DEFAULT_PROFILE_CONFIGS = {
       schemaValidation: "strict",
       retryOnSchemaError: true,
       fallbackPolicy: "keep_structured"
-    },
-    watchdog: {
-      enabled: true,
-      tokenDriftThreshold: 1800,
-      overToolingSpikeThreshold: 0.6,
-      contextRotThreshold: 0.55,
-      autoCompaction: true,
-      autoReplanOnFailure: true
     }
   },
   deep: {
@@ -1923,14 +1907,6 @@ var DEFAULT_PROFILE_CONFIGS = {
       schemaValidation: "strict",
       retryOnSchemaError: true,
       fallbackPolicy: "keep_structured"
-    },
-    watchdog: {
-      enabled: true,
-      tokenDriftThreshold: 2200,
-      overToolingSpikeThreshold: 0.55,
-      contextRotThreshold: 0.5,
-      autoCompaction: true,
-      autoReplanOnFailure: true
     }
   },
   research: {
@@ -1988,14 +1964,6 @@ var DEFAULT_PROFILE_CONFIGS = {
       schemaValidation: "strict",
       retryOnSchemaError: true,
       fallbackPolicy: "keep_structured"
-    },
-    watchdog: {
-      enabled: true,
-      tokenDriftThreshold: 2600,
-      overToolingSpikeThreshold: 0.5,
-      contextRotThreshold: 0.45,
-      autoCompaction: true,
-      autoReplanOnFailure: true
     }
   }
 };
@@ -2072,11 +2040,6 @@ function normalizeSmartAgentOptions(opts) {
         ...opts.toolResponses?.toolResponseRetentionByTool || {}
       },
       criticalTools: opts.toolResponses?.criticalTools ?? customProfile.toolResponses?.criticalTools ?? preset.toolResponses.criticalTools
-    },
-    watchdog: {
-      ...preset.watchdog,
-      ...customProfile.watchdog || {},
-      ...opts.watchdog || {}
     }
   };
 }
@@ -2307,7 +2270,7 @@ function createToolsNode(initialTools, opts) {
       if (toolCallId) {
         ctx.__toolSchemaError = { toolName, toolCallId, message };
       }
-      if (resolved.watchdog.autoReplanOnFailure) {
+      if (resolved.planning.replanPolicy === "on_failure") {
         ctx.__planNeedsReplan = true;
       }
       state.ctx = ctx;
@@ -2912,6 +2875,24 @@ function getLastAssistantMessage(messages) {
   }
   return void 0;
 }
+function getActiveSummarizationThreshold(opts) {
+  const summarization = opts.summarization;
+  if (!summarization || typeof summarization !== "object" || summarization.enable === false) {
+    return void 0;
+  }
+  return summarization.summaryTriggerTokens || summarization.maxTokens || opts.limits?.maxContextTokens || 5e4;
+}
+function clearNeedsSummarization(state) {
+  if (!state.ctx?.__needsSummarization) {
+    return state;
+  }
+  const nextCtx = { ...state.ctx || {} };
+  delete nextCtx.__needsSummarization;
+  return {
+    ...state,
+    ctx: Object.keys(nextCtx).length > 0 ? nextCtx : void 0
+  };
+}
 function createAgent(opts) {
   const resolver = createResolverNode();
   const agentCore = createAgentCoreNode(opts);
@@ -2960,6 +2941,7 @@ function createAgent(opts) {
     outputSchema: opts.outputSchema,
     tracing: opts.tracing
   };
+  const summarizationThreshold = getActiveSummarizationThreshold(opts);
   async function runLoop(initial, config, emit) {
     let state = await resolver(initial);
     if (state.ctx?.__paused) {
@@ -2976,6 +2958,9 @@ function createAgent(opts) {
       delete nextCtx.__resumeStage;
       state = { ...state, ctx: Object.keys(nextCtx).length > 0 ? nextCtx : void 0 };
     }
+    if (summarizationThreshold === void 0) {
+      state = clearNeedsSummarization(state);
+    }
     const mergedLimits = {
       ...opts.limits || {},
       ...config?.limits || {}
@@ -3084,25 +3069,18 @@ function createAgent(opts) {
             }
           }
         }
-        let maxTok;
-        if (opts.summarization && typeof opts.summarization === "object") {
-          maxTok = opts.summarization.summaryTriggerTokens || opts.summarization.maxTokens;
-        }
-        if (maxTok === void 0) {
-          maxTok = 5e4;
-        }
-        if (maxTok) {
+        if (summarizationThreshold !== void 0) {
           const tokCountMessages = (state.messages || []).filter((message) => {
             if (isSyntheticSummaryMessage(message)) return false;
             if (message.role === "system" && (message.name === "context_summary" || message.name === "memory_context")) return false;
             return true;
           });
           const tokenCount = countMessagesTokens(tokCountMessages);
-          if (tokenCount > maxTok) {
+          if (tokenCount > summarizationThreshold) {
             const hasFreshSummary = (state.messages || []).some(
               (m) => m.role === "tool" && typeof m.content === "string" && (m.content === "SUMMARIZED" || m.content.startsWith("SUMMARIZED_TOOL_RESPONSE"))
             );
-            if (hasFreshSummary && tokenCount <= maxTok * 1.15) ; else {
+            if (hasFreshSummary && tokenCount <= summarizationThreshold * 1.15) ; else {
               const ctx = { ...state.ctx || {}, __needsSummarization: true };
               state = { ...state, ctx };
               break;
@@ -3591,15 +3569,6 @@ function buildModelMessages(state, config) {
   ];
   return clampToBudget(assembled, config.limits.maxContextTokens);
 }
-function estimateContextRotScore(state) {
-  const summaryCount = state.summaryRecords?.length || 0;
-  const archivedCount = state.toolHistoryArchived?.length || 0;
-  const messageCount = state.messages?.length || 0;
-  const summarizedMessages = state.messages?.filter((message) => message.role === "tool" && typeof message.content === "string" && (message.content === "SUMMARIZED" || message.content.startsWith("SUMMARIZED_TOOL_RESPONSE"))).length || 0;
-  const numerator = summaryCount * 2 + archivedCount + summarizedMessages;
-  const denominator = Math.max(messageCount, 1);
-  return Number((numerator / denominator).toFixed(3));
-}
 
 // src/nodes/contextSummarize.ts
 var systemMessage = (content) => ({ role: "system", content });
@@ -3744,6 +3713,9 @@ function findToolHistoryEntry(state, toolCallId, toolName) {
       }
     }
   }
+  if (!toolName) {
+    return void 0;
+  }
   for (let index = entries.length - 1; index >= 0; index -= 1) {
     const entry = entries[index];
     if (entry?.toolName === toolName) {
@@ -3832,8 +3804,20 @@ function createContextSummarizeNode(opts) {
     const messages = state.messages || [];
     const onEvent = state.ctx?.__onEvent;
     const traceSession = state.ctx?.__traceSession;
+    const toolResponsesConfigEarly = resolved.toolResponses;
     const compressableMessages = messages.filter(
-      (m) => m.role === "tool" && !isSummarizedToolPlaceholder(m.content) && !isSyntheticSummaryToolMessage(m)
+      (m) => {
+        if (m.role !== "tool") return false;
+        if (isSummarizedToolPlaceholder(m.content)) return false;
+        if (isSyntheticSummaryToolMessage(m)) return false;
+        const histEntry = findToolHistoryEntry(state, m.tool_call_id, m.name);
+        if (histEntry?.retentionPolicy === "keep_full") return false;
+        if (!histEntry?.retentionPolicy) {
+          const byTool = toolResponsesConfigEarly.toolResponseRetentionByTool?.[m.name || ""];
+          if ((byTool || toolResponsesConfigEarly.defaultPolicy) === "keep_full") return false;
+        }
+        return true;
+      }
     );
     if (compressableMessages.length === 0) {
       return {};
@@ -3854,6 +3838,13 @@ function createContextSummarizeNode(opts) {
       else if (Array.isArray(m.content)) {
         content = m.content.map((c) => c.text || JSON.stringify(c)).join(" ");
       }
+      if (m.role === "tool" && isSummarizedToolPlaceholder(content)) {
+        const historyEntry = findToolHistoryEntry(state, m.tool_call_id, m.name);
+        if (historyEntry) {
+          const original = historyEntry.rawOutput ?? historyEntry.output;
+          content = typeof original === "string" ? original : JSON.stringify(original);
+        }
+      }
       const maxContentLength = 2e3;
       if (content.length > maxContentLength) {
         content = content.substring(0, maxContentLength) + "... [TRUNCATED]";
@@ -3989,6 +3980,23 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
         });
       }
     });
+    const protectedToolCallIds = /* @__PURE__ */ new Set();
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const m = messages[i];
+      if (m.role === "assistant" && Array.isArray(m.tool_calls) && m.tool_calls.length > 0) {
+        const thisToolCallIds = new Set(m.tool_calls.map((tc) => tc.id).filter(Boolean));
+        const otherCompressable = compressableMessages.filter(
+          (cm) => !thisToolCallIds.has(cm.tool_call_id)
+        );
+        if (otherCompressable.length > 0) {
+          for (const tc of m.tool_calls) {
+            if (tc.id) protectedToolCallIds.add(tc.id);
+          }
+        }
+        break;
+      }
+    }
+    const toolResponsesConfig = resolved.toolResponses;
     const newMessages = messages.map((m) => {
       if (m.role === "tool") {
         if (isSyntheticSummaryToolMessage(m)) {
@@ -3999,6 +4007,16 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
           return null;
         }
         if (isSummarizedToolPlaceholder(m.content)) return m;
+        if (toolCallId && protectedToolCallIds.has(toolCallId)) return m;
+        const toolName = m.name || "";
+        const historyEntry = findToolHistoryEntry(state, toolCallId, toolName);
+        const entryRetention = historyEntry?.retentionPolicy;
+        if (entryRetention === "keep_full") return m;
+        if (!entryRetention) {
+          const byTool = toolResponsesConfig.toolResponseRetentionByTool?.[toolName];
+          const configPolicy = byTool || toolResponsesConfig.defaultPolicy;
+          if (configPolicy === "keep_full") return m;
+        }
         return {
           ...m,
           content: buildSummarizedToolMessage(m, state)
@@ -4006,6 +4024,12 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       }
       return m;
     }).filter((m) => m !== null);
+    const actuallyCompressed = newMessages.filter(
+      (m) => m.role === "tool" && typeof m.content === "string" && m.content.startsWith("SUMMARIZED_TOOL_RESPONSE") && !messages.some((orig) => orig.tool_call_id === m.tool_call_id && isSummarizedToolPlaceholder(orig.content))
+    );
+    if (actuallyCompressed.length === 0) {
+      return {};
+    }
     const validatedMessages = validateMessageSequence(newMessages);
     const summaryToolCallId = `call_summary_${Date.now()}`;
     const assistantSummaryCall = {
@@ -4081,12 +4105,7 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
     return {
       messages: finalMessages,
       summaries,
-      summaryRecords,
-      watchdog: {
-        ...state.watchdog || {},
-        compactions: (state.watchdog?.compactions || 0) + 1,
-        lastAction: "summarized"
-      }
+      summaryRecords
     };
   };
 }
@@ -4385,8 +4404,7 @@ function needsSummarization(state, opts, summarizationEnabled) {
       return "";
     }).join("\n");
     const tokenCount = countApproxTokens(allText);
-    const contextRotScore = state.watchdog?.contextRotScore || 0;
-    const needsSum = tokenCount > maxTok || resolved.watchdog.enabled && resolved.watchdog.autoCompaction && contextRotScore >= resolved.watchdog.contextRotThreshold;
+    const needsSum = tokenCount > maxTok;
     if (process.env.DEBUG_SUMMARIZATION) {
       const msgCount = (state.messages || []).length;
       const toolMsgs = (state.messages || []).filter((m) => m.role === "tool");
@@ -4396,7 +4414,7 @@ function needsSummarization(state, opts, summarizationEnabled) {
       });
       console.log(`[Summarization] Messages: ${msgCount}, Tool messages: ${toolMsgs.length}`);
       toolDetails.forEach((d) => console.log(`  ${d}`));
-      console.log(`[Summarization] Token count: ${tokenCount}, Max: ${maxTok}, Context rot: ${contextRotScore}, Needs summarization: ${needsSum}`);
+      console.log(`[Summarization] Token count: ${tokenCount}, Max: ${maxTok}, Needs summarization: ${needsSum}`);
     }
     return needsSum;
   } catch {
@@ -4550,7 +4568,6 @@ function createSmartAgent(opts) {
     delegation: resolved.delegation,
     memory: { ...resolved.memory, store: memoryStore },
     toolResponses: resolved.toolResponses,
-    watchdog: resolved.watchdog,
     useTodoList: planningEnabled
   };
   const stateRef = { toolHistory: void 0, toolHistoryArchived: void 0, todoList: void 0, planVersion: 0, adherenceScore: 0 };
@@ -4633,7 +4650,6 @@ function createSmartAgent(opts) {
       let rawMessages = [...seedMessages];
       const effectiveMaxToolCalls = config?.limits?.maxToolCalls ?? resolved.limits.maxToolCalls ?? 10;
       const iterationLimit = Math.max(effectiveMaxToolCalls * 3 + 5, 30);
-      let previousTokenCount = countApproxTokens(rawMessages.map((message) => typeof message.content === "string" ? message.content : JSON.stringify(message.content)).join("\n"));
       for (let i = 0; i < iterationLimit; i++) {
         state = { ...state, messages: rawMessages };
         const next = summarizationEnabled ? decideBefore(state) : "agent";
@@ -4667,27 +4683,6 @@ function createSmartAgent(opts) {
         state = syncPlanState(await syncMemory(state));
         stateRef.toolHistory = state.toolHistory;
         stateRef.toolHistoryArchived = state.toolHistoryArchived;
-        const currentTokenCount = countApproxTokens(rawMessages.map((message) => typeof message.content === "string" ? message.content : JSON.stringify(message.content)).join("\n"));
-        const tokenDrift = Math.max(0, currentTokenCount - previousTokenCount);
-        const toolMessagesThisTurn = appendedMessages.filter((message) => message.role === "tool").length;
-        const overToolingRate = appendedMessages.length === 0 ? 0 : Number((toolMessagesThisTurn / appendedMessages.length).toFixed(2));
-        state = {
-          ...state,
-          watchdog: {
-            tokenDrift,
-            contextRotScore: estimateContextRotScore(state),
-            overToolingRate,
-            compactions: state.watchdog?.compactions || 0,
-            lastAction: appendedMessages.length > 0 ? "agent_turn" : state.watchdog?.lastAction
-          }
-        };
-        previousTokenCount = currentTokenCount;
-        if (resolved.watchdog.enabled && resolved.watchdog.autoCompaction) {
-          const shouldCompact = tokenDrift >= resolved.watchdog.tokenDriftThreshold || (state.watchdog?.overToolingRate || 0) >= resolved.watchdog.overToolingSpikeThreshold || (state.watchdog?.contextRotScore || 0) >= resolved.watchdog.contextRotThreshold;
-          if (shouldCompact) {
-            state = { ...state, ctx: { ...state.ctx || {}, __needsSummarization: true } };
-          }
-        }
         if (state.ctx?.__needsSummarization && summarizer) {
           const ctx = { ...state.ctx || {} };
           delete ctx.__needsSummarization;
@@ -4719,12 +4714,12 @@ function createSmartAgent(opts) {
       }
       if (state.summaries && state.summaries.length > 0) {
         if (lastResult.state) {
-          lastResult = { ...lastResult, state: { ...lastResult.state, summaries: state.summaries, summaryRecords: state.summaryRecords, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, watchdog: state.watchdog, messages: rawMessages } };
+          lastResult = { ...lastResult, state: { ...lastResult.state, summaries: state.summaries, summaryRecords: state.summaryRecords, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, messages: rawMessages } };
         } else {
-          lastResult = { ...lastResult, state: { ...state, summaries: state.summaries, summaryRecords: state.summaryRecords, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, watchdog: state.watchdog, messages: rawMessages } };
+          lastResult = { ...lastResult, state: { ...state, summaries: state.summaries, summaryRecords: state.summaryRecords, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, messages: rawMessages } };
         }
       } else if (lastResult.state) {
-        lastResult = { ...lastResult, state: { ...lastResult.state, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, watchdog: state.watchdog, messages: rawMessages } };
+        lastResult = { ...lastResult, state: { ...lastResult.state, memoryFacts: state.memoryFacts, plan: state.plan, planVersion: state.planVersion, messages: rawMessages } };
       }
       return lastResult;
     },