@cognipeer/agent-sdk 0.18.0 → 0.20.0

package/dist/index.cjs

@@ -1258,7 +1258,7 @@ function cacheWriteTokensOf(raw) {
 init_content();
 
 // src/version.ts
-var SDK_VERSION = "0.18.0";
+var SDK_VERSION = "0.20.0";
 
 // src/utils/otlp/content.ts
 init_content();
@@ -1752,7 +1752,7 @@ function sessionToOtlp(file, options) {
   const agentName = file.agent?.name;
   const everyEvent = file.events;
   const exportedEvents = foldSupersededSpans(everyEvent).filter(
-    (event) => options.terminal || !(event.type === "span" && event.status === "pending")
+    (event) => options.terminal || !(event.type === "span" && (event.status === "pending" || event.status === "running"))
   );
   const startMs = parseMs(file.startedAt);
   const nodes = exportedEvents.map((event) => {
@@ -3766,7 +3766,9 @@ function foldSupersededSpans(events) {
     const current = winner.get(id);
     if (current === void 0 || rank(events[i], i) >= rank(events[current], current)) winner.set(id, i);
   }
-  return events.filter((event, i) => !(event.spanId && event.status === "pending" && winner.get(event.spanId) !== i));
+  return events.filter(
+    (event, i) => !(event.spanId && (event.status === "pending" || event.status === "running") && winner.get(event.spanId) !== i)
+  );
 }
 function recordOrDeferTraceEvent(ctx, event, options) {
   const session = ctx?.__traceSession;
@@ -10544,14 +10546,16 @@ function createAgent(opts) {
           const startedAt = Date.now();
           const recordSpan = (status, extra) => {
             if (!traced || !isTraceSessionOpen(parentSession)) return;
+            const running = status === "running";
             recordSubagentSpan(parentSession, {
               spanId: agentSpanId,
-              parentSpanId: context?.traceSpanId ?? parentIterationSpanId,
+              // The delegating tool_call span only exists once the tool returns.
+              parentSpanId: running ? parentIterationSpanId : context?.traceSpanId ?? parentIterationSpanId,
               name: opts.name || "agent",
               status,
-              durationMs: Date.now() - startedAt,
+              durationMs: running ? void 0 : Date.now() - startedAt,
               error: extra.error,
-              info: { name: opts.name || "agent", mode: "asTool", depth: currentDepth + 1, parentToolCallId: context?.toolCallId, status: status === "success" ? "ok" : "error" },
+              info: { name: opts.name || "agent", mode: "asTool", depth: currentDepth + 1, parentToolCallId: context?.toolCallId, status: running ? "running" : status === "success" ? "ok" : "error" },
               task: input,
               result: extra.result
             });
@@ -10571,6 +10575,7 @@ function createAgent(opts) {
             stream: Boolean(parentCtx.__streaming),
             cancellationToken: parentCtx.__cancellationToken ?? parentCtx.__abortSignal
           };
+          recordSpan("running", {});
           let res;
           try {
             res = await instance.invoke({ messages: childMessages, ctx: childCtx }, childConfig);
@@ -10870,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) {
@@ -11196,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 {};
@@ -11387,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;
@@ -11433,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)) {
@@ -11542,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"
@@ -11574,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,
@@ -11591,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",
@@ -11653,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,
@@ -11662,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,
@@ -11680,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
@@ -11848,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;
@@ -11859,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;
@@ -12771,14 +12824,16 @@ async function runChild(spec, input, deps, ref, opts) {
   const traceScope = traced ? createSubagentTraceScope(parentSession, { agentSpanId, name: spec.name, spawnId, mode: spec.mode, hitl: opts.allowHITL, iterations: opts.record?.iterations }) : void 0;
   const recordSegment = (status, info) => {
     if (!traced || !isTraceSessionOpen(parentSession)) return;
+    const unfinished = status === "pending" || status === "running";
     recordSubagentSpan(parentSession, {
       spanId: agentSpanId,
-      // The tool_call span only exists once the tool returns; a paused child
-      // has not returned, so its span hangs off the iteration.
-      parentSpanId: status === "pending" ? parentIterationSpanId : opts.parentSpanId ?? parentIterationSpanId,
+      // The tool_call span only exists once the tool returns; a running or
+      // paused child has not returned, so its span hangs off the iteration.
+      parentSpanId: unfinished ? parentIterationSpanId : opts.parentSpanId ?? parentIterationSpanId,
       name: spec.name,
       status,
-      durationMs: elapsedMs(),
+      // A starting segment has no extent yet; a reader draws it over its children.
+      durationMs: status === "running" ? void 0 : elapsedMs(),
       error: info.error,
       info: {
         id: spawnId,
@@ -12789,7 +12844,7 @@ async function runChild(spec, input, deps, ref, opts) {
         parentToolCallId: opts.parentToolCallId,
         status: info.subagentStatus,
         // Only a run that paused has more than one segment to tell about.
-        ...segment > 1 || status === "pending" ? { segments: segment } : {},
+        ...segment > 1 || unfinished ? { segments: segment } : {},
         ...info.note ? { note: info.note } : {},
         ...spec.metadata ? { metadata: spec.metadata } : {}
       },
@@ -12913,6 +12968,7 @@ async function runChild(spec, input, deps, ref, opts) {
   }
   if (traceScope) childInput = { ...childInput, ctx: { ...childInput.ctx || {}, __traceSession: traceScope } };
   const usageBaseline = childInput.usage?.perRequest?.length ?? 0;
+  recordSegment("running", { subagentStatus: "running", input });
   let res;
   try {
     res = await child.invoke(childInput, childConfig);
@@ -13517,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 || []);
@@ -13537,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 || {},
@@ -13549,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 }));
@@ -13772,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);
@@ -13807,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);
@@ -13819,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);
@@ -22070,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;