fix: properly pass initialMessages to LocalRuntime (#1090)

Co-Authored-By: igo <[email protected]>
Yonom · Oct 26, 2024 · 5d43e55 · 5d43e55
1 parent 006ff20
commit 5d43e55
Show file tree

Hide file tree

Showing 12 changed files with 28 additions and 11 deletions.
diff --git a/apps/docs/app/(home)/page.tsx b/apps/docs/app/(home)/page.tsx
@@ -138,6 +138,17 @@ export type AssistantProps = {
 const MyRuntimeProvider = ({ children }: { children: React.ReactNode }) => {
   const runtime = useEdgeRuntime({
     api: "/api/chat",
+    initialMessages: [
+      {
+        role: "system",
+        content: [{ type: "text", text: "You are weather assistant" }],
+      },
+      {
+        role: "user",
+        content: [{ type: "text", text: "What's the weather in Paris?" }],
+      },
+      { role: "assistant", content: [{ type: "text", text: "It's sunny." }] },
+    ],
     adapters: {
       attachments: new CompositeAttachmentAdapter([
         new SimpleImageAttachmentAdapter(),

diff --git a/packages/react-ai-sdk/package.json b/packages/react-ai-sdk/package.json
@@ -34,7 +34,7 @@
     "zustand": "^5.0.0"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@types/react": "*",
     "react": "^18 || ^19"
   },

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

diff --git a/packages/react-langgraph/package.json b/packages/react-langgraph/package.json
@@ -30,7 +30,7 @@
     "zod": "^3.23.8"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@types/react": "*",
     "react": "^18 || ^19"
   },

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.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@types/react": "*",
     "react": "^18 || ^19",
     "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.0"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@types/react": "*",
     "react": "^18 || ^19",
     "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.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@assistant-ui/react-markdown": "^0.2.25",
     "@types/react": "*",
     "@types/react-syntax-highlighter": "*",

diff --git a/packages/react-trieve/package.json b/packages/react-trieve/package.json
@@ -46,7 +46,7 @@
     "unist-util-visit": "^5.0.0"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.95",
+    "@assistant-ui/react": "^0.5.96",
     "@assistant-ui/react-markdown": "^0.2.25",
     "trieve-ts-sdk": "^0.0.13",
     "@types/react": "*",

diff --git a/packages/react/CHANGELOG.md b/packages/react/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @assistant-ui/react
 
+## 0.5.96
+
+### Patch Changes
+
+- fix: properly pass initialMessages to LocalRuntime
+
 ## 0.5.95
 
 ### Patch Changes

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

diff --git a/packages/react/src/runtimes/local/LocalRuntimeCore.tsx b/packages/react/src/runtimes/local/LocalRuntimeCore.tsx
@@ -25,7 +25,7 @@ export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
 
   constructor(
     options: LocalRuntimeOptionsBase,
-    initialMessages?: CoreMessage[],
+    initialMessages: readonly CoreMessage[] | undefined,
   ) {
     super();
 

diff --git a/packages/react/src/runtimes/local/useLocalRuntime.tsx b/packages/react/src/runtimes/local/useLocalRuntime.tsx
@@ -37,7 +37,7 @@ class LocalRuntimeImpl extends AssistantRuntimeImpl implements LocalRuntime {
 
 export const useLocalRuntime = (
   adapter: ChatModelAdapter,
-  options: LocalRuntimeOptions = {},
+  { initialMessages, ...options }: LocalRuntimeOptions = {},
 ) => {
   const opt = {
     ...options,
@@ -47,7 +47,7 @@ export const useLocalRuntime = (
     },
   };
 
-  const [runtime] = useState(() => new LocalRuntimeCore(opt));
+  const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
 
   useEffect(() => {
     runtime.setOptions(opt);