Migrate gpt-4-turbo-preview to gpt-4-turbo

tristandostaler · Jun 3, 2024 · 167287f · 167287f
1 parent a1ef8ca
commit 167287f
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 1 deletion.
diff --git a/src/constants/stateVersion.ts b/src/constants/stateVersion.ts
@@ -1,3 +1,3 @@
-const stateVersion: number = 13;
+const stateVersion: number = 14;
 
 export default stateVersion;
diff --git a/src/store/migrate.ts b/src/store/migrate.ts
@@ -5,6 +5,7 @@ import {
   FolderCollection,
   LocalStorageInterfaceV0ToV1,
   LocalStorageInterfaceV12ToV13,
+  LocalStorageInterfaceV13ToV14,
   LocalStorageInterfaceV1ToV2,
   LocalStorageInterfaceV2ToV3,
   LocalStorageInterfaceV3ToV4,
@@ -134,4 +135,22 @@ export const migrateV13 = (persistedState: LocalStorageInterfaceV12ToV13) => {
       }
     })
   });
+};
+
+export const migrateV14 = (persistedState: LocalStorageInterfaceV13ToV14) => {
+  persistedState.chats.forEach((chat) => {
+    if(chat.config.model.toString() == "gpt-4-turbo-preview") {
+      chat.config.model = 'gpt-4-turbo';
+    }
+  });
+  if(persistedState.defaultChatConfig.model.toString() == "gpt-4-turbo-preview") {
+    persistedState.defaultChatConfig.model = 'gpt-4-turbo';
+  }
+  persistedState.prompts.forEach((prompt) => {
+    prompt.prompt = prompt.prompt.replaceAll("~ Model: gpt-4-turbo-preview ~", "~ Model: gpt-4-turbo ~")
+  });
+
+  persistedState.publicPrompts.forEach((prompt) => {
+    prompt.prompt = prompt.prompt.replaceAll("~ Model: gpt-4-turbo-preview ~", "~ Model: gpt-4-turbo ~")
+  });
 };
diff --git a/src/store/store.ts b/src/store/store.ts
@@ -18,6 +18,7 @@ import {
   LocalStorageInterfaceV8ToV9,
   LocalStorageInterfaceV8ToV12,
   LocalStorageInterfaceV12ToV13,
+  LocalStorageInterfaceV13ToV14,
 } from '@type/chat';
 import {
   migrateV0,
@@ -31,6 +32,7 @@ import {
   migrateV8,
   migrateV12,
   migrateV13,
+  migrateV14,
 } from './migrate';
 import stateVersion from '@constants/stateVersion';
 import { createPublicPromptSlice, PublicPromptSlice } from './public-prompt-sync-slice';
@@ -120,6 +122,9 @@ const useStore = create<StoreState>()(
           case 12:
             migrateV13(persistedState as LocalStorageInterfaceV12ToV13);
             break;
+          case 13:
+            migrateV14(persistedState as LocalStorageInterfaceV13ToV14);
+            break;
         }
         return persistedState as StoreState;
       },

diff --git a/src/types/chat.ts b/src/types/chat.ts
@@ -162,3 +162,6 @@ export interface LocalStorageInterfaceV12ToV13
   extends LocalStorageInterfaceV8ToV12 {
 }
 
+export interface LocalStorageInterfaceV13ToV14
+  extends LocalStorageInterfaceV12ToV13 {
+}