diff --git a/extensions/kimi-coding/stream.test.ts b/extensions/kimi-coding/stream.test.ts index 230fd673f3c..42dc36c5145 100644 --- a/extensions/kimi-coding/stream.test.ts +++ b/extensions/kimi-coding/stream.test.ts @@ -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);