@cognipeer/agent-sdk 0.4.4 → 0.4.6

package/dist/index.mjs

@@ -2181,7 +2181,7 @@ function resolveToolResponsePolicy(toolName, output, config) {
   const byTool = config.toolResponses.toolResponseRetentionByTool[toolName];
   let retentionPolicy = byTool || config.toolResponses.defaultPolicy;
   if (classification === "critical") {
-    retentionPolicy = byTool || "keep_structured";
+    retentionPolicy = byTool || config.toolResponses.fallbackPolicy;
   } else if (classification === "verbose") {
     retentionPolicy = byTool || config.toolResponses.largeResponsePolicy;
   } else if (classification === "redundant") {
@@ -2887,6 +2887,21 @@ 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 createAgent(opts) {
   const resolver = createResolverNode();
   const agentCore = createAgentCoreNode(opts);
@@ -3075,7 +3090,7 @@ function createAgent(opts) {
           const tokenCount = countMessagesTokens(tokCountMessages);
           if (tokenCount > maxTok) {
             const hasFreshSummary = (state.messages || []).some(
-              (m) => m.role === "tool" && m.content === "SUMMARIZED"
+              (m) => m.role === "tool" && typeof m.content === "string" && (m.content === "SUMMARIZED" || m.content.startsWith("SUMMARIZED_TOOL_RESPONSE"))
             );
             if (hasFreshSummary && tokenCount <= maxTok * 1.15) ; else {
               const ctx = { ...state.ctx || {}, __needsSummarization: true };
@@ -3190,10 +3205,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 +3293,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 +3315,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 });
@@ -3557,7 +3585,7 @@ 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" && message.content === "SUMMARIZED").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));
@@ -3674,6 +3702,69 @@ function extractCanonicalFacts(messages) {
   }
   return [...facts.values()];
 }
+function flattenToolMessageContent(message) {
+  if (typeof message.content === "string") {
+    return message.content;
+  }
+  if (Array.isArray(message.content)) {
+    return message.content.map((part) => typeof part === "string" ? part : part?.text ?? part?.content ?? String(part ?? "")).join(" ");
+  }
+  return "";
+}
+function isSummarizedToolPlaceholder(content) {
+  return typeof content === "string" && (content === "SUMMARIZED" || content.startsWith("SUMMARIZED_TOOL_RESPONSE"));
+}
+function summarizeToolMessagePreview(message) {
+  const preview = flattenToolMessageContent(message).trim();
+  if (!preview) {
+    return "No summary available.";
+  }
+  return preview.length > 240 ? `${preview.slice(0, 240)}...` : preview;
+}
+function findToolHistoryEntry(state, toolCallId, toolName) {
+  const entries = [
+    ...state.toolHistory || [],
+    ...state.toolHistoryArchived || []
+  ];
+  if (toolCallId) {
+    for (let index = entries.length - 1; index >= 0; index -= 1) {
+      const entry = entries[index];
+      if (entry?.tool_call_id === toolCallId) {
+        return entry;
+      }
+    }
+  }
+  for (let index = entries.length - 1; index >= 0; index -= 1) {
+    const entry = entries[index];
+    if (entry?.toolName === toolName) {
+      return entry;
+    }
+  }
+  return void 0;
+}
+function buildSummarizedToolMessage(message, state) {
+  const toolName = message.name || "unknown_tool";
+  const toolCallId = message.tool_call_id;
+  const historyEntry = findToolHistoryEntry(state, toolCallId, toolName);
+  const executionId = historyEntry?.executionId;
+  const summary = historyEntry?.summary || summarizeToolMessagePreview(message);
+  const references = [
+    `toolName=${toolName}`,
+    toolCallId ? `toolCallId=${toolCallId}` : null,
+    executionId ? `executionId=${executionId}` : null
+  ].filter(Boolean).join("; ");
+  let retrievalHint = "Use get_tool_response if exact details are still needed.";
+  if (executionId && toolCallId) {
+    retrievalHint = `Use get_tool_response with executionId "${executionId}" or tool call id "${toolCallId}" if exact details are still needed.`;
+  } else if (executionId || toolCallId) {
+    retrievalHint = `Use get_tool_response with executionId "${executionId || toolCallId}" if exact details are still needed.`;
+  }
+  return [
+    `SUMMARIZED_TOOL_RESPONSE [${references}]`,
+    `Summary: ${summary}`,
+    retrievalHint
+  ].join("\n");
+}
 function mergeStableFacts(summary, additionalFacts) {
   if (additionalFacts.length === 0) return summary;
   const merged = new Map(summary.stable_facts.map((fact) => [fact.key, fact]));
@@ -3732,7 +3823,7 @@ function createContextSummarizeNode(opts) {
     const onEvent = state.ctx?.__onEvent;
     const traceSession = state.ctx?.__traceSession;
     const compressableMessages = messages.filter(
-      (m) => m.role === "tool" && m.content !== "SUMMARIZED" && !isSyntheticSummaryToolMessage(m)
+      (m) => m.role === "tool" && !isSummarizedToolPlaceholder(m.content) && !isSyntheticSummaryToolMessage(m)
     );
     if (compressableMessages.length === 0) {
       return {};
@@ -3897,10 +3988,10 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
         if (!toolCallId || !toolCallIdToAssistantIdx.has(toolCallId)) {
           return null;
         }
-        if (m.content === "SUMMARIZED") return m;
+        if (isSummarizedToolPlaceholder(m.content)) return m;
         return {
           ...m,
-          content: "SUMMARIZED"
+          content: buildSummarizedToolMessage(m, state)
         };
       }
       return m;
@@ -4210,8 +4301,8 @@ function createContextTools(stateRef, opts) {
   }
   const getTool = createTool({
     name: "get_tool_response",
-    description: "RETRIEVE the full output of a tool execution whose response was archived or dropped. When a tool response appears as 'ARCHIVED_TOOL_RESPONSE [executionId=xxx]' or 'DROPPED_TOOL_RESPONSE [executionId=xxx]', pass that exact executionId to this tool to get the complete data. You can also pass the tool_call_id from your original tool call.",
-    schema: z.object({ executionId: z.string().describe("Tool execution id") }),
+    description: "RETRIEVE the full output of a tool execution whose response was archived, dropped, or later compacted in conversation history. When a tool response appears as 'ARCHIVED_TOOL_RESPONSE [executionId=xxx]', 'DROPPED_TOOL_RESPONSE [executionId=xxx]', or 'SUMMARIZED_TOOL_RESPONSE [...]', pass the referenced executionId to this tool to get the complete data. You can also pass the original tool_call_id from the tool call.",
+    schema: z.object({ executionId: z.string().describe("Tool execution id or original tool_call_id") }),
     func: async ({ executionId }) => {
       const matchesExecution = (t) => t?.executionId === executionId || t?.tool_call_id === executionId;
       let execution = stateRef.toolHistory?.find((t) => matchesExecution(t));