ai-cr 3.0.0__tar.gz → 3.1.1__tar.gz

{ai_cr-3.0.0 → ai_cr-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ai-cr
-Version: 3.0.0
+Version: 3.1.1
 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

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/bootstrap.py

@@ -41,7 +41,10 @@ def bootstrap(verbosity: int = 1):
     Env.verbosity = verbosity
     Env.logging_level = log_levels_by_verbosity.get(verbosity, logging.INFO)
     setup_logging(Env.logging_level)
-    logging.info("Bootstrapping... "+mc.ui.gray(f"[verbosity={verbosity}]"))
+    logging.info(
+        f"Bootstrapping Gito v{Env.gito_version}... "
+        + mc.ui.gray(f"[verbosity={verbosity}]")
+    )
 
     # cp1251 is used on Windows when redirecting output
     if sys.stdout.encoding and sys.stdout.encoding.lower() != "utf-8":

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/cli.py

@@ -26,7 +26,7 @@ from .utils import no_subcommand, extract_gh_owner_repo, remove_html_comments
 from .gh_api import resolve_gh_token
 
 # Import fix command to register it
-from .commands import fix, gh_react_to_comment, repl, deploy  # noqa
+from .commands import fix, gh_react_to_comment, repl, deploy, version  # noqa
 from .commands.gh_post_review_comment import post_github_cr_comment
 from .commands.linear_comment import linear_comment
 
@@ -92,6 +92,7 @@ def cmd_review(
     out: str = arg_out()
 ):
     _what, _against = args_to_target(refs, what, against)
+    pr = pr or os.getenv("PR_NUMBER_FROM_WORKFLOW_DISPATCH")
     with get_repo_context(url, _what) as (repo, out_folder):
         asyncio.run(review(
             repo=repo,
@@ -100,6 +101,7 @@ def cmd_review(
             filters=filters,
             use_merge_base=merge_base,
             out_folder=out or out_folder,
+            pr=pr,
         ))
         if post_comment:
             try:
@@ -134,8 +136,13 @@ def cmd_answer(
         default=None,
         show_default=False
     ),
+    pr: int = typer.Option(
+        default=None,
+        help="GitHub Pull Request number"
+    ),
 ):
     _what, _against = args_to_target(refs, what, against)
+    pr = pr or os.getenv("PR_NUMBER_FROM_WORKFLOW_DISPATCH")
     if str(question).startswith("tpl:"):
         prompt_file = str(question)[4:]
         question = ""
@@ -149,6 +156,7 @@ def cmd_answer(
         use_merge_base=merge_base,
         prompt_file=prompt_file,
         use_pipeline=use_pipeline,
+        pr=pr,
     )
     if post_to == 'linear':
         logging.info("Posting answer to Linear...")

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/commands/gh_react_to_comment.py

@@ -25,6 +25,30 @@ from ..utils import extract_gh_owner_repo
 from .fix import fix
 
 
+def cleanup_comment_addressed_to_gito(text):
+    if not text:
+        return text
+    patterns = [
+        r'^\s*gito\b',
+        r'^\s*ai\b',
+        r'^\s*bot\b',
+        r'^\s*@gito\b',
+        r'^\s*@ai\b',
+        r'^\s*@bot\b'
+    ]
+    result = text
+    # Remove each pattern from the beginning
+    for pattern in patterns:
+        result = re.sub(pattern, '', result, flags=re.IGNORECASE)
+
+    # Remove leading comma and spaces that may be left after prefix removal
+    result = re.sub(r'^\s*,\s*', '', result)
+
+    # Clean up extra whitespace
+    result = re.sub(r'\s+', ' ', result).strip()
+    return result
+
+
 @app.command(hidden=True)
 def react_to_comment(
     comment_id: int = typer.Argument(),
@@ -99,7 +123,8 @@ def react_to_comment(
         )
     else:
         if cfg.answer_github_comments:
-            response = answer(comment.body, repo=repo)
+            question = cleanup_comment_addressed_to_gito(comment.body)
+            response = answer(question, repo=repo, pr=pr)
             post_gh_comment(
                 gh_repository=f"{owner}/{repo_name}",
                 pr_or_issue_number=pr,

ai_cr-3.1.1/gito/commands/version.py

@@ -0,0 +1,8 @@
+from ..cli_base import app
+from ..env import Env
+
+
+@app.command(name='version', help='Show the version of gito.bot')
+def version():
+    print(Env.gito_version)
+    return Env.gito_version

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/constants.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from .env import Env
 
 PROJECT_GITO_FOLDER = ".gito"
 PROJECT_CONFIG_FILE_NAME = "config.toml"
@@ -9,5 +10,5 @@ JSON_REPORT_FILE_NAME = "code-review-report.json"
 GITHUB_MD_REPORT_FILE_NAME = "code-review-report.md"
 EXECUTABLE = "gito"
 TEXT_ICON_URL = 'https://raw.githubusercontent.com/Nayjest/Gito/main/press-kit/logo/gito-bot-1_64top.png'  # noqa: E501
-HTML_TEXT_ICON = f'<a href="https://github.com/Nayjest/Gito"><img src="{TEXT_ICON_URL}" align="left" width=64 height=50 /></a>'  # noqa: E501
+HTML_TEXT_ICON = f'<a href="https://github.com/Nayjest/Gito"><img src="{TEXT_ICON_URL}" align="left" width=64 height=50 title="Gito v{Env.gito_version}"/></a>'  # noqa: E501
 HTML_CR_COMMENT_MARKER = '<!-- GITO_COMMENT:CODE_REVIEW_REPORT -->'

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/core.py

@@ -1,11 +1,12 @@
+import os
 import fnmatch
 import logging
-from os import PathLike
 from typing import Iterable
 from pathlib import Path
 from functools import partial
 
 import microcore as mc
+from gito.gh_api import gh_api
 from microcore import ui
 from git import Repo, Commit
 from git.exc import GitCommandError
@@ -66,16 +67,57 @@ def commit_in_branch(repo: Repo, commit: Commit, target_branch: str) -> bool:
     return False
 
 
+def get_base_branch(repo: Repo, pr: int | str = None):
+    if os.getenv('GITHUB_ACTIONS'):
+
+        # triggered from PR
+        if base_ref := os.getenv('GITHUB_BASE_REF'):
+            logging.info(f"Using GITHUB_BASE_REF:{base_ref} as base branch")
+            return f'origin/{base_ref}'
+        logging.info("GITHUB_BASE_REF is not available...")
+        if pr:
+            api = gh_api(repo=repo)
+            pr_obj = api.pulls.get(pr)
+            logging.info(
+                f"Using 'origin/{pr_obj.base.ref}' as base branch "
+                f"(received via GH API for PR#{pr})"
+            )
+            return f'origin/{pr_obj.base.ref}'
+
+    try:
+        logging.info(
+            "Trying to resolve base branch from repo.remotes.origin.refs.HEAD.reference.name..."
+        )
+        # 'origin/main', 'origin/master', etc
+        # Stopped working in github actions since 07/2025
+        return repo.remotes.origin.refs.HEAD.reference.name
+    except AttributeError:
+        try:
+            logging.info(
+                "Checking if repo has 'main' or 'master' branchs to use as --against branch..."
+            )
+            remote_refs = repo.remotes.origin.refs
+            for branch_name in ['main', 'master']:
+                if hasattr(remote_refs, branch_name):
+                    return f'origin/{branch_name}'
+        except Exception:
+            pass
+
+        logging.error("Could not determine default branch from remote refs.")
+        raise ValueError("No default branch found in the repository.")
+
+
 def get_diff(
     repo: Repo = None,
     what: str = None,
     against: str = None,
     use_merge_base: bool = True,
+    pr: str | int = None
 ) -> PatchSet | list[PatchedFile]:
     repo = repo or Repo(".")
     if not against:
         # 'origin/main', 'origin/master', etc
-        against = repo.remotes.origin.refs.HEAD.reference.name
+        against = get_base_branch(repo, pr=pr)
     if review_subject_is_index(what):
         what = None  # working copy
     if use_merge_base:
@@ -175,7 +217,7 @@ def get_diff(
         )
         if file_path == DEV_NULL:
             continue
-        if is_binary_file(repo, file_path.lstrip("b/")):
+        if is_binary_file(repo, file_path.removeprefix("b/")):
             logging.info(f"Skipping binary file: {patched_file.path}")
             continue
         non_binary_diff.append(patched_file)
@@ -230,6 +272,7 @@ def make_cr_summary(ctx: Context, **kwargs) -> str:
             diff=mc.tokenizing.fit_to_token_size(ctx.diff, ctx.config.max_code_tokens)[0],
             issues=ctx.report.issues,
             pipeline_out=ctx.pipeline_out,
+            env=Env,
             **ctx.config.prompt_vars,
             **kwargs,
         ).to_llm()
@@ -250,11 +293,12 @@ def _prepare(
     against: str = None,
     filters: str | list[str] = "",
     use_merge_base: bool = True,
+    pr: str | int = None,
 ):
     repo = repo or Repo(".")
     cfg = ProjectConfig.load_for_repo(repo)
     diff = get_diff(
-        repo=repo, what=what, against=against, use_merge_base=use_merge_base
+        repo=repo, what=what, against=against, use_merge_base=use_merge_base, pr=pr,
     )
     diff = filter_diff(diff, filters)
     if not diff:
@@ -317,11 +361,17 @@ async def review(
     against: str = None,
     filters: str | list[str] = "",
     use_merge_base: bool = True,
-    out_folder: str | PathLike | None = None,
+    out_folder: str | os.PathLike | None = None,
+    pr: str | int = None
 ):
     try:
         repo, cfg, diff, lines = _prepare(
-            repo=repo, what=what, against=against, filters=filters, use_merge_base=use_merge_base
+            repo=repo,
+            what=what,
+            against=against,
+            filters=filters,
+            use_merge_base=use_merge_base,
+            pr=pr,
         )
     except NoChangesInContextError:
         logging.error("No changes to review")
@@ -377,10 +427,16 @@ def answer(
     use_merge_base: bool = True,
     use_pipeline: bool = True,
     prompt_file: str = None,
+    pr: str | int = None,
 ) -> str | None:
     try:
         repo, config, diff, lines = _prepare(
-            repo=repo, what=what, against=against, filters=filters, use_merge_base=use_merge_base
+            repo=repo,
+            what=what,
+            against=against,
+            filters=filters,
+            use_merge_base=use_merge_base,
+            pr=pr
         )
     except NoChangesInContextError:
         logging.error("No changes to review")

ai_cr-3.1.1/gito/env.py

@@ -0,0 +1,7 @@
+from importlib.metadata import version
+
+
+class Env:
+    logging_level: int = 1
+    verbosity: int = 1
+    gito_version: str = version("gito.bot")

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/gh_api.py

@@ -2,7 +2,38 @@ import os
 import logging
 
 import requests
+import git
 from fastcore.basics import AttrDict  # objects returned by ghapi
+from ghapi.core import GhApi
+
+from .project_config import ProjectConfig
+from .utils import extract_gh_owner_repo
+
+
+def gh_api(
+    repo: git.Repo = None,  # used to resolve owner/repo
+    config: ProjectConfig | None = None,  # used to resolve owner/repo
+    token: str | None = None
+) -> GhApi:
+    if repo:
+        # resolve owner/repo from repo.remotes.origin.url
+        owner, repo_name = extract_gh_owner_repo(repo)
+    else:
+        if not config:
+            config = ProjectConfig.load()
+        # resolve owner/repo from github env vars (github actions)
+        gh_env = config.prompt_vars.get("github_env", {})
+        gh_repo = gh_env.get("github_repo")
+        if not gh_repo:
+            raise ValueError("GitHub repository not specified and not found in project config.")
+        parts = gh_repo.split('/')
+        if len(parts) != 2:
+            raise ValueError(f"Invalid GitHub repository format: {gh_repo}. Expected 'owner/repo'.")
+        owner, repo_name = parts
+
+    token = resolve_gh_token(token)
+    api = GhApi(owner, repo_name, token=token)
+    return api
 
 
 def resolve_gh_token(token_or_none: str | None = None) -> str | None:

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/tpl/github_workflows/components/env-vars.j2

@@ -7,4 +7,5 @@
         JIRA_URL: ${{ secrets.JIRA_URL }}
         JIRA_USER: ${{ secrets.JIRA_USER }}
         LINEAR_API_KEY: ${{ secrets.LINEAR_API_KEY }}
+        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         {%- endraw %}

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/tpl/github_workflows/gito-code-review.yml.j2

@@ -22,8 +22,8 @@ jobs:
         {%- include("github_workflows/components/env-vars.j2") %}
         PR_NUMBER_FROM_WORKFLOW_DISPATCH: {% raw %}${{ github.event.inputs.pr_number }}{% endraw %}
       run: |{% raw %}
-        gito --verbose review ${{ github.event_name == 'pull_request' && github.event.pull_request.base.ref && format(' --against="origin/{0}"', github.event.pull_request.base.ref) || '' }}
-        gito github-comment --token ${{ secrets.GITHUB_TOKEN }}{% endraw %}
+        gito --verbose review
+        gito github-comment{% endraw %}
 
     - uses: actions/upload-artifact@v4
       with:

{ai_cr-3.0.0 → ai_cr-3.1.1}/gito/tpl/github_workflows/gito-react-to-comments.yml.j2

@@ -67,4 +67,4 @@ jobs:
         # Otherwise, use LLM_API_TYPE: none
         {%- include("github_workflows/components/env-vars.j2") %}
       run: |
-        {% raw %}gito react-to-comment ${{ github.event.comment.id }} --token ${{ secrets.GITHUB_TOKEN }}{%- endraw %}
+        {% raw %}gito react-to-comment ${{ github.event.comment.id }}{%- endraw %}

ai_cr-3.1.1/gito/tpl/questions/changes_summary.j2

@@ -0,0 +1,55 @@
+{{ self_id }}
+----TASK----
+
+## Subtask 1:
+Write a codebase changes summary clearly describing and explaining it for the engineering managers.
+- It is intended to be posted into bug tracker for estimating work done in the pull request.
+- Summary should be in a form of compact changelist where each change is described by one sentence.
+- Caption should be: ## PR Summary: <3-10 words>
+- Each change summary BP list item should start from effort estimation icon:
+- - ◆◇◇ (up to 2 hours)
+- - ◆◆◇ (half day)
+- - ◆◆◆ (1-2 days and more)
+
+Also include this estimation legend below as "Pure Codebase Work Estimation Legend".
+
+
+## Subtask 1 (Issue alignment sentence)
+Include one sentence about how the code changes address the requirements of the associated issue listed below.
+- Use ✅ or ⚠️ to indicate whether the implementation fully satisfies the issue requirements.
+- Put this sentence immediately below the PR Summary title
+Examples:
+
+If the implementation fully delivers the requested functionality:
+```
+✅  Implementation Satisfies [<ISSUE_KEY>](<ISSUE_URL>).
+```
+If there are concerns about how thoroughly the code covers the requirements and technical description from the associated issue:
+```
+⚠️ <Describe specific gap or concern>.
+⚠️ <Describe additional limitation or missing feature>.
+```
+
+## Subtask 3:
+Write release notes for public documentation.
+- Caption should be: ## Release Notes Proposal
+Summarize the following changes, focusing on what is new, improved, or fixed for the end user.
+Do not include internal or technical details.
+Structure release notes using clear sections: Added, Changed, Fixed.
+Avoid internal technical jargon or developer-specific details.
+
+----RELATED CODEBASE CHANGES----
+{% for part in diff %}{{ part }}\n{% endfor %}
+
+----FULL FILE CONTENT AFTER APPLYING CHANGES----
+{% for file, file_lines in all_file_lines.items() %}
+--FILE: {{ file }}--
+{{ file_lines }}
+{% endfor %}
+
+{%- if pipeline_out.associated_issue and pipeline_out.associated_issue.title %}
+----ASSOCIATED ISSUE----
+# {{ pipeline_out.associated_issue.title }}
+{{ pipeline_out.associated_issue.description }}
+URL: {{ pipeline_out.associated_issue.url }}
+{%- endif -%}{{ '\n' }}

ai_cr-3.1.1/gito/tpl/questions/testing_guide.j2

@@ -0,0 +1,23 @@
+{{ self_id }}
+----TASK----
+Create a concise comment for QA specialists that explains how to test the code-base changes.
+Include:
+- User journeys that could be disrupted by failures
+- Relevant security scenarios and edge cases
+- Potential risk areas introduced by the changes, with suggestions for focused testing to mitigate them
+
+----RELATED CODEBASE CHANGES----
+{% for part in diff %}{{ part }}\n{% endfor %}
+
+----FULL FILE CONTENT AFTER APPLYING CHANGES----
+{% for file, file_lines in all_file_lines.items() %}
+--FILE: {{ file }}--
+{{ file_lines }}
+{% endfor %}
+
+{%- if pipeline_out.associated_issue and pipeline_out.associated_issue.title %}
+----ASSOCIATED ISSUE----
+# {{ pipeline_out.associated_issue.title }}
+{{ pipeline_out.associated_issue.description }}
+URL: {{ pipeline_out.associated_issue.url }}
+{%- endif -%}{{ '\n' }}

{ai_cr-3.0.0 → ai_cr-3.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ai-cr"
-version = "3.0.0"
+version = "3.1.1"
 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"

ai_cr-3.0.0/gito/env.py

@@ -1,3 +0,0 @@
-class Env:
-    logging_level: int = 1
-    verbosity: int = 1