clear only model history

bit-gpt · Nov 10, 2023 · 46da8fc · 46da8fc
1 parent bc315f9
commit 46da8fc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/src/modules/prem-chat/components/PremChatSidebar.tsx b/src/modules/prem-chat/components/PremChatSidebar.tsx
@@ -14,7 +14,6 @@ import NoPrompts from "shared/components/NoPrompts";
 import WarningIcon from "shared/components/WarningIcon";
 import usePremChatStore from "shared/store/prem-chat";
 import type { HamburgerMenuProps } from "shared/types";
-import { shallow } from "zustand/shallow";
 
 import LeftSidebar from "../../../shared/components/LeftSidebar";
 
@@ -25,15 +24,9 @@ const PremChatSidebar = ({
   serviceType,
   historyId,
 }: HamburgerMenuProps) => {
-  // TODO: shallow will only check for reference equality, not deep equality
-  const { history, deleteHistory, clearHistory } = usePremChatStore(
-    (state) => ({
-      history: state.history,
-      deleteHistory: state.deleteHistory,
-      clearHistory: state.clearHistory,
-    }),
-    shallow,
-  );
+  const history = usePremChatStore((state) => state.history);
+  const deleteHistory = usePremChatStore((state) => state.deleteHistory);
+  const clearHistory = usePremChatStore((state) => state.clearHistory);
 
   const [search, setSearch] = useState("");
   const navigate = useNavigate();
@@ -47,7 +40,7 @@ const PremChatSidebar = ({
   };
 
   const onClearClick = () => {
-    clearHistory();
+    clearHistory(serviceId);
   };
 
   const filteredHistory = history.filter((item) => {

diff --git a/src/shared/store/prem-chat.ts b/src/shared/store/prem-chat.ts
@@ -9,7 +9,10 @@ const usePremChatStore = create<PremChatStore>()(
     persist(
       (set) => ({
         history: [],
-        clearHistory: () => set(() => ({ history: [] })),
+        clearHistory: (serviceId: string) =>
+          set((state) => ({
+            history: state.history.filter((_history) => _history.model !== serviceId),
+          })),
         addHistory: (newHistory: PremChatHistory) =>
           set((state) => ({ history: [...state.history, newHistory] })),
         updateHistoryMessages: (id, messages) =>

diff --git a/src/shared/types/index.ts b/src/shared/types/index.ts
@@ -78,7 +78,7 @@ export type ModalProps = {
 
 export type PremChatStore = {
   history: PremChatHistory[];
-  clearHistory: () => void;
+  clearHistory: (serviceId: string) => void;
   addHistory: (newHistory: PremChatHistory) => void;
   updateHistoryMessages: (id: string, messages: Message[]) => void;
   deleteHistory: (id: string) => void;