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

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

@@ -22,15 +22,20 @@ class BitbucketPRCommentSchema(BaseModel):
 
 
 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 BitbucketCreatePRCommentRequestSchema(BaseModel):

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

@@ -5,11 +5,13 @@ 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):

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)

ai_review/libs/http/paginate.py

@@ -0,0 +1,43 @@
+from typing import Awaitable, Callable, TypeVar
+
+from httpx import Response
+from pydantic import BaseModel
+
+from ai_review.libs.logger import get_logger
+
+T = TypeVar("T", bound=BaseModel)
+
+logger = get_logger("PAGINATE")
+
+
+async def paginate(
+        fetch_page: Callable[[int], Awaitable[Response]],
+        extract_items: Callable[[Response], list[T]],
+        has_next_page: Callable[[Response], bool],
+        max_pages: int | None = None,
+) -> list[T]:
+    page = 1
+    items: list[T] = []
+
+    while True:
+        response = await fetch_page(page)
+
+        try:
+            extracted = extract_items(response)
+        except Exception as error:
+            logger.error(f"Failed to extract items on {page=}")
+            raise RuntimeError(f"Failed to extract items on {page=}") from error
+
+        logger.debug(f"Page {page}: extracted {len(extracted)} items (total={len(items) + len(extracted)})")
+        items.extend(extracted)
+
+        if not has_next_page(response):
+            logger.debug(f"Pagination finished after {page} page(s), total items={len(items)}")
+            break
+
+        page += 1
+        if max_pages and (page > max_pages):
+            logger.error(f"Pagination exceeded {max_pages=}")
+            raise RuntimeError(f"Pagination exceeded {max_pages=}")
+
+    return items

ai_review/tests/suites/clients/bitbucket/test_client.py

@@ -0,0 +1,14 @@
+import pytest
+from httpx import AsyncClient
+
+from ai_review.clients.bitbucket.client import get_bitbucket_http_client, BitbucketHTTPClient
+from ai_review.clients.bitbucket.pr.client import BitbucketPullRequestsHTTPClient
+
+
+@pytest.mark.usefixtures("bitbucket_http_client_config")
+def test_get_bitbucket_http_client_builds_ok():
+    bitbucket_http_client = get_bitbucket_http_client()
+
+    assert isinstance(bitbucket_http_client, BitbucketHTTPClient)
+    assert isinstance(bitbucket_http_client.pr, BitbucketPullRequestsHTTPClient)
+    assert isinstance(bitbucket_http_client.pr.client, AsyncClient)

ai_review/tests/suites/clients/bitbucket/test_tools.py

@@ -0,0 +1,31 @@
+from httpx import Response, Request
+
+from ai_review.clients.bitbucket.tools import bitbucket_has_next_page
+
+
+def make_response(data: dict) -> Response:
+    return Response(
+        json=data,
+        request=Request("GET", "http://bitbucket.test"),
+        status_code=200,
+    )
+
+
+def test_bitbucket_has_next_page_true():
+    resp = make_response({"next": "https://api.bitbucket.org/2.0/repositories/test/repo?page=2"})
+    assert bitbucket_has_next_page(resp) is True
+
+
+def test_bitbucket_has_next_page_false_none():
+    resp = make_response({"next": None})
+    assert bitbucket_has_next_page(resp) is False
+
+
+def test_bitbucket_has_next_page_false_missing():
+    resp = make_response({})
+    assert bitbucket_has_next_page(resp) is False
+
+
+def test_bitbucket_has_next_page_false_empty_string():
+    resp = make_response({"next": ""})
+    assert bitbucket_has_next_page(resp) is False

ai_review/tests/suites/clients/github/test_tools.py

@@ -0,0 +1,31 @@
+from httpx import Response, Request
+
+from ai_review.clients.github.tools import github_has_next_page
+
+
+def make_response(headers: dict) -> Response:
+    return Response(
+        request=Request("GET", "http://test"),
+        headers=headers,
+        status_code=200,
+    )
+
+
+def test_github_has_next_page_true():
+    response = make_response({
+        "Link": '<https://api.github.com/resource?page=2>; rel="next", '
+                '<https://api.github.com/resource?page=5>; rel="last"'
+    })
+    assert github_has_next_page(response) is True
+
+
+def test_github_has_next_page_false_no_next():
+    response = make_response({
+        "Link": '<https://api.github.com/resource?page=5>; rel="last"'
+    })
+    assert github_has_next_page(response) is False
+
+
+def test_github_has_next_page_false_no_header():
+    resp = make_response({})
+    assert github_has_next_page(resp) is False

ai_review/tests/suites/clients/gitlab/test_tools.py

@@ -0,0 +1,26 @@
+from httpx import Response, Request
+
+from ai_review.clients.gitlab.tools import gitlab_has_next_page
+
+
+def make_response(headers: dict) -> Response:
+    return Response(
+        request=Request("GET", "http://gitlab.test"),
+        headers=headers,
+        status_code=200,
+    )
+
+
+def test_gitlab_has_next_page_true():
+    resp = make_response({"X-Next-Page": "2"})
+    assert gitlab_has_next_page(resp) is True
+
+
+def test_gitlab_has_next_page_false_empty():
+    resp = make_response({"X-Next-Page": ""})
+    assert gitlab_has_next_page(resp) is False
+
+
+def test_gitlab_has_next_page_false_missing():
+    resp = make_response({})
+    assert gitlab_has_next_page(resp) is False

ai_review/tests/suites/libs/http/test_paginate.py

@@ -0,0 +1,95 @@
+import pytest
+from httpx import Response, Request
+from pydantic import BaseModel
+
+from ai_review.libs.http.paginate import paginate
+
+
+class DummySchema(BaseModel):
+    value: int
+
+
+def make_response(data: dict) -> Response:
+    return Response(
+        json=data,
+        request=Request("GET", "http://test"),
+        status_code=200,
+    )
+
+
+@pytest.mark.asyncio
+async def test_single_page():
+    async def fetch_page(_: int) -> Response:
+        return make_response({"items": [1, 2, 3]})
+
+    def extract_items(response: Response) -> list[DummySchema]:
+        return [DummySchema(value=value) for value in response.json()["items"]]
+
+    def has_next_page(_: Response) -> bool:
+        return False
+
+    items = await paginate(fetch_page, extract_items, has_next_page)
+    assert len(items) == 3
+    assert [item.value for item in items] == [1, 2, 3]
+
+
+@pytest.mark.asyncio
+async def test_multiple_pages():
+    async def fetch_page(page: int) -> Response:
+        return make_response({"items": [page]})
+
+    def extract_items(response: Response):
+        return [DummySchema(value=value) for value in response.json()["items"]]
+
+    def has_next_page(response: Response) -> bool:
+        return response.json()["items"][0] < 3
+
+    items = await paginate(fetch_page, extract_items, has_next_page)
+    assert [item.value for item in items] == [1, 2, 3]
+
+
+@pytest.mark.asyncio
+async def test_extract_items_error():
+    async def fetch_page(_: int) -> Response:
+        return make_response({"items": [1]})
+
+    def extract_items(_: Response):
+        raise ValueError("bad json")
+
+    def has_next_page(_: Response) -> bool:
+        return False
+
+    with pytest.raises(RuntimeError) as exc:
+        await paginate(fetch_page, extract_items, has_next_page)
+    assert "Failed to extract items" in str(exc.value)
+
+
+@pytest.mark.asyncio
+async def test_max_pages_exceeded():
+    async def fetch_page(page: int) -> Response:
+        return make_response({"items": [page]})
+
+    def extract_items(response: Response):
+        return [DummySchema(value=value) for value in response.json()["items"]]
+
+    def has_next_page(_: Response) -> bool:
+        return True
+
+    with pytest.raises(RuntimeError) as exc:
+        await paginate(fetch_page, extract_items, has_next_page, max_pages=2)
+    assert "Pagination exceeded" in str(exc.value)
+
+
+@pytest.mark.asyncio
+async def test_empty_items():
+    async def fetch_page(_: int) -> Response:
+        return make_response({"items": []})
+
+    def extract_items(_: Response):
+        return []
+
+    def has_next_page(_: Response) -> bool:
+        return False
+
+    result = await paginate(fetch_page, extract_items, has_next_page)
+    assert result == []

{xai_review-0.26.0.dist-info → xai_review-0.27.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.26.0
+Version: 0.27.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.26.0
+      - uses: Nikita-Filonov/ai-review@v0.27.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

{xai_review-0.26.0.dist-info → xai_review-0.27.0.dist-info}/RECORD

@@ -10,12 +10,13 @@ ai_review/cli/commands/run_summary_review.py,sha256=NqjepGH5cbqczPzcuMEAxO4dI58F
 ai_review/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/bitbucket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/bitbucket/client.py,sha256=VaqaQ5USMPTOEeS5XPdr-RkMKsxUpJ2SBE6lcemkz-g,1174
+ai_review/clients/bitbucket/tools.py,sha256=UGBCurb8MQECivWDJDNXr7ej7rwA5-_kXTT4zF8RXIQ,147
 ai_review/clients/bitbucket/pr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/bitbucket/pr/client.py,sha256=9C6vXBz8o0Df76N9WW4hORN-Q39Vd8I575AaidyW_HM,4359
+ai_review/clients/bitbucket/pr/client.py,sha256=YnA5GCw_qqrjraNH_LjtE7t7cCmrMTY9IrcxDT2uyZ0,5836
 ai_review/clients/bitbucket/pr/types.py,sha256=ZICV4ghYChj1Jl9Nlwyw1_kwmGybX51GhGdGzkRaLCk,1296
 ai_review/clients/bitbucket/pr/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/bitbucket/pr/schema/comments.py,sha256=DLi3LhThXfHB9MJ5Akv7Yf_n-VttjvJAausSMoksHTY,1152
-ai_review/clients/bitbucket/pr/schema/files.py,sha256=A-h9Cgi0iJ6e9pGr5TcbpgSb3y9SMTqNi5FxJ7ySxpk,546
+ai_review/clients/bitbucket/pr/schema/comments.py,sha256=p3HQyi-M7DLXcD0RErVlqsnKhmo2JVqdA8ejVYF_Kzg,1337
+ai_review/clients/bitbucket/pr/schema/files.py,sha256=qBDfIv370TDMKM8yoLlm1c-BTHNvEZRuhnBo18nkx9g,750
 ai_review/clients/bitbucket/pr/schema/pull_request.py,sha256=buGULgaCkxCUFSdiw0XTwaSIYP_p1rAEuKXUyJ_Mzi8,863
 ai_review/clients/claude/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/claude/client.py,sha256=uEadbBNBJnzjHDczbxXiiw1V1H1PdUWKu-Gn-eIDEmw,1890
@@ -27,23 +28,25 @@ ai_review/clients/gemini/schema.py,sha256=5oVvbI-h_sw8bFreS4JUmMj-aXa_frvxK3H8sg
 ai_review/clients/gemini/types.py,sha256=D-P0THorrQ8yq5P-NKAC65zzhEYRa9HkiXTORG9QoIk,267
 ai_review/clients/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/github/client.py,sha256=pprQcCYdrhRYtuqRsTFiCbj54Qb1Ll6_jmlm7AJg8pk,1149
+ai_review/clients/github/tools.py,sha256=sD2eS_iNhzT4ZLgTRmO2uBLuId-fa2aSvbu6VFeKSlc,201
 ai_review/clients/github/pr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/github/pr/client.py,sha256=YvGc4ClK615PgWslD8bL54pk1YakXHivabjnPZ8p6iI,6642
+ai_review/clients/github/pr/client.py,sha256=HysL1u-WxLdhGKi-6HAx6fN-50XSk3yK55gZ9Nii72U,8851
 ai_review/clients/github/pr/types.py,sha256=mI5Vtlxc25iymOTPkNl55IQygXd4ti3B1Ydpx3q1fps,1726
 ai_review/clients/github/pr/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/github/pr/schema/comments.py,sha256=IjBtpcJABlql97_N7_CMAKuAozbVnSkaD16l3KueGqk,719
-ai_review/clients/github/pr/schema/files.py,sha256=mXfmjaJuVbbo2PPTppCyYlc4LSd04fb-_x_9aJkNW04,335
+ai_review/clients/github/pr/schema/comments.py,sha256=dIlX6tYPseAtLquk-UIie7jxBxrCbmPbDyB8bC9OFPo,743
+ai_review/clients/github/pr/schema/files.py,sha256=ARirokToZ1zqMWh2jpDtVIs4qw5rlL7xfoRDJBJF71o,359
 ai_review/clients/github/pr/schema/pull_request.py,sha256=EwOworYQY4kCmL6QFKEXK7r2fpINK8o-4-FEy9-nTpg,688
-ai_review/clients/github/pr/schema/reviews.py,sha256=tRHqGPzZ-LTbAZhtoVeBdl-LH34tgI0zPCtJTfFe2Mc,356
+ai_review/clients/github/pr/schema/reviews.py,sha256=qRJpBu03CBwC_tPng9q9tkqT_u2WcdDELblAi-OzxWI,380
 ai_review/clients/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/gitlab/client.py,sha256=MIZed-V4JUzntkU9vJq7Fkp5VAjnzjvep_5QM5ni_Co,1151
+ai_review/clients/gitlab/tools.py,sha256=ed8t2-dy2XtSRpYWhLxUobYvbBX_1zMZTPuzjbgPzTg,136
 ai_review/clients/gitlab/mr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/gitlab/mr/client.py,sha256=WdaJCJdSHeZZUQ9Hz8172RCgACwm-_NmcC91LHT7YwU,5158
+ai_review/clients/gitlab/mr/client.py,sha256=39EjS0bvvi8-kiBOkBa7N1q0wHb0qTgo_KvQise61js,6350
 ai_review/clients/gitlab/mr/types.py,sha256=tZ3rDDkzJM8v4zPt4gTyKCL30htiPQE9F_kVR6K5JHA,1280
 ai_review/clients/gitlab/mr/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/gitlab/mr/schema/changes.py,sha256=ZqSPb8zO0z_V8cEjxoTqnwbjRLxo6OTV4LeQEAg91cU,835
-ai_review/clients/gitlab/mr/schema/discussions.py,sha256=JgvxKfHoYxmp86aP4MpIczK-arU0hc-BZLASWDWBIRs,790
-ai_review/clients/gitlab/mr/schema/notes.py,sha256=yfnnRt69fALKfapzZpVtvCvNwPkq5jBFI7fbPMq1w1c,424
+ai_review/clients/gitlab/mr/schema/discussions.py,sha256=owaWWD5lMG9GXkT8kgHiU_-khYTWGHanbWgSKDiJknA,814
+ai_review/clients/gitlab/mr/schema/notes.py,sha256=DX9juurMYKcaYobh4n4M49dvHtUiLw5IzZKsGdBMxZg,448
 ai_review/clients/ollama/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/ollama/client.py,sha256=KoJ9J5_Vfpv5XNJREshE_gA46uo9J0Z3qVC7wJPEcX8,1720
 ai_review/clients/ollama/schema.py,sha256=A6oKwkkEVrduyzMR_lhLnaLyvKXqlfsXjkMIF2eXaYw,1310
@@ -74,10 +77,11 @@ ai_review/libs/config/llm/meta.py,sha256=cEcAHOwy-mQBKo9_KJrQe0I7qppq6h99lSmoWX4
 ai_review/libs/config/llm/ollama.py,sha256=M6aiPb5GvYvkiGcgHTsh9bOw5JsBLqmfSKoIbHCejrU,372
 ai_review/libs/config/llm/openai.py,sha256=jGVL4gJ2wIacoKeK9Zc9LCgY95TxdeYOThdglVPErFU,262
 ai_review/libs/config/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/libs/config/vcs/base.py,sha256=ks9lrSalkPUuG8ijlaw-8d-F-dv59GdSywHS2TsIKjs,1085
+ai_review/libs/config/vcs/base.py,sha256=B0kKc6-A3mmV7dvHpo47u-1yTvdUCvLj_g4oCBo_NyY,1214
 ai_review/libs/config/vcs/bitbucket.py,sha256=on5sQaE57kM_zSmqdDUNrttVtTPGOzqLHM5s7eFN7DA,275
 ai_review/libs/config/vcs/github.py,sha256=hk-kuDLd8wecqtEb8PSqF7Yy_pkihplJhi6nB6FZID4,256
 ai_review/libs/config/vcs/gitlab.py,sha256=ecYfU158VgVlM6P5mgZn8FOqk3Xt60xx7gUqT5e22a4,252
+ai_review/libs/config/vcs/pagination.py,sha256=S-XxWQYkIjhu3ffpHQ44d7UtRHH81fh9GaJ-xQXUUy4,175
 ai_review/libs/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/constants/llm_provider.py,sha256=k7GzctIZ-TDsRlhTPbpGYgym_CO2YKVFp_oXG9dTBW0,143
 ai_review/libs/constants/vcs_provider.py,sha256=xJpRdJIdAf05iH2x2f362d1MuviOlPVP7In-JvDVotE,127
@@ -88,6 +92,7 @@ ai_review/libs/diff/tools.py,sha256=CZWRDlOW2YS-b8h9gv_uP1MG194_FLkKzcKTwwZHocI,
 ai_review/libs/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/http/client.py,sha256=JONzbhJWJKFz8Jy6p9pzq2hAMKlyJ2_WkBksuAlqW7k,453
 ai_review/libs/http/handlers.py,sha256=k1VvCIFjLzfH3qQ--aj4CZVgbU0oj78sYStMBrxo_Ek,1040
+ai_review/libs/http/paginate.py,sha256=yAryDaUkQd-wojXOaak9HyicT-QZwx3L49AJlUEins4,1305
 ai_review/libs/http/event_hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/http/event_hooks/base.py,sha256=cnSOOButTJYKeyb_OnGms1vXRfwfExP81L3ZfYWLufk,279
 ai_review/libs/http/event_hooks/logger.py,sha256=8_omfl6q3JijaBBIgzvzb4SayjNEDW-oxyck_Ky8wnI,603
@@ -191,6 +196,9 @@ ai_review/tests/fixtures/services/review/inline.py,sha256=k4IW6oy5JHMo9Kv0H97DLl
 ai_review/tests/fixtures/services/review/summary.py,sha256=Hkt8mq1ZmqMH5_mELrS1x0wtCoNPbBjOEQ9yIsMbRts,691
 ai_review/tests/suites/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/clients/bitbucket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/clients/bitbucket/test_client.py,sha256=1qEkznC01hkh6TPpzChY9glhK6OTX-pfnSF43KNBtrY,600
+ai_review/tests/suites/clients/bitbucket/test_tools.py,sha256=naHq0Xy7uP8pYdWR8JVrET5JJpNb0Z2-TEPkGHh2fFI,887
 ai_review/tests/suites/clients/claude/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/claude/test_client.py,sha256=jLGqK7lzYc8LjJO-3HjBtLqCdg-ubw-xZ4EvFeFGRhY,404
 ai_review/tests/suites/clients/claude/test_schema.py,sha256=MUZXvEROgLNpUVHfCsH5D3ruJPQwTx0OgeT3_BRVjgI,1671
@@ -199,8 +207,10 @@ ai_review/tests/suites/clients/gemini/test_client.py,sha256=f2R7KisiENrzf8gaK26N
 ai_review/tests/suites/clients/gemini/test_schema.py,sha256=88dU28m7sEWvGx6tqYl7if7weWYuVc8erlkFkKKI3bk,3109
 ai_review/tests/suites/clients/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/github/test_client.py,sha256=BiuLKCHIk83U1szYEZkB-n3vvyPgj6tAI5EqxKiT-CY,558
+ai_review/tests/suites/clients/github/test_tools.py,sha256=_RKMWNgfeynnpbiDebRLg-1Qz91Kyevf5drl4hCngzU,881
 ai_review/tests/suites/clients/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/gitlab/test_client.py,sha256=5QOkNvgm0XRKHh79FNIY9CTonAqYPXqCCxcxeiAHYCA,560
+ai_review/tests/suites/clients/gitlab/test_tools.py,sha256=-gS_kvZwNBkvYeYyYPld78F4ZuZPrpNORWVbg2eq5wM,678
 ai_review/tests/suites/clients/ollama/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/ollama/test_client.py,sha256=XZ8NAd1bS_ltTuYZPgqlutPRA6kbvH3_3SKTCbNBTgA,404
 ai_review/tests/suites/clients/ollama/test_schema.py,sha256=A93wCmxwGdvudfbA97VCPYP3gT6u6EYMetAg5fgURRA,1836
@@ -217,6 +227,8 @@ ai_review/tests/suites/libs/diff/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 ai_review/tests/suites/libs/diff/test_models.py,sha256=RBFQ97LWhU8TlupxXkJ97ryAvJrSuOHLtT9biUBUMXg,3321
 ai_review/tests/suites/libs/diff/test_parser.py,sha256=rvWEVGIdaLBlDAnSevjRY7I1Zikj12d5GOgMk9QyHQQ,3013
 ai_review/tests/suites/libs/diff/test_tools.py,sha256=XkHJZ-b5veFz5oLKO09P7npaLN8lOzCnGR7e83Zv_mg,1953
+ai_review/tests/suites/libs/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/libs/http/test_paginate.py,sha256=9PCn1mUQypPId9l8RToBQnLBJBV_WEYxBh785bqqmMA,2694
 ai_review/tests/suites/libs/template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/libs/template/test_render.py,sha256=n-ss5bd_hwc-RzYmqWmFM6KSlP1zLSnlsW1Yki12Bpw,1890
 ai_review/tests/suites/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -262,9 +274,9 @@ ai_review/tests/suites/services/vcs/github/__init__.py,sha256=47DEQpj8HBSa-_TImW
 ai_review/tests/suites/services/vcs/github/test_service.py,sha256=c2sjecm4qzqYXuO9j6j35NQyJzqDpnXIJImRTcpkyHo,4378
 ai_review/tests/suites/services/vcs/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/vcs/gitlab/test_service.py,sha256=0dqgL5whzjcP-AQ4adP_12QfkYm_ZtdtMotmYm8Se7Y,4449
-xai_review-0.26.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
-xai_review-0.26.0.dist-info/METADATA,sha256=RXLUNKPnkpxjCcXWbKwFI8RN0UHvVjfWwtqGpVsukIs,11150
-xai_review-0.26.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xai_review-0.26.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
-xai_review-0.26.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
-xai_review-0.26.0.dist-info/RECORD,,
+xai_review-0.27.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
+xai_review-0.27.0.dist-info/METADATA,sha256=5SnK3Ip4vzfwvS110fYNDSDYD4OzrmA5G1eUV2oGMbE,11150
+xai_review-0.27.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xai_review-0.27.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
+xai_review-0.27.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
+xai_review-0.27.0.dist-info/RECORD,,