xai-review 0.25.0__tar.gz → 0.26.0__tar.gz

{xai_review-0.25.0 → xai_review-0.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.25.0
+Version: 0.26.0
 Summary: AI-powered code review tool
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>
@@ -67,7 +67,7 @@ improve code quality, enforce consistency, and speed up the review process.
 ✨ Key features:
 
 - **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, or **Ollama**, and switch anytime.
-- **VCS integration** — works out of the box with GitLab, GitHub (more providers coming).
+- **VCS integration** — works out of the box with **GitLab**, **GitHub**, and **Bitbucket**.
 - **Customizable prompts** — adapt inline, context, and summary reviews to match your team’s coding guidelines.
 - **Flexible configuration** — supports `YAML`, `JSON`, and `ENV`, with seamless overrides in CI/CD pipelines.
 - **AI Review runs fully client-side** — it never proxies or inspects your requests.
@@ -170,7 +170,7 @@ Key things you can customize:
 
 - **LLM provider** — OpenAI, Gemini, Claude, or Ollama
 - **Model settings** — model name, temperature, max tokens
-- **VCS integration** — works out of the box with **GitLab** and **GitHub**.
+- **VCS integration** — works out of the box with **GitLab**, **GitHub**, and **Bitbucket**
 - **Review policy** — which files to include/exclude, review modes
 - **Prompts** — inline/context/summary prompt templates
 
@@ -209,7 +209,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.25.0
+      - uses: Nikita-Filonov/ai-review@v0.26.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

{xai_review-0.25.0 → xai_review-0.26.0}/README.md

@@ -33,7 +33,7 @@ improve code quality, enforce consistency, and speed up the review process.
 ✨ Key features:
 
 - **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, or **Ollama**, and switch anytime.
-- **VCS integration** — works out of the box with GitLab, GitHub (more providers coming).
+- **VCS integration** — works out of the box with **GitLab**, **GitHub**, and **Bitbucket**.
 - **Customizable prompts** — adapt inline, context, and summary reviews to match your team’s coding guidelines.
 - **Flexible configuration** — supports `YAML`, `JSON`, and `ENV`, with seamless overrides in CI/CD pipelines.
 - **AI Review runs fully client-side** — it never proxies or inspects your requests.
@@ -136,7 +136,7 @@ Key things you can customize:
 
 - **LLM provider** — OpenAI, Gemini, Claude, or Ollama
 - **Model settings** — model name, temperature, max tokens
-- **VCS integration** — works out of the box with **GitLab** and **GitHub**.
+- **VCS integration** — works out of the box with **GitLab**, **GitHub**, and **Bitbucket**
 - **Review policy** — which files to include/exclude, review modes
 - **Prompts** — inline/context/summary prompt templates
 
@@ -175,7 +175,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.25.0
+      - uses: Nikita-Filonov/ai-review@v0.26.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

xai_review-0.26.0/ai_review/clients/bitbucket/client.py

@@ -0,0 +1,31 @@
+from ai_review.clients.bitbucket.pr.client import BitbucketPullRequestsHTTPClient
+from httpx import AsyncClient, AsyncHTTPTransport
+
+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 BitbucketHTTPClient:
+    def __init__(self, client: AsyncClient):
+        self.pr = BitbucketPullRequestsHTTPClient(client)
+
+
+def get_bitbucket_http_client() -> BitbucketHTTPClient:
+    logger = get_logger("BITBUCKET_HTTP_CLIENT")
+    logger_event_hook = LoggerEventHook(logger=logger)
+    retry_transport = RetryTransport(logger=logger, 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 BitbucketHTTPClient(client=client)

xai_review-0.26.0/ai_review/clients/bitbucket/pr/client.py

@@ -0,0 +1,104 @@
+from httpx import Response, QueryParams
+
+from ai_review.clients.bitbucket.pr.schema.comments import (
+    BitbucketGetPRCommentsQuerySchema,
+    BitbucketGetPRCommentsResponseSchema,
+    BitbucketCreatePRCommentRequestSchema,
+    BitbucketCreatePRCommentResponseSchema,
+)
+from ai_review.clients.bitbucket.pr.schema.files import (
+    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.libs.http.client import HTTPClient
+from ai_review.libs.http.handlers import handle_http_error, HTTPClientError
+
+
+class BitbucketPullRequestsHTTPClientError(HTTPClientError):
+    pass
+
+
+class BitbucketPullRequestsHTTPClient(HTTPClient, BitbucketPullRequestsHTTPClientProtocol):
+    @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
+    async def get_pull_request_api(self, workspace: str, repo_slug: str, pull_request_id: str) -> Response:
+        return await self.get(f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}")
+
+    @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
+    async def get_diffstat_api(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str,
+            query: BitbucketGetPRFilesQuerySchema,
+    ) -> Response:
+        return await self.get(
+            f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}/diffstat",
+            query=QueryParams(**query.model_dump()),
+        )
+
+    @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
+    async def get_comments_api(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str,
+            query: BitbucketGetPRCommentsQuerySchema,
+    ) -> Response:
+        return await self.get(
+            f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}/comments",
+            query=QueryParams(**query.model_dump()),
+        )
+
+    @handle_http_error(client="BitbucketPullRequestsHTTPClient", exception=BitbucketPullRequestsHTTPClientError)
+    async def create_comment_api(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str,
+            request: BitbucketCreatePRCommentRequestSchema,
+    ) -> Response:
+        return await self.post(
+            f"/repositories/{workspace}/{repo_slug}/pullrequests/{pull_request_id}/comments",
+            json=request.model_dump(by_alias=True),
+        )
+
+    async def get_pull_request(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str
+    ) -> BitbucketGetPRResponseSchema:
+        resp = await self.get_pull_request_api(workspace, repo_slug, pull_request_id)
+        return BitbucketGetPRResponseSchema.model_validate_json(resp.text)
+
+    async def get_files(
+            self,
+            workspace: str,
+            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 get_comments(
+            self,
+            workspace: str,
+            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 create_comment(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str,
+            request: BitbucketCreatePRCommentRequestSchema
+    ) -> BitbucketCreatePRCommentResponseSchema:
+        response = await self.create_comment_api(workspace, repo_slug, pull_request_id, request)
+        return BitbucketCreatePRCommentResponseSchema.model_validate_json(response.text)

xai_review-0.26.0/ai_review/clients/bitbucket/pr/schema/comments.py

@@ -0,0 +1,44 @@
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class BitbucketCommentContentSchema(BaseModel):
+    raw: str
+    html: str | None = None
+    markup: str | None = None
+
+
+class BitbucketCommentInlineSchema(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    path: str
+    to_line: int | None = Field(alias="to", default=None)
+    from_line: int | None = Field(alias="from", default=None)
+
+
+class BitbucketPRCommentSchema(BaseModel):
+    id: int
+    inline: BitbucketCommentInlineSchema | None = None
+    content: BitbucketCommentContentSchema
+
+
+class BitbucketGetPRCommentsQuerySchema(BaseModel):
+    pagelen: int = 100
+
+
+class BitbucketGetPRCommentsResponseSchema(BaseModel):
+    size: int
+    page: int | None = None
+    next: str | None = None
+    values: list[BitbucketPRCommentSchema]
+    pagelen: int
+
+
+class BitbucketCreatePRCommentRequestSchema(BaseModel):
+    inline: BitbucketCommentInlineSchema | None = None
+    content: BitbucketCommentContentSchema
+
+
+class BitbucketCreatePRCommentResponseSchema(BaseModel):
+    id: int
+    inline: BitbucketCommentInlineSchema | None = None
+    content: BitbucketCommentContentSchema

xai_review-0.26.0/ai_review/clients/bitbucket/pr/schema/files.py

@@ -0,0 +1,25 @@
+from pydantic import BaseModel
+
+
+class BitbucketPRFilePathSchema(BaseModel):
+    path: str
+
+
+class BitbucketPRFileSchema(BaseModel):
+    new: BitbucketPRFilePathSchema | None = None
+    old: BitbucketPRFilePathSchema | None = None
+    status: str
+    lines_added: int
+    lines_removed: int
+
+
+class BitbucketGetPRFilesQuerySchema(BaseModel):
+    pagelen: int = 100
+
+
+class BitbucketGetPRFilesResponseSchema(BaseModel):
+    size: int
+    page: int | None = None
+    next: str | None = None
+    values: list[BitbucketPRFileSchema]
+    pagelen: int

xai_review-0.26.0/ai_review/clients/bitbucket/pr/schema/pull_request.py

@@ -0,0 +1,38 @@
+from pydantic import BaseModel, Field
+
+
+class BitbucketUserSchema(BaseModel):
+    uuid: str
+    nickname: str
+    display_name: str
+
+
+class BitbucketBranchSchema(BaseModel):
+    name: str
+
+
+class BitbucketCommitSchema(BaseModel):
+    hash: str
+
+
+class BitbucketRepositorySchema(BaseModel):
+    uuid: str
+    full_name: str
+
+
+class BitbucketPRLocationSchema(BaseModel):
+    branch: BitbucketBranchSchema
+    commit: BitbucketCommitSchema
+    repository: BitbucketRepositorySchema
+
+
+class BitbucketGetPRResponseSchema(BaseModel):
+    id: int
+    title: str
+    description: str | None = None
+    state: str
+    author: BitbucketUserSchema
+    source: BitbucketPRLocationSchema
+    destination: BitbucketPRLocationSchema
+    reviewers: list[BitbucketUserSchema] = Field(default_factory=list)
+    participants: list[BitbucketUserSchema] = Field(default_factory=list)

xai_review-0.26.0/ai_review/clients/bitbucket/pr/types.py

@@ -0,0 +1,44 @@
+from typing import Protocol
+
+from ai_review.clients.bitbucket.pr.schema.comments import (
+    BitbucketGetPRCommentsResponseSchema,
+    BitbucketCreatePRCommentRequestSchema,
+    BitbucketCreatePRCommentResponseSchema,
+)
+from ai_review.clients.bitbucket.pr.schema.files import BitbucketGetPRFilesResponseSchema
+from ai_review.clients.bitbucket.pr.schema.pull_request import BitbucketGetPRResponseSchema
+
+
+class BitbucketPullRequestsHTTPClientProtocol(Protocol):
+    async def get_pull_request(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str
+    ) -> BitbucketGetPRResponseSchema:
+        ...
+
+    async def get_files(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str
+    ) -> BitbucketGetPRFilesResponseSchema:
+        ...
+
+    async def get_comments(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str
+    ) -> BitbucketGetPRCommentsResponseSchema:
+        ...
+
+    async def create_comment(
+            self,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str,
+            request: BitbucketCreatePRCommentRequestSchema,
+    ) -> BitbucketCreatePRCommentResponseSchema:
+        ...

{xai_review-0.25.0 → xai_review-0.26.0}/ai_review/libs/config/vcs/base.py

@@ -2,6 +2,7 @@ from typing import Annotated, Literal
 
 from pydantic import BaseModel, Field
 
+from ai_review.libs.config.vcs.bitbucket import BitbucketPipelineConfig, BitbucketHTTPClientConfig
 from ai_review.libs.config.vcs.github import GitHubPipelineConfig, GitHubHTTPClientConfig
 from ai_review.libs.config.vcs.gitlab import GitLabPipelineConfig, GitLabHTTPClientConfig
 from ai_review.libs.constants.vcs_provider import VCSProvider
@@ -23,4 +24,13 @@ class GitHubVCSConfig(VCSConfigBase):
     http_client: GitHubHTTPClientConfig
 
 
-VCSConfig = Annotated[GitLabVCSConfig | GitHubVCSConfig, Field(discriminator="provider")]
+class BitbucketVCSConfig(VCSConfigBase):
+    provider: Literal[VCSProvider.BITBUCKET]
+    pipeline: BitbucketPipelineConfig
+    http_client: BitbucketHTTPClientConfig
+
+
+VCSConfig = Annotated[
+    GitLabVCSConfig | GitHubVCSConfig | BitbucketVCSConfig,
+    Field(discriminator="provider")
+]

xai_review-0.26.0/ai_review/libs/config/vcs/bitbucket.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel
+
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+
+
+class BitbucketPipelineConfig(BaseModel):
+    workspace: str
+    repo_slug: str
+    pull_request_id: str
+
+
+class BitbucketHTTPClientConfig(HTTPClientWithTokenConfig):
+    pass

{xai_review-0.25.0 → xai_review-0.26.0}/ai_review/libs/constants/vcs_provider.py

@@ -4,3 +4,4 @@ from enum import StrEnum
 class VCSProvider(StrEnum):
     GITHUB = "GITHUB"
     GITLAB = "GITLAB"
+    BITBUCKET = "BITBUCKET"

xai_review-0.26.0/ai_review/services/vcs/bitbucket/client.py

@@ -0,0 +1,185 @@
+from ai_review.clients.bitbucket.client import get_bitbucket_http_client
+from ai_review.clients.bitbucket.pr.schema.comments import (
+    BitbucketCommentInlineSchema,
+    BitbucketCommentContentSchema,
+    BitbucketCreatePRCommentRequestSchema,
+)
+from ai_review.config import settings
+from ai_review.libs.logger import get_logger
+from ai_review.services.vcs.types import (
+    VCSClientProtocol,
+    UserSchema,
+    BranchRefSchema,
+    ReviewInfoSchema,
+    ReviewCommentSchema,
+)
+
+logger = get_logger("BITBUCKET_VCS_CLIENT")
+
+
+class BitbucketVCSClient(VCSClientProtocol):
+    def __init__(self):
+        self.http_client = get_bitbucket_http_client()
+        self.workspace = settings.vcs.pipeline.workspace
+        self.repo_slug = settings.vcs.pipeline.repo_slug
+        self.pull_request_id = settings.vcs.pipeline.pull_request_id
+
+    async def get_review_info(self) -> ReviewInfoSchema:
+        try:
+            pr = await self.http_client.pr.get_pull_request(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+            )
+            files = await self.http_client.pr.get_files(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+            )
+
+            logger.info(f"Fetched PR info for {self.workspace}/{self.repo_slug}#{self.pull_request_id}")
+
+            return ReviewInfoSchema(
+                id=pr.id,
+                title=pr.title,
+                description=pr.description or "",
+                author=UserSchema(
+                    id=pr.author.uuid,
+                    name=pr.author.display_name,
+                    username=pr.author.nickname,
+                ),
+                labels=[],
+                base_sha=pr.destination.commit.hash,
+                head_sha=pr.source.commit.hash,
+                assignees=[
+                    UserSchema(
+                        id=user.uuid,
+                        name=user.display_name,
+                        username=user.nickname,
+                    )
+                    for user in pr.participants
+                ],
+                reviewers=[
+                    UserSchema(
+                        id=user.uuid,
+                        name=user.display_name,
+                        username=user.nickname,
+                    )
+                    for user in pr.reviewers
+                ],
+                source_branch=BranchRefSchema(
+                    ref=pr.source.branch.name,
+                    sha=pr.source.commit.hash,
+                ),
+                target_branch=BranchRefSchema(
+                    ref=pr.destination.branch.name,
+                    sha=pr.destination.commit.hash,
+                ),
+                changed_files=[
+                    file.new.path if file.new else file.old.path
+                    for file in files.values
+                ],
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch PR info {self.workspace}/{self.repo_slug}#{self.pull_request_id}: {error}"
+            )
+            return ReviewInfoSchema()
+
+    async def get_general_comments(self) -> list[ReviewCommentSchema]:
+        try:
+            response = await self.http_client.pr.get_comments(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+            )
+            logger.info(f"Fetched general comments for {self.workspace}/{self.repo_slug}#{self.pull_request_id}")
+
+            return [
+                ReviewCommentSchema(id=comment.id, body=comment.content.raw)
+                for comment in response.values
+                if comment.inline is None
+            ]
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch general comments for "
+                f"{self.workspace}/{self.repo_slug}#{self.pull_request_id}: {error}"
+            )
+            return []
+
+    async def get_inline_comments(self) -> list[ReviewCommentSchema]:
+        try:
+            response = await self.http_client.pr.get_comments(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+            )
+            logger.info(f"Fetched inline comments for {self.workspace}/{self.repo_slug}#{self.pull_request_id}")
+
+            return [
+                ReviewCommentSchema(
+                    id=comment.id,
+                    body=comment.content.raw,
+                    file=comment.inline.path,
+                    line=comment.inline.to_line,
+                )
+                for comment in response.values
+                if comment.inline is not None
+            ]
+        except Exception as error:
+            logger.exception(
+                f"Failed to fetch inline comments for "
+                f"{self.workspace}/{self.repo_slug}#{self.pull_request_id}: {error}"
+            )
+            return []
+
+    async def create_general_comment(self, message: str) -> None:
+        try:
+            logger.info(
+                f"Posting general comment to PR {self.workspace}/{self.repo_slug}#{self.pull_request_id}: {message}"
+            )
+            request = BitbucketCreatePRCommentRequestSchema(
+                content=BitbucketCommentContentSchema(raw=message)
+            )
+            await self.http_client.pr.create_comment(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+                request=request,
+            )
+            logger.info(
+                f"Created general comment in PR {self.workspace}/{self.repo_slug}#{self.pull_request_id}"
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to create general comment in PR "
+                f"{self.workspace}/{self.repo_slug}#{self.pull_request_id}: {error}"
+            )
+            raise
+
+    async def create_inline_comment(self, file: str, line: int, message: str) -> None:
+        try:
+            logger.info(
+                f"Posting inline comment in {self.workspace}/{self.repo_slug}#{self.pull_request_id} "
+                f"at {file}:{line}: {message}"
+            )
+            request = BitbucketCreatePRCommentRequestSchema(
+                content=BitbucketCommentContentSchema(raw=message),
+                inline=BitbucketCommentInlineSchema(path=file, to_line=line),
+            )
+            await self.http_client.pr.create_comment(
+                workspace=self.workspace,
+                repo_slug=self.repo_slug,
+                pull_request_id=self.pull_request_id,
+                request=request,
+            )
+            logger.info(
+                f"Created inline comment in {self.workspace}/{self.repo_slug}#{self.pull_request_id} "
+                f"at {file}:{line}"
+            )
+        except Exception as error:
+            logger.exception(
+                f"Failed to create inline comment in {self.workspace}/{self.repo_slug}#{self.pull_request_id} "
+                f"at {file}:{line}: {error}"
+            )
+            raise

{xai_review-0.25.0 → xai_review-0.26.0}/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.bitbucket.client import BitbucketVCSClient
 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 VCSClientProtocol
@@ -11,5 +12,7 @@ def get_vcs_client() -> VCSClientProtocol:
             return GitLabVCSClient()
         case VCSProvider.GITHUB:
             return GitHubVCSClient()
+        case VCSProvider.BITBUCKET:
+            return BitbucketVCSClient()
         case _:
             raise ValueError(f"Unsupported VCS provider: {settings.vcs.provider}")