fix: ensure message status is set on runResultStream flush (#1257)

Yonom · Dec 19, 2024 · 87c69c1 · 87c69c1
1 parent 2b1007b
commit 87c69c1
Show file tree

Hide file tree

Showing 10 changed files with 28 additions and 8 deletions.
diff --git a/packages/react-ai-sdk/package.json b/packages/react-ai-sdk/package.json
@@ -34,7 +34,7 @@
     "zustand": "^5.0.2"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },

diff --git a/packages/react-hook-form/package.json b/packages/react-hook-form/package.json
@@ -30,7 +30,7 @@
     "zod": "^3.24.1"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc",
     "react-hook-form": "^7"

diff --git a/packages/react-langgraph/package.json b/packages/react-langgraph/package.json
@@ -30,7 +30,7 @@
     "zod": "^3.24.1"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },

diff --git a/packages/react-markdown/package.json b/packages/react-markdown/package.json
@@ -41,7 +41,7 @@
     "react-markdown": "^9.0.1"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc",
     "tailwindcss": "^3.4.4"

diff --git a/packages/react-playground/package.json b/packages/react-playground/package.json
@@ -47,7 +47,7 @@
     "zustand": "^5.0.2"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc",
     "tailwindcss": "^3.4.4"

diff --git a/packages/react-syntax-highlighter/package.json b/packages/react-syntax-highlighter/package.json
@@ -27,7 +27,7 @@
     "build": "tsup src/index.ts --format cjs,esm --dts --sourcemap --clean"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@assistant-ui/react-markdown": "^0.7.5",
     "@types/react": "*",
     "@types/react-syntax-highlighter": "*",

diff --git a/packages/react-trieve/package.json b/packages/react-trieve/package.json
@@ -45,7 +45,7 @@
     "unist-util-visit": "^5.0.0"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.12",
+    "@assistant-ui/react": "^0.7.16",
     "@assistant-ui/react-markdown": "^0.7.5",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc",

diff --git a/packages/react/CHANGELOG.md b/packages/react/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @assistant-ui/react
 
+## 0.7.16
+
+### Patch Changes
+
+- fix: ensure message status is set on runResultStream flush
+
 ## 0.7.15
 
 ### Patch Changes

diff --git a/packages/react/package.json b/packages/react/package.json
@@ -29,7 +29,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.7.15",
+  "version": "0.7.16",
   "license": "MIT",
   "exports": {
     ".": {

diff --git a/packages/react/src/runtimes/edge/streams/runResultStream.ts b/packages/react/src/runtimes/edge/streams/runResultStream.ts
@@ -75,6 +75,20 @@ export function runResultStream() {
         }
       }
     },
+    flush(controller) {
+      if (message.status?.type === "running") {
+        const requiresAction = message.content?.at(-1)?.type === "tool-call";
+        message = appendOrUpdateFinish(message, {
+          type: "finish",
+          finishReason: requiresAction ? "tool-calls" : "unknown",
+          usage: {
+            promptTokens: 0,
+            completionTokens: 0,
+          },
+        });
+        controller.enqueue(message);
+      }
+    },
   });
 }