From 32ce39437fbde84404a9d87b892ee2bc7917e320 Mon Sep 17 00:00:00 2001
From: litongjava <litongjava@qq.com>
Date: Tue, 4 Feb 2025 22:58:28 -1000
Subject: [PATCH] feat: enable quality

---
 ui/components/ChatWindow.tsx                       | 14 +++++++++++++-
 ui/components/EmptyChatMessageInput.tsx            |  7 ++-----
 ui/components/MessageInputActions/Optimization.tsx | 12 +++++-------
 3 files changed, 20 insertions(+), 13 deletions(-)

diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index 54c4b49..27e23b6 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -492,6 +492,18 @@ const ChatWindow = ({id}: { id?: string }) => {
     setFocusMode(mode);
   };
 
+  useEffect(() => {
+    const mode = localStorage.getItem('optimizationMode');
+    if (mode) {
+      setOptimizationMode(mode);
+    }
+  }, [setOptimizationMode]);
+
+  const handleOptimizationModeChange = (mode: string) => {
+    localStorage.setItem('optimizationMode', mode);
+    setOptimizationMode(mode);
+  };
+
   const messagesRef = useRef<Message[]>([]);
 
   useEffect(() => {
@@ -716,7 +728,7 @@ const ChatWindow = ({id}: { id?: string }) => {
             setCopilotEnabled={setCopilotEnabled}
             setFocusMode={handleFocusModeChange}
             optimizationMode={optimizationMode}
-            setOptimizationMode={setOptimizationMode}
+            setOptimizationMode={handleOptimizationModeChange}
             fileIds={fileIds}
             setFileIds={setFileIds}
             files={files}
diff --git a/ui/components/EmptyChatMessageInput.tsx b/ui/components/EmptyChatMessageInput.tsx
index 36be1c3..96be337 100644
--- a/ui/components/EmptyChatMessageInput.tsx
+++ b/ui/components/EmptyChatMessageInput.tsx
@@ -98,11 +98,8 @@ const EmptyChatMessageInput = ({
               showText
             />
 
-            <CopilotToggle setCopilotEnabled={setCopilotEnabled} copilotEnabled={copilotEnabled}/>
-            {/*<Optimization*/}
-            {/*  optimizationMode={optimizationMode}*/}
-            {/*  setOptimizationMode={setOptimizationMode}*/}
-            {/*/>*/}
+            {/*<CopilotToggle setCopilotEnabled={setCopilotEnabled} copilotEnabled={copilotEnabled}/>*/}
+            <Optimization optimizationMode={optimizationMode} setOptimizationMode={setOptimizationMode}/>
             <button
               disabled={message.trim().length === 0}
               className="bg-[#24A0ED] text-white disabled:text-black/50 dark:disabled:text-white/50 disabled:bg-[#e0e0dc] dark:disabled:bg-[#ececec21] hover:bg-opacity-85 transition duration-100 rounded-full p-2"
diff --git a/ui/components/MessageInputActions/Optimization.tsx b/ui/components/MessageInputActions/Optimization.tsx
index ac8a7b0..465129f 100644
--- a/ui/components/MessageInputActions/Optimization.tsx
+++ b/ui/components/MessageInputActions/Optimization.tsx
@@ -23,7 +23,7 @@ const OptimizationModes = [
   },
   {
     key: 'quality',
-    title: 'Quality (Soon)',
+    title: 'Quality',
     description: 'Get the most thorough and accurate answer',
     icon: (
       <Star
@@ -49,13 +49,11 @@ const Optimization = ({
       >
         <div className="flex flex-row items-center space-x-1">
           {
-            OptimizationModes.find((mode) => mode.key === optimizationMode)
-              ?.icon
+            OptimizationModes.find((mode) => mode.key === optimizationMode)?.icon
           }
           <p className="text-xs font-medium">
             {
-              OptimizationModes.find((mode) => mode.key === optimizationMode)
-                ?.title
+              OptimizationModes.find((mode) => mode.key === optimizationMode)?.title
             }
           </p>
           <ChevronDown size={20} />
@@ -76,13 +74,13 @@ const Optimization = ({
               <PopoverButton
                 onClick={() => setOptimizationMode(mode.key)}
                 key={i}
-                disabled={mode.key === 'quality'}
+                disabled={mode.key === 'quality1'}
                 className={cn(
                   'p-2 rounded-lg flex flex-col items-start justify-start text-start space-y-1 duration-200 cursor-pointer transition',
                   optimizationMode === mode.key
                     ? 'bg-light-secondary dark:bg-dark-secondary'
                     : 'hover:bg-light-secondary dark:hover:bg-dark-secondary',
-                  mode.key === 'quality' && 'opacity-50 cursor-not-allowed',
+                  mode.key === 'quality1' && 'opacity-50 cursor-not-allowed',
                 )}
               >
                 <div className="flex flex-row items-center space-x-1 text-black dark:text-white">