Add GPT-4o to the list of supported models

davidmyersdev · May 13, 2024 · 156c78d · 156c78d
1 parent cacb9ae
commit 156c78d
Showing 1 changed file with 42 additions and 35 deletions.
diff --git a/composables/useAssistant.ts b/composables/useAssistant.ts
@@ -8,6 +8,46 @@ import {
   useStorage,
 } from 'vellma/peripherals'
 
+const defaultLanguageModels = [
+  {
+    id: 'gpt-4o',
+    label: 'GPT-4o',
+    contextWindow: '128k',
+  },
+  {
+    id: 'gpt-4-turbo',
+    label: 'GPT-4 Turbo',
+    contextWindow: '128k',
+  },
+  {
+    id: 'gpt-4-turbo-preview',
+    label: 'GPT-4 Turbo (Preview)',
+    contextWindow: '128k',
+  },
+  {
+    id: 'gpt-4-vision-preview',
+    label: 'GPT-4 Turbo Vision',
+    contextWindow: '128k',
+    comingSoon: true,
+    disabled: true,
+  },
+  {
+    id: 'gpt-4',
+    label: 'GPT-4',
+    contextWindow: '8k',
+  },
+  {
+    id: 'gpt-4-32k',
+    label: 'GPT-4 (32k)',
+    contextWindow: '32k',
+  },
+  {
+    id: 'gpt-3.5-turbo-0125',
+    label: 'GPT-3.5 Turbo',
+    contextWindow: '16k',
+  },
+] as const
+
 export const dexieStorage = (): StorageAdapter => {
   const { db } = useDatabase()
 
@@ -50,7 +90,7 @@ export const dexieStorage = (): StorageAdapter => {
   }
 }
 
-export const useAssistant = ({ chatId, languageModel = ref({ id: 'gpt-4-1106-preview', label: 'GPT-4 Turbo' }) }: { chatId: Ref<string>, languageModel?: Ref<{ id: string, label: string }> }) => {
+export const useAssistant = ({ chatId, languageModel = ref(defaultLanguageModels[0]) }: { chatId: Ref<string>, languageModel?: Ref<{ id: string, label: string }> }) => {
   const storageAdapter = dexieStorage()
   const apiKey = useLocalStorage<string>('openAiApiKey', '', {
     initOnMounted: true,
@@ -63,40 +103,7 @@ export const useAssistant = ({ chatId, languageModel = ref({ id: 'gpt-4-1106-pre
   const chatModel = computed(() => chatInterface.value.model)
   const chatFactory = computed(() => chatInterface.value.factory)
 
-  const languageModels = ref([
-    {
-      id: 'gpt-4-turbo',
-      label: 'GPT-4 Turbo',
-      contextWindow: '128k',
-    },
-    {
-      id: 'gpt-4-turbo-preview',
-      label: 'GPT-4 Turbo (Preview)',
-      contextWindow: '128k',
-    },
-    {
-      id: 'gpt-4-vision-preview',
-      label: 'GPT-4 Turbo Vision',
-      contextWindow: '128k',
-      comingSoon: true,
-      disabled: true,
-    },
-    {
-      id: 'gpt-4',
-      label: 'GPT-4',
-      contextWindow: '8k',
-    },
-    {
-      id: 'gpt-4-32k',
-      label: 'GPT-4 (32k)',
-      contextWindow: '32k',
-    },
-    {
-      id: 'gpt-3.5-turbo-0125',
-      label: 'GPT-3.5 Turbo',
-      contextWindow: '16k',
-    },
-  ])
+  const languageModels = ref(defaultLanguageModels)
 
   return {
     apiKey,