gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix retrieval testing and query settings

Browse Source
This commit is contained in:
ArnoChen
2025-02-16 21:58:30 +08:00
parent fc3ca2777f
commit e5b89c5400
2 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag_webui/src/components/retrieval/QuerySettings.tsx
View File

@@ -75,7 +75,7 @@ export default function QuerySettings() {
</SelectTrigger> </SelectTrigger>
<SelectContent> <SelectContent>
<SelectGroup> <SelectGroup>
<SelectItem value="Multiple Paragraphs">Single Paragraph</SelectItem> <SelectItem value="Multiple Paragraphs">Multiple Paragraphs</SelectItem>
<SelectItem value="Single Paragraph">Single Paragraph</SelectItem> <SelectItem value="Single Paragraph">Single Paragraph</SelectItem>
<SelectItem value="Bullet Points">Bullet Points</SelectItem> <SelectItem value="Bullet Points">Bullet Points</SelectItem>
</SelectGroup> </SelectGroup>

15
lightrag_webui/src/features/RetrievalTesting.tsx
View File

@@ -37,11 +37,10 @@ export default function RetrievalTesting() {
role: 'assistant' role: 'assistant'
} }
const prevMessages = [...messages]
// Add messages to chatbox // Add messages to chatbox
setMessages((prev) => { setMessages([...prevMessages, userMessage, assistantMessage])
const newMessages = [...prev, userMessage, assistantMessage]
return newMessages
})
// Clear input and set loading // Clear input and set loading
setInputValue('') setInputValue('')
@@ -65,7 +64,7 @@ export default function RetrievalTesting() {
const queryParams = { const queryParams = {
...state.querySettings, ...state.querySettings,
query: userMessage.content, query: userMessage.content,
conversation_history: messages conversation_history: prevMessages
} }
try { try {
@@ -84,11 +83,7 @@ export default function RetrievalTesting() {
setIsLoading(false) setIsLoading(false)
useSettingsStore useSettingsStore
.getState() .getState()
.setRetrievalHistory([ .setRetrievalHistory([...prevMessages, userMessage, assistantMessage])
...useSettingsStore.getState().retrievalHistory,
userMessage,
assistantMessage
])
} }
}, },
[inputValue, isLoading, messages, setMessages] [inputValue, isLoading, messages, setMessages]