Skip to content

Commit f251633

Browse files
committed
Add token and prompt debugging print statements to Gemini review script
1 parent 2232cdf commit f251633

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

hack/gemini_review.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import google.generativeai as genai
22
import os
33
from github import Github
4-
import re
54
from google.cloud import storage
65

76
def get_pr_latest_commit_diff_files(repo_name, pr_number, github_token):
@@ -69,6 +68,8 @@ def generate_gemini_review_with_annotations(diff_file, api_key, guidelines):
6968
{diff}
7069
```
7170
"""
71+
print("prompt: ", prompt)
72+
print("total_tokens: ", model.count_tokens(prompt))
7273
response = model.generate_content(prompt)
7374
return response.text if response.text else None
7475

0 commit comments

Comments
 (0)