From d9b8a5a41cf32ab0f3e2704598a4115cb9327864 Mon Sep 17 00:00:00 2001
From: SC23 <jerrynh770@gmail.com>
Date: Sat, 24 Jan 2026 21:38:03 +0800
Subject: [PATCH 1/2] fix: rename _options to options to indicate the parameter
 is now used

---
 src/server.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/server.ts b/src/server.ts
index f3adf21..0c10ece 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -34,7 +34,7 @@ export class Chat extends AIChatAgent<Env> {
    */
   async onChatMessage(
     onFinish: StreamTextOnFinishCallback<ToolSet>,
-    _options?: { abortSignal?: AbortSignal }
+    options?: { abortSignal?: AbortSignal }
   ) {
     // const mcpConnection = await this.mcp.connect(
     //   "https://path-to-mcp-server/sse"

From d7e7411d08aa7750697ed1f7bb59e18edd0694b5 Mon Sep 17 00:00:00 2001
From: SC23 <jerrynh770@gmail.com>
Date: Sat, 24 Jan 2026 22:23:54 +0800
Subject: [PATCH 2/2] fix: pass abortSignal to streamText to enable stop
 streaming

---
 src/server.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/server.ts b/src/server.ts
index 0c10ece..9539722 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -76,7 +76,8 @@ If the user asks to schedule a task, use the schedule tool to schedule the task.
           onFinish: onFinish as unknown as StreamTextOnFinishCallback<
             typeof allTools
           >,
-          stopWhen: stepCountIs(10)
+          stopWhen: stepCountIs(10),
+          abortSignal: options?.abortSignal
         });
 
         writer.merge(result.toUIMessageStream());