xai-review 0.24.0__py3-none-any.whl → 0.26.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,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)

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

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

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/claude/client.py

@@ -1,6 +1,7 @@
 from httpx import AsyncClient, Response, AsyncHTTPTransport
 
 from ai_review.clients.claude.schema import ClaudeChatRequestSchema, ClaudeChatResponseSchema
+from ai_review.clients.claude.types import ClaudeHTTPClientProtocol
 from ai_review.config import settings
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.event_hooks.logger import LoggerEventHook
@@ -13,7 +14,7 @@ class ClaudeHTTPClientError(HTTPClientError):
     pass
 
 
-class ClaudeHTTPClient(HTTPClient):
+class ClaudeHTTPClient(HTTPClient, ClaudeHTTPClientProtocol):
     @handle_http_error(client="ClaudeHTTPClient", exception=ClaudeHTTPClientError)
     async def chat_api(self, request: ClaudeChatRequestSchema) -> Response:
         return await self.post("/v1/messages", json=request.model_dump())

ai_review/clients/claude/types.py

@@ -0,0 +1,8 @@
+from typing import Protocol
+
+from ai_review.clients.claude.schema import ClaudeChatRequestSchema, ClaudeChatResponseSchema
+
+
+class ClaudeHTTPClientProtocol(Protocol):
+    async def chat(self, request: ClaudeChatRequestSchema) -> ClaudeChatResponseSchema:
+        ...

ai_review/clients/gemini/client.py

@@ -1,6 +1,7 @@
 from httpx import Response, AsyncHTTPTransport, AsyncClient
 
 from ai_review.clients.gemini.schema import GeminiChatRequestSchema, GeminiChatResponseSchema
+from ai_review.clients.gemini.types import GeminiHTTPClientProtocol
 from ai_review.config import settings
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.event_hooks.logger import LoggerEventHook
@@ -13,7 +14,7 @@ class GeminiHTTPClientError(HTTPClientError):
     pass
 
 
-class GeminiHTTPClient(HTTPClient):
+class GeminiHTTPClient(HTTPClient, GeminiHTTPClientProtocol):
     @handle_http_error(client="GeminiHTTPClient", exception=GeminiHTTPClientError)
     async def chat_api(self, request: GeminiChatRequestSchema) -> Response:
         meta = settings.llm.meta

ai_review/clients/gemini/types.py

@@ -0,0 +1,8 @@
+from typing import Protocol
+
+from ai_review.clients.gemini.schema import GeminiChatRequestSchema, GeminiChatResponseSchema
+
+
+class GeminiHTTPClientProtocol(Protocol):
+    async def chat(self, request: GeminiChatRequestSchema) -> GeminiChatResponseSchema:
+        ...

ai_review/clients/ollama/client.py

@@ -0,0 +1,41 @@
+from httpx import AsyncClient, Response, AsyncHTTPTransport
+
+from ai_review.clients.ollama.schema import OllamaChatRequestSchema, OllamaChatResponseSchema
+from ai_review.clients.ollama.types import OllamaHTTPClientProtocol
+from ai_review.config import settings
+from ai_review.libs.http.client import HTTPClient
+from ai_review.libs.http.event_hooks.logger import LoggerEventHook
+from ai_review.libs.http.handlers import HTTPClientError, handle_http_error
+from ai_review.libs.http.transports.retry import RetryTransport
+from ai_review.libs.logger import get_logger
+
+
+class OllamaHTTPClientError(HTTPClientError):
+    pass
+
+
+class OllamaHTTPClient(HTTPClient, OllamaHTTPClientProtocol):
+    @handle_http_error(client="OllamaHTTPClient", exception=OllamaHTTPClientError)
+    async def chat_api(self, request: OllamaChatRequestSchema) -> Response:
+        return await self.post("/api/chat", json=request.model_dump())
+
+    async def chat(self, request: OllamaChatRequestSchema) -> OllamaChatResponseSchema:
+        response = await self.chat_api(request)
+        return OllamaChatResponseSchema.model_validate_json(response.text)
+
+
+def get_ollama_http_client() -> OllamaHTTPClient:
+    logger = get_logger("OLLAMA_HTTP_CLIENT")
+    logger_event_hook = LoggerEventHook(logger=logger)
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
+
+    client = AsyncClient(
+        timeout=settings.llm.http_client.timeout,
+        base_url=settings.llm.http_client.api_url_value,
+        transport=retry_transport,
+        event_hooks={
+            "request": [logger_event_hook.request],
+            "response": [logger_event_hook.response],
+        },
+    )
+    return OllamaHTTPClient(client=client)

ai_review/clients/ollama/schema.py

@@ -0,0 +1,47 @@
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class OllamaMessageSchema(BaseModel):
+    role: Literal["system", "user", "assistant"]
+    content: str
+
+
+class OllamaOptionsSchema(BaseModel):
+    stop: list[str] | None = None
+    seed: int | None = None
+    top_p: float | None = Field(default=None, ge=0.0, le=1.0)
+    temperature: float | None = Field(default=None, ge=0.0, le=2.0)
+    num_predict: int | None = Field(default=None, ge=1)
+    repeat_penalty: float | None = Field(default=None, ge=0.0)
+
+
+class OllamaChatRequestSchema(BaseModel):
+    model: str
+    stream: bool = False
+    options: OllamaOptionsSchema | None = None
+    messages: list[OllamaMessageSchema]
+
+
+class OllamaUsageSchema(BaseModel):
+    prompt_tokens: int | None = None
+    completion_tokens: int | None = None
+
+    @property
+    def total_tokens(self) -> int | None:
+        if (self.prompt_tokens is not None) and (self.completion_tokens is not None):
+            return self.prompt_tokens + self.completion_tokens
+
+        return None
+
+
+class OllamaChatResponseSchema(BaseModel):
+    done: bool = Field(default=True)
+    usage: OllamaUsageSchema | None = None
+    model: str
+    message: OllamaMessageSchema
+
+    @property
+    def first_text(self) -> str:
+        return (self.message.content or "").strip()

ai_review/clients/ollama/types.py

@@ -0,0 +1,8 @@
+from typing import Protocol
+
+from ai_review.clients.ollama.schema import OllamaChatRequestSchema, OllamaChatResponseSchema
+
+
+class OllamaHTTPClientProtocol(Protocol):
+    async def chat(self, request: OllamaChatRequestSchema) -> OllamaChatResponseSchema:
+        ...

ai_review/clients/openai/client.py

@@ -1,6 +1,7 @@
 from httpx import Response, AsyncHTTPTransport, AsyncClient
 
 from ai_review.clients.openai.schema import OpenAIChatRequestSchema, OpenAIChatResponseSchema
+from ai_review.clients.openai.types import OpenAIHTTPClientProtocol
 from ai_review.config import settings
 from ai_review.libs.http.client import HTTPClient
 from ai_review.libs.http.event_hooks.logger import LoggerEventHook
@@ -13,7 +14,7 @@ class OpenAIHTTPClientError(HTTPClientError):
     pass
 
 
-class OpenAIHTTPClient(HTTPClient):
+class OpenAIHTTPClient(HTTPClient, OpenAIHTTPClientProtocol):
     @handle_http_error(client='OpenAIHTTPClient', exception=OpenAIHTTPClientError)
     async def chat_api(self, request: OpenAIChatRequestSchema) -> Response:
         return await self.post("/chat/completions", json=request.model_dump())

ai_review/clients/openai/types.py

@@ -0,0 +1,8 @@
+from typing import Protocol
+
+from ai_review.clients.openai.schema import OpenAIChatRequestSchema, OpenAIChatResponseSchema
+
+
+class OpenAIHTTPClientProtocol(Protocol):
+    async def chat(self, request: OpenAIChatRequestSchema) -> OpenAIChatResponseSchema:
+        ...

ai_review/libs/config/http.py

@@ -4,12 +4,15 @@ from pydantic import BaseModel, HttpUrl, SecretStr
 class HTTPClientConfig(BaseModel):
     timeout: float = 120
     api_url: HttpUrl
-    api_token: SecretStr
 
     @property
     def api_url_value(self) -> str:
         return str(self.api_url)
 
+
+class HTTPClientWithTokenConfig(HTTPClientConfig):
+    api_token: SecretStr
+
     @property
     def api_token_value(self) -> str:
         return self.api_token.get_secret_value()

ai_review/libs/config/llm/base.py

@@ -6,6 +6,7 @@ from pydantic import BaseModel, Field, FilePath
 
 from ai_review.libs.config.llm.claude import ClaudeHTTPClientConfig, ClaudeMetaConfig
 from ai_review.libs.config.llm.gemini import GeminiHTTPClientConfig, GeminiMetaConfig
+from ai_review.libs.config.llm.ollama import OllamaHTTPClientConfig, OllamaMetaConfig
 from ai_review.libs.config.llm.openai import OpenAIHTTPClientConfig, OpenAIMetaConfig
 from ai_review.libs.constants.llm_provider import LLMProvider
 from ai_review.libs.resources import load_resource
@@ -55,7 +56,13 @@ class ClaudeLLMConfig(LLMConfigBase):
     http_client: ClaudeHTTPClientConfig
 
 
+class OllamaLLMConfig(LLMConfigBase):
+    meta: OllamaMetaConfig
+    provider: Literal[LLMProvider.OLLAMA]
+    http_client: OllamaHTTPClientConfig
+
+
 LLMConfig = Annotated[
-    OpenAILLMConfig | GeminiLLMConfig | ClaudeLLMConfig,
+    OpenAILLMConfig | GeminiLLMConfig | ClaudeLLMConfig | OllamaLLMConfig,
     Field(discriminator="provider")
 ]

ai_review/libs/config/llm/claude.py

@@ -1,13 +1,10 @@
-from pydantic import BaseModel
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
 
-from ai_review.libs.config.http import HTTPClientConfig
 
-
-class ClaudeMetaConfig(BaseModel):
+class ClaudeMetaConfig(LLMMetaConfig):
     model: str = "claude-3-sonnet"
-    max_tokens: int = 1200
-    temperature: float = 0.3
 
 
-class ClaudeHTTPClientConfig(HTTPClientConfig):
+class ClaudeHTTPClientConfig(HTTPClientWithTokenConfig):
     api_version: str = "2023-06-01"

ai_review/libs/config/llm/gemini.py

@@ -1,13 +1,10 @@
-from pydantic import BaseModel
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
 
-from ai_review.libs.config.http import HTTPClientConfig
 
-
-class GeminiMetaConfig(BaseModel):
+class GeminiMetaConfig(LLMMetaConfig):
     model: str = "gemini-2.0-pro"
-    max_tokens: int = 1200
-    temperature: float = 0.3
 
 
-class GeminiHTTPClientConfig(HTTPClientConfig):
+class GeminiHTTPClientConfig(HTTPClientWithTokenConfig):
     pass

ai_review/libs/config/llm/meta.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel, Field
+
+
+class LLMMetaConfig(BaseModel):
+    model: str
+    max_tokens: int = Field(default=5000, ge=1)
+    temperature: float = Field(default=0.3, ge=0.0, le=2.0)

ai_review/libs/config/llm/ollama.py

@@ -0,0 +1,14 @@
+from ai_review.libs.config.http import HTTPClientConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
+
+
+class OllamaMetaConfig(LLMMetaConfig):
+    stop: list[str] | None = None
+    seed: int | None = None
+    model: str = "llama2"
+    top_p: float | None = None
+    repeat_penalty: float | None = None
+
+
+class OllamaHTTPClientConfig(HTTPClientConfig):
+    pass

ai_review/libs/config/llm/openai.py

@@ -1,13 +1,10 @@
-from pydantic import BaseModel
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
 
-from ai_review.libs.config.http import HTTPClientConfig
 
-
-class OpenAIMetaConfig(BaseModel):
+class OpenAIMetaConfig(LLMMetaConfig):
     model: str = "gpt-4o-mini"
-    max_tokens: int = 1200
-    temperature: float = 0.3
 
 
-class OpenAIHTTPClientConfig(HTTPClientConfig):
+class OpenAIHTTPClientConfig(HTTPClientWithTokenConfig):
     pass

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")
+]

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/github.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ai_review.libs.config.http import HTTPClientConfig
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
 
 
 class GitHubPipelineConfig(BaseModel):
@@ -9,5 +9,5 @@ class GitHubPipelineConfig(BaseModel):
     pull_number: str
 
 
-class GitHubHTTPClientConfig(HTTPClientConfig):
+class GitHubHTTPClientConfig(HTTPClientWithTokenConfig):
     pass

ai_review/libs/config/vcs/gitlab.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ai_review.libs.config.http import HTTPClientConfig
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
 
 
 class GitLabPipelineConfig(BaseModel):
@@ -8,5 +8,5 @@ class GitLabPipelineConfig(BaseModel):
     merge_request_id: str
 
 
-class GitLabHTTPClientConfig(HTTPClientConfig):
+class GitLabHTTPClientConfig(HTTPClientWithTokenConfig):
     pass