@cognipeer/agent-sdk 0.2.3 → 0.2.4

package/dist/index.d.cts

@@ -165,9 +165,15 @@ type TraceSinkHttpConfig = {
     headers?: Record<string, string>;
 };
 type TraceSinkConfig = TraceSinkFileConfig | TraceSinkCustomConfig | TraceSinkCognipeerConfig | TraceSinkHttpConfig;
+type TracingMode = "batched" | "streaming";
 type TracingConfig = {
     enabled: boolean;
     logData?: boolean;
+    /** Tracing mode:
+     * - "batched" (default): Events are collected and sent together when session ends
+     * - "streaming": Session starts immediately, events are sent in real-time as they occur
+     */
+    mode?: TracingMode;
     sink?: TraceSinkConfig;
 };
 type SmartAgentTracingConfig = TracingConfig;
@@ -372,10 +378,30 @@ type ResolvedTraceSink = {
 };
 type ResolvedTraceConfig = {
     enabled: boolean;
-    mode: "batched";
+    mode: TracingMode;
     logData: boolean;
     sink: ResolvedTraceSink;
 };
+type TraceSessionRuntime = {
+    sessionId: string;
+    startedAt: number;
+    resolvedConfig: ResolvedTraceConfig;
+    events: TraceEventRecord[];
+    summary: TraceSessionSummary;
+    status: TraceSessionStatus;
+    errors: TraceErrorRecord[];
+    fileBaseDir?: string;
+    fileSessionDir?: string;
+    /** For streaming mode: whether session start has been sent to sink */
+    sessionStarted?: boolean;
+    /** Agent info for streaming mode session start */
+    agentInfo?: {
+        name?: string;
+        version?: string;
+        model?: string;
+        provider?: string;
+    };
+};
 type HandoffDescriptor<TParsed = any> = {
     type: "handoff";
     toolName: string;
@@ -478,6 +504,25 @@ type PlanEvent = {
 type SummarizationEvent = {
     type: "summarization";
     summary: string;
+    /** Number of messages that were compressed/summarized */
+    messagesCompressed?: number;
+    /** Input tokens used for summarization prompt (from model response if available, otherwise estimated) */
+    inputTokens?: number;
+    /** Output tokens from summarization response (from model response if available, otherwise estimated) */
+    outputTokens?: number;
+    /** Cached input tokens from model response (prompt cache hit) */
+    cachedInputTokens?: number;
+    /** Total tokens (input + output) */
+    totalTokens?: number;
+    /** Duration of the summarization call in milliseconds */
+    durationMs?: number;
+    /** Previous summary content (if incremental summarization) */
+    previousSummary?: string;
+    /** Total token count before summarization */
+    tokenCountBefore?: number;
+    /** Total token count after summarization */
+    tokenCountAfter?: number;
+    /** @deprecated Use messagesCompressed instead */
     archivedCount?: number;
 };
 type FinalAnswerEvent = {
@@ -853,5 +898,10 @@ declare function customSink(handler: CustomSinkArg): TraceSinkConfig;
 declare function cognipeerSink(apiKey: string): TraceSinkConfig;
 declare function cognipeerSink(url: string | undefined, apiKey: string): TraceSinkConfig;
 declare function httpSink(url: string, headers?: Record<string, string>): TraceSinkConfig;
+/**
+ * Start a streaming trace session. Should be called before recording events in streaming mode.
+ * This sends the session start to the remote endpoint immediately.
+ */
+declare function startStreamingSession(session: TraceSessionRuntime | undefined, agentRuntime?: AgentRuntimeConfig): Promise<void>;
 
-export { type AIMessage, type AgentInstance, type AgentInvokeResult, type AgentLimits, type AgentOptions, type AgentRuleOptions, type AgentRuntimeConfig, type AgentSnapshot, type AgentSnapshotMetadata, type AgentState, type BaseChatMessage, type BaseChatMessagePart, type BaseChatModel, type BaseMessage, type CodeDetectionRuleOptions, type CodeGuardrailOptions, type ConversationGuardrail, type CustomRuleCallback, type CustomRuleOptions, type GuardrailContext, type GuardrailDisposition, type GuardrailEvent, type GuardrailIncident, type GuardrailOutcome, GuardrailPhase, type GuardrailRule, type HandoffDescriptor, type InvokeConfig, type JsonGuardrailOptions, type JsonSchemaRuleOptions, type Message, type PendingToolApproval, type RegexGuardrailOptions, type RegexRuleOptions, type ResolvedTraceConfig, type ResolvedTraceSink, type RestoreSnapshotOptions, type SerializableSmartState, type SmartAgentEvent, type SmartAgentInstance, type SmartAgentLimits, type SmartAgentOptions, type SmartAgentTracingConfig, type SmartModel, type SmartState, type SmartToolFn, type SnapshotOptions, type SnapshotRuntimeHint, type TokenLimits, type ToolApprovalEvent, type ToolApprovalResolution, type ToolInterface, type TraceDataSection, type TraceErrorRecord, type TraceEventRecord, type TraceMessageSection, type TraceMetadataSection, type TraceSessionConfigSnapshot, type TraceSessionFile, type TraceSessionStatus, type TraceSessionSummary, type TraceSinkCognipeerConfig, type TraceSinkConfig, type TraceSinkCustomConfig, type TraceSinkFileConfig, type TraceSinkHttpConfig, type TraceSinkSnapshot, type TraceSummarySection, type TraceToolCallSection, type TraceToolResultSection, type TracingConfig, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, withTools };
+export { type AIMessage, type AgentInstance, type AgentInvokeResult, type AgentLimits, type AgentOptions, type AgentRuleOptions, type AgentRuntimeConfig, type AgentSnapshot, type AgentSnapshotMetadata, type AgentState, type BaseChatMessage, type BaseChatMessagePart, type BaseChatModel, type BaseMessage, type CodeDetectionRuleOptions, type CodeGuardrailOptions, type ConversationGuardrail, type CustomRuleCallback, type CustomRuleOptions, type GuardrailContext, type GuardrailDisposition, type GuardrailEvent, type GuardrailIncident, type GuardrailOutcome, GuardrailPhase, type GuardrailRule, type HandoffDescriptor, type InvokeConfig, type JsonGuardrailOptions, type JsonSchemaRuleOptions, type Message, type PendingToolApproval, type RegexGuardrailOptions, type RegexRuleOptions, type ResolvedTraceConfig, type ResolvedTraceSink, type RestoreSnapshotOptions, type SerializableSmartState, type SmartAgentEvent, type SmartAgentInstance, type SmartAgentLimits, type SmartAgentOptions, type SmartAgentTracingConfig, type SmartModel, type SmartState, type SmartToolFn, type SnapshotOptions, type SnapshotRuntimeHint, type TokenLimits, type ToolApprovalEvent, type ToolApprovalResolution, type ToolInterface, type TraceDataSection, type TraceErrorRecord, type TraceEventRecord, type TraceMessageSection, type TraceMetadataSection, type TraceSessionConfigSnapshot, type TraceSessionFile, type TraceSessionStatus, type TraceSessionSummary, type TraceSinkCognipeerConfig, type TraceSinkConfig, type TraceSinkCustomConfig, type TraceSinkFileConfig, type TraceSinkHttpConfig, type TraceSinkSnapshot, type TraceSummarySection, type TraceToolCallSection, type TraceToolResultSection, type TracingConfig, type TracingMode, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, startStreamingSession, withTools };

package/dist/index.d.ts

@@ -165,9 +165,15 @@ type TraceSinkHttpConfig = {
     headers?: Record<string, string>;
 };
 type TraceSinkConfig = TraceSinkFileConfig | TraceSinkCustomConfig | TraceSinkCognipeerConfig | TraceSinkHttpConfig;
+type TracingMode = "batched" | "streaming";
 type TracingConfig = {
     enabled: boolean;
     logData?: boolean;
+    /** Tracing mode:
+     * - "batched" (default): Events are collected and sent together when session ends
+     * - "streaming": Session starts immediately, events are sent in real-time as they occur
+     */
+    mode?: TracingMode;
     sink?: TraceSinkConfig;
 };
 type SmartAgentTracingConfig = TracingConfig;
@@ -372,10 +378,30 @@ type ResolvedTraceSink = {
 };
 type ResolvedTraceConfig = {
     enabled: boolean;
-    mode: "batched";
+    mode: TracingMode;
     logData: boolean;
     sink: ResolvedTraceSink;
 };
+type TraceSessionRuntime = {
+    sessionId: string;
+    startedAt: number;
+    resolvedConfig: ResolvedTraceConfig;
+    events: TraceEventRecord[];
+    summary: TraceSessionSummary;
+    status: TraceSessionStatus;
+    errors: TraceErrorRecord[];
+    fileBaseDir?: string;
+    fileSessionDir?: string;
+    /** For streaming mode: whether session start has been sent to sink */
+    sessionStarted?: boolean;
+    /** Agent info for streaming mode session start */
+    agentInfo?: {
+        name?: string;
+        version?: string;
+        model?: string;
+        provider?: string;
+    };
+};
 type HandoffDescriptor<TParsed = any> = {
     type: "handoff";
     toolName: string;
@@ -478,6 +504,25 @@ type PlanEvent = {
 type SummarizationEvent = {
     type: "summarization";
     summary: string;
+    /** Number of messages that were compressed/summarized */
+    messagesCompressed?: number;
+    /** Input tokens used for summarization prompt (from model response if available, otherwise estimated) */
+    inputTokens?: number;
+    /** Output tokens from summarization response (from model response if available, otherwise estimated) */
+    outputTokens?: number;
+    /** Cached input tokens from model response (prompt cache hit) */
+    cachedInputTokens?: number;
+    /** Total tokens (input + output) */
+    totalTokens?: number;
+    /** Duration of the summarization call in milliseconds */
+    durationMs?: number;
+    /** Previous summary content (if incremental summarization) */
+    previousSummary?: string;
+    /** Total token count before summarization */
+    tokenCountBefore?: number;
+    /** Total token count after summarization */
+    tokenCountAfter?: number;
+    /** @deprecated Use messagesCompressed instead */
     archivedCount?: number;
 };
 type FinalAnswerEvent = {
@@ -853,5 +898,10 @@ declare function customSink(handler: CustomSinkArg): TraceSinkConfig;
 declare function cognipeerSink(apiKey: string): TraceSinkConfig;
 declare function cognipeerSink(url: string | undefined, apiKey: string): TraceSinkConfig;
 declare function httpSink(url: string, headers?: Record<string, string>): TraceSinkConfig;
+/**
+ * Start a streaming trace session. Should be called before recording events in streaming mode.
+ * This sends the session start to the remote endpoint immediately.
+ */
+declare function startStreamingSession(session: TraceSessionRuntime | undefined, agentRuntime?: AgentRuntimeConfig): Promise<void>;
 
-export { type AIMessage, type AgentInstance, type AgentInvokeResult, type AgentLimits, type AgentOptions, type AgentRuleOptions, type AgentRuntimeConfig, type AgentSnapshot, type AgentSnapshotMetadata, type AgentState, type BaseChatMessage, type BaseChatMessagePart, type BaseChatModel, type BaseMessage, type CodeDetectionRuleOptions, type CodeGuardrailOptions, type ConversationGuardrail, type CustomRuleCallback, type CustomRuleOptions, type GuardrailContext, type GuardrailDisposition, type GuardrailEvent, type GuardrailIncident, type GuardrailOutcome, GuardrailPhase, type GuardrailRule, type HandoffDescriptor, type InvokeConfig, type JsonGuardrailOptions, type JsonSchemaRuleOptions, type Message, type PendingToolApproval, type RegexGuardrailOptions, type RegexRuleOptions, type ResolvedTraceConfig, type ResolvedTraceSink, type RestoreSnapshotOptions, type SerializableSmartState, type SmartAgentEvent, type SmartAgentInstance, type SmartAgentLimits, type SmartAgentOptions, type SmartAgentTracingConfig, type SmartModel, type SmartState, type SmartToolFn, type SnapshotOptions, type SnapshotRuntimeHint, type TokenLimits, type ToolApprovalEvent, type ToolApprovalResolution, type ToolInterface, type TraceDataSection, type TraceErrorRecord, type TraceEventRecord, type TraceMessageSection, type TraceMetadataSection, type TraceSessionConfigSnapshot, type TraceSessionFile, type TraceSessionStatus, type TraceSessionSummary, type TraceSinkCognipeerConfig, type TraceSinkConfig, type TraceSinkCustomConfig, type TraceSinkFileConfig, type TraceSinkHttpConfig, type TraceSinkSnapshot, type TraceSummarySection, type TraceToolCallSection, type TraceToolResultSection, type TracingConfig, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, withTools };
+export { type AIMessage, type AgentInstance, type AgentInvokeResult, type AgentLimits, type AgentOptions, type AgentRuleOptions, type AgentRuntimeConfig, type AgentSnapshot, type AgentSnapshotMetadata, type AgentState, type BaseChatMessage, type BaseChatMessagePart, type BaseChatModel, type BaseMessage, type CodeDetectionRuleOptions, type CodeGuardrailOptions, type ConversationGuardrail, type CustomRuleCallback, type CustomRuleOptions, type GuardrailContext, type GuardrailDisposition, type GuardrailEvent, type GuardrailIncident, type GuardrailOutcome, GuardrailPhase, type GuardrailRule, type HandoffDescriptor, type InvokeConfig, type JsonGuardrailOptions, type JsonSchemaRuleOptions, type Message, type PendingToolApproval, type RegexGuardrailOptions, type RegexRuleOptions, type ResolvedTraceConfig, type ResolvedTraceSink, type RestoreSnapshotOptions, type SerializableSmartState, type SmartAgentEvent, type SmartAgentInstance, type SmartAgentLimits, type SmartAgentOptions, type SmartAgentTracingConfig, type SmartModel, type SmartState, type SmartToolFn, type SnapshotOptions, type SnapshotRuntimeHint, type TokenLimits, type ToolApprovalEvent, type ToolApprovalResolution, type ToolInterface, type TraceDataSection, type TraceErrorRecord, type TraceEventRecord, type TraceMessageSection, type TraceMetadataSection, type TraceSessionConfigSnapshot, type TraceSessionFile, type TraceSessionStatus, type TraceSessionSummary, type TraceSinkCognipeerConfig, type TraceSinkConfig, type TraceSinkCustomConfig, type TraceSinkFileConfig, type TraceSinkHttpConfig, type TraceSinkSnapshot, type TraceSummarySection, type TraceToolCallSection, type TraceToolResultSection, type TracingConfig, type TracingMode, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, startStreamingSession, withTools };

package/dist/index.mjs

@@ -550,6 +550,7 @@ function ensureDir(p) {
 function withDefaults(config) {
   const enabled = Boolean(config?.enabled);
   const logData = config?.logData ?? DEFAULT_TRACE_CONFIG.logData;
+  const mode = config?.mode === "streaming" ? "streaming" : "batched";
   let sink;
   try {
     sink = resolveSink(config?.sink);
@@ -562,7 +563,7 @@ function withDefaults(config) {
   return {
     enabled,
     logData,
-    mode: "batched",
+    mode,
     sink
   };
 }
@@ -661,6 +662,108 @@ async function postTraceSession(url, headers, payload) {
     throw err;
   }
 }
+async function postStreamingSessionStart(baseUrl, headers, payload) {
+  const fetchFn = typeof globalThis.fetch === "function" ? globalThis.fetch.bind(globalThis) : void 0;
+  if (!fetchFn) {
+    throw new Error("HTTP sink requires fetch to be available in this runtime.");
+  }
+  const url = `${baseUrl.replace(/\/$/, "")}/${payload.sessionId}/start`;
+  const finalHeaders = { ...headers || {} };
+  if (!Object.keys(finalHeaders).some((key) => key.toLowerCase() === "content-type")) {
+    finalHeaders["content-type"] = "application/json";
+  }
+  try {
+    const response = await fetchFn(url, {
+      method: "POST",
+      headers: finalHeaders,
+      body: JSON.stringify(payload)
+    });
+    if (!response.ok) {
+      let responseText = "";
+      try {
+        responseText = await response.text();
+      } catch {
+      }
+      const statusLine = `HTTP ${response.status}${response.statusText ? ` ${response.statusText}` : ""}`;
+      const bodyPreview = responseText ? ` - ${responseText.slice(0, 200)}` : "";
+      throw new Error(`${statusLine}${bodyPreview}`);
+    }
+  } catch (err) {
+    const errMsg = err instanceof Error ? err.message : String(err);
+    if (typeof console !== "undefined" && typeof console.error === "function") {
+      console.error("[Tracing] Failed to post session start:", errMsg);
+    }
+    throw err;
+  }
+}
+async function postStreamingEvent(baseUrl, headers, sessionId, event) {
+  const fetchFn = typeof globalThis.fetch === "function" ? globalThis.fetch.bind(globalThis) : void 0;
+  if (!fetchFn) {
+    throw new Error("HTTP sink requires fetch to be available in this runtime.");
+  }
+  const url = `${baseUrl.replace(/\/$/, "")}/${sessionId}/events`;
+  const finalHeaders = { ...headers || {} };
+  if (!Object.keys(finalHeaders).some((key) => key.toLowerCase() === "content-type")) {
+    finalHeaders["content-type"] = "application/json";
+  }
+  try {
+    const response = await fetchFn(url, {
+      method: "POST",
+      headers: finalHeaders,
+      body: JSON.stringify({ event })
+    });
+    if (!response.ok) {
+      let responseText = "";
+      try {
+        responseText = await response.text();
+      } catch {
+      }
+      const statusLine = `HTTP ${response.status}${response.statusText ? ` ${response.statusText}` : ""}`;
+      const bodyPreview = responseText ? ` - ${responseText.slice(0, 200)}` : "";
+      throw new Error(`${statusLine}${bodyPreview}`);
+    }
+  } catch (err) {
+    const errMsg = err instanceof Error ? err.message : String(err);
+    if (typeof console !== "undefined" && typeof console.error === "function") {
+      console.error("[Tracing] Failed to post streaming event:", errMsg);
+    }
+    throw err;
+  }
+}
+async function postStreamingSessionEnd(baseUrl, headers, payload) {
+  const fetchFn = typeof globalThis.fetch === "function" ? globalThis.fetch.bind(globalThis) : void 0;
+  if (!fetchFn) {
+    throw new Error("HTTP sink requires fetch to be available in this runtime.");
+  }
+  const url = `${baseUrl.replace(/\/$/, "")}/${payload.sessionId}/end`;
+  const finalHeaders = { ...headers || {} };
+  if (!Object.keys(finalHeaders).some((key) => key.toLowerCase() === "content-type")) {
+    finalHeaders["content-type"] = "application/json";
+  }
+  try {
+    const response = await fetchFn(url, {
+      method: "POST",
+      headers: finalHeaders,
+      body: JSON.stringify(payload)
+    });
+    if (!response.ok) {
+      let responseText = "";
+      try {
+        responseText = await response.text();
+      } catch {
+      }
+      const statusLine = `HTTP ${response.status}${response.statusText ? ` ${response.statusText}` : ""}`;
+      const bodyPreview = responseText ? ` - ${responseText.slice(0, 200)}` : "";
+      throw new Error(`${statusLine}${bodyPreview}`);
+    }
+  } catch (err) {
+    const errMsg = err instanceof Error ? err.message : String(err);
+    if (typeof console !== "undefined" && typeof console.error === "function") {
+      console.error("[Tracing] Failed to post session end:", errMsg);
+    }
+    throw err;
+  }
+}
 function createEmptySummary() {
   return {
     totalDurationMs: 0,
@@ -837,7 +940,8 @@ function createTraceSession(opts) {
     events: [],
     summary: createEmptySummary(),
     status: "in_progress",
-    errors: []
+    errors: [],
+    sessionStarted: false
   };
   if (cfg.sink.type === "file") {
     const baseDir = cfg.sink.baseDir;
@@ -849,6 +953,49 @@ function createTraceSession(opts) {
   }
   return runtime;
 }
+async function startStreamingSession(session, agentRuntime) {
+  if (!session) return;
+  if (session.resolvedConfig.mode !== "streaming") return;
+  if (session.sessionStarted) return;
+  const sink = session.resolvedConfig.sink;
+  const agentInfo = agentRuntime ? {
+    name: agentRuntime.name || "unknown-agent",
+    version: agentRuntime.version,
+    model: getModelName(agentRuntime.model) || "unknown-model",
+    provider: getProviderName(agentRuntime.model)
+  } : void 0;
+  session.agentInfo = agentInfo;
+  const configSnapshot = buildConfigSnapshot(session);
+  const startedAtIso = new Date(session.startedAt).toISOString();
+  const payload = {
+    sessionId: session.sessionId,
+    startedAt: startedAtIso,
+    agent: agentInfo,
+    config: configSnapshot
+  };
+  if (sink.type === "cognipeer" || sink.type === "http") {
+    try {
+      const headers = sink.type === "cognipeer" ? { Authorization: `Bearer ${sink.apiKey}` } : sink.headers;
+      await postStreamingSessionStart(sink.url, headers, payload);
+      session.sessionStarted = true;
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      session.errors.push({
+        eventId: "session_start",
+        message,
+        type: "sink",
+        timestamp: startedAtIso
+      });
+      if (typeof console !== "undefined" && typeof console.warn === "function") {
+        console.warn("[Tracing] Failed to start streaming session, falling back to batched:", message);
+      }
+    }
+  } else if (sink.type === "custom" && typeof sink.onEvent === "function") {
+    session.sessionStarted = true;
+  } else if (sink.type === "file") {
+    session.sessionStarted = true;
+  }
+}
 function recordTraceEvent(session, event) {
   if (!session) return void 0;
   const sequence = session.events.length + 1;
@@ -951,6 +1098,23 @@ function recordTraceEvent(session, event) {
       }
     }
   }
+  if (session.resolvedConfig.mode === "streaming" && session.sessionStarted) {
+    if (sink.type === "cognipeer" || sink.type === "http") {
+      const headers = sink.type === "cognipeer" ? { Authorization: `Bearer ${sink.apiKey}` } : sink.headers;
+      postStreamingEvent(sink.url, headers, session.sessionId, record).catch((err) => {
+        const errMsg = err instanceof Error ? err.message : String(err);
+        if (typeof console !== "undefined" && typeof console.warn === "function") {
+          console.warn("[Tracing] Failed to send streaming event:", errMsg);
+        }
+        session.errors.push({
+          eventId: record.id,
+          message: errMsg,
+          type: "sink",
+          timestamp: timestampIso
+        });
+      });
+    }
+  }
   return record;
 }
 async function finalizeTraceSession(session, params = {}) {
@@ -985,7 +1149,7 @@ async function finalizeTraceSession(session, params = {}) {
     startedAt: startedAtIso,
     endedAt: endedAtIso,
     durationMs,
-    agent: agentInfo,
+    agent: agentInfo || session.agentInfo,
     config: configSnapshot,
     summary: session.summary,
     events: session.events,
@@ -995,10 +1159,22 @@ async function finalizeTraceSession(session, params = {}) {
   const payloadForSink = buildPayload(initialStatus);
   let sinkFailed = false;
   const sink = session.resolvedConfig.sink;
+  const isStreamingMode = session.resolvedConfig.mode === "streaming";
   if (sink.type === "cognipeer" || sink.type === "http") {
     try {
       const headers = sink.type === "cognipeer" ? { Authorization: `Bearer ${sink.apiKey}` } : sink.headers;
-      await postTraceSession(sink.url, headers, payloadForSink);
+      if (isStreamingMode && session.sessionStarted) {
+        await postStreamingSessionEnd(sink.url, headers, {
+          sessionId: session.sessionId,
+          endedAt: endedAtIso,
+          durationMs,
+          status: initialStatus,
+          summary: session.summary,
+          errors: session.errors
+        });
+      } else {
+        await postTraceSession(sink.url, headers, payloadForSink);
+      }
     } catch (err) {
       sinkFailed = true;
       const message = err instanceof Error ? err.message : String(err);
@@ -2458,10 +2634,14 @@ function createContextSummarizeNode(opts) {
     if (!config.enabled) return {};
     const model = opts.model;
     const messages = state.messages || [];
+    const onEvent = state.ctx?.__onEvent;
+    const traceSession = state.ctx?.__traceSession;
     const compressableMessages = messages.filter((m) => m.role === "tool" && m.content !== "SUMMARIZED");
     if (compressableMessages.length === 0) {
       return {};
     }
+    const allTextBefore = messages.map((m) => typeof m.content === "string" ? m.content : Array.isArray(m.content) ? m.content.map((c) => typeof c === "string" ? c : c?.text ?? c?.content ?? "").join("") : "").join("\n");
+    const tokenCountBefore = countApproxTokens(allTextBefore);
     const summaryPromptMaxTokens = config.summaryPromptMaxTokens;
     const availableTokensForConversation = summaryPromptMaxTokens - 500;
     const messageTexts = [];
@@ -2510,12 +2690,54 @@ Summary:`;
       systemMessage("You are a helpful assistant that summarizes conversation history efficiently."),
       humanMessage(promptBody)
     ];
+    const inputTokensEstimate = countApproxTokens(summaryPrompt.map((m) => m.content).join("\n"));
     let summaryText = "Summary unavailable.";
+    let outputTokensEstimate = 0;
+    let durationMs = 0;
+    let inputTokensActual;
+    let outputTokensActual;
+    let cachedInputTokens;
+    let totalTokensActual;
+    const startTime = Date.now();
     try {
       const response = await model.invoke(summaryPrompt);
+      durationMs = Date.now() - startTime;
       summaryText = typeof response.content === "string" ? response.content : Array.isArray(response.content) ? response.content.map((c) => c.text || "").join("") : JSON.stringify(response.content);
+      outputTokensEstimate = countApproxTokens(summaryText);
+      const rawUsage = response?.usage || response?.response_metadata?.token_usage || response?.response_metadata?.tokenUsage || response?.response_metadata?.usage;
+      const normalized = normalizeUsage(rawUsage);
+      if (normalized) {
+        inputTokensActual = normalized.prompt_tokens;
+        outputTokensActual = normalized.completion_tokens;
+        totalTokensActual = normalized.total_tokens;
+        cachedInputTokens = normalized.prompt_tokens_details?.cached_tokens;
+      }
     } catch (err) {
+      durationMs = Date.now() - startTime;
       console.error("[ContextSummarize] Failed to generate summary:", err);
+      const errorEvent = {
+        type: "summarization",
+        summary: "",
+        messagesCompressed: 0,
+        inputTokens: inputTokensEstimate,
+        outputTokens: 0,
+        durationMs,
+        previousSummary: previousSummary || void 0,
+        tokenCountBefore,
+        tokenCountAfter: tokenCountBefore
+      };
+      onEvent?.(errorEvent);
+      recordTraceEvent(traceSession, {
+        type: "summarization",
+        label: "Context Summarization Failed",
+        actor: { scope: "system", name: "summarizer", role: "summarization" },
+        status: "error",
+        durationMs,
+        inputTokens: inputTokensEstimate,
+        outputTokens: 0,
+        error: { message: err?.message || String(err), stack: err?.stack },
+        messageList: sanitizeTracePayload(summaryPrompt)
+      });
       return {};
     }
     const toolCallIdToAssistantIdx = /* @__PURE__ */ new Map();
@@ -2567,8 +2789,51 @@ Summary:`;
       content: summaryText
     };
     const summaries = Array.isArray(state.summaries) ? [...state.summaries, summaryText] : [summaryText];
+    const finalMessages = [...validatedMessages, assistantSummaryCall, toolSummaryResponse];
+    const allTextAfter = finalMessages.map((m) => typeof m.content === "string" ? m.content : Array.isArray(m.content) ? m.content.map((c) => typeof c === "string" ? c : c?.text ?? c?.content ?? "").join("") : "").join("\n");
+    const tokenCountAfter = countApproxTokens(allTextAfter);
+    const summarizationEvent = {
+      type: "summarization",
+      summary: summaryText,
+      messagesCompressed: compressableMessages.length,
+      inputTokens: inputTokensActual ?? inputTokensEstimate,
+      outputTokens: outputTokensActual ?? outputTokensEstimate,
+      cachedInputTokens,
+      totalTokens: totalTokensActual ?? (inputTokensActual ?? inputTokensEstimate) + (outputTokensActual ?? outputTokensEstimate),
+      durationMs,
+      previousSummary: previousSummary || void 0,
+      tokenCountBefore,
+      tokenCountAfter,
+      archivedCount: compressableMessages.length
+      // deprecated field for backward compatibility
+    };
+    onEvent?.(summarizationEvent);
+    recordTraceEvent(traceSession, {
+      type: "summarization",
+      label: "Context Summarization",
+      actor: { scope: "system", name: "summarizer", role: "summarization" },
+      status: "success",
+      durationMs,
+      inputTokens: inputTokensActual ?? inputTokensEstimate,
+      outputTokens: outputTokensActual ?? outputTokensEstimate,
+      cachedInputTokens,
+      totalTokens: totalTokensActual,
+      messageList: sanitizeTracePayload([
+        ...summaryPrompt,
+        { role: "assistant", content: summaryText, name: "summarization_result" }
+      ]),
+      debug: {
+        messagesCompressed: compressableMessages.length,
+        tokenCountBefore,
+        tokenCountAfter,
+        tokensSaved: tokenCountBefore - tokenCountAfter,
+        previousSummaryLength: previousSummary?.length || 0,
+        usageFromModel: inputTokensActual !== void 0,
+        cachedInputTokens
+      }
+    });
     return {
-      messages: [...validatedMessages, assistantSummaryCall, toolSummaryResponse],
+      messages: finalMessages,
       summaries
     };
   };
@@ -3095,6 +3360,6 @@ function createCodeGuardrail(options) {
   });
 }
 
-export { GuardrailPhase, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, withTools };
+export { GuardrailPhase, agentVerdictRule, applyTokenLimits, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, createAgent, createAgentNode, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGuardrail, createJsonGuardrail, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, evaluateGuardrails, fileSink, fromLangchainModel, fromLangchainTools, httpSink, isSmartModel, jsonSchemaRule, regexRule, resolveToolApprovalState, restoreSnapshot, startStreamingSession, withTools };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map