@cognipeer/agent-sdk 0.4.7 → 0.4.9

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;
@@ -3194,7 +3157,23 @@ function createAgent(opts) {
         state = { ...state, ...await finalizeNode(state) };
         continue;
       }
-      if (toolCalls.length === 0) break;
+      if (toolCalls.length === 0) {
+        if (opts.outputSchema && !state.ctx?.__finalizedDueToStructuredOutput && !state.ctx?.__structuredOutputForceFinalize) {
+          const forceMsg = {
+            role: "system",
+            content: [
+              "A structured output schema is active.",
+              "You MUST now call tool `response` with the final JSON object that matches the schema.",
+              "Do not write the JSON in the assistant message.",
+              "Call `response` exactly once, then stop."
+            ].join("\n")
+          };
+          const ctx = { ...state.ctx || {}, __structuredOutputForceFinalize: true };
+          state = { ...state, messages: [...state.messages, forceMsg], ctx };
+          continue;
+        }
+        break;
+      }
       onProgress?.({ stage: "tools", message: "Running tools" });
       if (cancelIfRequested("before_tools")) break;
       state = { ...state, ...await toolsNode(state) };
@@ -3207,31 +3186,31 @@ function createAgent(opts) {
       const last = state.messages[state.messages.length - 1];
       const lastHasToolCalls = Array.isArray(last?.tool_calls) && last.tool_calls.length > 0;
       if (!lastHasToolCalls) {
-        const forceMsg = {
-          role: "system",
-          content: [
-            "A structured output schema is active.",
-            "You MUST now call tool `response` with the final JSON object that matches the schema.",
-            "Do not write the JSON in the assistant message.",
-            "Call `response` exactly once, then stop."
-          ].join("\n")
-        };
         const alreadyForced = Boolean(state.ctx?.__structuredOutputForceFinalize);
-        if (!alreadyForced) {
-          const ctx = { ...state.ctx || {}, __structuredOutputForceFinalize: true };
-          state = { ...state, messages: [...state.messages, forceMsg], ctx };
-          try {
-            state = { ...state, ...await agentCore(state) };
-            const lastAfter = state.messages[state.messages.length - 1];
-            const toolCallsAfter = Array.isArray(lastAfter?.tool_calls) ? lastAfter.tool_calls : [];
-            if (toolCallsAfter.length > 0) {
-              state = { ...state, ...await toolsNode(state) };
-            }
-          } catch (err) {
-            const errMsg = err instanceof Error ? err.message : String(err);
-            emit?.({ type: "metadata", error: `Structured output force-finalize failed: ${errMsg}` });
-            throw err;
+        if (alreadyForced) {
+          const finalNudge = {
+            role: "system",
+            content: [
+              "A structured output schema is active.",
+              "You MUST now call tool `response` with the final JSON object that matches the schema.",
+              "Do not write the JSON in the assistant message.",
+              "Call `response` exactly once, then stop.",
+              "Do NOT call any other tools. Produce the response NOW based on whatever information you have gathered so far."
+            ].join("\n")
+          };
+          state = { ...state, messages: [...state.messages, finalNudge] };
+        }
+        try {
+          state = { ...state, ...await agentCore(state) };
+          const lastAfter = state.messages[state.messages.length - 1];
+          const toolCallsAfter = Array.isArray(lastAfter?.tool_calls) ? lastAfter.tool_calls : [];
+          if (toolCallsAfter.length > 0) {
+            state = { ...state, ...await toolsNode(state) };
           }
+        } catch (err) {
+          const errMsg = err instanceof Error ? err.message : String(err);
+          emit?.({ type: "metadata", error: `Structured output force-finalize failed: ${errMsg}` });
+          throw err;
         }
       }
     }
@@ -3606,15 +3585,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 });
@@ -3759,6 +3729,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) {
@@ -3847,8 +3820,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 {};
@@ -3869,6 +3854,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]";
@@ -4004,6 +3996,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)) {
@@ -4014,6 +4023,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)
@@ -4021,6 +4040,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 = {
@@ -4096,12 +4121,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
     };
   };
 }
@@ -4400,8 +4420,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");
@@ -4411,7 +4430,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 {
@@ -4565,7 +4584,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 };
@@ -4648,7 +4666,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";
@@ -4682,27 +4699,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;
@@ -4734,12 +4730,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;
     },