@cognipeer/agent-sdk 0.4.5 → 0.4.7

package/dist/index.mjs

@@ -2887,6 +2887,39 @@ function isSyntheticSummaryMessage(message) {
   }
   return false;
 }
+function getMessageText(message) {
+  if (typeof message?.content === "string") return message.content;
+  if (Array.isArray(message?.content)) {
+    return message.content.map((chunk) => typeof chunk === "string" ? chunk : chunk?.text ?? chunk?.content ?? "").join("");
+  }
+  return "";
+}
+function getLastAssistantMessage(messages) {
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    if (messages[index]?.role === "assistant") {
+      return messages[index];
+    }
+  }
+  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);
@@ -2935,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) {
@@ -2951,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 || {}
@@ -3059,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;
@@ -3190,10 +3220,25 @@ function createAgent(opts) {
           } catch (err) {
             const errMsg = err instanceof Error ? err.message : String(err);
             emit?.({ type: "metadata", error: `Structured output force-finalize failed: ${errMsg}` });
+            throw err;
           }
         }
       }
     }
+    if (state.ctx?.__needsSummarization && !opts?.summarization) {
+      throw new Error(
+        "Agent context exceeded the available budget before a final assistant response could be generated. Reduce tool response size or use createSmartAgent with summarization enabled."
+      );
+    }
+    const lastAfterLoop = state.messages[state.messages.length - 1];
+    if (lastAfterLoop?.role === "tool" && !pausedStage) {
+      const isExpectedExit = state.ctx?.__awaitingApproval || state.ctx?.__cancelled || state.ctx?.__finalizedDueToStructuredOutput || state.ctx?.__finalizedDueToToolLimit || state.ctx?.__needsSummarization;
+      if (!isExpectedExit) {
+        throw new Error(
+          "Agent loop terminated with a pending tool response but no subsequent model invocation. This usually indicates a model provider error or exhausted iteration budget."
+        );
+      }
+    }
     if (!pausedStage && typeof onStateChange === "function" && onStateChange(state)) {
       checkpointIfRequested("after_loop");
     }
@@ -3263,10 +3308,8 @@ function createAgent(opts) {
       agentRuntime: res.agent || runtime,
       status: "success"
     });
-    const finalMsg = res.messages[res.messages.length - 1];
-    let content = "";
-    if (typeof finalMsg?.content === "string") content = finalMsg.content;
-    else if (Array.isArray(finalMsg?.content)) content = finalMsg.content.map((c) => typeof c === "string" ? c : c?.text ?? c?.content ?? "").join("");
+    const finalAssistantMsg = getLastAssistantMessage(res.messages);
+    const content = getMessageText(finalAssistantMsg);
     let parsed = void 0;
     const schema = opts.outputSchema;
     if (schema && res.ctx?.__structuredOutputParsed) {
@@ -3287,7 +3330,7 @@ function createAgent(opts) {
       } catch {
       }
     }
-    emit({ type: "finalAnswer", content: typeof finalMsg?.content === "string" ? finalMsg.content : content });
+    emit({ type: "finalAnswer", content });
     if (streamEnabled && content) {
       onStream?.({ text: content, isFinal: true });
       emit({ type: "stream", text: content, isFinal: true });