ai review test

Author: siquanlv

.github/scripts/ai_code_review.py

@@ -2,21 +2,27 @@
 import requests
 import openai
 
-def load_diff(file_path='pr.diff'):
+
+def load_diff(file_path="pr.diff"):
     """Load the diff content from file."""
     if not os.path.exists(file_path):
         print("❌ Diff file not found.")
         return ""
-    with open(file_path, 'r', encoding='utf-8') as f:
+    with open(file_path, "r", encoding="utf-8") as f:
         return f.read()
 
+
 def call_gpt(diff_text):
     """Call AI to review the diff."""
-    api_key = os.getenv("OPENAI_API_KEY")
+    api_key = os.environ.get("OPENAI_API_KEY")
     if not api_key:
         raise ValueError("❌ OPENAI_API_KEY is not set.")
 
-    client = openai.OpenAI(api_key=api_key, base_url='https://llm-proxy.us-east-2.int.infra.intelligence.webex.com/azure/v1?api-version=2024-10-21',default_headers={"api-key": api_key})
+    client = openai.OpenAI(
+        api_key=api_key,
+        base_url="https://llm-proxy.us-east-2.int.infra.intelligence.webex.com/azure/v1?api-version=2024-10-21",
+        default_headers={"api-key": api_key},
+    )
 
     prompt = f"""
 You are a senior Java code reviewer.
@@ -52,12 +58,13 @@ def call_gpt(diff_text):
         model="gpt-4o",
         messages=[
             {"role": "system", "content": "You are a Java code review expert."},
-            {"role": "user", "content": prompt}
+            {"role": "user", "content": prompt},
         ],
-        temperature=0.2
+        temperature=0.2,
     )
     return response.choices[0].message.content
 
+
 def post_pr_comment(review_text):
     """Post review result as a comment on the PR."""
     repo = os.getenv("GITHUB_REPOSITORY")
@@ -67,7 +74,7 @@ def post_pr_comment(review_text):
     url = f"https://api.github.com/repos/{repo}/issues/{pr_number}/comments"
     headers = {
         "Authorization": f"token {token}",
-        "Accept": "application/vnd.github.v3+json"
+        "Accept": "application/vnd.github.v3+json",
     }
     payload = {"body": review_text}
 
@@ -79,6 +86,7 @@ def post_pr_comment(review_text):
     else:
         print("✅ Review comment posted successfully.")
 
+
 def main():
     diff = load_diff()
     if not diff.strip():
@@ -87,6 +95,6 @@ def main():
     review = call_gpt(diff)
     post_pr_comment(review)
 
+
 if __name__ == "__main__":
     main()
-