Allow GH action to post review as comment

Author: banstack

.github/workflows/code-review.yml

@@ -10,7 +10,7 @@ jobs:
 
     permissions:
       contents: read
-      pull-requests: read
+      pull-requests: write
 
     steps:
       - name: Checkout code

reviewbot/main.py

@@ -1,6 +1,6 @@
 import os
 import requests
-from reviewbot.utils import get_diff
+from reviewbot.utils import get_diff, post_pr_comment, get_pr_number
 
 def review_with_groq(diff):
     api_key = os.environ.get("GROQ_API_KEY")
@@ -12,23 +12,21 @@ def review_with_groq(diff):
         "model": "meta-llama/llama-4-scout-17b-16e-instruct",
         "messages": [
             {"role": "system", "content": "You are a senior software engineer doing code reviews."},
-            {"role": "user", "content": f"Please review this code diff:\n\n{diff}"}
+            {"role": "user", "content": f"Please review this code diff, list improvements in bullet points:\n\n{diff}"}
         ],
         "temperature": 0.3,
     }
 
     response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
 
-    # 👇 Print the response for debugging
     print("Raw Groq response:", response.status_code, response.text)
-
     response.raise_for_status()
     return response.json()["choices"][0]["message"]["content"]
 
 def main():
     diff = get_diff()
     review = review_with_groq(diff)
-    print("::notice file=test_module/hello.py,line=1::" + review.replace("\n", " "))
+    post_pr_comment(review)
 
 if __name__ == "__main__":
     main()

reviewbot/utils.py

@@ -1,19 +1,20 @@
 import os
 import requests
+import json
 
-def get_diff():
-    repo = os.environ.get("GITHUB_REPOSITORY")  # e.g. "owner/repo"
-    ref = os.environ.get("GITHUB_REF")          # e.g. "refs/pull/123/merge"
+def get_pr_number():
+    event_path = os.getenv("GITHUB_EVENT_PATH")
+    if not event_path:
+        raise Exception("GITHUB_EVENT_PATH not set")
 
-    if not repo or not ref:
-        raise Exception("Missing GitHub repository or ref environment variables.")
+    with open(event_path, 'r') as f:
+        event = json.load(f)
+    
+    return event["pull_request"]["number"]
 
-    # Extract PR number safely:
-    parts = ref.split("/")
-    if len(parts) >= 3 and parts[-3] == "pull":
-        pr_number = parts[-2]
-    else:
-        raise Exception(f"Unexpected GITHUB_REF format: {ref}")
+def get_diff():
+    repo = os.environ.get("GITHUB_REPOSITORY")  # e.g. "owner/repo"
+    pr_number = get_pr_number()
 
     token = os.environ.get("GITHUB_TOKEN")
     if not token:
@@ -38,8 +39,26 @@ def get_diff():
         for f in files:
             patch = f.get("patch")
             if patch:
-                diffs.append(patch)
+                diffs.append(f"File: {f['filename']}\n{patch}")
 
         page += 1
 
-    return "\n".join(diffs)
+    return "\n\n".join(diffs)
+
+def post_pr_comment(body: str):
+    repo = os.getenv("GITHUB_REPOSITORY")
+    token = os.getenv("GITHUB_TOKEN")
+    pr_number = get_pr_number()
+
+    url = f"https://api.github.com/repos/{repo}/issues/{pr_number}/comments"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Accept": "application/vnd.github+json"
+    }
+    payload = {
+        "body": body
+    }
+
+    response = requests.post(url, headers=headers, json=payload)
+    print("GitHub API response:", response.status_code, response.text)
+    response.raise_for_status()

test_module/hello.py

@@ -1,10 +1,6 @@
 # Hello.py: A sample python script used for testing reviews
 
-def twoSum(array, targetSum):
-    for i in range(0, len(array)):
-        for j in range(i+1, len(array)):
-            if array[i] + array[j] == targetSum:
-                return ([array[i], array[j]])
-        return []
+def helloWorld():
+    print("Hello World")
 
-twoSum([2,7,11,15], 9)
+helloWorld()