pullwise 0.4.0__tar.gz → 0.5.0__tar.gz

{pullwise-0.4.0 → pullwise-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pullwise
-Version: 0.4.0
+Version: 0.5.0
 Summary: AI-powered PR descriptions, labels, and code review using OpenAI
 License: MIT
 Project-URL: Homepage, https://github.com/albertusreza/pr-pilot
@@ -122,6 +122,9 @@ pr-pilot describe --markdown pr_description.md
 | `pr-pilot todos` | Scan for TODO/FIXME comments and create GitHub issues from them |
 | `pr-pilot commit` | Generate a Conventional Commit message from staged changes |
 | `pr-pilot release` | Full release: changelog + git tag + GitHub release in one command |
+| `pr-pilot docs` | Generate docstrings for functions in changed files |
+| `pr-pilot branch` | Suggest a git branch name from a plain-English task description |
+| `pr-pilot explain` | Explain what a file or function does in plain English |
 
 ## Usage
 
@@ -142,6 +145,26 @@ pr-pilot changelog
 pr-pilot changelog --output CHANGELOG.md
 ```
 
+```bash
+# Generate docstrings for all functions changed vs main
+pr-pilot docs
+
+# Document a specific file
+pr-pilot docs src/auth.py
+
+# Suggest a branch name from a task description
+pr-pilot branch "add rate limiting to upload endpoint"
+
+# Create the branch immediately
+pr-pilot branch "fix login timeout on mobile" --checkout
+
+# Explain what a file does
+pr-pilot explain src/auth.py
+
+# Explain a specific function
+pr-pilot explain src/auth.py --function authenticate
+```
+
 ```bash
 # Generate a commit message from staged changes
 pr-pilot commit

{pullwise-0.4.0 → pullwise-0.5.0}/README.md

@@ -102,6 +102,9 @@ pr-pilot describe --markdown pr_description.md
 | `pr-pilot todos` | Scan for TODO/FIXME comments and create GitHub issues from them |
 | `pr-pilot commit` | Generate a Conventional Commit message from staged changes |
 | `pr-pilot release` | Full release: changelog + git tag + GitHub release in one command |
+| `pr-pilot docs` | Generate docstrings for functions in changed files |
+| `pr-pilot branch` | Suggest a git branch name from a plain-English task description |
+| `pr-pilot explain` | Explain what a file or function does in plain English |
 
 ## Usage
 
@@ -122,6 +125,26 @@ pr-pilot changelog
 pr-pilot changelog --output CHANGELOG.md
 ```
 
+```bash
+# Generate docstrings for all functions changed vs main
+pr-pilot docs
+
+# Document a specific file
+pr-pilot docs src/auth.py
+
+# Suggest a branch name from a task description
+pr-pilot branch "add rate limiting to upload endpoint"
+
+# Create the branch immediately
+pr-pilot branch "fix login timeout on mobile" --checkout
+
+# Explain what a file does
+pr-pilot explain src/auth.py
+
+# Explain a specific function
+pr-pilot explain src/auth.py --function authenticate
+```
+
 ```bash
 # Generate a commit message from staged changes
 pr-pilot commit

pullwise-0.5.0/pr_pilot/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.5.0"

{pullwise-0.4.0 → pullwise-0.5.0}/pr_pilot/analyzer.py

@@ -13,6 +13,9 @@ from .templates import (
     ISSUE_SYSTEM, ISSUE_USER,
     COMMIT_SYSTEM, COMMIT_USER,
     RELEASE_NOTES_SYSTEM, RELEASE_NOTES_USER,
+    DOCSTRING_SYSTEM, DOCSTRING_USER,
+    BRANCH_SYSTEM, BRANCH_USER,
+    EXPLAIN_SYSTEM, EXPLAIN_USER,
 )
 
 _MAX_DIFF_CHARS = 24_000   # stay well within context limits
@@ -548,3 +551,170 @@ def run_release(
         body=release_body,
         prerelease=False,
     )
+
+
+# ── Docstring generator ───────────────────────────────────────────────────────
+
+@dataclass
+class DocstringResult:
+    language: str
+    function_name: str
+    docstring: str
+    placement: str   # "above" | "inside"
+
+
+def _detect_language(path: str) -> str:
+    ext = path.rsplit(".", 1)[-1].lower()
+    if ext == "py":
+        return "python"
+    if ext in {"ts", "tsx"}:
+        return "typescript"
+    return "javascript"
+
+
+def _extract_functions(code: str, language: str) -> list[tuple[str, int]]:
+    """Return list of (function_source, start_line) for top-level functions."""
+    import re
+    lines = code.splitlines()
+    results = []
+    if language == "python":
+        pattern = re.compile(r"^(def |async def )")
+        i = 0
+        while i < len(lines):
+            if pattern.match(lines[i]):
+                start = i
+                # collect until next top-level def/class or EOF
+                j = i + 1
+                while j < len(lines) and (not lines[j] or lines[j][0] in " \t#"):
+                    j += 1
+                results.append(("\n".join(lines[start:j]), start + 1))
+                i = j
+            else:
+                i += 1
+    else:
+        pattern = re.compile(r"^(export\s+)?(async\s+)?function\s+\w+|^\s*(const|let|var)\s+\w+\s*=\s*(async\s+)?\(")
+        for i, line in enumerate(lines):
+            if pattern.match(line):
+                end = min(i + 30, len(lines))
+                results.append(("\n".join(lines[i:end]), i + 1))
+    return results
+
+
+def generate_docstrings(
+    api_key: str, file_path: str, model: str = _MODEL
+) -> list[DocstringResult]:
+    """Generate docstrings for all functions in a file changed in the diff."""
+    from pathlib import Path
+    client = OpenAI(api_key=api_key)
+    code = Path(file_path).read_text(errors="replace")
+    language = _detect_language(file_path)
+    functions = _extract_functions(code, language)
+    if not functions:
+        return []
+    results = []
+    for func_code, _lineno in functions[:10]:  # cap at 10 per file
+        resp = client.chat.completions.create(
+            model=model,
+            max_tokens=512,
+            messages=[
+                {"role": "system", "content": DOCSTRING_SYSTEM},
+                {"role": "user", "content": DOCSTRING_USER.format(
+                    language=language, code=func_code[:3000]
+                )},
+            ],
+        )
+        raw = (resp.choices[0].message.content or "{}").strip()
+        if raw.startswith("```"):
+            raw = "\n".join(raw.splitlines()[1:])
+        if raw.endswith("```"):
+            raw = "\n".join(raw.splitlines()[:-1])
+        try:
+            data = json.loads(raw.strip())
+            results.append(DocstringResult(
+                language=data.get("language", language),
+                function_name=data.get("function_name", "unknown"),
+                docstring=data.get("docstring", ""),
+                placement=data.get("placement", "inside"),
+            ))
+        except json.JSONDecodeError:
+            continue
+    return results
+
+
+# ── Branch namer ──────────────────────────────────────────────────────────────
+
+@dataclass
+class BranchSuggestion:
+    suggestions: list[str]
+    recommended: int   # index into suggestions
+
+    @property
+    def best(self) -> str:
+        return self.suggestions[self.recommended]
+
+
+def suggest_branch(api_key: str, task: str, model: str = _MODEL) -> BranchSuggestion:
+    client = OpenAI(api_key=api_key)
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=200,
+        messages=[
+            {"role": "system", "content": BRANCH_SYSTEM},
+            {"role": "user", "content": BRANCH_USER.format(task=task)},
+        ],
+    )
+    raw = (resp.choices[0].message.content or "{}").strip()
+    if raw.startswith("```"):
+        raw = "\n".join(raw.splitlines()[1:])
+    if raw.endswith("```"):
+        raw = "\n".join(raw.splitlines()[:-1])
+    data = json.loads(raw.strip())
+    suggestions = data.get("suggestions", [f"feat/{task[:40].lower().replace(' ', '-')}"])
+    recommended = data.get("recommended", 0)
+    return BranchSuggestion(suggestions=suggestions, recommended=recommended)
+
+
+# ── Code explainer ────────────────────────────────────────────────────────────
+
+def explain_code(
+    api_key: str,
+    file_path: str,
+    selector: str | None = None,
+    model: str = _MODEL,
+) -> str:
+    """Explain a file or specific function in plain English."""
+    from pathlib import Path
+    client = OpenAI(api_key=api_key)
+    code = Path(file_path).read_text(errors="replace")
+
+    # If selector given, try to extract just that function/class
+    if selector:
+        import re
+        pattern = re.compile(
+            rf"^(def |async def |class |\w+ = (async )?function )"
+            rf".*{re.escape(selector)}",
+            re.MULTILINE
+        )
+        m = pattern.search(code)
+        if m:
+            start = m.start()
+            # grab next ~60 lines
+            snippet = "\n".join(code[start:].splitlines()[:60])
+            code = snippet
+
+    if len(code) > _MAX_DIFF_CHARS:
+        code = code[:_MAX_DIFF_CHARS] + "\n\n[truncated]"
+
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=512,
+        messages=[
+            {"role": "system", "content": EXPLAIN_SYSTEM},
+            {"role": "user", "content": EXPLAIN_USER.format(
+                file_path=file_path,
+                selector=f"Function/class: {selector}" if selector else "Whole file",
+                code=code,
+            )},
+        ],
+    )
+    return (resp.choices[0].message.content or "").strip()

{pullwise-0.4.0 → pullwise-0.5.0}/pr_pilot/cli.py

@@ -7,6 +7,7 @@ from .analyzer import (
     describe_pr, suggest_labels, review_pr, review_pr_as_comment,
     generate_changelog, suggest_reviewers, generate_standup, create_issues_from_todos,
     generate_commit_message, run_release,
+    generate_docstrings, suggest_branch, explain_code,
 )
 
 _RESET = "\033[0m"
@@ -169,6 +170,78 @@ def cmd_release(args: argparse.Namespace) -> None:
     print()
 
 
+def cmd_docs(args: argparse.Namespace) -> None:
+    import os as _os
+    files = args.files
+    if not files:
+        # Default: Python/JS/TS files changed vs base
+        from .analyzer import _git
+        changed = _git("diff", f"{args.base}...HEAD", "--name-only").splitlines()
+        files = [f for f in changed
+                 if f.endswith((".py", ".js", ".ts", ".tsx", ".jsx")) and _os.path.exists(f)]
+    if not files:
+        print("  No changed source files found. Pass file paths as arguments.")
+        return
+
+    for file_path in files:
+        print(f"\n  {_DIM}Generating docstrings for {file_path}...{_RESET}\n")
+        results = generate_docstrings(_key(), file_path, model=args.model)
+        if not results:
+            print(f"  No functions found in {file_path}")
+            continue
+        for r in results:
+            delim = '"""' if r.language == "python" else "/**"
+            delim_end = '"""' if r.language == "python" else " */"
+            print(f"  {_BOLD}{r.function_name}{_RESET}  {_DIM}({r.placement}){_RESET}")
+            print(f"  {_DIM}{delim}{_RESET}")
+            for line in r.docstring.splitlines():
+                prefix = "   " if r.language != "python" else "    "
+                print(f"  {prefix}{line}")
+            print(f"  {_DIM}{delim_end}{_RESET}\n")
+    print()
+
+
+def cmd_branch(args: argparse.Namespace) -> None:
+    task = " ".join(args.task)
+    if not task:
+        print("pr-pilot branch: provide a task description", file=sys.stderr)
+        sys.exit(1)
+    print(f"\n  {_DIM}Generating branch names for: \"{task}\"{_RESET}\n")
+    suggestion = suggest_branch(_key(), task=task, model=args.model)
+    for i, name in enumerate(suggestion.suggestions):
+        marker = f"{_GREEN}★{_RESET}" if i == suggestion.recommended else " "
+        print(f"  {marker} {_BOLD}{name}{_RESET}")
+
+    if args.checkout:
+        import subprocess
+        best = suggestion.best
+        result = subprocess.run(["git", "checkout", "-b", best])
+        if result.returncode == 0:
+            print(f"\n  {_GREEN}✓{_RESET} Switched to new branch '{best}'")
+        else:
+            print(f"\n  Could not create branch — it may already exist.")
+    elif args.copy:
+        try:
+            import subprocess
+            subprocess.run(["pbcopy"], input=suggestion.best.encode(), check=True)
+            print(f"\n  {_GREEN}✓{_RESET} '{suggestion.best}' copied to clipboard")
+        except Exception:
+            pass
+    print()
+
+
+def cmd_explain(args: argparse.Namespace) -> None:
+    import os as _os
+    if not _os.path.exists(args.file):
+        print(f"pr-pilot explain: file not found: {args.file}", file=sys.stderr)
+        sys.exit(1)
+    selector_label = f" → {args.function}" if args.function else ""
+    print(f"\n  {_DIM}Explaining {args.file}{selector_label}...{_RESET}\n")
+    explanation = explain_code(_key(), args.file, selector=args.function, model=args.model)
+    print(explanation)
+    print()
+
+
 def cmd_reviewers(args: argparse.Namespace) -> None:
     from .github_client import upsert_comment
     from .templates import REVIEWER_COMMENT_HEADER
@@ -312,6 +385,28 @@ def main() -> None:
     p_rev.add_argument("--model", default="gpt-4o", help="OpenAI model to use")
     p_rev.set_defaults(func=cmd_review)
 
+    # --- docs ---
+    p_docs = sub.add_parser("docs", help="Generate docstrings for functions in changed files")
+    p_docs.add_argument("files", nargs="*", help="Files to document (default: changed files vs base)")
+    p_docs.add_argument("--base", default="main")
+    p_docs.add_argument("--model", default="gpt-4o")
+    p_docs.set_defaults(func=cmd_docs)
+
+    # --- branch ---
+    p_branch = sub.add_parser("branch", help="Suggest a git branch name from a task description")
+    p_branch.add_argument("task", nargs="+", help="Plain-English task description")
+    p_branch.add_argument("--model", default="gpt-4o")
+    p_branch.add_argument("--checkout", action="store_true", help="Run git checkout -b with the best suggestion")
+    p_branch.add_argument("--copy", action="store_true", help="Copy best suggestion to clipboard (macOS)")
+    p_branch.set_defaults(func=cmd_branch)
+
+    # --- explain ---
+    p_explain = sub.add_parser("explain", help="Explain what a file or function does in plain English")
+    p_explain.add_argument("file", help="File to explain")
+    p_explain.add_argument("--function", "-f", default=None, help="Specific function or class to explain")
+    p_explain.add_argument("--model", default="gpt-4o")
+    p_explain.set_defaults(func=cmd_explain)
+
     # --- commit ---
     p_commit = sub.add_parser("commit", help="Generate a conventional commit message from staged changes")
     p_commit.add_argument("--model", default="gpt-4o")

{pullwise-0.4.0 → pullwise-0.5.0}/pr_pilot/templates.py

@@ -229,3 +229,80 @@ Version: {version}
 Changelog entry:
 {changelog_md}
 """
+
+# ── Docstring generator ───────────────────────────────────────────────────────
+
+DOCSTRING_SYSTEM = """\
+You are an expert technical writer generating docstrings for source code functions.
+
+Given a function's source code and language, write a concise, accurate docstring.
+
+Return a JSON object:
+{
+  "language": "python" | "javascript" | "typescript",
+  "function_name": "the function name",
+  "docstring": "the full docstring text to insert (without surrounding quotes/delimiters)",
+  "placement": "above" | "inside"
+}
+
+Rules:
+- Python: use Google-style docstrings. placement = "inside" (first line after def)
+- JS/TS: use JSDoc format. placement = "above" (/** ... */ block before the function)
+- Include Args/Parameters, Returns, and Raises/Throws only if non-trivial
+- Be concise — max 6 lines for simple functions
+- Return ONLY the JSON object, no markdown fences
+"""
+
+DOCSTRING_USER = """\
+Language: {language}
+Function:
+{code}
+"""
+
+# ── Branch namer ──────────────────────────────────────────────────────────────
+
+BRANCH_SYSTEM = """\
+You are a developer assistant suggesting clean git branch names.
+
+Given a task description, return 3 branch name suggestions.
+
+Return a JSON object:
+{
+  "suggestions": [
+    "type/short-kebab-description",
+    "type/alternative-name",
+    "type/another-option"
+  ],
+  "recommended": 0
+}
+
+Types: feat | fix | chore | docs | refactor | test | hotfix
+Rules:
+- lowercase only
+- use hyphens, no underscores or spaces
+- max 50 chars total
+- be specific, not generic (not "fix/bug" but "fix/login-timeout-safari")
+- Return ONLY the JSON object, no markdown fences
+"""
+
+BRANCH_USER = "Task: {task}"
+
+# ── Code explainer ────────────────────────────────────────────────────────────
+
+EXPLAIN_SYSTEM = """\
+You are a senior engineer explaining code to a teammate.
+Given source code (a file or function), explain:
+1. What it does — in 2-3 plain sentences
+2. Key design decisions or patterns used (if any)
+3. Any gotchas, assumptions, or side effects worth knowing
+
+Be direct. Write like you're doing a quick code walkthrough over Slack, not writing docs.
+No headers. No bullet points for the overview — save bullets for gotchas only.
+"""
+
+EXPLAIN_USER = """\
+File: {file_path}
+{selector}
+Code:
+{code}
+"""

{pullwise-0.4.0 → pullwise-0.5.0}/pullwise.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pullwise
-Version: 0.4.0
+Version: 0.5.0
 Summary: AI-powered PR descriptions, labels, and code review using OpenAI
 License: MIT
 Project-URL: Homepage, https://github.com/albertusreza/pr-pilot
@@ -122,6 +122,9 @@ pr-pilot describe --markdown pr_description.md
 | `pr-pilot todos` | Scan for TODO/FIXME comments and create GitHub issues from them |
 | `pr-pilot commit` | Generate a Conventional Commit message from staged changes |
 | `pr-pilot release` | Full release: changelog + git tag + GitHub release in one command |
+| `pr-pilot docs` | Generate docstrings for functions in changed files |
+| `pr-pilot branch` | Suggest a git branch name from a plain-English task description |
+| `pr-pilot explain` | Explain what a file or function does in plain English |
 
 ## Usage
 
@@ -142,6 +145,26 @@ pr-pilot changelog
 pr-pilot changelog --output CHANGELOG.md
 ```
 
+```bash
+# Generate docstrings for all functions changed vs main
+pr-pilot docs
+
+# Document a specific file
+pr-pilot docs src/auth.py
+
+# Suggest a branch name from a task description
+pr-pilot branch "add rate limiting to upload endpoint"
+
+# Create the branch immediately
+pr-pilot branch "fix login timeout on mobile" --checkout
+
+# Explain what a file does
+pr-pilot explain src/auth.py
+
+# Explain a specific function
+pr-pilot explain src/auth.py --function authenticate
+```
+
 ```bash
 # Generate a commit message from staged changes
 pr-pilot commit

{pullwise-0.4.0 → pullwise-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pullwise"
-version = "0.4.0"
+version = "0.5.0"
 description = "AI-powered PR descriptions, labels, and code review using OpenAI"
 readme = "README.md"
 license = { text = "MIT" }

{pullwise-0.4.0 → pullwise-0.5.0}/tests/test_analyzer.py

@@ -7,6 +7,8 @@ from pr_pilot.analyzer import (
     review_pr_as_comment, generate_changelog, ChangelogEntry,
     suggest_reviewers, generate_standup, create_issues_from_todos, _bump_version,
     generate_commit_message, CommitMessage, run_release,
+    generate_docstrings, suggest_branch, explain_code,
+    _extract_functions, _detect_language,
 )
 from pr_pilot.templates import REVIEW_COMMENT_HEADER, REVIEWER_COMMENT_HEADER
 
@@ -292,7 +294,7 @@ def test_commit_message_format_subject_only():
 @patch("pr_pilot.analyzer.get_diff", return_value="+ new code")
 @patch("pr_pilot.analyzer.get_commits", return_value="")
 @patch("pr_pilot.analyzer.get_branch", return_value="main")
-def test_run_release_dry_run(mock_branch, mock_commits, mock_diff, mock_ver, mock_ctag, tmp_path):
+def test_run_release_dry_run(mock_branch, mock_commits, mock_diff, mock_ver, mock_ctag, tmp_path):  # noqa: E501
     changelog_payload = {
         "version": "minor", "highlights": "New features.",
         "added": ["Dark mode"], "changed": [], "fixed": ["Login bug"],
@@ -318,3 +320,96 @@ def test_run_release_dry_run(mock_branch, mock_commits, mock_diff, mock_ver, moc
     # dry run: changelog file should NOT be written
     import pathlib
     assert not pathlib.Path(changelog_file).exists()
+
+
+# ── Docstring generator tests ─────────────────────────────────────────────────
+
+def test_detect_language():
+    assert _detect_language("app.py") == "python"
+    assert _detect_language("index.ts") == "typescript"
+    assert _detect_language("utils.tsx") == "typescript"
+    assert _detect_language("main.js") == "javascript"
+
+
+def test_extract_functions_python():
+    code = "x = 1\n\ndef foo(a, b):\n    return a + b\n\ndef bar():\n    pass\n"
+    funcs = _extract_functions(code, "python")
+    assert len(funcs) == 2
+    assert "def foo" in funcs[0][0]
+    assert funcs[0][1] == 3  # line number
+
+
+def test_generate_docstrings(tmp_path):
+    py_file = tmp_path / "utils.py"
+    py_file.write_text("def add(a, b):\n    return a + b\n")
+    payload = {
+        "language": "python",
+        "function_name": "add",
+        "docstring": "Add two numbers and return the result.",
+        "placement": "inside",
+    }
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(json.dumps(payload))
+        results = generate_docstrings("fake-key", str(py_file))
+    assert len(results) == 1
+    assert results[0].function_name == "add"
+    assert "Add two numbers" in results[0].docstring
+    assert results[0].placement == "inside"
+
+
+def test_generate_docstrings_no_functions(tmp_path):
+    py_file = tmp_path / "constants.py"
+    py_file.write_text("MAX = 100\nMIN = 0\n")
+    results = generate_docstrings("fake-key", str(py_file))
+    assert results == []
+
+
+# ── Branch namer tests ────────────────────────────────────────────────────────
+
+def test_suggest_branch_basic():
+    payload = {
+        "suggestions": ["feat/add-dark-mode", "feat/dark-mode-toggle", "feat/theme-switcher"],
+        "recommended": 0,
+    }
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(json.dumps(payload))
+        result = suggest_branch("fake-key", task="add dark mode to settings page")
+    assert result.best == "feat/add-dark-mode"
+    assert len(result.suggestions) == 3
+    assert result.recommended == 0
+
+
+def test_suggest_branch_recommended_index():
+    payload = {
+        "suggestions": ["fix/login-bug", "fix/auth-timeout", "fix/session-expiry"],
+        "recommended": 2,
+    }
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(json.dumps(payload))
+        result = suggest_branch("fake-key", task="fix session expiry on mobile")
+    assert result.best == "fix/session-expiry"
+
+
+# ── Code explainer tests ──────────────────────────────────────────────────────
+
+def test_explain_code_file(tmp_path):
+    py_file = tmp_path / "auth.py"
+    py_file.write_text("def authenticate(user, password):\n    return user == 'admin'\n")
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response("This file handles basic authentication by comparing credentials.")
+        result = explain_code("fake-key", str(py_file))
+    assert "authentication" in result.lower()
+
+
+def test_explain_code_with_selector(tmp_path):
+    py_file = tmp_path / "utils.py"
+    py_file.write_text("def add(a, b):\n    return a + b\n\ndef sub(a, b):\n    return a - b\n")
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response("The add function takes two numbers and returns their sum.")
+        result = explain_code("fake-key", str(py_file), selector="add")
+    assert "add" in result.lower() or "sum" in result.lower()

pullwise-0.4.0/pr_pilot/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.4.0"