@cognipeer/agent-sdk 0.4.6 → 0.4.7

package/dist/index.mjs

@@ -2902,6 +2902,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);
@@ -2950,6 +2968,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) {
@@ -2966,6 +2985,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 || {}
@@ -3074,25 +3096,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;