Using model gemini-1.5-pro-002

truongnh1992 · Oct 29, 2024 · 0e32094 · 0e32094
1 parent 78138b2
commit 0e32094
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         )