feat: support new OpenAI model GPT-4o

normal-computing · May 13, 2024 · 1c8c4aa · 1c8c4aa
1 parent 60e11f5
commit 1c8c4aa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/helpers/aiSdkUtils.ts b/src/helpers/aiSdkUtils.ts
@@ -9,6 +9,7 @@ export enum SupportedModels {
   Gpt4TurboPreview = "gpt-4-turbo-preview",
   Gpt4VisionPreview = "gpt-4-vision-preview",
   Gpt4Turbo = "gpt-4-turbo",
+  Gpt4O = "gpt-4o",
   Claude3Sonnet = "claude-3-sonnet-20240229",
 }
 
@@ -24,13 +25,15 @@ export const DisplayName = {
   [SupportedModels.Gpt4TurboPreview]: "GPT-4 Turbo (Preview)",
   [SupportedModels.Gpt4VisionPreview]: "GPT-4 Vision (Preview)",
   [SupportedModels.Gpt4Turbo]: "GPT-4 Turbo",
+  [SupportedModels.Gpt4O]: "GPT-4o",
   [SupportedModels.Claude3Sonnet]: "Claude 3 Sonnet",
 };
 
 export function hasVisionSupport(model: SupportedModels) {
   return (
     model === SupportedModels.Gpt4VisionPreview ||
     model === SupportedModels.Gpt4Turbo ||
+    model === SupportedModels.Gpt4O ||
     model === SupportedModels.Claude3Sonnet
   );
 }