@cognipeer/agent-sdk 0.19.0 → 0.20.0

package/dist/index.cjs

@@ -1258,7 +1258,7 @@ function cacheWriteTokensOf(raw) {
 init_content();
 
 // src/version.ts
-var SDK_VERSION = "0.19.0";
+var SDK_VERSION = "0.20.0";
 
 // src/utils/otlp/content.ts
 init_content();
@@ -10875,6 +10875,13 @@ function buildModelMessages(state, config) {
 var systemMessage = (content) => ({ role: "system", content });
 var humanMessage = (content) => ({ role: "user", content });
 var LAST_SUMMARIZATION_CALL_KEY = "__lastSummarizationCall";
+var SUMMARIZATION_TRIGGER_KEY = "__summarizationTrigger";
+var SUMMARIZATION_TRIGGERS = ["pre_leg", "mid_leg", "post_tools"];
+var MAX_EVENT_ERROR_CHARS = 300;
+function eventErrorText(message) {
+  if (!message) return void 0;
+  return message.length > MAX_EVENT_ERROR_CHARS ? `${message.slice(0, MAX_EVENT_ERROR_CHARS - 3)}...` : message;
+}
 var MAX_MESSAGE_CHARS = 2e3;
 var MAX_USER_MESSAGE_CHARS = 4e3;
 function validateMessageSequence(messages) {
@@ -11201,7 +11208,7 @@ function repairStructuredSummary(previous, current, integrity) {
   return { ...current, user_directives: directives, stable_facts: filteredFacts };
 }
 function createContextSummarizeNode(opts) {
-  return async (state) => {
+  const summarizePass = async (state, events) => {
     const resolved = getResolvedSmartConfig(opts, state.agent);
     const config = resolved.summarization;
     if (!config.enable) return {};
@@ -11392,11 +11399,25 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       ...summaryPrompt,
       ...rawModelOutput !== void 0 ? [{ role: "assistant", content: rawModelOutput, name: "summarization_result" }] : []
     ]));
+    const summaryToolCallId = `call_summary_${Date.now()}_${nanoid.nanoid(6)}`;
+    const passNumber = (Array.isArray(state.summaries) ? state.summaries.length : 0) + 1;
+    const rawTrigger = state.ctx?.[SUMMARIZATION_TRIGGER_KEY];
+    const trigger = SUMMARIZATION_TRIGGERS.includes(rawTrigger) ? rawTrigger : void 0;
+    const passIdentity = {
+      id: summaryToolCallId,
+      pass: passNumber,
+      ...trigger ? { trigger } : {}
+    };
     const startTime = Date.now();
     try {
       if (!model || typeof model.invoke !== "function") {
         throw new Error("Summarization model is unavailable.");
       }
+      events.started = passIdentity;
+      try {
+        onEvent?.({ type: "summarization", phase: "start", ...passIdentity, summary: "" });
+      } catch {
+      }
       const langchainRunOptions = model?._lc ? { runName: "summarize_context", tags: ["cognipeer:summarization"], metadata: { cognipeer_event: "summarization" } } : void 0;
       const response = langchainRunOptions ? await model.invoke(summaryPrompt, langchainRunOptions) : await model.invoke(summaryPrompt);
       durationMs = Date.now() - startTime;
@@ -11438,24 +11459,9 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       structuredSummary = groundDirectives(structuredSummary);
       integrity = noteDroppedDirectives({ passed: true, criticalFactLoss: false, obsoleteFactRevived: false, notes: ["Fallback summary generated locally."] });
       summaryText = renderStructuredSummary(structuredSummary);
-      const errorEvent = {
-        type: "summarization",
-        summary: summaryText,
-        messagesCompressed: compressableMessages.length,
-        inputTokens: inputTokensEstimate,
-        outputTokens: countApproxTokens(summaryText),
-        durationMs,
-        previousSummary: previousSummary || void 0,
-        tokenCountBefore,
-        tokenCountAfter: tokenCountBefore,
-        structuredSummary,
-        integrity,
-        ...modelName ? { model: modelName } : {},
-        ...droppedDirectives.length > 0 ? { droppedDirectives } : {}
-      };
-      onEvent?.(errorEvent);
       summarizerError = traceErrorOf(err);
     }
+    const eventError = eventErrorText(summarizerError?.message);
     const toolCallIdToAssistantIdx = /* @__PURE__ */ new Map();
     messages.forEach((m, idx) => {
       if (m.role === "assistant" && m.tool_calls && Array.isArray(m.tool_calls)) {
@@ -11547,10 +11553,19 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       (m) => m.role === "tool" && isSummarizedToolPlaceholder(m.content) && !messages.some((orig) => orig.tool_call_id === m.tool_call_id && isSummarizedToolPlaceholder(orig.content))
     );
     if (actuallyCompressed.length === 0) {
+      if (events.started) {
+        events.ended = true;
+        onEvent?.({
+          type: "summarization",
+          phase: "skipped",
+          ...passIdentity,
+          summary: "",
+          ...fallback ? { status: "fallback", ...eventError ? { error: eventError } : {} } : {}
+        });
+      }
       return {};
     }
     const validatedMessages = validateMessageSequence(newMessages);
-    const summaryToolCallId = `call_summary_${Date.now()}`;
     const assistantSummaryCall = {
       role: "assistant",
       // "assistant"
@@ -11579,6 +11594,11 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
     const tokenCountAfter = countMessagesTokens(finalMessages);
     const summarizationEvent = {
       type: "summarization",
+      phase: "done",
+      ...passIdentity,
+      status: fallback ? "fallback" : "ok",
+      ...fallback && eventError ? { error: eventError } : {},
+      compactedTools: actuallyCompressed.filter((m) => typeof m.tool_call_id === "string" && m.tool_call_id).map((m) => ({ toolCallId: m.tool_call_id, toolName: m.name || "" })),
       summary: summaryText,
       messagesCompressed: compressableMessages.length,
       inputTokens: inputTokensActual ?? inputTokensEstimate,
@@ -11596,6 +11616,7 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       ...modelName ? { model: modelName } : {},
       ...droppedDirectives.length > 0 ? { droppedDirectives } : {}
     };
+    events.ended = true;
     onEvent?.(summarizationEvent);
     traceSummarization({
       type: "summarization",
@@ -11658,6 +11679,7 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
         fallback
       }
     });
+    const { [SUMMARIZATION_TRIGGER_KEY]: _passTrigger, ...carriedCtx } = ctxWithDeferredTrace ?? state.ctx ?? {};
     return {
       messages: finalMessages,
       summaries,
@@ -11667,7 +11689,7 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       // held for the next leg. The smart driver merges ctx and removes the
       // call record once the hook has seen it.
       ctx: {
-        ...ctxWithDeferredTrace ?? state.ctx ?? {},
+        ...carriedCtx,
         [LAST_SUMMARIZATION_CALL_KEY]: {
           ...modelName ? { model: modelName } : {},
           durationMs,
@@ -11685,6 +11707,30 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
       }
     };
   };
+  return async (state) => {
+    const events = {};
+    try {
+      return await summarizePass(state, events);
+    } catch (err) {
+      if (events.started && !events.ended) {
+        events.ended = true;
+        const onEvent = state.ctx?.__onEvent;
+        const error = eventErrorText(traceErrorOf(err).message);
+        try {
+          onEvent?.({
+            type: "summarization",
+            phase: "skipped",
+            ...events.started,
+            summary: "",
+            status: "error",
+            ...error ? { error } : {}
+          });
+        } catch {
+        }
+      }
+      throw err;
+    }
+  };
 }
 
 // src/index.ts
@@ -11853,9 +11899,10 @@ function createPlanTool(stateRef, toolName, description) {
       expectedVersion: zod.z.number().int().min(0).optional(),
       todoList: zod.z.array(todoItemWireSchema).optional()
     }),
-    func: async ({ operation, expectedVersion, todoList }) => {
+    func: async ({ operation, expectedVersion, todoList }, context) => {
       const toolStateRef = manageTodo._stateRef;
       const onEvent = toolStateRef?.__onEvent;
+      const toolCallId = context?.toolCallId || toolStateRef?.__currentToolCallId || void 0;
       const emitPlanEvent = (list) => {
         const version = stateRef.planVersion || 1;
         const adherenceScore = stateRef.adherenceScore || 0;
@@ -11864,7 +11911,8 @@ function createPlanTool(stateRef, toolName, description) {
           operation,
           version,
           adherenceScore,
-          count: Array.isArray(list) ? list.length : 0
+          count: Array.isArray(list) ? list.length : 0,
+          ...toolCallId ? { toolCallId } : {}
         };
         onEvent?.({ type: "plan", todoList: list, ...planData });
         const traceSession = toolStateRef?.ctx?.__traceSession;
@@ -13525,7 +13573,7 @@ ${pluginAppend}`;
     const toolCalls = Array.isArray(lastAssistant.tool_calls) ? lastAssistant.tool_calls : [];
     return toolCalls.length === 0;
   }
-  async function trySummarize(currentState, currentRawMessages, ref) {
+  async function trySummarize(currentState, currentRawMessages, ref, trigger) {
     if (!summarizer) return { state: currentState, rawMessages: currentRawMessages, compressed: false };
     const compactHost = currentState.ctx?.__plugins;
     const tokensBefore = countMessagesTokens(currentState.messages || []);
@@ -13545,7 +13593,9 @@ ${pluginAppend}`;
         return { state: { ...currentState, ctx: ctx2 }, rawMessages: currentRawMessages, compressed: false };
       }
     }
-    const delta = await summarizer(currentState);
+    const nodeState = { ...currentState, ctx: { ...currentState.ctx || {}, [SUMMARIZATION_TRIGGER_KEY]: trigger } };
+    const delta = await summarizer(nodeState);
+    if (nodeState.usage !== currentState.usage) currentState.usage = nodeState.usage;
     if (!delta || Object.keys(delta).length === 0) {
       const ctx2 = {
         ...currentState.ctx || {},
@@ -13557,6 +13607,7 @@ ${pluginAppend}`;
     const ctx = { ...currentState.ctx || {}, ...delta.ctx || {} };
     const summarizationCall = ctx[LAST_SUMMARIZATION_CALL_KEY];
     delete ctx[LAST_SUMMARIZATION_CALL_KEY];
+    delete ctx[SUMMARIZATION_TRIGGER_KEY];
     delete ctx.__summarizationExhausted;
     delete ctx.__summarizationExhaustedAtToolCount;
     const updated = await persistLatestSummary(syncPlanStateWith(ref, { ...currentState, ...delta, ctx }));
@@ -13780,7 +13831,7 @@ ${pluginAppend}`;
           state = syncRuntimeTools({ ...state, messages: rawMessages }, rawMessages, toolSet);
           const next = summarizationEnabled ? decideBefore(state) : "agent";
           if (next === "contextSummarize" && summarizer) {
-            const result = await trySummarize(state, rawMessages, stateRef);
+            const result = await trySummarize(state, rawMessages, stateRef, "pre_leg");
             state = result.state;
             rawMessages = result.rawMessages;
             noteTraceHandoff(state);
@@ -13815,7 +13866,7 @@ ${pluginAppend}`;
             const ctx = { ...state.ctx || {} };
             delete ctx.__needsSummarization;
             state = { ...state, ctx };
-            const result = await trySummarize(state, rawMessages, stateRef);
+            const result = await trySummarize(state, rawMessages, stateRef, "mid_leg");
             state = result.state;
             rawMessages = result.rawMessages;
             noteTraceHandoff(state);
@@ -13827,7 +13878,7 @@ ${pluginAppend}`;
           if (summarizationEnabled) {
             const after = decideAfter(state);
             if (after === "contextSummarize" && summarizer) {
-              const result = await trySummarize(state, rawMessages, stateRef);
+              const result = await trySummarize(state, rawMessages, stateRef, "post_tools");
               state = result.state;
               rawMessages = result.rawMessages;
               noteTraceHandoff(state);
@@ -22078,6 +22129,7 @@ exports.OpenAIProvider = OpenAIProvider;
 exports.PREOPEN_TOOL_CALL_PREFIX = PREOPEN_TOOL_CALL_PREFIX;
 exports.ProviderError = ProviderError;
 exports.SMALL_TIER_SKILL_POLICY = SMALL_TIER_SKILL_POLICY;
+exports.SUMMARIZATION_TRIGGER_KEY = SUMMARIZATION_TRIGGER_KEY;
 exports.StructuredOutputManager = StructuredOutputManager;
 exports.TOOL_INPUT_DIGEST_KEY = TOOL_INPUT_DIGEST_KEY;
 exports.ToolBasedStrategy = ToolBasedStrategy;