From 8175f199d2287ed8d488364b5da3a98934d9972b Mon Sep 17 00:00:00 2001
From: Rafi <rafiwx@gmail.com>
Date: Mon, 27 Mar 2023 22:17:19 +0800
Subject: [PATCH] Support GPT-4

---
 components/ModelParameters.vue | 10 +++++++---
 composables/states.js          |  2 +-
 pages/index.vue                |  5 +++--
 utils/enums.js                 | 30 ++++++++++++++++++++++--------
 utils/localStorage.js          | 17 +++++++++--------
 5 files changed, 42 insertions(+), 22 deletions(-)

diff --git a/components/ModelParameters.vue b/components/ModelParameters.vue
index 3c54f34..69c1e88 100644
--- a/components/ModelParameters.vue
+++ b/components/ModelParameters.vue
@@ -1,11 +1,15 @@
 <script setup>
+
 const dialog = ref(false)
 const currentModel = useCurrentModel()
 const availableModels = [
-  DEFAULT_MODEL.name
+    'gpt-3.5-turbo',
+    'gpt-4'
 ]
+const currentModelDefault = ref(MODELS[currentModel.value.name])
 
 watch(currentModel, (newVal, oldVal) => {
+  currentModelDefault.value = MODELS[newVal.name]
   saveCurrentModel(newVal)
 }, { deep: true })
 
@@ -83,7 +87,7 @@ watch(currentModel, (newVal, oldVal) => {
                   single-line
                   density="compact"
                   type="number"
-                  max="2048"
+                  :max="currentModelDefault.total_tokens"
                   step="1"
                   style="width: 100px"
                   class="flex-grow-0"
@@ -93,7 +97,7 @@ watch(currentModel, (newVal, oldVal) => {
           <v-col cols="12">
             <v-slider
                 v-model="currentModel.max_tokens"
-                :max="2048"
+                :max="currentModelDefault.total_tokens"
                 :step="1"
                 hide-details
             >
diff --git a/composables/states.js b/composables/states.js
index 9d38a7b..8f041f7 100644
--- a/composables/states.js
+++ b/composables/states.js
@@ -1,5 +1,5 @@
 
-export const useModels = () => useState('models', () => getStoredModels())
+// export const useModels = () => useState('models', () => getStoredModels())
 
 export const useCurrentModel = () => useState('currentModel', () => getCurrentModel())
 
diff --git a/pages/index.vue b/pages/index.vue
index 9217972..b472d2e 100644
--- a/pages/index.vue
+++ b/pages/index.vue
@@ -90,12 +90,13 @@ const fetchReply = async (message) => {
         throw err;
       },
       async onmessage(message) {
-        // console.log(message)
         const event = message.event
         const data = JSON.parse(message.data)
 
         if (event === 'error') {
-          throw new Error(data.error);
+          abortFetch()
+          showSnackbar(data.error)
+          return;
         }
 
         if (event === 'userMessageId') {
diff --git a/utils/enums.js b/utils/enums.js
index 29b4348..b7bf124 100644
--- a/utils/enums.js
+++ b/utils/enums.js
@@ -5,11 +5,25 @@ export const STORAGE_KEY = {
     OPENAI_API_KEY: 'openai_api_key',
 }
 
-export const DEFAULT_MODEL = {
-    name: 'gpt-3.5-turbo',
-    frequency_penalty: 0.0,
-    presence_penalty: 0.0,
-    max_tokens: 1000,
-    temperature: 0.7,
-    top_p: 1.0
-}
\ No newline at end of file
+export const MODELS = {
+    'gpt-3.5-turbo': {
+        name: 'gpt-3.5-turbo',
+        frequency_penalty: 0.0,
+        presence_penalty: 0.0,
+        total_tokens: 4096,
+        max_tokens: 1000,
+        temperature: 0.7,
+        top_p: 1.0
+    },
+    'gpt-4': {
+        name: 'gpt-4',
+        frequency_penalty: 0.0,
+        presence_penalty: 0.0,
+        total_tokens: 8192,
+        max_tokens: 2000,
+        temperature: 0.7,
+        top_p: 1.0
+    }
+}
+
+export const DEFAULT_MODEL_NAME = 'gpt-3.5-turbo'
\ No newline at end of file
diff --git a/utils/localStorage.js b/utils/localStorage.js
index 9d32e5d..5bf4123 100644
--- a/utils/localStorage.js
+++ b/utils/localStorage.js
@@ -1,3 +1,4 @@
+import {MODELS} from "~/utils/enums";
 
 const get = (key) => {
     let val = localStorage.getItem(key)
@@ -17,13 +18,13 @@ export const setModels = (val) => {
     models.value = val
 }
 
-export const getStoredModels = () => {
-    let models = get(STORAGE_KEY.MODELS)
-    if (!models) {
-        models = [DEFAULT_MODEL]
-    }
-    return models
-}
+// export const getStoredModels = () => {
+//     let models = get(STORAGE_KEY.MODELS)
+//     if (!models) {
+//         models = [DEFAULT_MODEL]
+//     }
+//     return models
+// }
 
 export const saveCurrentModel = (val) => {
     set(STORAGE_KEY.CURRENT_MODEL, val)
@@ -32,7 +33,7 @@ export const saveCurrentModel = (val) => {
 export const getCurrentModel = () => {
     let model = get(STORAGE_KEY.CURRENT_MODEL)
     if (!model) {
-        model = DEFAULT_MODEL
+        model = MODELS[DEFAULT_MODEL_NAME]
     }
     return model
 }