Initial Python Code

Author: banstack

.github/workflows/code-review.yml

@@ -0,0 +1,25 @@
+name: LLM Code Review
+
+on:
+  pull_request:
+    types: [opened, synchronize]
+
+jobs:
+  review:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+
+      - name: Install dependencies
+        run: |
+          pip install -r requirements.txt
+
+      - name: Run code review
+        run: |
+          python reviewbot/main.py

README.md

@@ -0,0 +1 @@
+# LLM Code Review Bot

reviewbot/main.py

@@ -0,0 +1,29 @@
+import os
+import requests
+from reviewbot.utils import get_diff
+
+def review_with_groq(diff):
+    api_key = os.environ.get("GROQ_API_KEY")
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json",
+    }
+    payload = {
+        "model": "mixtral-8x7b-32768",
+        "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}"}
+        ],
+        "temperature": 0.3,
+    }
+
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
+    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", " "))
+
+if __name__ == "__main__":
+    main()

reviewbot/utils.py

@@ -0,0 +1,10 @@
+# reviewbot/utils.py
+
+def get_diff():
+    # In real setup, parse `git diff` or PR files
+    return "def hello(name):\n    print('Hello ' + name)"
+
+def fake_llm_review(diff):
+    if "print" in diff:
+        return "Consider using logging instead of print for better production code quality."
+    return "No issues found."

test_module/hello.py

@@ -0,0 +1,6 @@
+# test_module/hello.py
+
+def hello(name):
+    print("Hello " + name)
+
+hello("World")