@cognipeer/agent-sdk 0.4.4 → 0.4.5

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") {
@@ -3075,7 +3075,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 };
@@ -3557,7 +3557,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 +3674,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 +3795,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 +3960,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 +4273,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));