github-code-review 3.3.1__tar.gz → 3.3.2__tar.gz

{github_code_review-3.3.1 → github_code_review-3.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: github-code-review
-Version: 3.3.1
+Version: 3.3.2
 Summary: AI code review tool that works with any language model provider. It detects issues in GitHub pull requests or local changes—instantly, reliably, and without vendor lock-in.
 License: MIT
 Keywords: static code analysis,code review,code quality,ai,coding,assistant,llm,github,automation,devops,developer tools,github actions,workflows,git
@@ -21,6 +21,7 @@ Requires-Dist: anthropic (>=0.57.1,<0.58.0)
 Requires-Dist: ghapi (>=1.0.6,<1.1.0)
 Requires-Dist: google-generativeai (>=0.8.5,<0.9.0)
 Requires-Dist: jira (>=3.8.0,<4.0.0)
+Requires-Dist: pydantic (>=2.12.3,<3.0.0)
 Requires-Dist: typer (>=0.16.0,<0.17.0)
 Requires-Dist: unidiff (>=0.7.5,<0.8.0)
 Project-URL: Homepage, https://github.com/Nayjest/Gito

{github_code_review-3.3.1 → github_code_review-3.3.2}/gito/cli.py

@@ -202,8 +202,8 @@ def setup():
     mc.interactive_setup(HOME_ENV_PATH)
 
 
-@app.command(name="render", help="Render and display code review report.")
-@app.command(name="report", hidden=True)
+@app.command(name="report", help="Render and display code review report.")
+@app.command(name="render", hidden=True)
 def render(
     format: str = typer.Argument(default=Report.Format.CLI),
     source: str = typer.Option(
@@ -238,7 +238,8 @@ def files(
             f"Changed files: "
             f"{mc.ui.green(_what or 'INDEX')} vs "
             f"{mc.ui.yellow(_against or repo.remotes.origin.refs.HEAD.reference.name)}"
-            f"{' filtered by ' + mc.ui.cyan(filters) if filters else ''}"
+            f"{' filtered by ' + mc.ui.cyan(filters) if filters else ''} --> "
+            f"{mc.ui.cyan(len(patch_set or []))} file(s)."
         )
 
         for patch in patch_set:

{github_code_review-3.3.1 → github_code_review-3.3.2}/gito/core.py

@@ -1,3 +1,6 @@
+"""
+Gito core business logic.
+"""
 import os
 import fnmatch
 import logging
@@ -6,8 +9,6 @@ from pathlib import Path
 from functools import partial
 
 import microcore as mc
-from gito.constants import REFS_VALUE_ALL
-from gito.gh_api import gh_api
 from microcore import ui
 from git import Repo, Commit
 from git.exc import GitCommandError
@@ -16,11 +17,12 @@ from unidiff.constants import DEV_NULL
 
 from .context import Context
 from .project_config import ProjectConfig
-from .report_struct import Report
-from .constants import JSON_REPORT_FILE_NAME
+from .report_struct import Report, RawIssue
+from .constants import JSON_REPORT_FILE_NAME, REFS_VALUE_ALL
 from .utils import make_streaming_function
 from .pipeline import Pipeline
 from .env import Env
+from .gh_api import gh_api
 
 
 def review_subject_is_index(what):
@@ -384,6 +386,19 @@ def provide_affected_code_blocks(issues: dict, repo: Repo):
                     i["affected_code"] = block
 
 
+def _llm_response_validator(parsed_response: list[dict]):
+    """
+    Validate that the LLM response is a list of dicts that can be converted to RawIssue.
+    """
+    if not isinstance(parsed_response, list):
+        raise ValueError("Response is not a list")
+    for item in parsed_response:
+        if not isinstance(item, dict):
+            raise ValueError("Response item is not a dict")
+        RawIssue(**item)
+    return True
+
+
 async def review(
     repo: Repo = None,
     what: str = None,
@@ -420,14 +435,15 @@ async def review(
             for file_diff in diff
         ],
         retries=cfg.retries,
-        parse_json=True,
+        parse_json={"validator": _llm_response_validator},
     )
     issues = {file.path: issues for file, issues in zip(diff, responses) if issues}
     provide_affected_code_blocks(issues, repo)
     exec(cfg.post_process, {"mc": mc, **locals()})
     out_folder = Path(out_folder or repo.working_tree_dir)
     out_folder.mkdir(parents=True, exist_ok=True)
-    report = Report(issues=issues, number_of_processed_files=len(diff))
+    report = Report(number_of_processed_files=len(diff))
+    report.register_issues(issues)
     ctx = Context(
         report=report,
         config=cfg,
@@ -496,7 +512,7 @@ def answer(
             config.max_code_tokens // 2
         )
     else:
-        aux_files_dict = dict()
+        aux_files_dict = {}
 
     if not prompt_file and config.answer_prompt.startswith("tpl:"):
         prompt_file = str(config.answer_prompt)[4:]

{github_code_review-3.3.1 → github_code_review-3.3.2}/gito/gh_api.py

@@ -67,9 +67,9 @@ def post_gh_comment(
     if 200 <= resp.status_code < 300:
         logging.info(f"Posted review comment to #{pr_or_issue_number} in {gh_repository}")
         return True
-    else:
-        logging.error(f"Failed to post comment: {resp.status_code} {resp.reason}\n{resp.text}")
-        return False
+
+    logging.error(f"Failed to post comment: {resp.status_code} {resp.reason}\n{resp.text}")
+    return False
 
 
 def hide_gh_comment(

{github_code_review-3.3.1 → github_code_review-3.3.2}/gito/report_struct.py

@@ -1,14 +1,15 @@
 import json
 import logging
-from dataclasses import dataclass, field, asdict
+from dataclasses import field, asdict, is_dataclass
 from datetime import datetime
 from enum import StrEnum
 from pathlib import Path
 
+import textwrap
 import microcore as mc
-from colorama import Fore, Style, Back
 from microcore.utils import file_link
-import textwrap
+from colorama import Fore, Style, Back
+from pydantic.dataclasses import dataclass
 
 from .constants import JSON_REPORT_FILE_NAME, HTML_TEXT_ICON, HTML_CR_COMMENT_MARKER
 from .project_config import ProjectConfig
@@ -16,33 +17,47 @@ from .utils import syntax_hint, block_wrap_lr, max_line_len, remove_html_comment
 
 
 @dataclass
-class Issue:
+class RawIssue:
     @dataclass
     class AffectedCode:
         start_line: int = field()
         end_line: int | None = field(default=None)
+        proposal: str | None = field(default="")
+
+    title: str = field()
+    details: str | None = field(default="")
+    severity: int | None = field(default=None)
+    confidence: int | None = field(default=None)
+    tags: list[str] = field(default_factory=list)
+    affected_lines: list[AffectedCode] = field(default_factory=list)
+
+
+@dataclass
+class Issue(RawIssue):
+    @dataclass
+    class AffectedCode(RawIssue.AffectedCode):
         file: str = field(default="")
-        proposal: str = field(default="")
         affected_code: str = field(default="")
 
         @property
         def syntax_hint(self) -> str:
             return syntax_hint(self.file)
 
-    id: str = field()
-    title: str = field()
-    details: str = field(default="")
-    severity: int | None = field(default=None)
-    confidence: int | None = field(default=None)
-    tags: list[str] = field(default_factory=list)
+    id: int | str = field(kw_only=True)
     file: str = field(default="")
     affected_lines: list[AffectedCode] = field(default_factory=list)
 
-    def __post_init__(self):
-        self.affected_lines = [
-            Issue.AffectedCode(**filter_kwargs(Issue.AffectedCode, dict(file=self.file) | i))
-            for i in self.affected_lines
-        ]
+    @staticmethod
+    def from_raw_issue(file: str, raw_issue: RawIssue | dict, issue_id: int | str) -> "Issue":
+        if is_dataclass(raw_issue):
+            raw_issue = asdict(raw_issue)
+        params = filter_kwargs(Issue, raw_issue | {"file": file, "id": issue_id})
+        for i, obj in enumerate(params["affected_lines"]):
+            d = obj if isinstance(obj, dict) else asdict(obj)
+            params["affected_lines"][i] = Issue.AffectedCode(
+                **filter_kwargs(Issue.AffectedCode, {"file": file} | d)
+            )
+        return Issue(**params)
 
     def github_code_link(self, github_env: dict) -> str:
         url = (
@@ -63,7 +78,7 @@ class Report:
         MARKDOWN = "md"
         CLI = "cli"
 
-    issues: dict[str, list[Issue | dict]] = field(default_factory=dict)
+    issues: dict[str, list[Issue]] = field(default_factory=dict)
     summary: str = field(default="")
     number_of_processed_files: int = field(default=0)
     total_issues: int = field(init=False)
@@ -79,19 +94,20 @@ class Report:
             for issue in issues
         ]
 
+    def register_issues(self, issues: dict[str, list[RawIssue | dict]]):
+        for file, file_issues in issues.items():
+            for issue in file_issues:
+                self.register_issue(file, issue)
+
+    def register_issue(self, file: str, issue: RawIssue | dict):
+        if file not in self.issues:
+            self.issues[file] = []
+        total = len(self.plain_issues)
+        self.issues[file].append(Issue.from_raw_issue(file, issue, issue_id=total + 1))
+        self.total_issues = total + 1
+
     def __post_init__(self):
-        issue_id: int = 0
-        for file in self.issues.keys():
-            self.issues[file] = [
-                Issue(
-                    **filter_kwargs(Issue, {
-                        "id": (issue_id := issue_id + 1),
-                        "file": file,
-                    } | issue)
-                )
-                for issue in self.issues[file]
-            ]
-        self.total_issues = issue_id
+        self.total_issues = len(self.plain_issues)
 
     def save(self, file_name: str = ""):
         file_name = file_name or JSON_REPORT_FILE_NAME

{github_code_review-3.3.1 → github_code_review-3.3.2}/gito/utils.py

@@ -2,7 +2,7 @@ import logging
 import re
 import sys
 import os
-from dataclasses import fields
+from dataclasses import fields, is_dataclass
 from pathlib import Path
 import importlib.metadata
 from typing import Optional
@@ -254,6 +254,9 @@ def filter_kwargs(cls, kwargs, log_warnings=True):
     Returns:
         A dictionary containing only the fields that are defined in the dataclass.
     """
+    if not is_dataclass(cls):
+        raise TypeError(f"{cls.__name__} is not a dataclass or pydantic dataclass")
+
     cls_fields = {f.name for f in fields(cls)}
     filtered = {}
     for k, v in kwargs.items():

{github_code_review-3.3.1 → github_code_review-3.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "github-code-review"
-version = "3.3.1"
+version = "3.3.2"
 description = "AI code review tool that works with any language model provider. It detects issues in GitHub pull requests or local changes—instantly, reliably, and without vendor lock-in."
 authors = ["Nayjest <mail@vitaliy.in>"]
 readme = "README.md"
@@ -29,6 +29,7 @@ anthropic = "^0.57.1"
 typer = "^0.16.0"
 ghapi = "~=1.0.6"
 jira = "^3.8.0"
+pydantic = "^2.12.3"
 
 [tool.poetry.group.dev.dependencies]
 flake8 = "*"