xai-review 0.27.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/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,8 +17,14 @@ 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
 
@@ -38,12 +46,18 @@ class BitbucketGetPRCommentsResponseSchema(BaseModel):
     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/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/github/pr/client.py

@@ -2,10 +2,13 @@ from httpx import Response, QueryParams
 
 from ai_review.clients.github.pr.schema.comments import (
     GitHubPRCommentSchema,
+    GitHubIssueCommentSchema,
     GitHubGetPRCommentsQuerySchema,
     GitHubGetPRCommentsResponseSchema,
+    GitHubGetIssueCommentsResponseSchema,
     GitHubCreateIssueCommentRequestSchema,
     GitHubCreateIssueCommentResponseSchema,
+    GitHubCreateReviewReplyRequestSchema,
     GitHubCreateReviewCommentRequestSchema,
     GitHubCreateReviewCommentResponseSchema
 )
@@ -76,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,
@@ -136,13 +152,18 @@ class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProto
         )
         return GitHubGetPRFilesResponseSchema(root=items)
 
-    async def get_issue_comments(self, owner: str, repo: str, issue_number: str) -> GitHubGetPRCommentsResponseSchema:
+    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[GitHubPRCommentSchema]:
-            result = GitHubGetPRCommentsResponseSchema.model_validate_json(response.text)
+        def extract_items(response: Response) -> list[GitHubIssueCommentSchema]:
+            result = GitHubGetIssueCommentsResponseSchema.model_validate_json(response.text)
             return result.root
 
         items = await paginate(
@@ -151,7 +172,7 @@ class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProto
             extract_items=extract_items,
             has_next_page=github_has_next_page
         )
-        return GitHubGetPRCommentsResponseSchema(root=items)
+        return GitHubGetIssueCommentsResponseSchema(root=items)
 
     async def get_review_comments(self, owner: str, repo: str, pull_number: str) -> GitHubGetPRCommentsResponseSchema:
         async def fetch_page(page: int) -> Response:
@@ -187,6 +208,16 @@ class GitHubPullRequestsHTTPClient(HTTPClient, GitHubPullRequestsHTTPClientProto
         )
         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,
             owner: str,

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

@@ -1,11 +1,23 @@
 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):
@@ -17,6 +29,10 @@ class GitHubGetPRCommentsResponseSchema(RootModel[list[GitHubPRCommentSchema]]):
     root: list[GitHubPRCommentSchema]
 
 
+class GitHubGetIssueCommentsResponseSchema(RootModel[list[GitHubIssueCommentSchema]]):
+    root: list[GitHubIssueCommentSchema]
+
+
 class GitHubCreateIssueCommentRequestSchema(BaseModel):
     body: str
 
@@ -26,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/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/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/gitlab/mr/client.py

@@ -6,7 +6,9 @@ from ai_review.clients.gitlab.mr.schema.discussions import (
     GitLabGetMRDiscussionsQuerySchema,
     GitLabGetMRDiscussionsResponseSchema,
     GitLabCreateMRDiscussionRequestSchema,
-    GitLabCreateMRDiscussionResponseSchema
+    GitLabCreateMRDiscussionResponseSchema,
+    GitLabCreateMRDiscussionReplyRequestSchema,
+    GitLabCreateMRDiscussionReplyResponseSchema
 )
 from ai_review.clients.gitlab.mr.schema.notes import (
     GitLabNoteSchema,
@@ -82,6 +84,19 @@ class GitLabMergeRequestsHTTPClient(HTTPClient, GitLabMergeRequestsHTTPClientPro
             json=request.model_dump(),
         )
 
+    @handle_http_error(client="GitLabMergeRequestsHTTPClient", exception=GitLabMergeRequestsHTTPClientError)
+    async def create_discussion_reply_api(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            discussion_id: str,
+            request: GitLabCreateMRDiscussionReplyRequestSchema,
+    ) -> Response:
+        return await self.post(
+            f"/api/v4/projects/{project_id}/merge_requests/{merge_request_id}/discussions/{discussion_id}/notes",
+            json=request.model_dump(),
+        )
+
     async def get_changes(self, project_id: str, merge_request_id: str) -> GitLabGetMRChangesResponseSchema:
         response = await self.get_changes_api(project_id, merge_request_id)
         return GitLabGetMRChangesResponseSchema.model_validate_json(response.text)
@@ -154,3 +169,19 @@ class GitLabMergeRequestsHTTPClient(HTTPClient, GitLabMergeRequestsHTTPClientPro
             merge_request_id=merge_request_id
         )
         return GitLabCreateMRDiscussionResponseSchema.model_validate_json(response.text)
+
+    async def create_discussion_reply(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            discussion_id: str,
+            body: str,
+    ) -> GitLabCreateMRDiscussionReplyResponseSchema:
+        request = GitLabCreateMRDiscussionReplyRequestSchema(body=body)
+        response = await self.create_discussion_reply_api(
+            project_id=project_id,
+            merge_request_id=merge_request_id,
+            discussion_id=discussion_id,
+            request=request,
+        )
+        return GitLabCreateMRDiscussionReplyResponseSchema.model_validate_json(response.text)

ai_review/clients/gitlab/mr/schema/changes.py

@@ -1,10 +1,6 @@
 from pydantic import BaseModel, Field
 
-
-class GitLabUserSchema(BaseModel):
-    id: int
-    name: str
-    username: str
+from ai_review.clients.gitlab.mr.schema.user import GitLabUserSchema
 
 
 class GitLabDiffRefsSchema(BaseModel):

ai_review/clients/gitlab/mr/schema/discussions.py

@@ -1,13 +1,8 @@
-from pydantic import BaseModel, RootModel
+from pydantic import BaseModel, RootModel, Field
 
 from ai_review.clients.gitlab.mr.schema.notes import GitLabNoteSchema
 
 
-class GitLabDiscussionSchema(BaseModel):
-    id: str
-    notes: list[GitLabNoteSchema]
-
-
 class GitLabDiscussionPositionSchema(BaseModel):
     position_type: str = "text"
     base_sha: str
@@ -17,6 +12,12 @@ class GitLabDiscussionPositionSchema(BaseModel):
     new_line: int
 
 
+class GitLabDiscussionSchema(BaseModel):
+    id: str
+    notes: list[GitLabNoteSchema]
+    position: GitLabDiscussionPositionSchema | None = None
+
+
 class GitLabGetMRDiscussionsQuerySchema(BaseModel):
     page: int = 1
     per_page: int = 100
@@ -33,4 +34,13 @@ class GitLabCreateMRDiscussionRequestSchema(BaseModel):
 
 class GitLabCreateMRDiscussionResponseSchema(BaseModel):
     id: str
-    body: str | None = None
+    notes: list[GitLabNoteSchema] = Field(default_factory=list)
+
+
+class GitLabCreateMRDiscussionReplyRequestSchema(BaseModel):
+    body: str
+
+
+class GitLabCreateMRDiscussionReplyResponseSchema(BaseModel):
+    id: int
+    body: str

ai_review/clients/gitlab/mr/schema/notes.py

@@ -1,9 +1,12 @@
 from pydantic import BaseModel, RootModel
 
+from ai_review.clients.gitlab.mr.schema.user import GitLabUserSchema
+
 
 class GitLabNoteSchema(BaseModel):
     id: int
     body: str
+    author: GitLabUserSchema | None = None
 
 
 class GitLabGetMRNotesQuerySchema(BaseModel):

ai_review/clients/gitlab/mr/schema/user.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class GitLabUserSchema(BaseModel):
+    id: int
+    name: str
+    username: str

ai_review/clients/gitlab/mr/types.py

@@ -3,8 +3,9 @@ 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,
+    GitLabCreateMRDiscussionResponseSchema,
+    GitLabCreateMRDiscussionReplyResponseSchema
 )
 from ai_review.clients.gitlab.mr.schema.notes import GitLabGetMRNotesResponseSchema, GitLabCreateMRNoteResponseSchema
 
@@ -14,12 +15,6 @@ class GitLabMergeRequestsHTTPClientProtocol(Protocol):
 
     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,
@@ -27,9 +22,23 @@ class GitLabMergeRequestsHTTPClientProtocol(Protocol):
             merge_request_id: str,
     ) -> GitLabCreateMRNoteResponseSchema: ...
 
+    async def get_discussions(
+            self,
+            project_id: str,
+            merge_request_id: str
+    ) -> GitLabGetMRDiscussionsResponseSchema: ...
+
     async def create_discussion(
             self,
             project_id: str,
             merge_request_id: str,
             request: GitLabCreateMRDiscussionRequestSchema,
     ) -> GitLabCreateMRDiscussionResponseSchema: ...
+
+    async def create_discussion_reply(
+            self,
+            project_id: str,
+            merge_request_id: str,
+            discussion_id: str,
+            body: str,
+    ) -> GitLabCreateMRDiscussionReplyResponseSchema: ...