From 0e32094870dac3817507eb2184c99aefe3ae6ebc Mon Sep 17 00:00:00 2001
From: Alex Nguyen <truong@clare.ai>
Date: Tue, 29 Oct 2024 15:30:42 +0700
Subject: [PATCH] Using model gemini-1.5-pro-002

---
 src/review_code_gemini.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/review_code_gemini.py b/src/review_code_gemini.py
index 67fab31..05e6ad3 100644
--- a/src/review_code_gemini.py
+++ b/src/review_code_gemini.py
@@ -120,7 +120,7 @@ def get_ai_response(prompt: str) -> List[Dict[str, str]]:
     try:
         response = gemini_client.generate_text(
             prompt=prompt,
-            model="models/code-bison-001",
+            model="gemini-1.5-pro-002",
             temperature=0.2,
             max_output_tokens=700,
         )