From 08fb31862153a6343c8937011abd525d975a3b1c Mon Sep 17 00:00:00 2001
From: Naman <1608naman@gmail.com>
Date: Mon, 20 May 2024 09:05:03 +0530
Subject: [PATCH] add gemini 1.5 pro to lib models

---
 frontend/src/lib/models.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/frontend/src/lib/models.ts b/frontend/src/lib/models.ts
index d6fd09d..9e1f7f6 100644
--- a/frontend/src/lib/models.ts
+++ b/frontend/src/lib/models.ts
@@ -5,6 +5,7 @@ export enum CodeGenerationModel {
   GPT_4_TURBO_2024_04_09 = "gpt-4-turbo-2024-04-09",
   GPT_4_VISION = "gpt_4_vision",
   CLAUDE_3_SONNET = "claude_3_sonnet",
+  GEMINI_1_5_PRO = "gemini-1.5-pro",
 }
 
 // Will generate a static error if a model in the enum above is not in the descriptions
@@ -15,4 +16,5 @@ export const CODE_GENERATION_MODEL_DESCRIPTIONS: {
   "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 },
+  "gemini-1.5-pro": { name: "Gemini 1.5 Pro", inBeta: true },
 };