xai-review 0.26.0__py3-none-any.whl → 0.28.0__py3-none-any.whl

ai_review/cli/commands/run_inline_reply_review.py

@@ -0,0 +1,7 @@
+from ai_review.services.review.service import ReviewService
+
+
+async def run_inline_reply_review_command():
+    review_service = ReviewService()
+    await review_service.run_inline_reply_review()
+    review_service.report_total_cost()

ai_review/cli/commands/run_summary_reply_review.py

@@ -0,0 +1,7 @@
+from ai_review.services.review.service import ReviewService
+
+
+async def run_summary_reply_review_command():
+    review_service = ReviewService()
+    await review_service.run_summary_reply_review()
+    review_service.report_total_cost()

ai_review/cli/main.py

@@ -3,8 +3,10 @@ import asyncio
 import typer
 
 from ai_review.cli.commands.run_context_review import run_context_review_command
+from ai_review.cli.commands.run_inline_reply_review import run_inline_reply_review_command
 from ai_review.cli.commands.run_inline_review import run_inline_review_command
 from ai_review.cli.commands.run_review import run_review_command
+from ai_review.cli.commands.run_summary_reply_review import run_summary_reply_review_command
 from ai_review.cli.commands.run_summary_review import run_summary_review_command
 from ai_review.config import settings
 
@@ -43,6 +45,21 @@ def run_summary():
     typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
 
 
+@app.command("run-inline-reply")
+def run_inline_reply():
+    """Run only the inline reply review"""
+    typer.secho("Starting inline reply AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_inline_reply_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-summary-reply")
+def run_summary_reply():
+    typer.secho("Starting summary reply AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_summary_reply_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
 @app.command("show-config")
 def show_config():
     """Show the current resolved configuration"""

ai_review/clients/bitbucket/pr/client.py

@@ -1,19 +1,24 @@
 from httpx import Response, QueryParams
 
 from ai_review.clients.bitbucket.pr.schema.comments import (
+    BitbucketPRCommentSchema,
     BitbucketGetPRCommentsQuerySchema,
     BitbucketGetPRCommentsResponseSchema,
     BitbucketCreatePRCommentRequestSchema,
     BitbucketCreatePRCommentResponseSchema,
 )
 from ai_review.clients.bitbucket.pr.schema.files import (
+    BitbucketPRFileSchema,
     BitbucketGetPRFilesQuerySchema,
     BitbucketGetPRFilesResponseSchema,
 )
 from ai_review.clients.bitbucket.pr.schema.pull_request import BitbucketGetPRResponseSchema
 from ai_review.clients.bitbucket.pr.types import BitbucketPullRequestsHTTPClientProtocol
+from ai_review.clients.bitbucket.tools import bitbucket_has_next_page
+from ai_review.config import settings
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.handlers import handle_http_error, HTTPClientError
+from ai_review.libs.http.paginate import paginate
 
 
 class BitbucketPullRequestsHTTPClientError(HTTPClientError):
@@ -35,7 +40,7 @@ class BitbucketPullRequestsHTTPClient(HTTPClient, BitbucketPullRequestsHTTPClien
     ) -> Response:
         return await self.get(
             f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}/diffstat",
-            query=QueryParams(**query.model_dump()),
+            query=QueryParams(**query.model_dump(by_alias=True)),
         )
 
     @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
@@ -48,7 +53,7 @@ class BitbucketPullRequestsHTTPClient(HTTPClient, BitbucketPullRequestsHTTPClien
     ) -> Response:
         return await self.get(
             f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}/comments",
-            query=QueryParams(**query.model_dump()),
+            query=QueryParams(**query.model_dump(by_alias=True)),
         )
 
     @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
@@ -79,9 +84,25 @@ class BitbucketPullRequestsHTTPClient(HTTPClient, BitbucketPullRequestsHTTPClien
             repo_slug: str,
             pull_request_id: str
     ) -> BitbucketGetPRFilesResponseSchema:
-        query = BitbucketGetPRFilesQuerySchema(pagelen=100)
-        resp = await self.get_diffstat_api(workspace, repo_slug, pull_request_id, query)
-        return BitbucketGetPRFilesResponseSchema.model_validate_json(resp.text)
+        async def fetch_page(page: int) -> Response:
+            query = BitbucketGetPRFilesQuerySchema(page=page, page_len=settings.vcs.pagination.per_page)
+            return await self.get_diffstat_api(workspace, repo_slug, pull_request_id, query)
+
+        def extract_items(response: Response) -> list[BitbucketPRFileSchema]:
+            result = BitbucketGetPRFilesResponseSchema.model_validate_json(response.text)
+            return result.values
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=bitbucket_has_next_page
+        )
+        return BitbucketGetPRFilesResponseSchema(
+            size=len(items),
+            values=items,
+            page_len=settings.vcs.pagination.per_page
+        )
 
     async def get_comments(
             self,
@@ -89,9 +110,25 @@ class BitbucketPullRequestsHTTPClient(HTTPClient, BitbucketPullRequestsHTTPClien
             repo_slug: str,
             pull_request_id: str
     ) -> BitbucketGetPRCommentsResponseSchema:
-        query = BitbucketGetPRCommentsQuerySchema(pagelen=100)
-        response = await self.get_comments_api(workspace, repo_slug, pull_request_id, query)
-        return BitbucketGetPRCommentsResponseSchema.model_validate_json(response.text)
+        async def fetch_page(page: int) -> Response:
+            query = BitbucketGetPRCommentsQuerySchema(page=page, page_len=settings.vcs.pagination.per_page)
+            return await self.get_comments_api(workspace, repo_slug, pull_request_id, query)
+
+        def extract_items(response: Response) -> list[BitbucketPRCommentSchema]:
+            result = BitbucketGetPRCommentsResponseSchema.model_validate_json(response.text)
+            return result.values
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=bitbucket_has_next_page
+        )
+        return BitbucketGetPRCommentsResponseSchema(
+            size=len(items),
+            values=items,
+            page_len=settings.vcs.pagination.per_page
+        )
 
     async def create_comment(
             self,

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

@@ -1,5 +1,7 @@
 from pydantic import BaseModel, Field, ConfigDict
 
+from ai_review.clients.bitbucket.pr.schema.user import BitbucketUserSchema
+
 
 class BitbucketCommentContentSchema(BaseModel):
     raw: str
@@ -15,30 +17,47 @@ class BitbucketCommentInlineSchema(BaseModel):
     from_line: int | None = Field(alias="from", default=None)
 
 
+class BitbucketCommentParentSchema(BaseModel):
+    id: int
+
+
 class BitbucketPRCommentSchema(BaseModel):
     id: int
+    user: BitbucketUserSchema | None = None
+    parent: BitbucketCommentParentSchema | None = None
     inline: BitbucketCommentInlineSchema | None = None
     content: BitbucketCommentContentSchema
 
 
 class BitbucketGetPRCommentsQuerySchema(BaseModel):
-    pagelen: int = 100
+    model_config = ConfigDict(populate_by_name=True)
+
+    page: int = 1
+    page_len: int = Field(alias="pagelen", default=100)
 
 
 class BitbucketGetPRCommentsResponseSchema(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
     size: int
     page: int | None = None
     next: str | None = None
     values: list[BitbucketPRCommentSchema]
-    pagelen: int
+    page_len: int = Field(alias="pagelen")
+
+
+class BitbucketParentSchema(BaseModel):
+    id: int
 
 
 class BitbucketCreatePRCommentRequestSchema(BaseModel):
+    parent: BitbucketParentSchema | None = None
     inline: BitbucketCommentInlineSchema | None = None
     content: BitbucketCommentContentSchema
 
 
 class BitbucketCreatePRCommentResponseSchema(BaseModel):
     id: int
+    parent: BitbucketParentSchema | None = None
     inline: BitbucketCommentInlineSchema | None = None
     content: BitbucketCommentContentSchema

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

@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, Field, ConfigDict
 
 
 class BitbucketPRFilePathSchema(BaseModel):
@@ -14,12 +14,17 @@ class BitbucketPRFileSchema(BaseModel):
 
 
 class BitbucketGetPRFilesQuerySchema(BaseModel):
-    pagelen: int = 100
+    model_config = ConfigDict(populate_by_name=True)
+
+    page: int = 1
+    page_len: int = Field(alias="pagelen", default=100)
 
 
 class BitbucketGetPRFilesResponseSchema(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
     size: int
     page: int | None = None
     next: str | None = None
     values: list[BitbucketPRFileSchema]
-    pagelen: int
+    page_len: int = Field(alias="pagelen")

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

@@ -1,10 +1,6 @@
 from pydantic import BaseModel, Field
 
-
-class BitbucketUserSchema(BaseModel):
-    uuid: str
-    nickname: str
-    display_name: str
+from ai_review.clients.bitbucket.pr.schema.user import BitbucketUserSchema
 
 
 class BitbucketBranchSchema(BaseModel):

ai_review/clients/bitbucket/pr/schema/user.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class BitbucketUserSchema(BaseModel):
+    uuid: str
+    nickname: str
+    display_name: str

ai_review/clients/bitbucket/tools.py

@@ -0,0 +1,6 @@
+from httpx import Response
+
+
+def bitbucket_has_next_page(response: Response) -> bool:
+    data = response.json()
+    return bool(data.get("next"))

ai_review/clients/github/pr/client.py

@@ -1,25 +1,34 @@
 from httpx import Response, QueryParams
 
 from ai_review.clients.github.pr.schema.comments import (
+    GitHubPRCommentSchema,
+    GitHubIssueCommentSchema,
     GitHubGetPRCommentsQuerySchema,
     GitHubGetPRCommentsResponseSchema,
+    GitHubGetIssueCommentsResponseSchema,
     GitHubCreateIssueCommentRequestSchema,
     GitHubCreateIssueCommentResponseSchema,
+    GitHubCreateReviewReplyRequestSchema,
     GitHubCreateReviewCommentRequestSchema,
     GitHubCreateReviewCommentResponseSchema
 )
 from ai_review.clients.github.pr.schema.files import (
+    GitHubPRFileSchema,
     GitHubGetPRFilesQuerySchema,
     GitHubGetPRFilesResponseSchema
 )
 from ai_review.clients.github.pr.schema.pull_request import GitHubGetPRResponseSchema
 from ai_review.clients.github.pr.schema.reviews import (
+    GitHubPRReviewSchema,
     GitHubGetPRReviewsQuerySchema,
     GitHubGetPRReviewsResponseSchema
 )
 from ai_review.clients.github.pr.types import GitHubPullRequestsHTTPClientProtocol
+from ai_review.clients.github.tools import github_has_next_page
+from ai_review.config import settings
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.handlers import HTTPClientError, handle_http_error
+from ai_review.libs.http.paginate import paginate
 
 
 class GitHubPullRequestsHTTPClientError(HTTPClientError):
@@ -70,6 +79,19 @@ class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProto
             query=QueryParams(**query.model_dump())
         )
 
+    @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
+    async def create_review_reply_api(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            request: GitHubCreateReviewReplyRequestSchema,
+    ) -> Response:
+        return await self.post(
+            f"/repos/{owner}/{repo}/pulls/{pull_number}/comments",
+            json=request.model_dump(),
+        )
+
     @handle_http_error(client="GitHubPullRequestsHTTPClient", exception=GitHubPullRequestsHTTPClientError)
     async def create_review_comment_api(
             self,
@@ -114,24 +136,87 @@ class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProto
         return GitHubGetPRResponseSchema.model_validate_json(response.text)
 
     async def get_files(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRFilesResponseSchema:
-        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 fetch_page(page: int) -> Response:
+            query = GitHubGetPRFilesQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_files_api(owner, repo, pull_number, query)
+
+        def extract_items(response: Response) -> list[GitHubPRFileSchema]:
+            result = GitHubGetPRFilesResponseSchema.model_validate_json(response.text)
+            return result.root
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=github_has_next_page
+        )
+        return GitHubGetPRFilesResponseSchema(root=items)
 
-    async def get_issue_comments(self, owner: str, repo: str, issue_number: str) -> GitHubGetPRCommentsResponseSchema:
-        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_issue_comments(
+            self,
+            owner: str,
+            repo: str,
+            issue_number: str
+    ) -> GitHubGetIssueCommentsResponseSchema:
+        async def fetch_page(page: int) -> Response:
+            query = GitHubGetPRCommentsQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_issue_comments_api(owner, repo, issue_number, query)
+
+        def extract_items(response: Response) -> list[GitHubIssueCommentSchema]:
+            result = GitHubGetIssueCommentsResponseSchema.model_validate_json(response.text)
+            return result.root
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=github_has_next_page
+        )
+        return GitHubGetIssueCommentsResponseSchema(root=items)
 
     async def get_review_comments(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRCommentsResponseSchema:
-        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 fetch_page(page: int) -> Response:
+            query = GitHubGetPRCommentsQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_review_comments_api(owner, repo, pull_number, query)
+
+        def extract_items(response: Response) -> list[GitHubPRCommentSchema]:
+            result = GitHubGetPRCommentsResponseSchema.model_validate_json(response.text)
+            return result.root
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=github_has_next_page
+        )
+        return GitHubGetPRCommentsResponseSchema(root=items)
 
     async def get_reviews(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRReviewsResponseSchema:
-        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 fetch_page(page: int) -> Response:
+            query = GitHubGetPRReviewsQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_reviews_api(owner, repo, pull_number, query)
+
+        def extract_items(response: Response) -> list[GitHubPRReviewSchema]:
+            result = GitHubGetPRReviewsResponseSchema.model_validate_json(response.text)
+            return result.root
+
+        items = await paginate(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            has_next_page=github_has_next_page
+        )
+        return GitHubGetPRReviewsResponseSchema(root=items)
+
+    async def create_review_reply(
+            self,
+            owner: str,
+            repo: str,
+            pull_number: str,
+            request: GitHubCreateReviewReplyRequestSchema,
+    ) -> GitHubCreateReviewCommentResponseSchema:
+        response = await self.create_review_reply_api(owner, repo, pull_number, request)
+        return GitHubCreateReviewCommentResponseSchema.model_validate_json(response.text)
 
     async def create_review_comment(
             self,

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

@@ -1,21 +1,38 @@
 from pydantic import BaseModel, RootModel
 
+from ai_review.clients.github.pr.schema.user import GitHubUserSchema
+
+
+class GitHubIssueCommentSchema(BaseModel):
+    """Represents a top-level comment in a PR discussion (issue-level)."""
+    id: int
+    body: str
+    user: GitHubUserSchema | None = None
+
 
 class GitHubPRCommentSchema(BaseModel):
+    """Represents an inline code review comment on a specific line in a PR."""
     id: int
     body: str
     path: str | None = None
     line: int | None = None
+    user: GitHubUserSchema | None = None
+    in_reply_to_id: int | None = None
 
 
 class GitHubGetPRCommentsQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitHubGetPRCommentsResponseSchema(RootModel[list[GitHubPRCommentSchema]]):
     root: list[GitHubPRCommentSchema]
 
 
+class GitHubGetIssueCommentsResponseSchema(RootModel[list[GitHubIssueCommentSchema]]):
+    root: list[GitHubIssueCommentSchema]
+
+
 class GitHubCreateIssueCommentRequestSchema(BaseModel):
     body: str
 
@@ -25,6 +42,11 @@ class GitHubCreateIssueCommentResponseSchema(BaseModel):
     body: str
 
 
+class GitHubCreateReviewReplyRequestSchema(BaseModel):
+    body: str
+    in_reply_to: int
+
+
 class GitHubCreateReviewCommentRequestSchema(BaseModel):
     body: str
     path: str

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

@@ -9,7 +9,8 @@ class GitHubPRFileSchema(BaseModel):
 
 
 class GitHubGetPRFilesQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitHubGetPRFilesResponseSchema(RootModel[list[GitHubPRFileSchema]]):

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

@@ -1,9 +1,6 @@
 from pydantic import BaseModel, Field
 
-
-class GitHubUserSchema(BaseModel):
-    id: int
-    login: str
+from ai_review.clients.github.pr.schema.user import GitHubUserSchema
 
 
 class GitHubLabelSchema(BaseModel):

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

@@ -10,7 +10,8 @@ class GitHubPRReviewSchema(BaseModel):
 
 
 class GitHubGetPRReviewsQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitHubGetPRReviewsResponseSchema(RootModel[list[GitHubPRReviewSchema]]):

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

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class GitHubUserSchema(BaseModel):
+    id: int | None = None
+    login: str

ai_review/clients/github/pr/types.py

@@ -2,7 +2,9 @@ from typing import Protocol
 
 from ai_review.clients.github.pr.schema.comments import (
     GitHubGetPRCommentsResponseSchema,
+    GitHubGetIssueCommentsResponseSchema,
     GitHubCreateIssueCommentResponseSchema,
+    GitHubCreateReviewReplyRequestSchema,
     GitHubCreateReviewCommentResponseSchema,
     GitHubCreateReviewCommentRequestSchema,
 )
@@ -21,7 +23,7 @@ class GitHubPullRequestsHTTPClientProtocol(Protocol):
             owner: str,
             repo: str,
             issue_number: str
-    ) -> GitHubGetPRCommentsResponseSchema: ...
+    ) -> GitHubGetIssueCommentsResponseSchema: ...
 
     async def get_review_comments(
             self,
@@ -32,6 +34,14 @@ class GitHubPullRequestsHTTPClientProtocol(Protocol):
 
     async def get_reviews(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRReviewsResponseSchema: ...
 
+    async def create_review_reply(
+            self,
+            owner: str,
+            repo: str,
+            comment_id: str,
+            request: GitHubCreateReviewReplyRequestSchema,
+    ) -> GitHubCreateReviewCommentResponseSchema: ...
+
     async def create_review_comment(
             self,
             owner: str,

ai_review/clients/github/tools.py

@@ -0,0 +1,6 @@
+from httpx import Response
+
+
+def github_has_next_page(response: Response) -> bool:
+    link_header = response.headers.get("Link")
+    return (link_header is not None) and ('rel="next"' in link_header)