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