Expose setMessages for useAssistant hook (#1175)

vercel · Mar 18, 2024 · f51e438 · f51e438
1 parent 500b1fd
commit f51e438
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/docs/pages/docs/api-reference/use-assistant.mdx b/docs/pages/docs/api-reference/use-assistant.mdx
@@ -129,6 +129,11 @@ The `experimental_useAssistant` hook returns an object with several helper metho
 <OptionTable
   options={[
     ['messages', 'Message[]', 'The current array of chat messages.'],
+    [
+      'setMessages',
+      'React.Dispatch<React.SetStateAction<Message>>',
+      'Function to update the `messages` array.',
+    ],
     ['threadId', 'string | undefined', 'The current thread ID.'],
     ['input', 'string', 'The current value of the input field.'],
     [

diff --git a/packages/core/react/use-assistant.ts b/packages/core/react/use-assistant.ts
@@ -1,9 +1,10 @@
 /* eslint-disable react-hooks/rules-of-hooks */
 
+import { nanoid } from 'nanoid';
 import { useState } from 'react';
+
 import { readDataStream } from '../shared/read-data-stream';
 import { Message } from '../shared/types';
-import { nanoid } from 'nanoid';
 
 export type AssistantStatus = 'in_progress' | 'awaiting_message';
 
@@ -13,6 +14,11 @@ export type UseAssistantHelpers = {
    */
   messages: Message[];
 
+  /**
+   * setState-powered method to update the messages array.
+   */
+  setMessages: React.Dispatch<React.SetStateAction<Message[]>>;
+
   /**
    * The current thread ID.
    */
@@ -236,6 +242,7 @@ export function experimental_useAssistant({
 
   return {
     messages,
+    setMessages,
     threadId,
     input,
     setInput,