From 551019d87e02c51971eee80169aa7ddcf870c8eb Mon Sep 17 00:00:00 2001
From: Nicolas Bonamy <nicolas@Bonamy.fr>
Date: Mon, 13 May 2024 15:00:37 -0500
Subject: [PATCH] GPT-4o support

---
 README.md              | 1 +
 src/services/openai.ts | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 980993a..59a2ceb 100644
--- a/README.md
+++ b/README.md
@@ -73,6 +73,7 @@ To use Internet search you need a [Tavily API key](https://app.tavily.com/home).
 
 ## DONE
 
+- [x] GPT-4o support
 - [x] Different default engine/model for commands
 - [x] Text attachments (TXT, PDF, DOCX, PPTX, XLSX)
 - [x] MistralAI function calling
diff --git a/src/services/openai.ts b/src/services/openai.ts
index 1e6fcf1..fab147d 100644
--- a/src/services/openai.ts
+++ b/src/services/openai.ts
@@ -33,7 +33,7 @@ export default class extends LlmEngine {
   }
 
   getVisionModels(): string[] {
-    return ['gpt-4-turbo', 'gpt-4-vision', 'gpt-4-vision-preview', '*vision']
+    return [/*'*4o*', */ 'gpt-4-turbo', 'gpt-4-vision', '*vision*']
   }
 
   isVisionModel(model: string): boolean {