test(kimi): preserve async stream arguments

This commit is contained in:
Peter Steinberger
2026-04-20 18:19:49 +01:00
parent 1fc7dd2fc1
commit 1471f25b45

View File

@@ -197,7 +197,8 @@ describe("kimi tool-call markup wrapper", () => {
it("supports async stream functions", async () => {
const finalMessage = createAssistantTextMessage(KIMI_TOOL_TEXT);
const baseStreamFn: StreamFn = async () => createResultStreamFn(finalMessage)();
const baseStreamFn: StreamFn = async (model, context, options) =>
createResultStreamFn(finalMessage)(model, context, options);
const wrapped = createKimiToolCallMarkupWrapper(baseStreamFn);
const stream = await callKimiStream(wrapped);