fix: send request data when reload (#35)

run-llama · Nov 29, 2024 · 8214bcd · 8214bcd
1 parent 23482ea
commit 8214bcd
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/.changeset/red-pumas-cheat.md b/.changeset/red-pumas-cheat.md
@@ -0,0 +1,5 @@
+---
+'@llamaindex/chat-ui': patch
+---
+
+fix: send request data when reload
diff --git a/packages/chat-ui/src/chat/chat-messages.tsx b/packages/chat-ui/src/chat/chat-messages.tsx
@@ -146,7 +146,7 @@ function ChatMessagesLoading(props: ChatMessagesLoadingProps) {
 }
 
 function ChatActions(props: ChatActionsProps) {
-  const { reload, stop } = useChatUI()
+  const { reload, stop, requestData } = useChatUI()
   const { showReload, showStop } = useChatMessages()
   if (!showStop && !showReload) return null
 
@@ -159,7 +159,11 @@ function ChatActions(props: ChatActionsProps) {
         </Button>
       )}
       {showReload && (
-        <Button variant="outline" size="sm" onClick={reload}>
+        <Button
+          variant="outline"
+          size="sm"
+          onClick={() => reload?.({ data: requestData })}
+        >
           <RefreshCw className="mr-2 h-4 w-4" />
           Regenerate
         </Button>

diff --git a/packages/chat-ui/src/chat/chat.interface.ts b/packages/chat-ui/src/chat/chat.interface.ts
@@ -11,7 +11,7 @@ export type ChatHandler = {
   setInput: (input: string) => void
   isLoading: boolean
   messages: Message[]
-  reload?: () => void
+  reload?: (chatRequestOptions?: { data?: any }) => void
   stop?: () => void
   append: (
     message: Message,