From 1811e5492015c2d1b1c51f99ddbc598068c646d2 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sun, 5 Apr 2026 11:58:02 +0900
Subject: [PATCH] test: fix plugin stream typing assertions

---
 extensions/openrouter/index.test.ts | 4 +++-
 extensions/xai/stream.test.ts       | 9 ++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/extensions/openrouter/index.test.ts b/extensions/openrouter/index.test.ts
index ff633d3f9d5..51690e84d25 100644
--- a/extensions/openrouter/index.test.ts
+++ b/extensions/openrouter/index.test.ts
@@ -85,7 +85,9 @@ describe("openrouter provider hooks", () => {
     );
 
     expect(baseStreamFn).toHaveBeenCalledOnce();
-    expect(baseStreamFn.mock.calls[0]?.[0]).toMatchObject({
+    const firstCall = baseStreamFn.mock.calls[0];
+    const firstModel = firstCall?.[0];
+    expect(firstModel).toMatchObject({
       compat: {
         openRouterRouting: {
           order: ["moonshot"],
diff --git a/extensions/xai/stream.test.ts b/extensions/xai/stream.test.ts
index c73ef03cd1f..4787442a2b1 100644
--- a/extensions/xai/stream.test.ts
+++ b/extensions/xai/stream.test.ts
@@ -10,7 +10,10 @@ import {
 type ToolPayload = {
   function?: Record<string, unknown>;
 };
-
+type XaiTestPayload = Record<string, unknown> & {
+  tools?: Array<{ type?: string; function?: Record<string, unknown> }>;
+  input?: unknown[];
+};
 function captureWrappedModelId(params: {
   modelId: string;
   fastMode: boolean;
@@ -63,10 +66,10 @@ describe("xai stream wrappers", () => {
 
   it("composes the xai provider stream chain from extra params", () => {
     let capturedModelId = "";
-    let capturedPayload: Record<string, unknown> | undefined;
+    let capturedPayload: XaiTestPayload | undefined;
     const baseStreamFn: StreamFn = (model, _context, options) => {
       capturedModelId = String(model.id);
-      const payload: Record<string, unknown> = {
+      const payload: XaiTestPayload = {
         reasoning: { effort: "high" },
         tools: [
           {