fix(agent): decouple agent streams from compat

closes #6851
2026-07-20 17:54:17 +02:00
parent 3a40794ea1
commit 1235c0ec64
29 changed files with 219 additions and 105 deletions
@@ -2,7 +2,7 @@ import { existsSync, mkdtempSync, rmSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
 import { Agent } from "@earendil-works/pi-agent-core";
-import { fauxAssistantMessage, registerFauxProvider } from "@earendil-works/pi-ai/compat";
+import { fauxAssistantMessage, registerFauxProvider, streamSimple } from "@earendil-works/pi-ai/compat";
 import { afterEach, describe, expect, it } from "vitest";
 import { AgentSession } from "../../../src/core/agent-session.ts";
 import { AuthStorage } from "../../../src/core/auth-storage.ts";
@@ -61,6 +61,7 @@ describe("regression #5596: missing configured theme export", () => {
 				tools: [],
 			},
 			convertToLlm,
+			streamFunction: streamSimple,
 		});
 		const session = new AgentSession({
 			agent,
@@ -17,7 +17,7 @@ describe("issue #6324 branch summary ambient auth", () => {
 		harnesses.push(harness);

 		let streamCallCount = 0;
-		harness.session.agent.streamFn = (model, _context, options) => {
+		harness.session.agent.streamFunction = (model, _context, options) => {
 			streamCallCount++;
 			expect(options?.apiKey).toBeUndefined();