Skip to content

Commit 0c0a937

Browse files
committed
Update Gemini review script to use latest flash model
1 parent f251633 commit 0c0a937

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hack/gemini_review.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def download_and_combine_guidelines(bucket_name, prefix):
4646
def generate_gemini_review_with_annotations(diff_file, api_key, guidelines):
4747
"""Generates a code review with annotations, incorporating guidelines."""
4848
genai.configure(api_key=api_key)
49-
model = genai.GenerativeModel('gemini-pro')
49+
model = genai.GenerativeModel('gemini-2.0-flash')
5050

5151
diff = diff_file.patch
5252
max_diff_length = 20000 # Adjust based on token count

0 commit comments

Comments
 (0)