From 474859aaaa3dd256d1ebe19751591dcfd994eb51 Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Tue, 28 Apr 2026 01:24:53 -0700
Subject: [PATCH] test(agents): cover raw model cache trace stage

---
 src/agents/cache-trace.test.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/agents/cache-trace.test.ts b/src/agents/cache-trace.test.ts
index a1aa70c53f6..f0d751072fb 100644
--- a/src/agents/cache-trace.test.ts
+++ b/src/agents/cache-trace.test.ts
@@ -91,6 +91,19 @@ describe("createCacheTrace", () => {
     expect(event.system).toBe("");
   });
 
+  it("records raw model run session stages", () => {
+    const { lines, trace } = createMemoryTraceForTest();
+
+    trace?.recordStage("session:raw-model-run", {
+      messages: [],
+      system: "",
+    });
+
+    const event = JSON.parse(lines[0]?.trim() ?? "{}") as Record<string, unknown>;
+    expect(event.stage).toBe("session:raw-model-run");
+    expect(event.system).toBe("");
+  });
+
   it("records stream context from systemPrompt when wrapping stream functions", () => {
     const lines: string[] = [];
     const trace = createCacheTrace({