test(agents): align replay test harness with sanitizer

steipete · steipete · commit 51507e152ee5 · 2026-05-31T14:01:13.000+01:00
diff --git a/src/agents/embedded-agent-runner.sanitize-session-history.test-harness.ts b/src/agents/embedded-agent-runner.sanitize-session-history.test-harness.ts
@@ -4,17 +4,8 @@ import type { SessionManager } from "./sessions/index.js";
 import type { TranscriptPolicy } from "./transcript-policy.js";
 
 type SessionEntry = { type: string; customType: string; data: unknown };
-export type SanitizeSessionHistoryFn = (params: {
-  messages: AgentMessage[];
-  modelApi: string;
-  provider: string;
-  allowedToolNames?: Iterable<string>;
-  sessionManager: SessionManager;
-  sessionId: string;
-  modelId?: string;
-  policy?: TranscriptPolicy;
-  preserveLatestAssistantThinking?: boolean;
-}) => Promise<AgentMessage[]>;
+export type SanitizeSessionHistoryFn =
+  typeof import("./embedded-agent-runner/replay-history.js").sanitizeSessionHistory;
 type SanitizeSessionHistoryMockedHelpers = typeof import("./embedded-agent-helpers.js");
 export type SanitizeSessionHistoryHarness = {
   sanitizeSessionHistory: SanitizeSessionHistoryFn;
diff --git a/src/agents/openai-transport-stream.ts b/src/agents/openai-transport-stream.ts
@@ -3524,7 +3524,7 @@ function shouldPreserveOpenRouterReasoningReplay(model: OpenAIModeModel): boolea
 }
 
 function shouldTrustReasoningContentReplayMetadata(model: OpenAIModeModel): boolean {
-  if (model.reasoning !== true || isGemma4ModelId(model.id)) {
+  if (!model.reasoning || isGemma4ModelId(model.id)) {
     return false;
   }
   const provider = model.provider.trim().toLowerCase();

Original file line number	Diff line number	Diff line change
`@@ -3524,7 +3524,7 @@ function shouldPreserveOpenRouterReasoningReplay(model: OpenAIModeModel): boolea`
`3524`	`3524`	`}`
`3525`	`3525`
`3526`	`3526`	`function shouldTrustReasoningContentReplayMetadata(model: OpenAIModeModel): boolean {`
`3527`		`- if (model.reasoning !== true \|\| isGemma4ModelId(model.id)) {`
	`3527`	`+ if (!model.reasoning \|\| isGemma4ModelId(model.id)) {`
`3528`	`3528`	`return false;`
`3529`	`3529`	`}`
`3530`	`3530`	`const provider = model.provider.trim().toLowerCase();`