From e9d9726f2d8f51847b3944db0935bd00b7b2130a Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 25 Apr 2026 19:54:18 +0100
Subject: [PATCH] fix: handle context assembled diagnostics

---
 src/logging/diagnostic-stability.ts | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/logging/diagnostic-stability.ts b/src/logging/diagnostic-stability.ts
index 50b0357a3e9..58b41f8cd41 100644
--- a/src/logging/diagnostic-stability.ts
+++ b/src/logging/diagnostic-stability.ts
@@ -318,6 +318,13 @@ function sanitizeDiagnosticEvent(event: DiagnosticEventPayload): DiagnosticStabi
       record.durationMs = event.durationMs;
       assignReasonCode(record, event.errorCategory);
       break;
+    case "context.assembled":
+      record.provider = event.provider;
+      record.model = event.model;
+      record.channel = event.channel;
+      record.count = event.messageCount;
+      record.bytes = event.promptChars;
+      break;
     case "log.record":
       record.level = event.level;
       record.source = event.loggerName;