xai-review 0.25.0__py3-none-any.whl → 0.27.0__py3-none-any.whl

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)

ai_review/clients/bitbucket/pr/client.py

@@ -0,0 +1,141 @@
+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):
+    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(by_alias=True)),
+        )
+
+    @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(by_alias=True)),
+        )
+
+    @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:
+        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,
+            workspace: str,
+            repo_slug: str,
+            pull_request_id: str
+    ) -> BitbucketGetPRCommentsResponseSchema:
+        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,
+            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)

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

@@ -0,0 +1,49 @@
+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):
+    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]
+    page_len: int = Field(alias="pagelen")
+
+
+class BitbucketCreatePRCommentRequestSchema(BaseModel):
+    inline: BitbucketCommentInlineSchema | None = None
+    content: BitbucketCommentContentSchema
+
+
+class BitbucketCreatePRCommentResponseSchema(BaseModel):
+    id: int
+    inline: BitbucketCommentInlineSchema | None = None
+    content: BitbucketCommentContentSchema

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

@@ -0,0 +1,30 @@
+from pydantic import BaseModel, Field, ConfigDict
+
+
+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):
+    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]
+    page_len: int = Field(alias="pagelen")

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)

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:
+        ...

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,6 +1,7 @@
 from httpx import Response, QueryParams
 
 from ai_review.clients.github.pr.schema.comments import (
+    GitHubPRCommentSchema,
     GitHubGetPRCommentsQuerySchema,
     GitHubGetPRCommentsResponseSchema,
     GitHubCreateIssueCommentRequestSchema,
@@ -9,17 +10,22 @@ from ai_review.clients.github.pr.schema.comments import (
     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):
@@ -114,24 +120,72 @@ 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 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)
+            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_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_comment(
             self,

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

@@ -9,7 +9,8 @@ class GitHubPRCommentSchema(BaseModel):
 
 
 class GitHubGetPRCommentsQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitHubGetPRCommentsResponseSchema(RootModel[list[GitHubPRCommentSchema]]):

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/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/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)

ai_review/clients/gitlab/mr/client.py

@@ -2,20 +2,25 @@ from httpx import Response, QueryParams
 
 from ai_review.clients.gitlab.mr.schema.changes import GitLabGetMRChangesResponseSchema
 from ai_review.clients.gitlab.mr.schema.discussions import (
+    GitLabDiscussionSchema,
     GitLabGetMRDiscussionsQuerySchema,
     GitLabGetMRDiscussionsResponseSchema,
     GitLabCreateMRDiscussionRequestSchema,
     GitLabCreateMRDiscussionResponseSchema
 )
 from ai_review.clients.gitlab.mr.schema.notes import (
+    GitLabNoteSchema,
     GitLabGetMRNotesQuerySchema,
     GitLabGetMRNotesResponseSchema,
     GitLabCreateMRNoteRequestSchema,
     GitLabCreateMRNoteResponseSchema,
 )
 from ai_review.clients.gitlab.mr.types import GitLabMergeRequestsHTTPClientProtocol
+from ai_review.clients.gitlab.tools import gitlab_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 GitLabMergeRequestsHTTPClientError(HTTPClientError):
@@ -86,18 +91,42 @@ class GitLabMergeRequestsHTTPClient(HTTPClient, GitLabMergeRequestsHTTPClientPro
             project_id: str,
             merge_request_id: str
     ) -> GitLabGetMRNotesResponseSchema:
-        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 fetch_page(page: int) -> Response:
+            query = GitLabGetMRNotesQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_notes_api(project_id, merge_request_id, query)
+
+        def extract_items(response: Response) -> list[GitLabNoteSchema]:
+            result = GitLabGetMRNotesResponseSchema.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=gitlab_has_next_page
+        )
+        return GitLabGetMRNotesResponseSchema(root=items)
 
     async def get_discussions(
             self,
             project_id: str,
             merge_request_id: str
     ) -> GitLabGetMRDiscussionsResponseSchema:
-        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 fetch_page(page: int) -> Response:
+            query = GitLabGetMRDiscussionsQuerySchema(page=page, per_page=settings.vcs.pagination.per_page)
+            return await self.get_discussions_api(project_id, merge_request_id, query)
+
+        def extract_items(response: Response) -> list[GitLabDiscussionSchema]:
+            result = GitLabGetMRDiscussionsResponseSchema.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=gitlab_has_next_page
+        )
+        return GitLabGetMRDiscussionsResponseSchema(root=items)
 
     async def create_note(
             self,

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

@@ -18,7 +18,8 @@ class GitLabDiscussionPositionSchema(BaseModel):
 
 
 class GitLabGetMRDiscussionsQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitLabGetMRDiscussionsResponseSchema(RootModel[list[GitLabDiscussionSchema]]):

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

@@ -7,7 +7,8 @@ class GitLabNoteSchema(BaseModel):
 
 
 class GitLabGetMRNotesQuerySchema(BaseModel):
-    per_page: int
+    page: int = 1
+    per_page: int = 100
 
 
 class GitLabGetMRNotesResponseSchema(RootModel[list[GitLabNoteSchema]]):

ai_review/clients/gitlab/tools.py

@@ -0,0 +1,5 @@
+from httpx import Response
+
+
+def gitlab_has_next_page(response: Response) -> bool:
+    return bool(response.headers.get("X-Next-Page"))

ai_review/libs/config/vcs/base.py

@@ -2,13 +2,16 @@ 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.config.vcs.pagination import VCSPaginationConfig
 from ai_review.libs.constants.vcs_provider import VCSProvider
 
 
 class VCSConfigBase(BaseModel):
     provider: VCSProvider
+    pagination: VCSPaginationConfig = VCSPaginationConfig()
 
 
 class GitLabVCSConfig(VCSConfigBase):
@@ -23,4 +26,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")
+]

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

ai_review/libs/config/vcs/pagination.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel, Field
+
+
+class VCSPaginationConfig(BaseModel):
+    per_page: int = Field(default=100, ge=1, le=100)
+    max_pages: int = Field(default=5, ge=1)