From 1471f25b45beebbae93c9e731701ff4d306fb774 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 20 Apr 2026 18:19:49 +0100
Subject: [PATCH] test(kimi): preserve async stream arguments

---
 extensions/kimi-coding/stream.test.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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);