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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.24.0
+Version: 0.25.0
 Summary: AI-powered code review tool
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>
@@ -66,7 +66,7 @@ improve code quality, enforce consistency, and speed up the review process.
 
 ✨ Key features:
 
-- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, and **Gemini**, or switch anytime.
+- **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).
 - **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.
@@ -168,7 +168,7 @@ for complete, ready-to-use examples.
 
 Key things you can customize:
 
-- **LLM provider** — OpenAI, Gemini, or Claude
+- **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**.
 - **Review policy** — which files to include/exclude, review modes
@@ -209,7 +209,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.24.0
+      - uses: Nikita-Filonov/ai-review@v0.25.0
         with:
           review-command: ${{ inputs.review-command }}
         env:
@@ -288,6 +288,10 @@ provider** explicitly configured in your `.ai-review.yaml`.
 All data is sent **directly** from your CI/CD environment to the selected LLM API endpoint (e.g. OpenAI, Gemini,
 Claude). No intermediary servers or storage layers are involved.
 
+If you use **Ollama**, requests are sent to your **local or self-hosted Ollama runtime**  
+(by default `http://localhost:11434`). This allows you to run reviews completely **offline**, keeping all data strictly
+inside your infrastructure.
+
 > ⚠️ Please ensure you use proper API tokens and avoid exposing corporate or personal secrets.
 > If you accidentally leak private code or credentials due to incorrect configuration (e.g., using a personal key
 > instead of an enterprise one), it is **your responsibility** — the tool does not retain or share any data by itself.

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

@@ -32,7 +32,7 @@ improve code quality, enforce consistency, and speed up the review process.
 
 ✨ Key features:
 
-- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, and **Gemini**, or switch anytime.
+- **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).
 - **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.
@@ -134,7 +134,7 @@ for complete, ready-to-use examples.
 
 Key things you can customize:
 
-- **LLM provider** — OpenAI, Gemini, or Claude
+- **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**.
 - **Review policy** — which files to include/exclude, review modes
@@ -175,7 +175,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: Nikita-Filonov/ai-review@v0.24.0
+      - uses: Nikita-Filonov/ai-review@v0.25.0
         with:
           review-command: ${{ inputs.review-command }}
         env:
@@ -254,6 +254,10 @@ provider** explicitly configured in your `.ai-review.yaml`.
 All data is sent **directly** from your CI/CD environment to the selected LLM API endpoint (e.g. OpenAI, Gemini,
 Claude). No intermediary servers or storage layers are involved.
 
+If you use **Ollama**, requests are sent to your **local or self-hosted Ollama runtime**  
+(by default `http://localhost:11434`). This allows you to run reviews completely **offline**, keeping all data strictly
+inside your infrastructure.
+
 > ⚠️ Please ensure you use proper API tokens and avoid exposing corporate or personal secrets.
 > If you accidentally leak private code or credentials due to incorrect configuration (e.g., using a personal key
 > instead of an enterprise one), it is **your responsibility** — the tool does not retain or share any data by itself.

{xai_review-0.24.0 → xai_review-0.25.0}/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())

xai_review-0.25.0/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:
+        ...

{xai_review-0.24.0 → xai_review-0.25.0}/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

xai_review-0.25.0/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:
+        ...

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

xai_review-0.25.0/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()

xai_review-0.25.0/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:
+        ...

{xai_review-0.24.0 → xai_review-0.25.0}/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())

xai_review-0.25.0/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:
+        ...

{xai_review-0.24.0 → xai_review-0.25.0}/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()

{xai_review-0.24.0 → xai_review-0.25.0}/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")
 ]

xai_review-0.25.0/ai_review/libs/config/llm/claude.py

@@ -0,0 +1,10 @@
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
+
+
+class ClaudeMetaConfig(LLMMetaConfig):
+    model: str = "claude-3-sonnet"
+
+
+class ClaudeHTTPClientConfig(HTTPClientWithTokenConfig):
+    api_version: str = "2023-06-01"

xai_review-0.25.0/ai_review/libs/config/llm/gemini.py

@@ -0,0 +1,10 @@
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
+
+
+class GeminiMetaConfig(LLMMetaConfig):
+    model: str = "gemini-2.0-pro"
+
+
+class GeminiHTTPClientConfig(HTTPClientWithTokenConfig):
+    pass

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

xai_review-0.25.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

xai_review-0.25.0/ai_review/libs/config/llm/openai.py

@@ -0,0 +1,10 @@
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
+
+
+class OpenAIMetaConfig(LLMMetaConfig):
+    model: str = "gpt-4o-mini"
+
+
+class OpenAIHTTPClientConfig(HTTPClientWithTokenConfig):
+    pass

{xai_review-0.24.0 → xai_review-0.25.0}/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

{xai_review-0.24.0 → xai_review-0.25.0}/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

{xai_review-0.24.0 → xai_review-0.25.0}/ai_review/libs/constants/llm_provider.py

@@ -5,3 +5,4 @@ class LLMProvider(StrEnum):
     OPENAI = "OPENAI"
     GEMINI = "GEMINI"
     CLAUDE = "CLAUDE"
+    OLLAMA = "OLLAMA"

{xai_review-0.24.0 → xai_review-0.25.0}/ai_review/services/llm/factory.py

@@ -2,6 +2,7 @@ from ai_review.config import settings
 from ai_review.libs.constants.llm_provider import LLMProvider
 from ai_review.services.llm.claude.client import ClaudeLLMClient
 from ai_review.services.llm.gemini.client import GeminiLLMClient
+from ai_review.services.llm.ollama.client import OllamaLLMClient
 from ai_review.services.llm.openai.client import OpenAILLMClient
 from ai_review.services.llm.types import LLMClientProtocol
 
@@ -14,5 +15,7 @@ def get_llm_client() -> LLMClientProtocol:
             return GeminiLLMClient()
         case LLMProvider.CLAUDE:
             return ClaudeLLMClient()
+        case LLMProvider.OLLAMA:
+            return OllamaLLMClient()
         case _:
             raise ValueError(f"Unsupported LLM provider: {settings.llm.provider}")

xai_review-0.25.0/ai_review/services/llm/ollama/client.py

@@ -0,0 +1,34 @@
+from ai_review.clients.ollama.client import get_ollama_http_client
+from ai_review.clients.ollama.schema import OllamaChatRequestSchema, OllamaMessageSchema, OllamaOptionsSchema
+from ai_review.config import settings
+from ai_review.services.llm.types import LLMClientProtocol, ChatResultSchema
+
+
+class OllamaLLMClient(LLMClientProtocol):
+    def __init__(self):
+        self.http_client = get_ollama_http_client()
+
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        meta = settings.llm.meta
+        request = OllamaChatRequestSchema(
+            model=meta.model,
+            options=OllamaOptionsSchema(
+                stop=meta.stop,
+                seed=meta.seed,
+                top_p=meta.top_p,
+                temperature=meta.temperature,
+                num_predict=meta.max_tokens,
+                repeat_penalty=meta.repeat_penalty,
+            ),
+            messages=[
+                OllamaMessageSchema(role="system", content=prompt_system),
+                OllamaMessageSchema(role="user", content=prompt),
+            ],
+        )
+        response = await self.http_client.chat(request)
+        return ChatResultSchema(
+            text=response.first_text,
+            total_tokens=response.usage.total_tokens if response.usage else None,
+            prompt_tokens=response.usage.prompt_tokens if response.usage else None,
+            completion_tokens=response.usage.completion_tokens if response.usage else None,
+        )

xai_review-0.25.0/ai_review/tests/fixtures/clients/claude.py

@@ -0,0 +1,67 @@
+from typing import Any
+
+import pytest
+from pydantic import HttpUrl, SecretStr
+
+from ai_review.clients.claude.schema import (
+    ClaudeUsageSchema,
+    ClaudeContentSchema,
+    ClaudeChatRequestSchema,
+    ClaudeChatResponseSchema,
+)
+from ai_review.clients.claude.types import ClaudeHTTPClientProtocol
+from ai_review.config import settings
+from ai_review.libs.config.llm.base import ClaudeLLMConfig
+from ai_review.libs.config.llm.claude import ClaudeMetaConfig, ClaudeHTTPClientConfig
+from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.claude.client import ClaudeLLMClient
+
+
+class FakeClaudeHTTPClient(ClaudeHTTPClientProtocol):
+    def __init__(self, responses: dict[str, Any] | None = None) -> None:
+        self.calls: list[tuple[str, dict]] = []
+        self.responses = responses or {}
+
+    async def chat(self, request: ClaudeChatRequestSchema) -> ClaudeChatResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            ClaudeChatResponseSchema(
+                id="fake-id",
+                role="assistant",
+                usage=ClaudeUsageSchema(input_tokens=5, output_tokens=7),
+                content=[ClaudeContentSchema(type="text", text="FAKE_CLAUDE_RESPONSE")],
+            ),
+        )
+
+
+@pytest.fixture
+def fake_claude_http_client():
+    return FakeClaudeHTTPClient()
+
+
+@pytest.fixture
+def claude_llm_client(
+        monkeypatch: pytest.MonkeyPatch,
+        fake_claude_http_client: FakeClaudeHTTPClient
+) -> ClaudeLLMClient:
+    monkeypatch.setattr(
+        "ai_review.services.llm.claude.client.get_claude_http_client",
+        lambda: fake_claude_http_client,
+    )
+    return ClaudeLLMClient()
+
+
+@pytest.fixture
+def claude_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = ClaudeLLMConfig(
+        meta=ClaudeMetaConfig(),
+        provider=LLMProvider.CLAUDE,
+        http_client=ClaudeHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://api.anthropic.com"),
+            api_token=SecretStr("fake-token"),
+            api_version="2023-06-01",
+        )
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)

xai_review-0.25.0/ai_review/tests/fixtures/clients/gemini.py

@@ -0,0 +1,73 @@
+from typing import Any
+
+import pytest
+from pydantic import HttpUrl, SecretStr
+
+from ai_review.clients.gemini.schema import (
+    GeminiPartSchema,
+    GeminiUsageSchema,
+    GeminiContentSchema,
+    GeminiCandidateSchema,
+    GeminiChatRequestSchema,
+    GeminiChatResponseSchema,
+)
+from ai_review.clients.gemini.types import GeminiHTTPClientProtocol
+from ai_review.config import settings
+from ai_review.libs.config.llm.base import GeminiLLMConfig
+from ai_review.libs.config.llm.gemini import GeminiMetaConfig, GeminiHTTPClientConfig
+from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.gemini.client import GeminiLLMClient
+
+
+class FakeGeminiHTTPClient(GeminiHTTPClientProtocol):
+    def __init__(self, responses: dict[str, Any] | None = None) -> None:
+        self.calls: list[tuple[str, dict]] = []
+        self.responses = responses or {}
+
+    async def chat(self, request: GeminiChatRequestSchema) -> GeminiChatResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            GeminiChatResponseSchema(
+                usage=GeminiUsageSchema(prompt_token_count=2, total_tokens_count=10),
+                candidates=[
+                    GeminiCandidateSchema(
+                        content=GeminiContentSchema(
+                            role="model",
+                            parts=[GeminiPartSchema(text="FAKE_GEMINI_RESPONSE")]
+                        )
+                    )
+                ],
+            ),
+        )
+
+
+@pytest.fixture
+def fake_gemini_http_client() -> FakeGeminiHTTPClient:
+    return FakeGeminiHTTPClient()
+
+
+@pytest.fixture
+def gemini_llm_client(
+        monkeypatch: pytest.MonkeyPatch,
+        fake_gemini_http_client: FakeGeminiHTTPClient
+) -> GeminiLLMClient:
+    monkeypatch.setattr(
+        "ai_review.services.llm.gemini.client.get_gemini_http_client",
+        lambda: fake_gemini_http_client,
+    )
+    return GeminiLLMClient()
+
+
+@pytest.fixture
+def gemini_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = GeminiLLMConfig(
+        meta=GeminiMetaConfig(),
+        provider=LLMProvider.GEMINI,
+        http_client=GeminiHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://generativelanguage.googleapis.com"),
+            api_token=SecretStr("fake-token"),
+        )
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)

xai_review-0.25.0/ai_review/tests/fixtures/clients/ollama.py

@@ -0,0 +1,65 @@
+from typing import Any
+
+import pytest
+from pydantic import HttpUrl
+
+from ai_review.clients.ollama.schema import (
+    OllamaUsageSchema,
+    OllamaMessageSchema,
+    OllamaChatRequestSchema,
+    OllamaChatResponseSchema,
+)
+from ai_review.clients.ollama.types import OllamaHTTPClientProtocol
+from ai_review.config import settings
+from ai_review.libs.config.llm.base import OllamaLLMConfig
+from ai_review.libs.config.llm.ollama import OllamaMetaConfig, OllamaHTTPClientConfig
+from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.ollama.client import OllamaLLMClient
+
+
+class FakeOllamaHTTPClient(OllamaHTTPClientProtocol):
+    def __init__(self, responses: dict[str, Any] | None = None) -> None:
+        self.calls: list[tuple[str, dict]] = []
+        self.responses = responses or {}
+
+    async def chat(self, request: OllamaChatRequestSchema) -> OllamaChatResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            OllamaChatResponseSchema(
+                done=True,
+                model="llama2",
+                usage=OllamaUsageSchema(prompt_tokens=3, completion_tokens=5),
+                message=OllamaMessageSchema(role="assistant", content="FAKE_OLLAMA_RESPONSE"),
+            ),
+        )
+
+
+@pytest.fixture
+def fake_ollama_http_client():
+    return FakeOllamaHTTPClient()
+
+
+@pytest.fixture
+def ollama_llm_client(
+        monkeypatch: pytest.MonkeyPatch,
+        fake_ollama_http_client: FakeOllamaHTTPClient
+) -> OllamaLLMClient:
+    monkeypatch.setattr(
+        "ai_review.services.llm.ollama.client.get_ollama_http_client",
+        lambda: fake_ollama_http_client,
+    )
+    return OllamaLLMClient()
+
+
+@pytest.fixture
+def ollama_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = OllamaLLMConfig(
+        meta=OllamaMetaConfig(),
+        provider=LLMProvider.OLLAMA,
+        http_client=OllamaHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("http://localhost:11434")
+        )
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)