From 1642180039e20e0aff08cbb4a64e2bf707c4f9f2 Mon Sep 17 00:00:00 2001
From: yannikkellerde <yannik@kelnet.de>
Date: Sat, 30 Nov 2024 18:22:23 +0100
Subject: [PATCH] Add GPT-4o

---
 developergpt/config.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/developergpt/config.py b/developergpt/config.py
index ace8aec..2d88ad5 100644
--- a/developergpt/config.py
+++ b/developergpt/config.py
@@ -25,6 +25,7 @@
 ### Supported LLMs and Configuration ###
 GPT35 = "gpt35"
 GPT4 = "gpt4"
+GPT4o = "gpt4o"
 ZEPHYR = "zephyr"
 GEMMA = "gemma"
 GEMMA_BASE = "gemma-base"
@@ -40,6 +41,7 @@
     [
         GPT35,
         GPT4,
+        GPT4o,
         GEMINI,
         FLASH,
         ZEPHYR,
@@ -69,10 +71,7 @@
     ),
 }
 
-OPENAI_MODEL_MAP = {
-    GPT35: "gpt-3.5-turbo",
-    GPT4: "gpt-4-turbo",
-}
+OPENAI_MODEL_MAP = {GPT35: "gpt-3.5-turbo", GPT4: "gpt-4-turbo", GPT4o: "gpt-4o"}
 
 ANTHROPIC_MODEL_MAP = {
     SONNET: "claude-3-sonnet-20240229",