Add Gemini API-review workflow and script

Author: yongruilin

.github/workflows/gemini-api-review.yml

@@ -0,0 +1,35 @@
+name: Gemini API PR Review
+
+on:
+  pull_request:
+    types: [opened, synchronize, reopened, labeled]
+    branches:
+      - "gke-ai-hackathon-api-review-bot"
+
+jobs:
+  pr-review:
+    if: contains(github.event.pull_request.labels.*.name, 'api-review')
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout PR code
+        uses: actions/checkout@v3
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.x'
+
+      - name: Install dependencies
+        run: pip install requests
+
+      - name: Run Gemini API review script
+        env:
+          PR_NUMBER: ${{ github.event.pull_request.number }}
+          PR_TITLE: ${{ github.event.pull_request.title }}
+          PR_BODY: ${{ github.event.pull_request.body }}
+          GEMINI_API_ENDPOINT: ${{ secrets.GEMINI_API_ENDPOINT }}
+          GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          GITHUB_REPOSITORY: ${{ github.repository }}
+        run: python3 hack/gemini_api_review.py 

hack/gemini_api_review.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+import os
+import json
+import requests
+
+
+def gather_markdown_files(file_paths):
+    combined = ""
+    for file in file_paths:
+        if os.path.exists(file):
+            try:
+                with open(file, 'r') as f:
+                    content = f.read()
+                combined += f"\n\n---\nContent of {file}:\n{content}\n"
+            except Exception as e:
+                print(f"Error reading {file}: {e}")
+        else:
+            print(f"File {file} not found, skipping.")
+    return combined
+
+
+def post_comment_to_pr(repository, pr_number, comment_body, github_token):
+    url = f"https://api.github.com/repos/{repository}/issues/{pr_number}/comments"
+    headers = {
+        "Authorization": f"token {github_token}",
+        "Content-Type": "application/json",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    payload = {"body": comment_body}
+    response = requests.post(url, headers=headers, json=payload)
+    if response.status_code in (200, 201):
+        print("Successfully posted comment to PR.")
+    else:
+        print(f"Failed to post comment to PR (status code {response.status_code}): {response.text}")
+
+
+def main():
+    # Define the markdown files to be gathered
+    file_paths = ["README.md", "docs/api-guidelines.md", "docs/api-changes.md"]
+    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")
+    gemini_api_endpoint = os.environ.get("GEMINI_API_ENDPOINT")
+    gemini_api_key = os.environ.get("GEMINI_API_KEY")
+
+    # Retrieve GitHub info for posting a comment
+    github_token = os.environ.get("GITHUB_TOKEN")
+    repository = os.environ.get("GITHUB_REPOSITORY")
+
+    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}")
+
+        # If GitHub credentials are provided, post a comment with the Gemini API response
+        if github_token and repository:
+            comment_body = (
+                f"Gemini API Review Result:\n\n"
+                f"Response Status Code: {response.status_code}\n\n"
+                f"Response Body:\n{response.text}"
+            )
+            post_comment_to_pr(repository, pr_number, comment_body, github_token)
+        else:
+            print("GitHub token or repository environment variable is missing; skipping posting comment on PR.")
+    except Exception as e:
+        print(f"An error occurred while calling Gemini API: {e}")
+
+
+if __name__ == "__main__":
+    main()