From e5b89c54005e7da0eaa8a9d1cf8862ad91ca3318 Mon Sep 17 00:00:00 2001
From: ArnoChen <arnochen101@gmail.com>
Date: Sun, 16 Feb 2025 21:58:30 +0800
Subject: [PATCH] fix retrieval testing and query settings

---
 .../src/components/retrieval/QuerySettings.tsx    |  2 +-
 lightrag_webui/src/features/RetrievalTesting.tsx  | 15 +++++----------
 2 files changed, 6 insertions(+), 11 deletions(-)

diff --git a/lightrag_webui/src/components/retrieval/QuerySettings.tsx b/lightrag_webui/src/components/retrieval/QuerySettings.tsx
index 3051b33a..bef34857 100644
--- a/lightrag_webui/src/components/retrieval/QuerySettings.tsx
+++ b/lightrag_webui/src/components/retrieval/QuerySettings.tsx
@@ -75,7 +75,7 @@ export default function QuerySettings() {
                 </SelectTrigger>
                 <SelectContent>
                   <SelectGroup>
-                    <SelectItem value="Multiple Paragraphs">Single Paragraph</SelectItem>
+                    <SelectItem value="Multiple Paragraphs">Multiple Paragraphs</SelectItem>
                     <SelectItem value="Single Paragraph">Single Paragraph</SelectItem>
                     <SelectItem value="Bullet Points">Bullet Points</SelectItem>
                   </SelectGroup>
diff --git a/lightrag_webui/src/features/RetrievalTesting.tsx b/lightrag_webui/src/features/RetrievalTesting.tsx
index cbad51c9..e94b1085 100644
--- a/lightrag_webui/src/features/RetrievalTesting.tsx
+++ b/lightrag_webui/src/features/RetrievalTesting.tsx
@@ -37,11 +37,10 @@ export default function RetrievalTesting() {
         role: 'assistant'
       }
 
+      const prevMessages = [...messages]
+
       // Add messages to chatbox
-      setMessages((prev) => {
-        const newMessages = [...prev, userMessage, assistantMessage]
-        return newMessages
-      })
+      setMessages([...prevMessages, userMessage, assistantMessage])
 
       // Clear input and set loading
       setInputValue('')
@@ -65,7 +64,7 @@ export default function RetrievalTesting() {
       const queryParams = {
         ...state.querySettings,
         query: userMessage.content,
-        conversation_history: messages
+        conversation_history: prevMessages
       }
 
       try {
@@ -84,11 +83,7 @@ export default function RetrievalTesting() {
         setIsLoading(false)
         useSettingsStore
           .getState()
-          .setRetrievalHistory([
-            ...useSettingsStore.getState().retrievalHistory,
-            userMessage,
-            assistantMessage
-          ])
+          .setRetrievalHistory([...prevMessages, userMessage, assistantMessage])
       }
     },
     [inputValue, isLoading, messages, setMessages]