From b702be1f78c36c9a54407491963becb6d3773b8c Mon Sep 17 00:00:00 2001
From: Gustavo Madeira Santana <gumadeiras@gmail.com>
Date: Wed, 18 Feb 2026 16:14:21 -0500
Subject: [PATCH] tests: type chat.send block streaming options capture

---
 src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts b/src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts
index 630b9b3f1a0..a40a91b09a9 100644
--- a/src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts
+++ b/src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts
@@ -2,6 +2,7 @@ import fs from "node:fs/promises";
 import os from "node:os";
 import path from "node:path";
 import { describe, expect, test, vi } from "vitest";
+import type { GetReplyOptions } from "../auto-reply/types.js";
 import { __setMaxChatHistoryMessagesBytesForTest } from "./server-constants.js";
 import {
   connectOk,
@@ -138,7 +139,7 @@ describe("gateway server chat", () => {
 
   test("chat.send does not force-disable block streaming", async () => {
     await withGatewayChatHarness(async ({ ws, createSessionDir }) => {
-      const spy = vi.mocked(getReplyFromConfig);
+      const spy = vi.mocked(getReplyFromConfig) as unknown as ReturnType<typeof vi.fn>;
       await connectOk(ws);
 
       await createSessionDir();
@@ -146,8 +147,8 @@ describe("gateway server chat", () => {
       testState.agentConfig = { blockStreamingDefault: "on" };
       try {
         spy.mockReset();
-        let capturedOpts: Parameters<typeof getReplyFromConfig>[1];
-        spy.mockImplementationOnce(async (_ctx, opts) => {
+        let capturedOpts: GetReplyOptions | undefined;
+        spy.mockImplementationOnce(async (_ctx: unknown, opts?: GetReplyOptions) => {
           capturedOpts = opts;
         });