refactor AIHttpChatAgent to utilize a new ResumableStreamManager

Author: whoiskatrin

examples/resumable-stream-chat/src/server.ts

@@ -16,7 +16,8 @@ type Env = {
 
 export class ResumableChatAgent extends AIHttpChatAgent<Env> {
   async onChatMessage(
-    onFinish: StreamTextOnFinishCallback<ToolSet>
+    onFinish: StreamTextOnFinishCallback<ToolSet>,
+    _options?: { streamId?: string }
   ): Promise<Response | undefined> {
     const stream = createUIMessageStream({
       execute: async ({ writer }) => {
@@ -30,7 +31,8 @@ export class ResumableChatAgent extends AIHttpChatAgent<Env> {
       }
     });
 
-    return createUIMessageStreamResponse({ stream });
+    const response = createUIMessageStreamResponse({ stream });
+    return response;
   }
 }
 

examples/resumable-stream-chat/wrangler.jsonc

@@ -0,0 +1,26 @@
+{
+  "name": "resumable-chat",
+  "main": "./src/server.ts",
+  "compatibility_date": "2025-02-21",
+  "compatibility_flags": [
+    "nodejs_compat",
+    "nodejs_compat_populate_process_env"
+  ],
+  "assets": {
+    "directory": "public"
+  },
+  "durable_objects": {
+    "bindings": [
+      {
+        "name": "ResumableChatAgent",
+        "class_name": "ResumableChatAgent"
+      }
+    ]
+  },
+  "migrations": [
+    {
+      "tag": "v1",
+      "new_sqlite_classes": ["ResumableChatAgent"]
+    }
+  ]
+}