From 9b99cad286d2e5fb4ec2d8b816c9e9eab165caee Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Thu, 11 Jul 2024 16:26:26 -0400
Subject: [PATCH 1/2] auto-upgrade usage of older models

---
 backend/routes/generate_code.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index 6426f20..19ef382 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -94,6 +94,19 @@ async def stream_code(websocket: WebSocket):
     except:
         await throw_error(f"Invalid model: {code_generation_model_str}")
         raise Exception(f"Invalid model: {code_generation_model_str}")
+
+    # Auto-upgrade usage of older models
+    if code_generation_model in {Llm.GPT_4_VISION, Llm.GPT_4_TURBO_2024_04_09}:
+        print(
+            f"Initial deprecated model: {code_generation_model}. Auto-updating code generation model to GPT-4O-2024-05-13"
+        )
+        code_generation_model = Llm.GPT_4O_2024_05_13
+    elif code_generation_model == Llm.CLAUDE_3_SONNET:
+        print(
+            f"Initial deprecated model: {code_generation_model}. Auto-updating code generation model to CLAUDE-3.5-SONNET-2024-06-20"
+        )
+        code_generation_model = Llm.CLAUDE_3_5_SONNET_2024_06_20
+
     exact_llm_version = None
 
     print(

From 329b8e8e77dd379cb295331cce5b7c130851a7ff Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Thu, 11 Jul 2024 16:34:48 -0400
Subject: [PATCH 2/2] update front-end to indicate status of deprecated models

---
 frontend/src/App.tsx       |  4 ++--
 frontend/src/lib/models.ts | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index 44669cf..e3b5b64 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -454,8 +454,8 @@ function App() {
           {showBetterModelMessage && (
             <div className="rounded-lg p-2 bg-fuchsia-200">
               <p className="text-gray-800 text-sm">
-                Now supporting GPT-4o and Claude Sonnet 3.5. Higher quality and
-                2x faster. Give it a try!
+                We no longer support this model. Instead, code generation will
+                use GPT-4o or Claude Sonnet 3.5, the 2 state-of-the-art models.
               </p>
             </div>
           )}
diff --git a/frontend/src/lib/models.ts b/frontend/src/lib/models.ts
index c9be84b..06ff708 100644
--- a/frontend/src/lib/models.ts
+++ b/frontend/src/lib/models.ts
@@ -12,9 +12,9 @@ export enum CodeGenerationModel {
 export const CODE_GENERATION_MODEL_DESCRIPTIONS: {
   [key in CodeGenerationModel]: { name: string; inBeta: boolean };
 } = {
-  "gpt-4o-2024-05-13": { name: "GPT-4o 🌟", inBeta: false },
-  "claude-3-5-sonnet-20240620": { name: "Claude 3.5 Sonnet 🌟", inBeta: false },
-  "gpt-4-turbo-2024-04-09": { name: "GPT-4 Turbo (Apr 2024)", inBeta: false },
-  gpt_4_vision: { name: "GPT-4 Vision (Nov 2023)", inBeta: false },
-  claude_3_sonnet: { name: "Claude 3 Sonnet", inBeta: false },
+  "gpt-4o-2024-05-13": { name: "GPT-4o", inBeta: false },
+  "claude-3-5-sonnet-20240620": { name: "Claude 3.5 Sonnet", inBeta: false },
+  "gpt-4-turbo-2024-04-09": { name: "GPT-4 Turbo (deprecated)", inBeta: false },
+  gpt_4_vision: { name: "GPT-4 Vision (deprecated)", inBeta: false },
+  claude_3_sonnet: { name: "Claude 3 (deprecated)", inBeta: false },
 };