test: share provider stream capture helper

This commit is contained in:
Peter Steinberger
2026-04-20 22:58:37 +01:00
parent 614d0348a5
commit eb94d3af94
3 changed files with 26 additions and 26 deletions

View File

@@ -0,0 +1,18 @@
import type { StreamFn } from "@mariozechner/pi-agent-core";
export function createCapturedThinkingConfigStream() {
let capturedPayload: Record<string, unknown> | undefined;
const streamFn: StreamFn = (model, _context, options) => {
const payload = { config: { thinkingConfig: { thinkingBudget: -1 } } } as Record<
string,
unknown
>;
options?.onPayload?.(payload as never, model as never);
capturedPayload = payload;
return {} as never;
};
return {
streamFn,
getCapturedPayload: () => capturedPayload,
};
}