feat: Add query invalidation for sidebar settings after saving changes

This commit is contained in:
n4ze3m 2025-01-25 21:05:58 +05:30
parent 8c0706aa7c
commit 94af788365

View File

@ -1,4 +1,4 @@
import { useMutation, useQuery } from "@tanstack/react-query"
import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query"
import React from "react"
import {
getOllamaURL,
@ -7,7 +7,6 @@ import {
setOllamaURL as saveOllamaURL,
setPromptForRag,
setSystemPromptForNonRag,
getAllModels,
defaultEmbeddingChunkOverlap,
defaultEmbeddingChunkSize,
defaultEmbeddingModelForRag,
@ -54,11 +53,12 @@ export const SettingsBody = () => {
const [hideCurrentChatModelSettings, setHideCurrentChatModelSettings] =
useStorage("hideCurrentChatModelSettings", false)
const [ speechToTextLanguage, setSpeechToTextLanguage ] = useStorage(
const [speechToTextLanguage, setSpeechToTextLanguage] = useStorage(
"speechToTextLanguage",
"en-US"
)
const { mode, toggleDarkMode } = useDarkMode()
const queryClient = useQueryClient()
const { changeLocale, locale, supportLanguage } = useI18n()
@ -84,7 +84,6 @@ export const SettingsBody = () => {
defaultEmbeddingModelForRag(),
getTotalFilePerKB()
])
return {
url: ollamaURL,
normalSystemPrompt: systemPrompt,
@ -106,6 +105,7 @@ export const SettingsBody = () => {
overlap: number
}) => {
await saveForRag(f.model, f.chunkSize, f.overlap, data.totalFilePerKB)
await queryClient.invalidateQueries({ queryKey: ["sidebarSettings"] })
}
})