xai-review 0.21.0__tar.gz → 0.23.0__tar.gz

{xai_review-0.21.0/xai_review.egg-info → xai_review-0.23.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.21.0
+Version: 0.23.0
 Summary: AI-powered code review tool
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>
@@ -209,7 +209,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.21.0
+      - uses: Nikita-Filonov/ai-review@v0.23.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

{xai_review-0.21.0 → xai_review-0.23.0}/README.md

@@ -175,7 +175,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.21.0
+      - uses: Nikita-Filonov/ai-review@v0.23.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/claude/client.py

@@ -26,7 +26,7 @@ class ClaudeHTTPClient(HTTPClient):
 def get_claude_http_client() -> ClaudeHTTPClient:
     logger = get_logger("CLAUDE_HTTP_CLIENT")
     logger_event_hook = LoggerEventHook(logger=logger)
-    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
 
     client = AsyncClient(
         timeout=settings.llm.http_client.timeout,

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/gemini/client.py

@@ -29,7 +29,7 @@ class GeminiHTTPClient(HTTPClient):
 def get_gemini_http_client() -> GeminiHTTPClient:
     logger = get_logger("GEMINI_HTTP_CLIENT")
     logger_event_hook = LoggerEventHook(logger=logger)
-    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
 
     client = AsyncClient(
         timeout=settings.llm.http_client.timeout,

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/github/client.py

@@ -15,7 +15,7 @@ class GitHubHTTPClient:
 def get_github_http_client() -> GitHubHTTPClient:
     logger = get_logger("GITHUB_HTTP_CLIENT")
     logger_event_hook = LoggerEventHook(logger=logger)
-    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
 
     client = AsyncClient(
         timeout=settings.llm.http_client.timeout,

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/github/pr/client.py

@@ -1,15 +1,23 @@
-from httpx import Response
+from httpx import Response, QueryParams
 
 from ai_review.clients.github.pr.schema.comments import (
+    GitHubGetPRCommentsQuerySchema,
     GitHubGetPRCommentsResponseSchema,
     GitHubCreateIssueCommentRequestSchema,
     GitHubCreateIssueCommentResponseSchema,
     GitHubCreateReviewCommentRequestSchema,
     GitHubCreateReviewCommentResponseSchema
 )
-from ai_review.clients.github.pr.schema.files import GitHubGetPRFilesResponseSchema
+from ai_review.clients.github.pr.schema.files import (
+    GitHubGetPRFilesQuerySchema,
+    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.clients.github.pr.schema.reviews import (
+    GitHubGetPRReviewsQuerySchema,
+    GitHubGetPRReviewsResponseSchema
+)
+from ai_review.clients.github.pr.types import GitHubPullRequestsHTTPClientProtocol
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.handlers import HTTPClientError, handle_http_error
 
@@ -18,22 +26,49 @@ class GitHubPullRequestsHTTPClientError(HTTPClientError):
     pass
 
 
-class GitHubPullRequestsHTTPClient(HTTPClient):
+class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProtocol):
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
     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}")
 
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
-    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_files_api(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            query: GitHubGetPRFilesQuerySchema
+    ) -> Response:
+        return await self.get(
+            f"/repos/{owner}/{repo}/pulls/{pull_number}/files",
+            query=QueryParams(**query.model_dump())
+        )
 
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
-    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_issue_comments_api(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str,
+            query: GitHubGetPRCommentsQuerySchema,
+    ) -> Response:
+        return await self.get(
+            f"/repos/{owner}/{repo}/issues/{issue_number}/comments",
+            query=QueryParams(**query.model_dump())
+        )
 
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
-    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 get_review_comments_api(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            query: GitHubGetPRCommentsQuerySchema,
+    ) -> Response:
+        return await self.get(
+            f"/repos/{owner}/{repo}/pulls/{pull_number}/comments",
+            query=QueryParams(**query.model_dump())
+        )
 
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
     async def create_review_comment_api(
@@ -62,27 +97,40 @@ class GitHubPullRequestsHTTPClient(HTTPClient):
         )
 
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
-    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_reviews_api(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            query: GitHubGetPRReviewsQuerySchema
+    ) -> Response:
+        return await self.get(
+            f"/repos/{owner}/{repo}/pulls/{pull_number}/reviews",
+            query=QueryParams(**query.model_dump())
+        )
 
     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)
+        query = GitHubGetPRFilesQuerySchema(per_page=100)
+        response = await self.get_files_api(owner, repo, pull_number, query)
         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)
+        query = GitHubGetPRCommentsQuerySchema(per_page=100)
+        response = await self.get_issue_comments_api(owner, repo, issue_number, query)
         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)
+        query = GitHubGetPRCommentsQuerySchema(per_page=100)
+        response = await self.get_review_comments_api(owner, repo, pull_number, query)
         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)
+        query = GitHubGetPRReviewsQuerySchema(per_page=100)
+        response = await self.get_reviews_api(owner, repo, pull_number, query)
         return GitHubGetPRReviewsResponseSchema.model_validate_json(response.text)
 
     async def create_review_comment(

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/github/pr/schema/comments.py

@@ -8,6 +8,10 @@ class GitHubPRCommentSchema(BaseModel):
     line: int | None = None
 
 
+class GitHubGetPRCommentsQuerySchema(BaseModel):
+    per_page: int
+
+
 class GitHubGetPRCommentsResponseSchema(RootModel[list[GitHubPRCommentSchema]]):
     root: list[GitHubPRCommentSchema]
 

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/github/pr/schema/files.py

@@ -8,5 +8,9 @@ class GitHubPRFileSchema(BaseModel):
     filename: str
 
 
+class GitHubGetPRFilesQuerySchema(BaseModel):
+    per_page: int
+
+
 class GitHubGetPRFilesResponseSchema(RootModel[list[GitHubPRFileSchema]]):
     root: list[GitHubPRFileSchema]

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/github/pr/schema/reviews.py

@@ -9,5 +9,9 @@ class GitHubPRReviewSchema(BaseModel):
     state: str
 
 
+class GitHubGetPRReviewsQuerySchema(BaseModel):
+    per_page: int
+
+
 class GitHubGetPRReviewsResponseSchema(RootModel[list[GitHubPRReviewSchema]]):
     root: list[GitHubPRReviewSchema]

xai_review-0.23.0/ai_review/clients/github/pr/types.py

@@ -0,0 +1,49 @@
+from typing import Protocol
+
+from ai_review.clients.github.pr.schema.comments import (
+    GitHubGetPRCommentsResponseSchema,
+    GitHubCreateIssueCommentResponseSchema,
+    GitHubCreateReviewCommentResponseSchema,
+    GitHubCreateReviewCommentRequestSchema,
+)
+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
+
+
+class GitHubPullRequestsHTTPClientProtocol(Protocol):
+    async def get_pull_request(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRResponseSchema: ...
+
+    async def get_files(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRFilesResponseSchema: ...
+
+    async def get_issue_comments(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str
+    ) -> GitHubGetPRCommentsResponseSchema: ...
+
+    async def get_review_comments(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str
+    ) -> GitHubGetPRCommentsResponseSchema: ...
+
+    async def get_reviews(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRReviewsResponseSchema: ...
+
+    async def create_review_comment(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            request: GitHubCreateReviewCommentRequestSchema,
+    ) -> GitHubCreateReviewCommentResponseSchema: ...
+
+    async def create_issue_comment(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str,
+            body: str,
+    ) -> GitHubCreateIssueCommentResponseSchema: ...

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/gitlab/client.py

@@ -15,7 +15,7 @@ class GitLabHTTPClient:
 def get_gitlab_http_client() -> GitLabHTTPClient:
     logger = get_logger("GITLAB_HTTP_CLIENT")
     logger_event_hook = LoggerEventHook(logger=logger)
-    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
 
     client = AsyncClient(
         timeout=settings.llm.http_client.timeout,

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/gitlab/mr/client.py

@@ -1,16 +1,19 @@
-from httpx import Response
+from httpx import Response, QueryParams
 
 from ai_review.clients.gitlab.mr.schema.changes import GitLabGetMRChangesResponseSchema
 from ai_review.clients.gitlab.mr.schema.discussions import (
+    GitLabGetMRDiscussionsQuerySchema,
     GitLabGetMRDiscussionsResponseSchema,
     GitLabCreateMRDiscussionRequestSchema,
     GitLabCreateMRDiscussionResponseSchema
 )
 from ai_review.clients.gitlab.mr.schema.notes import (
+    GitLabGetMRNotesQuerySchema,
     GitLabGetMRNotesResponseSchema,
     GitLabCreateMRNoteRequestSchema,
     GitLabCreateMRNoteResponseSchema,
 )
+from ai_review.clients.gitlab.mr.types import GitLabMergeRequestsHTTPClientProtocol
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.handlers import handle_http_error, HTTPClientError
 
@@ -19,7 +22,7 @@ class GitLabMergeRequestsHTTPClientError(HTTPClientError):
     pass
 
 
-class GitLabMergeRequestsHTTPClient(HTTPClient):
+class GitLabMergeRequestsHTTPClient(HTTPClient, GitLabMergeRequestsHTTPClientProtocol):
     @handle_http_error(client="GitLabMergeRequestsHTTPClient", exception=GitLabMergeRequestsHTTPClientError)
     async def get_changes_api(self, project_id: str, merge_request_id: str) -> Response:
         return await self.get(
@@ -27,15 +30,27 @@ class GitLabMergeRequestsHTTPClient(HTTPClient):
         )
 
     @handle_http_error(client="GitLabMergeRequestsHTTPClient", exception=GitLabMergeRequestsHTTPClientError)
-    async def get_notes_api(self, project_id: str, merge_request_id: str) -> Response:
+    async def get_notes_api(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            query: GitLabGetMRNotesQuerySchema
+    ) -> Response:
         return await self.get(
-            f"/api/v4/projects/{project_id}/merge_requests/{merge_request_id}/notes"
+            f"/api/v4/projects/{project_id}/merge_requests/{merge_request_id}/notes",
+            query=QueryParams(**query.model_dump())
         )
 
     @handle_http_error(client="GitLabMergeRequestsHTTPClient", exception=GitLabMergeRequestsHTTPClientError)
-    async def get_discussions_api(self, project_id: str, merge_request_id: str) -> Response:
+    async def get_discussions_api(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            query: GitLabGetMRDiscussionsQuerySchema
+    ) -> Response:
         return await self.get(
-            f"/api/v4/projects/{project_id}/merge_requests/{merge_request_id}/discussions"
+            f"/api/v4/projects/{project_id}/merge_requests/{merge_request_id}/discussions",
+            query=QueryParams(**query.model_dump())
         )
 
     @handle_http_error(client="GitLabMergeRequestsHTTPClient", exception=GitLabMergeRequestsHTTPClientError)
@@ -71,7 +86,8 @@ class GitLabMergeRequestsHTTPClient(HTTPClient):
             project_id: str,
             merge_request_id: str
     ) -> GitLabGetMRNotesResponseSchema:
-        response = await self.get_notes_api(project_id, merge_request_id)
+        query = GitLabGetMRNotesQuerySchema(per_page=100)
+        response = await self.get_notes_api(project_id, merge_request_id, query)
         return GitLabGetMRNotesResponseSchema.model_validate_json(response.text)
 
     async def get_discussions(
@@ -79,7 +95,8 @@ class GitLabMergeRequestsHTTPClient(HTTPClient):
             project_id: str,
             merge_request_id: str
     ) -> GitLabGetMRDiscussionsResponseSchema:
-        response = await self.get_discussions_api(project_id, merge_request_id)
+        query = GitLabGetMRDiscussionsQuerySchema(per_page=100)
+        response = await self.get_discussions_api(project_id, merge_request_id, query)
         return GitLabGetMRDiscussionsResponseSchema.model_validate_json(response.text)
 
     async def create_note(

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/gitlab/mr/schema/discussions.py

@@ -17,6 +17,10 @@ class GitLabDiscussionPositionSchema(BaseModel):
     new_line: int
 
 
+class GitLabGetMRDiscussionsQuerySchema(BaseModel):
+    per_page: int
+
+
 class GitLabGetMRDiscussionsResponseSchema(RootModel[list[GitLabDiscussionSchema]]):
     root: list[GitLabDiscussionSchema]
 

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/gitlab/mr/schema/notes.py

@@ -6,6 +6,10 @@ class GitLabNoteSchema(BaseModel):
     body: str
 
 
+class GitLabGetMRNotesQuerySchema(BaseModel):
+    per_page: int
+
+
 class GitLabGetMRNotesResponseSchema(RootModel[list[GitLabNoteSchema]]):
     root: list[GitLabNoteSchema]
 

xai_review-0.23.0/ai_review/clients/gitlab/mr/types.py

@@ -0,0 +1,35 @@
+from typing import Protocol
+
+from ai_review.clients.gitlab.mr.schema.changes import GitLabGetMRChangesResponseSchema
+from ai_review.clients.gitlab.mr.schema.discussions import (
+    GitLabGetMRDiscussionsResponseSchema,
+    GitLabCreateMRDiscussionResponseSchema,
+    GitLabCreateMRDiscussionRequestSchema,
+)
+from ai_review.clients.gitlab.mr.schema.notes import GitLabGetMRNotesResponseSchema, GitLabCreateMRNoteResponseSchema
+
+
+class GitLabMergeRequestsHTTPClientProtocol(Protocol):
+    async def get_changes(self, project_id: str, merge_request_id: str) -> GitLabGetMRChangesResponseSchema: ...
+
+    async def get_notes(self, project_id: str, merge_request_id: str) -> GitLabGetMRNotesResponseSchema: ...
+
+    async def get_discussions(
+            self,
+            project_id: str,
+            merge_request_id: str
+    ) -> GitLabGetMRDiscussionsResponseSchema: ...
+
+    async def create_note(
+            self,
+            body: str,
+            project_id: str,
+            merge_request_id: str,
+    ) -> GitLabCreateMRNoteResponseSchema: ...
+
+    async def create_discussion(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            request: GitLabCreateMRDiscussionRequestSchema,
+    ) -> GitLabCreateMRDiscussionResponseSchema: ...

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/clients/openai/client.py

@@ -26,7 +26,7 @@ class OpenAIHTTPClient(HTTPClient):
 def get_openai_http_client() -> OpenAIHTTPClient:
     logger = get_logger("OPENAI_HTTP_CLIENT")
     logger_event_hook = LoggerEventHook(logger=logger)
-    retry_transport = RetryTransport(transport=AsyncHTTPTransport())
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
 
     client = AsyncClient(
         timeout=settings.llm.http_client.timeout,

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/config.py

@@ -12,6 +12,7 @@ from ai_review.libs.config.base import (
     get_yaml_config_file_or_default,
     get_json_config_file_or_default
 )
+from ai_review.libs.config.core import CoreConfig
 from ai_review.libs.config.llm import LLMConfig
 from ai_review.libs.config.logger import LoggerConfig
 from ai_review.libs.config.prompt import PromptConfig
@@ -36,6 +37,7 @@ class Settings(BaseSettings):
 
     llm: LLMConfig
     vcs: VCSConfig
+    core: CoreConfig = CoreConfig()
     prompt: PromptConfig = PromptConfig()
     review: ReviewConfig = ReviewConfig()
     logger: LoggerConfig = LoggerConfig()

xai_review-0.23.0/ai_review/libs/asynchronous/gather.py

@@ -0,0 +1,17 @@
+import asyncio
+from typing import Awaitable, Iterable, TypeVar
+
+from ai_review.config import settings
+
+T = TypeVar("T")
+
+
+async def bounded_gather(coroutines: Iterable[Awaitable[T]]) -> tuple[T, ...]:
+    sem = asyncio.Semaphore(settings.core.concurrency)
+
+    async def wrap(coro: Awaitable[T]) -> T:
+        async with sem:
+            return await coro
+
+    results = await asyncio.gather(*(wrap(coroutine) for coroutine in coroutines), return_exceptions=True)
+    return tuple(results)

xai_review-0.23.0/ai_review/libs/config/core.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class CoreConfig(BaseModel):
+    concurrency: int = 7

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/libs/http/event_hooks/logger.py

@@ -1,12 +1,15 @@
-from logging import Logger
+from typing import TYPE_CHECKING
 
 from httpx import Request, Response
 
 from ai_review.libs.http.event_hooks.base import BaseEventHook
 
+if TYPE_CHECKING:
+    from loguru import Logger
+
 
 class LoggerEventHook(BaseEventHook):
-    def __init__(self, logger: Logger):
+    def __init__(self, logger: "Logger"):
         self.logger = logger
 
     async def request(self, request: Request):

xai_review-0.23.0/ai_review/libs/http/transports/retry.py

@@ -0,0 +1,51 @@
+import asyncio
+from http import HTTPStatus
+from typing import TYPE_CHECKING
+
+from httpx import Request, Response, AsyncBaseTransport
+
+if TYPE_CHECKING:
+    from loguru import Logger
+
+
+class RetryTransport(AsyncBaseTransport):
+    def __init__(
+            self,
+            logger: "Logger",
+            transport: AsyncBaseTransport,
+            max_retries: int = 5,
+            retry_delay: float = 0.5,
+            retry_status_codes: tuple[HTTPStatus, ...] = (
+                    HTTPStatus.BAD_GATEWAY,
+                    HTTPStatus.GATEWAY_TIMEOUT,
+                    HTTPStatus.SERVICE_UNAVAILABLE,
+                    HTTPStatus.INTERNAL_SERVER_ERROR,
+            )
+    ):
+        self.logger = logger
+        self.transport = transport
+        self.max_retries = max_retries
+        self.retry_delay = retry_delay
+        self.retry_status_codes = retry_status_codes
+
+    async def handle_async_request(self, request: Request) -> Response:
+        last_response: Response | None = None
+        for attempt in range(self.max_retries):
+            last_response = await self.transport.handle_async_request(request)
+            if last_response.status_code not in self.retry_status_codes:
+                return last_response
+
+            self.logger.warning(
+                f"Attempt {attempt}/{self.max_retries} failed "
+                f"with status={last_response.status_code} for {request.method} {request.url}. "
+                f"Retrying in {self.retry_delay:.1f}s..."
+            )
+
+            await asyncio.sleep(self.retry_delay)
+
+        self.logger.error(
+            f"All {self.max_retries} attempts failed for "
+            f"{request.method} {request.url} (last status={last_response.status_code})"
+        )
+
+        return last_response

{xai_review-0.21.0 → xai_review-0.23.0}/ai_review/services/diff/renderers.py

@@ -116,14 +116,14 @@ def render_unified(
     added_new_positions = file.added_line_numbers()
     removed_old_positions = file.removed_line_numbers()
 
-    def in_context(old_no: int | None, new_no: int | None) -> bool:
+    def in_context(inner_old_no: int | None, inner_new_no: int | None) -> bool:
         """Check if an unchanged line falls within context radius."""
         if context <= 0:
             return False
-        if include_added and new_no is not None:
+        if include_added and inner_new_no is not None:
             if any(abs(new_no - a) <= context for a in added_new_positions):
                 return True
-        if include_removed and old_no is not None:
+        if include_removed and inner_old_no is not None:
             if any(abs(old_no - r) <= context for r in removed_old_positions):
                 return True
         return False

xai_review-0.23.0/ai_review/services/prompt/schema.py

@@ -0,0 +1,41 @@
+from pydantic import BaseModel, Field, field_serializer
+
+from ai_review.config import settings
+from ai_review.libs.template.render import render_template
+
+
+class PromptContextSchema(BaseModel):
+    review_title: str = ""
+    review_description: str = ""
+
+    review_author_name: str = ""
+    review_author_username: str = ""
+
+    review_reviewer: str = ""
+    review_reviewers: list[str] = Field(default_factory=list)
+    review_reviewers_usernames: list[str] = Field(default_factory=list)
+
+    review_assignees: list[str] = Field(default_factory=list)
+    review_assignees_usernames: list[str] = Field(default_factory=list)
+
+    source_branch: str = ""
+    target_branch: str = ""
+
+    labels: list[str] = Field(default_factory=list)
+    changed_files: list[str] = Field(default_factory=list)
+
+    @field_serializer(
+        "review_reviewers",
+        "review_reviewers_usernames",
+        "review_assignees",
+        "review_assignees_usernames",
+        "labels",
+        "changed_files",
+        when_used="always"
+    )
+    def list_of_strings_serializer(self, value: list[str]) -> str:
+        return ", ".join(value)
+
+    def apply_format(self, prompt: str) -> str:
+        values = {**self.model_dump(), **settings.prompt.context}
+        return render_template(prompt, values, settings.prompt.context_placeholder)

xai_review-0.23.0/ai_review/tests/fixtures/clients/claude.py

@@ -0,0 +1,22 @@
+import pytest
+from pydantic import HttpUrl, SecretStr
+
+from ai_review.config import settings
+from ai_review.libs.config.claude import ClaudeMetaConfig, ClaudeHTTPClientConfig
+from ai_review.libs.config.llm import ClaudeLLMConfig
+from ai_review.libs.constants.llm_provider import LLMProvider
+
+
+@pytest.fixture
+def claude_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = ClaudeLLMConfig(
+        meta=ClaudeMetaConfig(),
+        provider=LLMProvider.CLAUDE,
+        http_client=ClaudeHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://api.anthropic.com"),
+            api_token=SecretStr("fake-token"),
+            api_version="2023-06-01",
+        )
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)

xai_review-0.21.0/ai_review/tests/suites/clients/gemini/test_client.py → xai_review-0.23.0/ai_review/tests/fixtures/clients/gemini.py

@@ -1,16 +1,14 @@
 import pytest
-from httpx import AsyncClient
 from pydantic import HttpUrl, SecretStr
 
-from ai_review.clients.gemini.client import get_gemini_http_client, GeminiHTTPClient
 from ai_review.config import settings
 from ai_review.libs.config.gemini import GeminiMetaConfig, GeminiHTTPClientConfig
 from ai_review.libs.config.llm import GeminiLLMConfig
 from ai_review.libs.constants.llm_provider import LLMProvider
 
 
-@pytest.fixture(autouse=True)
-def gemini_http_client_config(monkeypatch):
+@pytest.fixture
+def gemini_http_client_config(monkeypatch: pytest.MonkeyPatch):
     fake_config = GeminiLLMConfig(
         meta=GeminiMetaConfig(),
         provider=LLMProvider.GEMINI,
@@ -21,10 +19,3 @@ def gemini_http_client_config(monkeypatch):
         )
     )
     monkeypatch.setattr(settings, "llm", fake_config)
-
-
-def test_get_gemini_http_client_builds_ok():
-    gemini_http_client = get_gemini_http_client()
-
-    assert isinstance(gemini_http_client, GeminiHTTPClient)
-    assert isinstance(gemini_http_client.client, AsyncClient)