Add Gemini API-review workflow and script

Author: yongruilin

hack/gemini_api_review.py

@@ -2,7 +2,10 @@
 import os
 import json
 import requests
+<<<<<<< HEAD
 from google import genai
+=======
+>>>>>>> a53a08dd494 (Add Gemini API-review workflow and script)
 
 
 def gather_markdown_files(file_paths):
@@ -37,20 +40,30 @@ def post_comment_to_pr(repository, pr_number, comment_body, github_token):
 
 def main():
     # Define the markdown files to be gathered
+<<<<<<< HEAD
     # file_paths = ["README.md", "docs/api-guidelines.md", "docs/api-changes.md"]  
     file_paths = ["README.md"]
+=======
+    file_paths = ["README.md", "docs/api-guidelines.md", "docs/api-changes.md"]
+>>>>>>> a53a08dd494 (Add Gemini API-review workflow and script)
     docs = gather_markdown_files(file_paths)
 
     # Retrieve environment variables for PR data and Gemini API credentials
     pr_number = os.environ.get("PR_NUMBER")
     pr_title = os.environ.get("PR_TITLE")
     pr_body = os.environ.get("PR_BODY")
+<<<<<<< HEAD
     gemini_api_key = os.environ.get("GEMINI_API_KEY")  # Now used with google-genai
+=======
+    gemini_api_endpoint = os.environ.get("GEMINI_API_ENDPOINT")
+    gemini_api_key = os.environ.get("GEMINI_API_KEY")
+>>>>>>> a53a08dd494 (Add Gemini API-review workflow and script)
 
     # Retrieve GitHub info for posting a comment
     github_token = os.environ.get("GITHUB_TOKEN")
     repository = os.environ.get("GITHUB_REPOSITORY")
 
+<<<<<<< HEAD
     if not all([pr_number, pr_title, pr_body, gemini_api_key]):
         print("Error: One or more required environment variables (PR_NUMBER, PR_TITLE, PR_BODY, GEMINI_API_KEY) are missing.")
         return
@@ -71,12 +84,42 @@ def main():
             contents=contents
         )
         print(f"Gemini API response: {response.text}")
+=======
+    if not all([pr_number, pr_title, pr_body, gemini_api_endpoint, gemini_api_key]):
+        print("Error: One or more required environment variables (PR_NUMBER, PR_TITLE, PR_BODY, GEMINI_API_ENDPOINT, GEMINI_API_KEY) are missing.")
+        return
+
+    # Build JSON payload for Gemini API request
+    payload = {
+        "pr_number": pr_number,
+        "pr_title": pr_title,
+        "pr_body": pr_body,
+        "docs": docs
+    }
+
+    headers = {
+        "Authorization": f"Bearer {gemini_api_key}",
+        "Content-Type": "application/json"
+    }
+
+    try:
+        print("Sending payload to Gemini API...")
+        response = requests.post(gemini_api_endpoint, headers=headers, json=payload)
+        print(f"Response status code: {response.status_code}")
+        print(f"Response body: {response.text}")
+>>>>>>> a53a08dd494 (Add Gemini API-review workflow and script)
 
         # If GitHub credentials are provided, post a comment with the Gemini API response
         if github_token and repository:
             comment_body = (
+<<<<<<< HEAD
                 "Gemini API Review Result:\n\n"
                 f"{response.text}"
+=======
+                f"Gemini API Review Result:\n\n"
+                f"Response Status Code: {response.status_code}\n\n"
+                f"Response Body:\n{response.text}"
+>>>>>>> a53a08dd494 (Add Gemini API-review workflow and script)
             )
             post_comment_to_pr(repository, pr_number, comment_body, github_token)
         else: