total cost of deleted state but no logic

jackschedel · Feb 9, 2024 · 8c1edd4 · 8c1edd4
1 parent 0bc2178
commit 8c1edd4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/components/SettingsMenu/TotalTokenCost.tsx b/src/components/SettingsMenu/TotalTokenCost.tsx
@@ -121,14 +121,16 @@ export const TotalTokenCostDisplay = () => {
   const totalTokenUsed = useStore((state) => state.totalTokenUsed);
 
   const [totalCost, setTotalCost] = useState<number>(0);
+  const modelDefs = useStore((state) => state.modelDefs);
+  const costOfDeleted = useStore((state) => state.costOfDeleted);
 
   useEffect(() => {
-    let updatedTotalCost = 0;
+    let updatedTotalCost = costOfDeleted;
 
     Object.entries(totalTokenUsed).forEach(([key, tokenCost]) => {
       const model = parseInt(key, 10);
       if (!isNaN(model)) {
-        updatedTotalCost += tokenCostToCost(tokenCost, model);
+        updatedTotalCost += tokenCostToCost(tokenCost, model, modelDefs);
       }
     });
 

diff --git a/src/store/config-slice.ts b/src/store/config-slice.ts
@@ -19,6 +19,7 @@ export interface ConfigSlice {
   markdownMode: boolean;
   countTotalTokens: boolean;
   totalTokenUsed: TotalTokenUsed;
+  costOfDeleted: number;
   setOpenConfig: (openConfig: boolean) => void;
   setTheme: (theme: Theme) => void;
   setAutoTitle: (autoTitle: boolean) => void;
@@ -34,6 +35,7 @@ export interface ConfigSlice {
   setMarkdownMode: (markdownMode: boolean) => void;
   setCountTotalTokens: (countTotalTokens: boolean) => void;
   setTotalTokenUsed: (totalTokenUsed: TotalTokenUsed) => void;
+  setCostOfDeleted: (costOfDeleted: number) => void;
 }
 
 export const createConfigSlice: StoreSlice<ConfigSlice> = (set) => ({
@@ -52,6 +54,7 @@ export const createConfigSlice: StoreSlice<ConfigSlice> = (set) => ({
   markdownMode: true,
   countTotalTokens: false,
   totalTokenUsed: {},
+  costOfDeleted: 0,
   setOpenConfig: (openConfig: boolean) => {
     set((prev: ConfigSlice) => ({
       ...prev,
@@ -142,4 +145,10 @@ export const createConfigSlice: StoreSlice<ConfigSlice> = (set) => ({
       totalTokenUsed: totalTokenUsed,
     }));
   },
+  setCostOfDeleted: (costOfDeleted: number) => {
+    set((prev: ConfigSlice) => ({
+      ...prev,
+      costOfDeleted: costOfDeleted,
+    }));
+  },
 });