EX-319: fixed test/created logic for github

Author: MUlanovych

.codereview.yml

@@ -1,18 +1,18 @@
-llm:
-  provider: 'openai'
-  models:
-    context_builder:
-      name: 'gpt-4o'
-    reviewer:
-      name: 'gpt-4o'
-
 # llm:
-#   provider: 'openrouter'
-#   base_url: 'https://openrouter.ai/api/v1'
-
+#   provider: 'openai'
 #   models:
-#     context_builder: 'google/gemini-flash-1.5'
-#     reviewer: 'google/gemini-1.5-pro'
+#     context_builder:
+#       name: 'gpt-4o'
+#     reviewer:
+#       name: 'gpt-4o'
+
+llm:
+  provider: 'openrouter'
+  base_url: 'https://openrouter.ai/api/v1'
+
+  models:
+    context_builder: 'google/gemini-flash-1.5'
+    reviewer: 'google/gemini-1.5-pro'
 
 max_context_iterations: 3
 max_context_files: 25

.github/workflows/code_review.yml

@@ -0,0 +1,34 @@
+# .github/workflows/code_review.yml
+
+name: AI Code Review
+
+on:
+  pull_request:
+    types: [opened, synchronize]
+
+permissions:
+  contents: read
+  pull-requests: write
+
+jobs:
+  review:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Pull Request Code
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - name: Run AI Code Review Agent
+        run: |
+          IMAGE_NAME="umykhailo/codereviewagent:latest"
+          docker run \
+            --volume ${{ github.workspace }}:/repo \
+            --workdir /repo \
+            --env LLM_API_KEY=${{ secrets.LLM_API_KEY }} \
+            --env GITHUB_TOKEN=${{ secrets.GITHUB_TOKEN }} \
+            --env GITHUB_PR_NUMBER=${{ github.event.pull_request.number }} \
+            --env GITHUB_REPOSITORY=${{ github.repository }} \
+            --env GITHUB_BASE_REF=${{ github.base_ref }} \
+            --env GITHUB_HEAD_REF=${{ github.head_ref }} \
+            $IMAGE_NAME \
+            review --repo-path "." --base-ref "origin/${{ github.base_ref }}" --head-ref "${{ github.head_ref }}"

poetry.lock

@@ -21,6 +21,7 @@ pydantic = "^2.7.1"
 python-dotenv = "^1.0.1"
 tiktoken = "^0.9.0"
 atlassian-python-api = "^4.0.4"
+pygithub = "^2.6.1"
 
 
 [tool.poetry.group.dev.dependencies]

pyproject.toml

@@ -4,41 +4,52 @@
 from collections import Counter
 from .models import CodeIssue
 
-USERNAME = os.environ["BITBUCKET_USERNAME"]
-APP_PASSWORD = os.environ["BITBUCKET_APP_PASSWORD"]
-WORKSPACE = os.environ["BITBUCKET_WORKSPACE"]
-REPO_SLUG = os.environ["BITBUCKET_REPO_SLUG"]
-PR_ID = os.environ["BITBUCKET_PR_ID"]
 
-BASE_URL = f"https://api.bitbucket.org/2.0/repositories/{WORKSPACE}/{REPO_SLUG}/pullrequests/{PR_ID}"
-AUTH = HTTPBasicAuth(USERNAME, APP_PASSWORD)
-HEADERS = {
-    "Content-Type": "application/json"
-}
+def _get_api_details():
+    """
+    Helper function to get all necessary details from environment variables.
+    This function should only be called when we know we are in a Bitbucket environment.
+    """
+    try:
+        username = os.environ["BITBUCKET_APP_USERNAME"]
+        app_password = os.environ["BITBUCKET_APP_PASSWORD"]
+        workspace = os.environ["BITBUCKET_WORKSPACE"]
+        repo_slug = os.environ["BITBUCKET_REPO_SLUG"]
+        pr_id = os.environ["BITBUCKET_PR_ID"]
+    except KeyError as e:
+        raise ValueError(f"Required Bitbucket environment variable is not set: {e}")
+
+    base_url = f"https://api.bitbucket.org/2.0/repositories/{workspace}/{repo_slug}/pullrequests/{pr_id}"
+    auth = HTTPBasicAuth(username, app_password)
+    headers = {"Content-Type": "application/json"}
+    
+    return base_url, auth, headers
 
 
 def post_pr_comment(issue: CodeIssue, file_path: str):
     """Posts a single review comment to a specific line in a Bitbucket Pull Request."""
-    url = f"{BASE_URL}/comments"
-
-    comment_body = f"**[{issue.issue_type}]**\n\n{issue.comment}"
-    if issue.suggestion:
-        comment_body += f"\n\n**Suggestion:**\n```\n{issue.suggestion}\n```"
-
-    payload = {
-        "content": {"raw": comment_body},
-        "inline": {
-            "path": file_path,
-            "to": issue.line_number
+    try:
+        base_url, auth, headers = _get_api_details()
+        url = f"{base_url}/comments"
+
+        comment_body = f"**[{issue.issue_type}]**\n\n{issue.comment}"
+        if issue.suggestion:
+            comment_body += f"\n\n**Suggestion:**\n```\n{issue.suggestion}\n```"
+
+        payload = {
+            "content": {"raw": comment_body},
+            "inline": {
+                "path": file_path,
+                "to": issue.line_number
+            }
         }
-    }
 
-    response = requests.post(url, headers=HEADERS, auth=AUTH, json=payload)
+        response = requests.post(url, headers=headers, auth=auth, json=payload)
+        response.raise_for_status()
 
-    if response.status_code == 201:
         print(f"✅ Successfully posted comment to PR on file {file_path}.")
-    else:
-        print(f"❌ Failed to post line comment: {response.status_code} — {response.text}")
+    except (ValueError, requests.exceptions.RequestException) as e:
+        print(f"❌ Failed to post line comment: {e}")
 
 
 def post_summary_comment(all_issues: list[CodeIssue]):
@@ -47,25 +58,26 @@ def post_summary_comment(all_issues: list[CodeIssue]):
         return
 
     print("📝 Generating and posting summary comment to Bitbucket...")
+    try:
+        total_issues = len(all_issues)
+        issue_counts = Counter(issue.issue_type for issue in all_issues)
+
+        summary_body = f"### 🤖 AI Code Review Summary\n\nFound **{total_issues} potential issue(s)**.\n\n"
+        if issue_counts:
+            summary_body += "**Issue Breakdown:**\n"
+            for issue_type, count in issue_counts.items():
+                summary_body += f"* **{issue_type}:** {count} issue(s)\n"
+        summary_body += "\n---\n*Please see the detailed inline comments on the \"Diff\" tab for more context.*"
+
+        base_url, auth, headers = _get_api_details()
+        url = f"{base_url}/comments"
+        payload = {
+            "content": {"raw": summary_body}
+        }
 
-    total_issues = len(all_issues)
-    issue_counts = Counter(issue.issue_type for issue in all_issues)
-
-    summary_body = f"### 🤖 AI Code Review Summary\n\nFound **{total_issues} potential issue(s)**.\n\n"
-    if issue_counts:
-        summary_body += "**Issue Breakdown:**\n"
-        for issue_type, count in issue_counts.items():
-            summary_body += f"* **{issue_type}:** {count} issue(s)\n"
-    summary_body += "\n---\n*Please see the detailed inline comments on the \"Diff\" tab for more context.*"
-
-    url = f"{BASE_URL}/comments"
-    payload = {
-        "content": {"raw": summary_body}
-    }
-
-    response = requests.post(url, headers=HEADERS, auth=AUTH, json=payload)
+        response = requests.post(url, headers=headers, auth=auth, json=payload)
+        response.raise_for_status()
 
-    if response.status_code == 201:
         print("✅ Successfully posted the summary comment to Bitbucket.")
-    else:
-        print(f"❌ Failed to post summary comment: {response.status_code} — {response.text}")
+    except (ValueError, requests.exceptions.RequestException) as e:
+        print(f"❌ Failed to post summary comment: {e}")