xai-review 0.11.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

ai_review/clients/github/client.py

@@ -0,0 +1,31 @@
+from httpx import AsyncClient, AsyncHTTPTransport
+
+from ai_review.clients.github.pr.client import GitHubPullRequestsHTTPClient
+from ai_review.config import settings
+from ai_review.libs.http.event_hooks.logger import LoggerEventHook
+from ai_review.libs.http.transports.retry import RetryTransport
+from ai_review.libs.logger import get_logger
+
+
+class GitHubHTTPClient:
+    def __init__(self, client: AsyncClient):
+        self.pr = GitHubPullRequestsHTTPClient(client)
+
+
+def get_github_http_client() -> GitHubHTTPClient:
+    logger = get_logger("GITHUB_HTTP_CLIENT")
+    logger_event_hook = LoggerEventHook(logger=logger)
+    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+
+    client = AsyncClient(
+        timeout=settings.llm.http_client.timeout,
+        headers={"Authorization": f"Bearer {settings.vcs.http_client.api_token_value}"},
+        base_url=settings.vcs.http_client.api_url_value,
+        transport=retry_transport,
+        event_hooks={
+            'request': [logger_event_hook.request],
+            'response': [logger_event_hook.response]
+        }
+    )
+
+    return GitHubHTTPClient(client=client)

ai_review/clients/github/pr/client.py

@@ -0,0 +1,101 @@
+from httpx import Response
+
+from ai_review.clients.github.pr.schema.comments import (
+    GitHubGetPRCommentsResponseSchema,
+    GitHubCreateIssueCommentRequestSchema,
+    GitHubCreateIssueCommentResponseSchema,
+    GitHubCreateReviewCommentRequestSchema,
+    GitHubCreateReviewCommentResponseSchema
+)
+from ai_review.clients.github.pr.schema.files import GitHubGetPRFilesResponseSchema
+from ai_review.clients.github.pr.schema.pull_request import GitHubGetPRResponseSchema
+from ai_review.clients.github.pr.schema.reviews import GitHubGetPRReviewsResponseSchema
+from ai_review.libs.http.client import HTTPClient
+
+
+class GitHubPullRequestsHTTPClient(HTTPClient):
+    async def get_pull_request_api(self, owner: str, repo: str, pull_number: str) -> Response:
+        return await self.get(f"/repos/{owner}/{repo}/pulls/{pull_number}")
+
+    async def get_files_api(self, owner: str, repo: str, pull_number: str) -> Response:
+        return await self.get(f"/repos/{owner}/{repo}/pulls/{pull_number}/files")
+
+    async def get_issue_comments_api(self, owner: str, repo: str, issue_number: str) -> Response:
+        return await self.get(f"/repos/{owner}/{repo}/issues/{issue_number}/comments")
+
+    async def get_review_comments_api(self, owner: str, repo: str, pull_number: str) -> Response:
+        return await self.get(f"/repos/{owner}/{repo}/pulls/{pull_number}/comments")
+
+    async def create_review_comment_api(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            request: GitHubCreateReviewCommentRequestSchema,
+    ) -> Response:
+        return await self.post(
+            f"/repos/{owner}/{repo}/pulls/{pull_number}/comments",
+            json=request.model_dump(),
+        )
+
+    async def create_issue_comment_api(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str,
+            request: GitHubCreateIssueCommentRequestSchema,
+    ) -> Response:
+        return await self.post(
+            f"/repos/{owner}/{repo}/issues/{issue_number}/comments",
+            json=request.model_dump(),
+        )
+
+    async def get_reviews_api(self, owner: str, repo: str, pull_number: str) -> Response:
+        return await self.get(f"/repos/{owner}/{repo}/pulls/{pull_number}/reviews")
+
+    async def get_pull_request(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRResponseSchema:
+        response = await self.get_pull_request_api(owner, repo, pull_number)
+        return GitHubGetPRResponseSchema.model_validate_json(response.text)
+
+    async def get_files(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRFilesResponseSchema:
+        response = await self.get_files_api(owner, repo, pull_number)
+        return GitHubGetPRFilesResponseSchema.model_validate_json(response.text)
+
+    async def get_issue_comments(self, owner: str, repo: str, issue_number: str) -> GitHubGetPRCommentsResponseSchema:
+        response = await self.get_issue_comments_api(owner, repo, issue_number)
+        return GitHubGetPRCommentsResponseSchema.model_validate_json(response.text)
+
+    async def get_review_comments(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRCommentsResponseSchema:
+        response = await self.get_review_comments_api(owner, repo, pull_number)
+        return GitHubGetPRCommentsResponseSchema.model_validate_json(response.text)
+
+    async def get_reviews(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRReviewsResponseSchema:
+        response = await self.get_reviews_api(owner, repo, pull_number)
+        return GitHubGetPRReviewsResponseSchema.model_validate_json(response.text)
+
+    async def create_review_comment(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            body: str,
+            commit_id: str,
+            path: str,
+            line: int,
+    ) -> GitHubCreateReviewCommentResponseSchema:
+        request = GitHubCreateReviewCommentRequestSchema(
+            body=body, commit_id=commit_id, path=path, line=line
+        )
+        response = await self.create_review_comment_api(owner, repo, pull_number, request)
+        return GitHubCreateReviewCommentResponseSchema.model_validate_json(response.text)
+
+    async def create_issue_comment(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str,
+            body: str,
+    ) -> GitHubCreateIssueCommentResponseSchema:
+        request = GitHubCreateIssueCommentRequestSchema(body=body)
+        response = await self.create_issue_comment_api(owner, repo, issue_number, request)
+        return GitHubCreateIssueCommentResponseSchema.model_validate_json(response.text)

ai_review/clients/github/pr/schema/comments.py

@@ -0,0 +1,33 @@
+from pydantic import BaseModel, RootModel
+
+
+class GitHubPRCommentSchema(BaseModel):
+    id: int
+    body: str
+    path: str | None = None
+    line: int | None = None
+
+
+class GitHubGetPRCommentsResponseSchema(RootModel[list[GitHubPRCommentSchema]]):
+    root: list[GitHubPRCommentSchema]
+
+
+class GitHubCreateIssueCommentRequestSchema(BaseModel):
+    body: str
+
+
+class GitHubCreateIssueCommentResponseSchema(BaseModel):
+    id: int
+    body: str
+
+
+class GitHubCreateReviewCommentRequestSchema(BaseModel):
+    body: str
+    path: str
+    line: int
+    commit_id: str
+
+
+class GitHubCreateReviewCommentResponseSchema(BaseModel):
+    id: int
+    body: str

ai_review/clients/github/pr/schema/files.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel, RootModel
+
+
+class GitHubPRFileSchema(BaseModel):
+    sha: str
+    patch: str | None = None
+    status: str
+    filename: str
+
+
+class GitHubGetPRFilesResponseSchema(RootModel[list[GitHubPRFileSchema]]):
+    root: list[GitHubPRFileSchema]

ai_review/clients/github/pr/schema/pull_request.py

@@ -0,0 +1,30 @@
+from pydantic import BaseModel
+
+
+class GitHubUserSchema(BaseModel):
+    id: int
+    login: str
+
+
+class GitHubLabelSchema(BaseModel):
+    id: int
+    name: str
+
+
+class GitHubBranchSchema(BaseModel):
+    ref: str
+    sha: str
+    label: str
+
+
+class GitHubGetPRResponseSchema(BaseModel):
+    id: int
+    number: int
+    title: str
+    body: str | None = None
+    user: GitHubUserSchema
+    labels: list[GitHubLabelSchema]
+    assignees: list[GitHubUserSchema] = []
+    requested_reviewers: list[GitHubUserSchema] = []
+    base: GitHubBranchSchema
+    head: GitHubBranchSchema

ai_review/clients/github/pr/schema/reviews.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+from pydantic import BaseModel, RootModel
+
+
+class GitHubPRReviewSchema(BaseModel):
+    id: int
+    body: Optional[str] = None
+    state: str
+
+
+class GitHubGetPRReviewsResponseSchema(RootModel[list[GitHubPRReviewSchema]]):
+    root: list[GitHubPRReviewSchema]

ai_review/libs/config/github.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel
+
+from ai_review.libs.config.http import HTTPClientConfig
+
+
+class GitHubPipelineConfig(BaseModel):
+    repo: str
+    owner: str
+    pull_number: str
+
+
+class GitHubHTTPClientConfig(HTTPClientConfig):
+    pass

ai_review/libs/config/prompt.py

@@ -8,6 +8,7 @@ from ai_review.libs.resources import load_resource
 
 class PromptConfig(BaseModel):
     context: dict[str, str] = Field(default_factory=dict)
+    normalize_prompts: bool = True
     context_placeholder: str = "<<{value}>>"
     inline_prompt_files: list[FilePath] | None = None
     context_prompt_files: list[FilePath] | None = None

ai_review/libs/config/vcs.py

@@ -2,6 +2,7 @@ from typing import Annotated, Literal
 
 from pydantic import BaseModel, Field
 
+from ai_review.libs.config.github import GitHubPipelineConfig, GitHubHTTPClientConfig
 from ai_review.libs.config.gitlab import GitLabPipelineConfig, GitLabHTTPClientConfig
 from ai_review.libs.constants.vcs_provider import VCSProvider
 
@@ -16,4 +17,10 @@ class GitLabVCSConfig(VCSConfigBase):
     http_client: GitLabHTTPClientConfig
 
 
-VCSConfig = Annotated[GitLabVCSConfig, Field(discriminator="provider")]
+class GitHubVCSConfig(VCSConfigBase):
+    provider: Literal[VCSProvider.GITHUB]
+    pipeline: GitHubPipelineConfig
+    http_client: GitHubHTTPClientConfig
+
+
+VCSConfig = Annotated[GitLabVCSConfig | GitHubVCSConfig, Field(discriminator="provider")]

ai_review/services/prompt/service.py

@@ -1,56 +1,57 @@
 from ai_review.config import settings
 from ai_review.services.diff.schema import DiffFileSchema
 from ai_review.services.prompt.schema import PromptContextSchema
+from ai_review.services.prompt.tools import normalize_prompt, format_file
 
 
-def format_file(diff: DiffFileSchema) -> str:
-    return f"# File: {diff.file}\n{diff.diff}\n"
+class PromptService:
+    @classmethod
+    def prepare_prompt(cls, prompts: list[str], context: PromptContextSchema) -> str:
+        prompt = "\n\n".join(prompts)
+        prompt = context.apply_format(prompt)
 
+        if settings.prompt.normalize_prompts:
+            prompt = normalize_prompt(prompt)
+
+        return prompt
 
-class PromptService:
     @classmethod
     def build_inline_request(cls, diff: DiffFileSchema, context: PromptContextSchema) -> str:
-        inline_prompts = "\n\n".join(settings.prompt.load_inline())
-        inline_prompts = context.apply_format(inline_prompts)
+        prompt = cls.prepare_prompt(settings.prompt.load_inline(), context)
         return (
-            f"{inline_prompts}\n\n"
+            f"{prompt}\n\n"
             f"## Diff\n\n"
             f"{format_file(diff)}"
         )
 
     @classmethod
     def build_summary_request(cls, diffs: list[DiffFileSchema], context: PromptContextSchema) -> str:
+        prompt = cls.prepare_prompt(settings.prompt.load_summary(), context)
         changes = "\n\n".join(map(format_file, diffs))
-        summary_prompts = "\n\n".join(settings.prompt.load_summary())
-        summary_prompts = context.apply_format(summary_prompts)
         return (
-            f"{summary_prompts}\n\n"
+            f"{prompt}\n\n"
             f"## Changes\n\n"
             f"{changes}\n"
         )
 
     @classmethod
     def build_context_request(cls, diffs: list[DiffFileSchema], context: PromptContextSchema) -> str:
+        prompt = cls.prepare_prompt(settings.prompt.load_context(), context)
         changes = "\n\n".join(map(format_file, diffs))
-        inline_prompts = "\n\n".join(settings.prompt.load_context())
-        inline_prompts = context.apply_format(inline_prompts)
         return (
-            f"{inline_prompts}\n\n"
+            f"{prompt}\n\n"
             f"## Diff\n\n"
             f"{changes}\n"
         )
 
     @classmethod
     def build_system_inline_request(cls, context: PromptContextSchema) -> str:
-        prompt = "\n\n".join(settings.prompt.load_system_inline())
-        return context.apply_format(prompt)
+        return cls.prepare_prompt(settings.prompt.load_system_inline(), context)
 
     @classmethod
     def build_system_context_request(cls, context: PromptContextSchema) -> str:
-        prompt = "\n\n".join(settings.prompt.load_system_context())
-        return context.apply_format(prompt)
+        return cls.prepare_prompt(settings.prompt.load_system_context(), context)
 
     @classmethod
     def build_system_summary_request(cls, context: PromptContextSchema) -> str:
-        prompt = "\n\n".join(settings.prompt.load_system_summary())
-        return context.apply_format(prompt)
+        return cls.prepare_prompt(settings.prompt.load_system_summary(), context)

ai_review/services/prompt/tools.py

@@ -0,0 +1,24 @@
+import re
+
+from ai_review.libs.logger import get_logger
+from ai_review.services.diff.schema import DiffFileSchema
+
+logger = get_logger("PROMPT_TOOLS")
+
+
+def format_file(diff: DiffFileSchema) -> str:
+    return f"# File: {diff.file}\n{diff.diff}\n"
+
+
+def normalize_prompt(text: str) -> str:
+    tails_stripped = [re.sub(r"[ \t]+$", "", line) for line in text.splitlines()]
+    text = "\n".join(tails_stripped)
+
+    text = re.sub(r"\n{3,}", "\n\n", text)
+
+    result = text.strip()
+    if len(text) > len(result):
+        logger.info(f"Prompt has been normalized from {len(text)} to {len(result)}")
+        return result
+
+    return text

ai_review/services/vcs/factory.py

@@ -1,5 +1,6 @@
 from ai_review.config import settings
 from ai_review.libs.constants.vcs_provider import VCSProvider
+from ai_review.services.vcs.github.client import GitHubVCSClient
 from ai_review.services.vcs.gitlab.client import GitLabVCSClient
 from ai_review.services.vcs.types import VCSClient
 
@@ -8,5 +9,7 @@ def get_vcs_client() -> VCSClient:
     match settings.vcs.provider:
         case VCSProvider.GITLAB:
             return GitLabVCSClient()
+        case VCSProvider.GITHUB:
+            return GitHubVCSClient()
         case _:
             raise ValueError(f"Unsupported provider: {settings.llm.provider}")

ai_review/services/vcs/github/client.py

@@ -0,0 +1,147 @@
+from ai_review.clients.github.client import get_github_http_client
+from ai_review.config import settings
+from ai_review.libs.logger import get_logger
+from ai_review.services.vcs.types import (
+    VCSClient,
+    MRNoteSchema,
+    MRUserSchema,
+    MRInfoSchema,
+    MRCommentSchema,
+    MRDiscussionSchema,
+)
+
+logger = get_logger("GITHUB_VCS_CLIENT")
+
+
+class GitHubVCSClient(VCSClient):
+    def __init__(self):
+        self.http_client = get_github_http_client()
+        self.owner = settings.vcs.pipeline.owner
+        self.repo = settings.vcs.pipeline.repo
+        self.pull_number = settings.vcs.pipeline.pull_number
+
+    async def get_mr_info(self) -> MRInfoSchema:
+        try:
+            pr = await self.http_client.pr.get_pull_request(
+                owner=self.owner, repo=self.repo, pull_number=self.pull_number
+            )
+            files = await self.http_client.pr.get_files(
+                owner=self.owner, repo=self.repo, pull_number=self.pull_number
+            )
+
+            logger.info(
+                f"Fetched PR info for {self.owner}/{self.repo}#{self.pull_number}"
+            )
+
+            return MRInfoSchema(
+                title=pr.title,
+                author=MRUserSchema(
+                    name=pr.user.login,
+                    username=pr.user.login,
+                ),
+                labels=[label.name for label in pr.labels],
+                base_sha=pr.base.sha,
+                head_sha=pr.head.sha,
+                assignees=[
+                    MRUserSchema(name=user.login, username=user.login)
+                    for user in pr.assignees
+                ],
+                reviewers=[
+                    MRUserSchema(name=user.login, username=user.login)
+                    for user in pr.requested_reviewers
+                ],
+                description=pr.body or "",
+                source_branch=pr.head.ref,
+                target_branch=pr.base.ref,
+                changed_files=[file.filename for file in files.root],
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch PR info {self.owner}/{self.repo}#{self.pull_number}: {error}"
+            )
+            return MRInfoSchema()
+
+    async def get_comments(self) -> list[MRCommentSchema]:
+        try:
+            response = await self.http_client.pr.get_issue_comments(
+                owner=self.owner,
+                repo=self.repo,
+                issue_number=self.pull_number,
+            )
+            logger.info(
+                f"Fetched issue comments for {self.owner}/{self.repo}#{self.pull_number}"
+            )
+
+            return [MRCommentSchema(id=comment.id, body=comment.body) for comment in response.root]
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch issue comments {self.owner}/{self.repo}#{self.pull_number}: {error}"
+            )
+            return []
+
+    async def get_discussions(self) -> list[MRDiscussionSchema]:
+        try:
+            response = await self.http_client.pr.get_review_comments(
+                owner=self.owner,
+                repo=self.repo,
+                pull_number=self.pull_number,
+            )
+            logger.info(
+                f"Fetched review comments for {self.owner}/{self.repo}#{self.pull_number}"
+            )
+
+            return [
+                MRDiscussionSchema(
+                    id=str(comment.id),
+                    notes=[MRNoteSchema(id=comment.id, body=comment.body or "")]
+                )
+                for comment in response.root
+            ]
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch review comments {self.owner}/{self.repo}#{self.pull_number}: {error}"
+            )
+            return []
+
+    async def create_comment(self, message: str) -> None:
+        try:
+            logger.info(
+                f"Posting general comment to PR {self.owner}/{self.repo}#{self.pull_number}: {message}"
+            )
+            await self.http_client.pr.create_issue_comment(
+                owner=self.owner,
+                repo=self.repo,
+                issue_number=self.pull_number,
+                body=message,
+            )
+            logger.info(
+                f"Created general comment in PR {self.owner}/{self.repo}#{self.pull_number}"
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to create general comment in PR {self.owner}/{self.repo}#{self.pull_number}: {error}"
+            )
+
+    async def create_discussion(self, file: str, line: int, message: str) -> None:
+        try:
+            pr = await self.http_client.pr.get_pull_request(
+                owner=self.owner, repo=self.repo, pull_number=self.pull_number
+            )
+            commit_id = pr.head.sha
+
+            await self.http_client.pr.create_review_comment(
+                owner=self.owner,
+                repo=self.repo,
+                pull_number=self.pull_number,
+                body=message,
+                path=file,
+                line=line,
+                commit_id=commit_id,
+            )
+            logger.info(
+                f"Created inline comment in {self.owner}/{self.repo}#{self.pull_number} at {file}:{line}"
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to create inline comment in {self.owner}/{self.repo}#{self.pull_number} at {file}:{line}: {error}"
+            )

ai_review/tests/suites/clients/github/test_client.py

@@ -0,0 +1,36 @@
+import pytest
+from httpx import AsyncClient
+from pydantic import HttpUrl, SecretStr
+
+from ai_review.clients.github.client import get_github_http_client, GitHubHTTPClient
+from ai_review.clients.github.pr.client import GitHubPullRequestsHTTPClient
+from ai_review.config import settings
+from ai_review.libs.config.github import GitHubPipelineConfig, GitHubHTTPClientConfig
+from ai_review.libs.config.vcs import GitHubVCSConfig
+from ai_review.libs.constants.vcs_provider import VCSProvider
+
+
+@pytest.fixture(autouse=True)
+def github_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = GitHubVCSConfig(
+        provider=VCSProvider.GITHUB,
+        pipeline=GitHubPipelineConfig(
+            repo="repo",
+            owner="owner",
+            pull_number="pull_number"
+        ),
+        http_client=GitHubHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://github.com"),
+            api_token=SecretStr("fake-token"),
+        )
+    )
+    monkeypatch.setattr(settings, "vcs", fake_config)
+
+
+def test_get_github_http_client_builds_ok():
+    github_http_client = get_github_http_client()
+
+    assert isinstance(github_http_client, GitHubHTTPClient)
+    assert isinstance(github_http_client.pr, GitHubPullRequestsHTTPClient)
+    assert isinstance(github_http_client.pr.client, AsyncClient)

ai_review/tests/suites/services/prompt/test_service.py

@@ -1,5 +1,6 @@
 import pytest
 
+from ai_review.config import settings
 from ai_review.libs.config.prompt import PromptConfig
 from ai_review.services.diff.schema import DiffFileSchema
 from ai_review.services.prompt.schema import PromptContextSchema
@@ -134,3 +135,35 @@ def test_diff_placeholders_are_not_replaced(dummy_context: PromptContextSchema)
 
     assert "<<merge_request_title>>" in result
     assert "Fix login bug" not in result
+
+
+def test_prepare_prompt_basic_substitution(dummy_context: PromptContextSchema) -> None:
+    prompts = ["Hello", "MR title: <<merge_request_title>>"]
+    result = PromptService.prepare_prompt(prompts, dummy_context)
+    assert "Hello" in result
+    assert "MR title: Fix login bug" in result
+
+
+def test_prepare_prompt_applies_normalization(
+        monkeypatch: pytest.MonkeyPatch,
+        dummy_context: PromptContextSchema
+) -> None:
+    monkeypatch.setattr(settings.prompt, "normalize_prompts", True)
+    prompts = ["Line with space   ", "", "", "Next line"]
+    result = PromptService.prepare_prompt(prompts, dummy_context)
+
+    assert "Line with space" in result
+    assert "Next line" in result
+    assert "\n\n\n" not in result
+
+
+def test_prepare_prompt_skips_normalization(
+        monkeypatch: pytest.MonkeyPatch,
+        dummy_context: PromptContextSchema
+) -> None:
+    monkeypatch.setattr(settings.prompt, "normalize_prompts", False)
+    prompts = ["Line with space   ", "", "", "Next line"]
+    result = PromptService.prepare_prompt(prompts, dummy_context)
+
+    assert "Line with space   " in result
+    assert "\n\n\n" in result

ai_review/tests/suites/services/prompt/test_tools.py

@@ -0,0 +1,72 @@
+from ai_review.services.diff.schema import DiffFileSchema
+from ai_review.services.prompt.tools import format_file, normalize_prompt
+
+
+def test_format_file_basic():
+    diff = DiffFileSchema(file="main.py", diff="+ print('hello')")
+    result = format_file(diff)
+    assert result == "# File: main.py\n+ print('hello')\n"
+
+
+def test_format_file_empty_diff():
+    diff = DiffFileSchema(file="empty.py", diff="")
+    result = format_file(diff)
+    assert result == "# File: empty.py\n\n"
+
+
+def test_format_file_multiline_diff():
+    diff = DiffFileSchema(
+        file="utils/helpers.py",
+        diff="- old line\n+ new line\n+ another line"
+    )
+    result = format_file(diff)
+    expected = (
+        "# File: utils/helpers.py\n"
+        "- old line\n"
+        "+ new line\n"
+        "+ another line\n"
+    )
+    assert result == expected
+
+
+def test_format_file_filename_with_path():
+    diff = DiffFileSchema(file="src/app/models/user.py", diff="+ class User:")
+    result = format_file(diff)
+    assert result.startswith("# File: src/app/models/user.py\n")
+    assert result.endswith("+ class User:\n")
+
+
+def test_trailing_spaces_are_removed():
+    text = "hello   \nworld\t\t"
+    result = normalize_prompt(text)
+    assert result == "hello\nworld"
+
+
+def test_multiple_empty_lines_collapsed():
+    text = "line1\n\n\n\nline2"
+    result = normalize_prompt(text)
+    assert result == "line1\n\nline2"
+
+
+def test_leading_and_trailing_whitespace_removed():
+    text = "\n\n   hello\nworld   \n\n"
+    result = normalize_prompt(text)
+    assert result == "hello\nworld"
+
+
+def test_internal_spaces_preserved():
+    text = "foo    bar\nbaz\t\tqux"
+    result = normalize_prompt(text)
+    assert result == "foo    bar\nbaz\t\tqux"
+
+
+def test_only_whitespace_string():
+    text = "   \n   \n"
+    result = normalize_prompt(text)
+    assert result == ""
+
+
+def test_no_changes_when_already_clean():
+    text = "line1\nline2"
+    result = normalize_prompt(text)
+    assert result == text

{xai_review-0.11.0.dist-info → xai_review-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.11.0
+Version: 0.13.0
 Summary: AI-powered code review tool
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>

{xai_review-0.11.0.dist-info → xai_review-0.13.0.dist-info}/RECORD

@@ -14,6 +14,15 @@ ai_review/clients/claude/schema.py,sha256=LE6KCjJKDXqBGU2Cno5XL5R8vUfScgskE9MqvE
 ai_review/clients/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/gemini/client.py,sha256=7ZPgqx77ER7gonxX0VoN4YrMpex3iBEQtd9Hi-bnDms,1780
 ai_review/clients/gemini/schema.py,sha256=5oVvbI-h_sw8bFreS4JUmMj-aXa_frvxK3H8sg4iJIA,2264
+ai_review/clients/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/clients/github/client.py,sha256=4uZsnMY-OZ9BNzMLqEHG80jgpyQdd61ePNdVuwGMcrI,1134
+ai_review/clients/github/pr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/clients/github/pr/client.py,sha256=Trv5MwmOOi5gAM8KHknbpf8NWNZ-Tnag-bi_74KIdu0,4678
+ai_review/clients/github/pr/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/clients/github/pr/schema/comments.py,sha256=K9KQ9TmWv9Hjw8uTrCPkzyAVbFohjCCf_rww6Ucj3wM,650
+ai_review/clients/github/pr/schema/files.py,sha256=mLHg1CfXUKCdQf5YUtuJ8n6xOROKoAjiJY5PL70kP-w,269
+ai_review/clients/github/pr/schema/pull_request.py,sha256=sdSvPgBkspW2DVO9GIyiqdhTngaVFFpYMCgcc5kFf8I,573
+ai_review/clients/github/pr/schema/reviews.py,sha256=v99DLYT5LOAcc18PATIse1mld8J0wKEAaTzUKI70s0c,288
 ai_review/clients/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/gitlab/client.py,sha256=acMflkHGp8mv0TVLdZ1gmdXkWQPcq609QjmkYWjEmys,1136
 ai_review/clients/gitlab/mr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -36,14 +45,15 @@ ai_review/libs/config/artifacts.py,sha256=8BzbQu5GxwV6i6qzrUKM1De1Ogb00Ph5WTqwZ3
 ai_review/libs/config/base.py,sha256=sPf3OKeF1ID0ouOwiVaUtvpWuZXJXQvIw5kbnPUyN9o,686
 ai_review/libs/config/claude.py,sha256=E9AJmszfY4TH8PkJjnDDDJYNAU9bLGsUThM3kriVA58,302
 ai_review/libs/config/gemini.py,sha256=sXHud43LWb4xTvhdkGQeHSLC7qvWl5LfU41fgcIVE5E,274
+ai_review/libs/config/github.py,sha256=1yFfvkTOt5ernIrxjqmiUKDpbEEHpa6lTpDiFQ5gVn4,238
 ai_review/libs/config/gitlab.py,sha256=VFvoVtni86tWky6Y34XCYdNrBuAtbgFFYGK3idPSOS4,234
 ai_review/libs/config/http.py,sha256=QsIj0yH1IYELOFBQ5AoqPZT0kGIIrQ19cxk1ozPRhLE,345
 ai_review/libs/config/llm.py,sha256=cK-e4NCQxnnixLATCsO8-r5k3zUWz1N0BdPCoqerORM,1824
 ai_review/libs/config/logger.py,sha256=oPmjpjf6EZwW7CgOjT8mOQdGnT98CLwXepiGB_ajZvU,384
 ai_review/libs/config/openai.py,sha256=vOYqhUq0ceEuNdQrQaHq44lVS5M648mB61Zc4YlfJVw,271
-ai_review/libs/config/prompt.py,sha256=e5jmHsfC6WpnkYZpTLT9TyKQfGtsbqJbxMkJBmLAWf0,4434
+ai_review/libs/config/prompt.py,sha256=8aO5WNnhVhQcpWzWxqzb9lq6PzormaJazVwPHuf_ia8,4469
 ai_review/libs/config/review.py,sha256=LEZni68iH_0m4URPfN0d3F6yrrK7KSn-BwXf-7w2al8,1058
-ai_review/libs/config/vcs.py,sha256=FduvJJkGObh2LgTSapaMB6FABIvjX7E63yTwZF4p8CU,517
+ai_review/libs/config/vcs.py,sha256=ULuLicuulFgG-_sTuDWsldyVWIT2SkiS8brPUU1svgk,778
 ai_review/libs/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/constants/llm_provider.py,sha256=sKnDLylCIIosYjq0-0r91LMiYJ4DlHVH2jeRDv_DlsQ,121
 ai_review/libs/constants/vcs_provider.py,sha256=mZMC8DWIDWQ1YeUZh1a1jduX5enOAe1rWeza0RBmpTY,99
@@ -98,7 +108,8 @@ ai_review/services/llm/openai/client.py,sha256=WhMXNfH_G1NTlFkdRK5sgYvrCIE5ZQNfP
 ai_review/services/prompt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/prompt/adapter.py,sha256=humGHLRVBu0JspeULgYHCs782BAy4YYKSf5yaG8aF24,1003
 ai_review/services/prompt/schema.py,sha256=erAecUYzOWyZfixt-pjmPSnvcMDh5DajMd1b7_SPm_o,2052
-ai_review/services/prompt/service.py,sha256=VsY8mj6UvY1a4Zsb8JlDJIg_8l7LBW6PXrObiHCwCzo,2128
+ai_review/services/prompt/service.py,sha256=D1PR2HC4cgrEND6mAhU5EPRAtp4mgEkOLEyD51WBc0g,2129
+ai_review/services/prompt/tools.py,sha256=-gS74mVM3OZPKWQkY9_QfStkfxaUfssDbJ3Bdi4AQ74,636
 ai_review/services/review/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/review/service.py,sha256=8YhRFqhZAk2pAnkDaytKSCENlOeOti1brAJq3R9tVMY,8394
 ai_review/services/review/inline/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -110,8 +121,10 @@ ai_review/services/review/summary/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 ai_review/services/review/summary/schema.py,sha256=GipVNWrEKtgZPkytNSrXwzvX9Zq8Pv2wxjXhfJq4D3g,364
 ai_review/services/review/summary/service.py,sha256=GB7-l4UyjZfUe6yP_8Q-SD1_uDKHM0W-CZJVMiEL8S0,449
 ai_review/services/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/services/vcs/factory.py,sha256=_xp8l1Pn2_yXi5xX1v13SROT3rRbVzFHXP1eTZxVcxI,451
+ai_review/services/vcs/factory.py,sha256=na1AOXgL9oUHqGIdRwT73BofxnkXEFnDr7fL3Sk_hkw,586
 ai_review/services/vcs/types.py,sha256=o3CJ8bZJ8unB9AKSpS66NwPVkFkweV4R02nCYsNqCko,1270
+ai_review/services/vcs/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/services/vcs/github/client.py,sha256=EfFa6DwQ527sfUmq0RWTc-y3t1I2GRRNRVG4A0U-xgY,5477
 ai_review/services/vcs/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/vcs/gitlab/client.py,sha256=-ZZFFlB7vv2DgEYAU016FP4CcYO8hp5LY1E2xokuCmU,6140
 ai_review/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -125,6 +138,8 @@ ai_review/tests/suites/clients/claude/test_schema.py,sha256=MUZXvEROgLNpUVHfCsH5
 ai_review/tests/suites/clients/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/gemini/test_client.py,sha256=6hxpK7r7iZVbOzAffRNDJnA63-3Zxvqw5ynANPhBhBg,1066
 ai_review/tests/suites/clients/gemini/test_schema.py,sha256=88dU28m7sEWvGx6tqYl7if7weWYuVc8erlkFkKKI3bk,3109
+ai_review/tests/suites/clients/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/clients/github/test_client.py,sha256=iCFQRaDPNota21No5SaCwMvWRW4VTJu_MgmeCC4Dk2o,1328
 ai_review/tests/suites/clients/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/gitlab/test_client.py,sha256=vXN7UZLC2yc7P7GZftpVvvUDycqR231ZFnfHZk97VLY,1325
 ai_review/tests/suites/clients/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -147,7 +162,8 @@ ai_review/tests/suites/services/diff/test_service.py,sha256=iFkGX9Vj2X44JU3eFsoH
 ai_review/tests/suites/services/diff/test_tools.py,sha256=HBQ3eCn-kLeb_k5iTgyr09x0VwLYSegSbxm0Qk9ZrCc,3543
 ai_review/tests/suites/services/prompt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/prompt/test_schema.py,sha256=XkJk4N9ovgod7G3i6oZwRBjpd71sv0vtVDJhSOfIwGA,2660
-ai_review/tests/suites/services/prompt/test_service.py,sha256=M8vvBhEbyHnXCSiIRu7231odn89sPDyCiRMOc2XufC4,5570
+ai_review/tests/suites/services/prompt/test_service.py,sha256=plJ8xDnBifCrLtHJO00cdl11U1EsqSw7lBrEGxu0AIw,6752
+ai_review/tests/suites/services/prompt/test_tools.py,sha256=_yNZoBATvPU5enWNIopbjY8lVVjfaB_46kNIKODhCW4,1981
 ai_review/tests/suites/services/review/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/review/inline/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/review/inline/test_schema.py,sha256=tIz-1UA_GgwcdsyUqgrodiiVVmd_jhoOVmtEwzRVWiY,2474
@@ -157,9 +173,9 @@ ai_review/tests/suites/services/review/policy/test_service.py,sha256=kRWT550OjWY
 ai_review/tests/suites/services/review/summary/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/review/summary/test_schema.py,sha256=xSoydvABZldHaVDa0OBFvYrj8wMuZqUDN3MO-XdvxOI,819
 ai_review/tests/suites/services/review/summary/test_service.py,sha256=8UMvi_NL9frm280vD6Q1NCDrdI7K8YbXzoViIus-I2g,541
-xai_review-0.11.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
-xai_review-0.11.0.dist-info/METADATA,sha256=HSYx4FRillzjL5CfB97QZZTxnCOlLRoRfDMg_Kufa4w,9618
-xai_review-0.11.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xai_review-0.11.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
-xai_review-0.11.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
-xai_review-0.11.0.dist-info/RECORD,,
+xai_review-0.13.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
+xai_review-0.13.0.dist-info/METADATA,sha256=mUB-mJITaTeO1i_y4DHOTFl1MyTxsqX6vayu4iBLZZM,9618
+xai_review-0.13.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xai_review-0.13.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
+xai_review-0.13.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
+xai_review-0.13.0.dist-info/RECORD,,