diff --git a/src/components/Sidepanel/Settings/body.tsx b/src/components/Sidepanel/Settings/body.tsx index dc36baa..710f47a 100644 --- a/src/components/Sidepanel/Settings/body.tsx +++ b/src/components/Sidepanel/Settings/body.tsx @@ -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"] }) } })