chore: Fix types in tests 26/N.

This commit is contained in:
cpojer
2026-02-17 14:31:40 +09:00
parent 6e5df1dc0f
commit 4235435309
14 changed files with 57 additions and 45 deletions

View File

@@ -1,6 +1,5 @@
import type { StreamFn } from "@mariozechner/pi-agent-core";
import type { Context, Model, SimpleStreamOptions } from "@mariozechner/pi-ai";
import { AssistantMessageEventStream } from "@mariozechner/pi-ai";
import { describe, expect, it } from "vitest";
import { applyExtraParamsToAgent, resolveExtraParams } from "./pi-embedded-runner.js";
@@ -77,7 +76,7 @@ describe("applyExtraParamsToAgent", () => {
const payload = { store: false };
const baseStreamFn: StreamFn = (_model, _context, options) => {
options?.onPayload?.(payload);
return new AssistantMessageEventStream();
return {} as ReturnType<StreamFn>;
};
const agent = { streamFn: baseStreamFn };
applyExtraParamsToAgent(agent, undefined, params.applyProvider, params.applyModelId);
@@ -90,7 +89,7 @@ describe("applyExtraParamsToAgent", () => {
const calls: Array<SimpleStreamOptions | undefined> = [];
const baseStreamFn: StreamFn = (_model, _context, options) => {
calls.push(options);
return new AssistantMessageEventStream();
return {} as ReturnType<StreamFn>;
};
const agent = { streamFn: baseStreamFn };
@@ -159,7 +158,7 @@ describe("applyExtraParamsToAgent", () => {
const payload = { store: false };
const baseStreamFn: StreamFn = (_model, _context, options) => {
options?.onPayload?.(payload);
return new AssistantMessageEventStream();
return {} as ReturnType<StreamFn>;
};
const agent = { streamFn: baseStreamFn };