diff --git a/offline/code_review_benchmark/step0_fork_prs.py b/offline/code_review_benchmark/step0_fork_prs.py index 3a41feb..3f14219 100644 --- a/offline/code_review_benchmark/step0_fork_prs.py +++ b/offline/code_review_benchmark/step0_fork_prs.py @@ -29,6 +29,7 @@ import tempfile import time +from dotenv import load_dotenv import requests @@ -268,6 +269,8 @@ def _load_pr_urls_from_file(path: str) -> list[str]: def main(): """CLI entrypoint: process a single PR or a batch file.""" + load_dotenv(".env", verbose=True) + parser = argparse.ArgumentParser(description="Clone PR(s) to your org for AI review") parser.add_argument("pr_url", nargs="?", help="GitHub PR URL (for single run)") parser.add_argument("--file", help="Path to golden comments JSON to batch process") diff --git a/offline/pyproject.toml b/offline/pyproject.toml index 429a426..9a0e5df 100644 --- a/offline/pyproject.toml +++ b/offline/pyproject.toml @@ -9,6 +9,9 @@ dependencies = [ "matplotlib>=3.8.0", "requests>=2.32.5", "tqdm>=4.66.0", + "pytest>=9.0.3", + "ruff>=0.14.11", + "python-dotenv>=1.2.2", ] [dependency-groups]