Merge pull request #227 from ZeekoZhu/patch-1

Add more gpt models
WongSaang · Jul 5, 2023 · 54330e0 · 54330e0
2 parents b626bde + 23f1cda
commit 54330e0
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 25 deletions.
diff --git a/components/ModelParameters.client.vue b/components/ModelParameters.client.vue
@@ -4,7 +4,9 @@ const dialog = ref(false)
 const currentModel = useCurrentModel()
 const availableModels = [
     'gpt-3.5-turbo',
-    'gpt-4'
+    'gpt-4',
+    'gpt-3.5-turbo-16k',
+    'gpt-4-32k'
 ]
 const currentModelDefault = ref(MODELS[currentModel.value.name])
 
@@ -198,4 +200,4 @@ onNuxtReady(() => {
 
 <style scoped>
 
-</style>
+</style>
diff --git a/utils/enums.js b/utils/enums.js
@@ -1,29 +1,46 @@
-
 export const STORAGE_KEY = {
-    MODELS: 'models',
-    CURRENT_MODEL: 'current_model',
-    OPENAI_API_KEY: 'openai_api_key',
+  MODELS: 'models',
+  CURRENT_MODEL: 'current_model',
+  OPENAI_API_KEY: 'openai_api_key',
 }
 
 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
-    }
+  '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-3.5-turbo-16k': {
+    name: 'gpt-3.5-16k',
+    frequency_penalty: 0.0,
+    presence_penalty: 0.0,
+    total_tokens: 16384,
+    max_tokens: 4000,
+    temperature: 0.7,
+    top_p: 1.0
+  },
+  'gpt-4-32k': {
+    name: 'gpt-4-32k',
+    frequency_penalty: 0.0,
+    presence_penalty: 0.0,
+    total_tokens: 32768,
+    max_tokens: 8000,
+    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'
+export const DEFAULT_MODEL_NAME = 'gpt-3.5-turbo'