From 94af7883652baf1e218ad671705ac0d69461aae1 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 25 Jan 2025 21:05:58 +0530
Subject: [PATCH] feat: Add query invalidation for sidebar settings after
 saving changes

---
 src/components/Sidepanel/Settings/body.tsx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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"] })
     }
   })