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

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

@@ -13,11 +13,11 @@ from ai_review.libs.config.base import (
     get_json_config_file_or_default
 )
 from ai_review.libs.config.core import CoreConfig
-from ai_review.libs.config.llm import LLMConfig
+from ai_review.libs.config.llm.base import LLMConfig
 from ai_review.libs.config.logger import LoggerConfig
 from ai_review.libs.config.prompt import PromptConfig
 from ai_review.libs.config.review import ReviewConfig
-from ai_review.libs.config.vcs import VCSConfig
+from ai_review.libs.config.vcs.base import VCSConfig
 
 
 class Settings(BaseSettings):

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.py → llm/base.py}

@@ -4,9 +4,10 @@ from typing import Annotated, Literal
 import yaml
 from pydantic import BaseModel, Field, FilePath
 
-from ai_review.libs.config.claude import ClaudeHTTPClientConfig, ClaudeMetaConfig
-from ai_review.libs.config.gemini import GeminiHTTPClientConfig, GeminiMetaConfig
-from ai_review.libs.config.openai import OpenAIHTTPClientConfig, OpenAIMetaConfig
+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

@@ -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"

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

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

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

ai_review/libs/config/{vcs.py → vcs/base.py}

@@ -2,8 +2,8 @@ from typing import Annotated, Literal
 
 from pydantic import BaseModel, Field
 
-from ai_review.libs.config.github import GitHubPipelineConfig, GitHubHTTPClientConfig
-from ai_review.libs.config.gitlab import GitLabPipelineConfig, GitLabHTTPClientConfig
+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
 
 

ai_review/libs/config/{github.py → 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/{gitlab.py → 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

ai_review/libs/constants/llm_provider.py

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

ai_review/libs/http/transports/retry.py

@@ -36,7 +36,7 @@ class RetryTransport(AsyncBaseTransport):
                 return last_response
 
             self.logger.warning(
-                f"Attempt {attempt}/{self.max_retries} failed "
+                f"Attempt {attempt + 1}/{self.max_retries} failed "
                 f"with status={last_response.status_code} for {request.method} {request.url}. "
                 f"Retrying in {self.retry_delay:.1f}s..."
             )

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}")

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,
+        )

ai_review/tests/fixtures/clients/claude.py

@@ -1,10 +1,55 @@
+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.claude import ClaudeMetaConfig, ClaudeHTTPClientConfig
-from ai_review.libs.config.llm import ClaudeLLMConfig
+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

ai_review/tests/fixtures/clients/gemini.py

@@ -1,10 +1,62 @@
+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.gemini import GeminiMetaConfig, GeminiHTTPClientConfig
-from ai_review.libs.config.llm import GeminiLLMConfig
+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

ai_review/tests/fixtures/clients/github.py

@@ -18,8 +18,8 @@ from ai_review.clients.github.pr.schema.pull_request import (
 from ai_review.clients.github.pr.schema.reviews import GitHubGetPRReviewsResponseSchema, GitHubPRReviewSchema
 from ai_review.clients.github.pr.types import GitHubPullRequestsHTTPClientProtocol
 from ai_review.config import settings
-from ai_review.libs.config.github import GitHubPipelineConfig, GitHubHTTPClientConfig
-from ai_review.libs.config.vcs import GitHubVCSConfig
+from ai_review.libs.config.vcs.base import GitHubVCSConfig
+from ai_review.libs.config.vcs.github import GitHubPipelineConfig, GitHubHTTPClientConfig
 from ai_review.libs.constants.vcs_provider import VCSProvider
 from ai_review.services.vcs.github.client import GitHubVCSClient
 

ai_review/tests/fixtures/clients/gitlab.py

@@ -20,8 +20,8 @@ from ai_review.clients.gitlab.mr.schema.notes import (
 )
 from ai_review.clients.gitlab.mr.types import GitLabMergeRequestsHTTPClientProtocol
 from ai_review.config import settings
-from ai_review.libs.config.gitlab import GitLabPipelineConfig, GitLabHTTPClientConfig
-from ai_review.libs.config.vcs import GitLabVCSConfig
+from ai_review.libs.config.vcs.base import GitLabVCSConfig
+from ai_review.libs.config.vcs.gitlab import GitLabPipelineConfig, GitLabHTTPClientConfig
 from ai_review.libs.constants.vcs_provider import VCSProvider
 from ai_review.services.vcs.gitlab.client import GitLabVCSClient
 

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)

ai_review/tests/fixtures/clients/openai.py

@@ -1,10 +1,58 @@
+from typing import Any
+
 import pytest
 from pydantic import HttpUrl, SecretStr
 
+from ai_review.clients.openai.schema import (
+    OpenAIUsageSchema,
+    OpenAIChoiceSchema,
+    OpenAIMessageSchema,
+    OpenAIChatRequestSchema,
+    OpenAIChatResponseSchema,
+)
+from ai_review.clients.openai.types import OpenAIHTTPClientProtocol
 from ai_review.config import settings
-from ai_review.libs.config.llm import OpenAILLMConfig
-from ai_review.libs.config.openai import OpenAIMetaConfig, OpenAIHTTPClientConfig
+from ai_review.libs.config.llm.base import OpenAILLMConfig
+from ai_review.libs.config.llm.openai import OpenAIMetaConfig, OpenAIHTTPClientConfig
 from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.openai.client import OpenAILLMClient
+
+
+class FakeOpenAIHTTPClient(OpenAIHTTPClientProtocol):
+    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: OpenAIChatRequestSchema) -> OpenAIChatResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            OpenAIChatResponseSchema(
+                usage=OpenAIUsageSchema(total_tokens=12, prompt_tokens=5, completion_tokens=7),
+                choices=[
+                    OpenAIChoiceSchema(
+                        message=OpenAIMessageSchema(role="assistant", content="FAKE_OPENAI_RESPONSE")
+                    )
+                ],
+            ),
+        )
+
+
+@pytest.fixture
+def fake_openai_http_client():
+    return FakeOpenAIHTTPClient()
+
+
+@pytest.fixture
+def openai_llm_client(
+        monkeypatch: pytest.MonkeyPatch,
+        fake_openai_http_client: FakeOpenAIHTTPClient
+) -> OpenAILLMClient:
+    monkeypatch.setattr(
+        "ai_review.services.llm.openai.client.get_openai_http_client",
+        lambda: fake_openai_http_client,
+    )
+    return OpenAILLMClient()
 
 
 @pytest.fixture

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

@@ -0,0 +1,12 @@
+import pytest
+from httpx import AsyncClient
+
+from ai_review.clients.ollama.client import get_ollama_http_client, OllamaHTTPClient
+
+
+@pytest.mark.usefixtures('ollama_http_client_config')
+def test_get_ollama_http_client_builds_ok():
+    ollama_http_client = get_ollama_http_client()
+
+    assert isinstance(ollama_http_client, OllamaHTTPClient)
+    assert isinstance(ollama_http_client.client, AsyncClient)

ai_review/tests/suites/clients/ollama/test_schema.py

@@ -0,0 +1,65 @@
+from ai_review.clients.ollama.schema import (
+    OllamaMessageSchema,
+    OllamaOptionsSchema,
+    OllamaChatRequestSchema,
+    OllamaUsageSchema,
+    OllamaChatResponseSchema,
+)
+
+
+# ---------- OllamaUsageSchema ----------
+
+def test_usage_total_tokens_sum_ok():
+    usage = OllamaUsageSchema(prompt_tokens=5, completion_tokens=7)
+    assert usage.total_tokens == 12
+
+
+def test_usage_total_tokens_none_if_missing():
+    usage = OllamaUsageSchema(prompt_tokens=3)
+    assert usage.total_tokens is None
+
+
+# ---------- OllamaChatResponseSchema ----------
+
+def test_first_text_returns_text():
+    resp = OllamaChatResponseSchema(
+        model="llama2",
+        message=OllamaMessageSchema(role="assistant", content=" hello ollama "),
+        usage=OllamaUsageSchema(prompt_tokens=2, completion_tokens=3),
+    )
+    assert resp.first_text == "hello ollama"
+
+
+def test_first_text_empty_if_content_empty():
+    resp = OllamaChatResponseSchema(
+        model="llama2",
+        message=OllamaMessageSchema(role="assistant", content="   "),
+        usage=OllamaUsageSchema(prompt_tokens=1, completion_tokens=1),
+    )
+    assert resp.first_text == ""
+
+
+# ---------- OllamaChatRequestSchema ----------
+
+def test_chat_request_schema_builds_ok():
+    msg = OllamaMessageSchema(role="user", content="hi ollama")
+    opts = OllamaOptionsSchema(
+        stop=["stop1", "stop2"],
+        seed=123,
+        top_p=0.9,
+        temperature=0.7,
+        num_predict=256,
+        repeat_penalty=1.1,
+    )
+    req = OllamaChatRequestSchema(
+        model="llama2",
+        stream=False,
+        options=opts,
+        messages=[msg],
+    )
+
+    assert req.model == "llama2"
+    assert req.options.temperature == 0.7
+    assert req.options.num_predict == 256
+    assert req.options.stop == ["stop1", "stop2"]
+    assert req.messages[0].content == "hi ollama"

ai_review/tests/suites/services/cost/test_service.py

@@ -1,6 +1,6 @@
 import pytest
 
-from ai_review.libs.config.llm import LLMPricingConfig, LLMConfigBase
+from ai_review.libs.config.llm.base import LLMPricingConfig, LLMConfigBase
 from ai_review.services.cost.schema import CostReportSchema
 from ai_review.services.cost.service import CostService
 from ai_review.services.llm.types import ChatResultSchema

ai_review/tests/suites/services/llm/claude/test_client.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ai_review.services.llm.claude.client import ClaudeLLMClient
+from ai_review.services.llm.types import ChatResultSchema
+from ai_review.tests.fixtures.clients.claude import FakeClaudeHTTPClient
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("claude_http_client_config")
+async def test_claude_llm_chat(
+        claude_llm_client: ClaudeLLMClient,
+        fake_claude_http_client: FakeClaudeHTTPClient
+):
+    result = await claude_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_CLAUDE_RESPONSE"
+    assert result.total_tokens == 12
+    assert result.prompt_tokens == 5
+    assert result.completion_tokens == 7
+
+    assert fake_claude_http_client.calls[0][0] == "chat"

ai_review/tests/suites/services/llm/gemini/test_client.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ai_review.services.llm.gemini.client import GeminiLLMClient
+from ai_review.services.llm.types import ChatResultSchema
+from ai_review.tests.fixtures.clients.gemini import FakeGeminiHTTPClient
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("gemini_http_client_config")
+async def test_gemini_llm_chat(
+        gemini_llm_client: GeminiLLMClient,
+        fake_gemini_http_client: FakeGeminiHTTPClient
+):
+    result = await gemini_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_GEMINI_RESPONSE"
+    assert result.total_tokens == 10
+    assert result.prompt_tokens == 2
+    assert result.completion_tokens is None
+
+    assert fake_gemini_http_client.calls[0][0] == "chat"

ai_review/tests/suites/services/llm/ollama/test_client.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ai_review.services.llm.ollama.client import OllamaLLMClient
+from ai_review.services.llm.types import ChatResultSchema
+from ai_review.tests.fixtures.clients.ollama import FakeOllamaHTTPClient
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("ollama_http_client_config")
+async def test_ollama_llm_chat(
+        ollama_llm_client: OllamaLLMClient,
+        fake_ollama_http_client: FakeOllamaHTTPClient
+):
+    result = await ollama_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_OLLAMA_RESPONSE"
+    assert result.total_tokens == 8
+    assert result.prompt_tokens == 3
+    assert result.completion_tokens == 5
+
+    assert fake_ollama_http_client.calls[0][0] == "chat"

ai_review/tests/suites/services/llm/openai/test_client.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ai_review.services.llm.openai.client import OpenAILLMClient
+from ai_review.services.llm.types import ChatResultSchema
+from ai_review.tests.fixtures.clients.openai import FakeOpenAIHTTPClient
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("openai_http_client_config")
+async def test_openai_llm_chat(
+        openai_llm_client: OpenAILLMClient,
+        fake_openai_http_client: FakeOpenAIHTTPClient
+):
+    result = await openai_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_OPENAI_RESPONSE"
+    assert result.total_tokens == 12
+    assert result.prompt_tokens == 5
+    assert result.completion_tokens == 7
+
+    assert fake_openai_http_client.calls[0][0] == "chat"

ai_review/tests/suites/services/llm/test_factory.py

@@ -3,6 +3,7 @@ import pytest
 from ai_review.services.llm.claude.client import ClaudeLLMClient
 from ai_review.services.llm.factory import get_llm_client
 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
 
 
@@ -24,7 +25,13 @@ def test_get_llm_client_returns_claude(monkeypatch: pytest.MonkeyPatch):
     assert isinstance(client, ClaudeLLMClient)
 
 
+@pytest.mark.usefixtures("ollama_http_client_config")
+def test_get_llm_client_returns_ollama(monkeypatch: pytest.MonkeyPatch):
+    client = get_llm_client()
+    assert isinstance(client, OllamaLLMClient)
+
+
 def test_get_llm_client_unsupported_provider(monkeypatch: pytest.MonkeyPatch):
-    monkeypatch.setattr("ai_review.services.llm.factory.settings.llm.provider", "DEEPSEEK")
+    monkeypatch.setattr("ai_review.services.llm.factory.settings.llm.provider", "UNSUPPORTED")
     with pytest.raises(ValueError):
         get_llm_client()

{xai_review-0.23.0.dist-info → xai_review-0.25.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.23.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.23.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.23.0.dist-info → xai_review-0.25.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ai_review/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/config.py,sha256=tPda0l_EX6MWHtKXE6HLFIkzYqLxA2yogSQDMeE5MAM,1954
+ai_review/config.py,sha256=LmHYNW44hi9qUhGxUjwoZBNrfu49-TcZD7WyGpfbpPI,1964
 ai_review/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/cli/main.py,sha256=rZ0LYSAt3AFT-wOHdop8lq2GOVAK48kJYp7nf801Mjs,1854
 ai_review/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -9,11 +9,13 @@ ai_review/cli/commands/run_review.py,sha256=i39IYNDE_lAiQQnKLmxG71Ao8WAIOSn82L9E
 ai_review/cli/commands/run_summary_review.py,sha256=NqjepGH5cbqczPzcuMEAxO4dI58FEUZl0b6uRVQ9SiA,224
 ai_review/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/clients/claude/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/claude/client.py,sha256=MKN7W5RRvV_zRWZLCLx75ATj-gtkef-YE7-Yd7C3ke8,1796
+ai_review/clients/claude/client.py,sha256=uEadbBNBJnzjHDczbxXiiw1V1H1PdUWKu-Gn-eIDEmw,1890
 ai_review/clients/claude/schema.py,sha256=LE6KCjJKDXqBGU2Cno5XL5R8vUfScgskE9MqvE0Pt2A,887
+ai_review/clients/claude/types.py,sha256=y_-yF7zQrTvyiowS2b9xjIlAzkF8i6OfOjqo9eB8Xo4,267
 ai_review/clients/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/gemini/client.py,sha256=a-QY_4lDgZgUcEZpaeNjrqthWhXQ7iqdE_8pQROxqR0,1795
+ai_review/clients/gemini/client.py,sha256=4G1LBcpiFcrITOysQbMwhY1db4hHcSGgyI-0XazZMV0,1889
 ai_review/clients/gemini/schema.py,sha256=5oVvbI-h_sw8bFreS4JUmMj-aXa_frvxK3H8sg4iJIA,2264
+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/pr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -33,9 +35,14 @@ ai_review/clients/gitlab/mr/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQe
 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/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
+ai_review/clients/ollama/types.py,sha256=9ES8K-EClKYU7UsaMKgXvZ3sUOF9o6reEvfL6wFOJ4M,267
 ai_review/clients/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/clients/openai/client.py,sha256=IWIl5um4chLiz45Hw1hd_KAz7jkXpJp0l2qf0ztMUV0,1722
+ai_review/clients/openai/client.py,sha256=jY1XG_5GtNboNjkXu3UtuXFx5V9rD6UskK7VT0lOzP8,1816
 ai_review/clients/openai/schema.py,sha256=glxwMtBrDA6W0BQgH-ruKe0bKH3Ps1P-Y1-2jGdqaUM,764
+ai_review/clients/openai/types.py,sha256=4VRY45ihKjii8w0d5XLnUGnHuBSh9wRsOP6lmkseC0Q,267
 ai_review/libs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/json.py,sha256=g-P5_pNUomQ-bGHCXASvPKj9Og0s9MaLFVEAkzqGp1A,350
 ai_review/libs/logger.py,sha256=LbXR2Zk1btJ-83I-vHee7cUETgT1mHToSsqEI_8uM0U,370
@@ -45,20 +52,24 @@ ai_review/libs/asynchronous/gather.py,sha256=wH65sqBfrnwA1A9Juc5MSyLCJrcxzRqk2m0
 ai_review/libs/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/config/artifacts.py,sha256=8BzbQu5GxwV6i6qzrUKM1De1Ogb00Ph5WTqwZ3fVpGg,483
 ai_review/libs/config/base.py,sha256=sPf3OKeF1ID0ouOwiVaUtvpWuZXJXQvIw5kbnPUyN9o,686
-ai_review/libs/config/claude.py,sha256=E9AJmszfY4TH8PkJjnDDDJYNAU9bLGsUThM3kriVA58,302
 ai_review/libs/config/core.py,sha256=ZQ2QtYr7vAF0tXbVLvVwk9QFE5h6JjAKAUQWcb9gHws,87
-ai_review/libs/config/gemini.py,sha256=sXHud43LWb4xTvhdkGQeHSLC7qvWl5LfU41fgcIVE5E,274
-ai_review/libs/config/github.py,sha256=1yFfvkTOt5ernIrxjqmiUKDpbEEHpa6lTpDiFQ5gVn4,238
-ai_review/libs/config/gitlab.py,sha256=VFvoVtni86tWky6Y34XCYdNrBuAtbgFFYGK3idPSOS4,234
-ai_review/libs/config/http.py,sha256=QsIj0yH1IYELOFBQ5AoqPZT0kGIIrQ19cxk1ozPRhLE,345
-ai_review/libs/config/llm.py,sha256=cK-e4NCQxnnixLATCsO8-r5k3zUWz1N0BdPCoqerORM,1824
+ai_review/libs/config/http.py,sha256=dx5PwgnGbPocUwf9QRhFmXmjfFDoeerOM04yB3B6S8w,398
 ai_review/libs/config/logger.py,sha256=oPmjpjf6EZwW7CgOjT8mOQdGnT98CLwXepiGB_ajZvU,384
-ai_review/libs/config/openai.py,sha256=vOYqhUq0ceEuNdQrQaHq44lVS5M648mB61Zc4YlfJVw,271
 ai_review/libs/config/prompt.py,sha256=8aO5WNnhVhQcpWzWxqzb9lq6PzormaJazVwPHuf_ia8,4469
 ai_review/libs/config/review.py,sha256=LEZni68iH_0m4URPfN0d3F6yrrK7KSn-BwXf-7w2al8,1058
-ai_review/libs/config/vcs.py,sha256=ULuLicuulFgG-_sTuDWsldyVWIT2SkiS8brPUU1svgk,778
+ai_review/libs/config/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/libs/config/llm/base.py,sha256=ovvULFhfwH66_705D1O87ZGMeaQOZO7ZQhRUzzfzguU,2089
+ai_review/libs/config/llm/claude.py,sha256=MoalXkBA6pEp01znS8ohTRopfea9RUcqhZX5lOIuek8,293
+ai_review/libs/config/llm/gemini.py,sha256=SKtlzsRuNWOlM9m3SFvcqOIjnml8lpPidp7FiGmIEz4,265
+ai_review/libs/config/llm/meta.py,sha256=cEcAHOwy-mQBKo9_KJrQe0I7qppq6h99lSmoWX4ElJI,195
+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=zkfqBnI9kF_wbU9Nan19CKciKwOpWvdcZEt57jHujbE,786
+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/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/libs/constants/llm_provider.py,sha256=sKnDLylCIIosYjq0-0r91LMiYJ4DlHVH2jeRDv_DlsQ,121
+ai_review/libs/constants/llm_provider.py,sha256=k7GzctIZ-TDsRlhTPbpGYgym_CO2YKVFp_oXG9dTBW0,143
 ai_review/libs/constants/vcs_provider.py,sha256=mZMC8DWIDWQ1YeUZh1a1jduX5enOAe1rWeza0RBmpTY,99
 ai_review/libs/diff/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/diff/models.py,sha256=RT4YJboOPA-AjNJGRj_HIZaJLEmROOhOgMh1wIGpIwY,2344
@@ -71,7 +82,7 @@ ai_review/libs/http/event_hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
 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
 ai_review/libs/http/transports/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/libs/http/transports/retry.py,sha256=OIhTplK22RigjF5A8M9FdMm01BLyWpvQnuK2__NC678,1768
+ai_review/libs/http/transports/retry.py,sha256=S66-SKvJvIgqBknDgEnqciTB0g2-ZQ7JubQtkzv8P44,1772
 ai_review/libs/template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/libs/template/render.py,sha256=PwLG46fXg8P3gZvmJB93P51G2IBdsEK2I8oDlLGmA-4,414
 ai_review/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -107,12 +118,14 @@ ai_review/services/hook/constants.py,sha256=uQJld5tJVUFk506h5RswTqLy-sIYxQfuQcUw
 ai_review/services/hook/service.py,sha256=InPoWBas6SPoy0KUyKJFg5xVk90jBlWdWtUTaX71G88,6364
 ai_review/services/hook/types.py,sha256=zwOmnZVGlg53vUoC2rHNhpEiNsTpf0Tnb-s3SRPKFys,1405
 ai_review/services/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/services/llm/factory.py,sha256=EHCRA5asqyE86Utn-EBYe_HbWRzck0S0UJG1gm5f2uo,741
+ai_review/services/llm/factory.py,sha256=AszDqufYPaZdVVR99UZBEFvnGdOildBFQ9uVOiBI1Tc,876
 ai_review/services/llm/types.py,sha256=OvbJWYRDThBgLhn9TWU0mliuanOW01CS3e8ermtuS-s,353
 ai_review/services/llm/claude/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/llm/claude/client.py,sha256=JJD0FWiXjCCpO7NW3vVoBMXhTQ9VBA4Q93QqkeQqON0,1082
 ai_review/services/llm/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/llm/gemini/client.py,sha256=TR4HshVxtDV8_luQKCM3aFNH9tjAjpzNeFBg-oxdsfA,1282
+ai_review/services/llm/ollama/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/services/llm/ollama/client.py,sha256=817nOQRsnaVqoY6LdO95l5JkRHkGvvS8TX7hezT2gqk,1479
 ai_review/services/llm/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/llm/openai/client.py,sha256=c3DWwLnwTheERdSGnMiQIbg5SaICouUAGClcQZSh1fE,1159
 ai_review/services/prompt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -146,11 +159,12 @@ ai_review/services/vcs/gitlab/client.py,sha256=LK95m-uFSxhDEVU-cBGct61NTKjul-ieL
 ai_review/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/fixtures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/fixtures/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/tests/fixtures/clients/claude.py,sha256=CpB65Twi2Q4kIOclflnyT--wGNrLPCEEgt9P_qyhu10,759
-ai_review/tests/fixtures/clients/gemini.py,sha256=UXlLEqyXlei3ojrVCYRX0ic3Lq7iBNrG6BFZ8Ec_K7c,737
-ai_review/tests/fixtures/clients/github.py,sha256=vZu_Tf8RlvYKDyMUnlP9Uc8fL95HRs1VTv7zWxIH_eQ,6856
-ai_review/tests/fixtures/clients/gitlab.py,sha256=qyzp6lYZ9EQz-ECujFa5WK_xSbE-hmy35cfnW11mS6g,5531
-ai_review/tests/fixtures/clients/openai.py,sha256=O30n-tCfpeCxeUQgrQPK5ll__buhbeShobK7LASFMls,721
+ai_review/tests/fixtures/clients/claude.py,sha256=6ldJlSSea0zsZV0hRDMi9mqWm0hWT3mp_ROwG_sVU1c,2203
+ai_review/tests/fixtures/clients/gemini.py,sha256=zhLJhm49keKEBCPOf_pLu8_zCatsKKAWM4-gXOhaXeM,2429
+ai_review/tests/fixtures/clients/github.py,sha256=Mzr8LcvVlYLhimzDMG4tEOQwj_6E6kTvYvSrq04R3YI,6865
+ai_review/tests/fixtures/clients/gitlab.py,sha256=_0JSN-ixA7nDOwY18BlL_L9fh_qmT1_6sxGx_CIRhmM,5540
+ai_review/tests/fixtures/clients/ollama.py,sha256=UUHDDPUraQAG8gBC-0UvftaK0BDYir5cJDlRKJymSQg,2109
+ai_review/tests/fixtures/clients/openai.py,sha256=UgfuRZWzl3X7ZVHMLKP4mZxNXVpcccitkc9tuUyffXE,2267
 ai_review/tests/fixtures/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/fixtures/services/artifacts.py,sha256=V5FvUnC9OAo0n-paxxJP5OxAgLz1Zz3OZ8zZvqu_01w,1462
 ai_review/tests/fixtures/services/cost.py,sha256=A6Ja0CtQ-k6pR2-B5LRE8EzkqPL34xHGXYtaILjhYvw,1612
@@ -174,6 +188,9 @@ ai_review/tests/suites/clients/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCe
 ai_review/tests/suites/clients/github/test_client.py,sha256=BiuLKCHIk83U1szYEZkB-n3vvyPgj6tAI5EqxKiT-CY,558
 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/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
 ai_review/tests/suites/clients/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/clients/openai/test_client.py,sha256=6Wsxw6-6Uk0uPYFkzpWSwsxfCYUZhT3UYznayo-xlPI,404
 ai_review/tests/suites/clients/openai/test_schema.py,sha256=x1tamS4GC9pOTpjieKDbK2D73CVV4BkATppytwMevLo,1599
@@ -192,7 +209,7 @@ ai_review/tests/suites/libs/template/test_render.py,sha256=n-ss5bd_hwc-RzYmqWmFM
 ai_review/tests/suites/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/cost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/cost/test_schema.py,sha256=AI3Wg1sR6nzLpkEqJGDu6nDYwiwzbbghsxhRNwRsUFA,3044
-ai_review/tests/suites/services/cost/test_service.py,sha256=fMW4Tg6BRMXKcqOO7MmSqJc1mpuguvFSl0GjS93m7u8,3253
+ai_review/tests/suites/services/cost/test_service.py,sha256=9_Mi5hu2cq3w2tIEPfhrn9x8SblCT5m1W-QUOc9BZds,3258
 ai_review/tests/suites/services/diff/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/diff/test_renderers.py,sha256=IKOpsGedONNW8ZfYTAk0Vq0hfFi7L6TpWs8vVVQroj0,6273
 ai_review/tests/suites/services/diff/test_service.py,sha256=evFe1I-ulOYdkhQS9NC8Bol4jXuhEnb2iArTAzckWrA,3195
@@ -200,7 +217,15 @@ ai_review/tests/suites/services/diff/test_tools.py,sha256=vsOSSIDZKkuD8dMCoBBEBt
 ai_review/tests/suites/services/hook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/hook/test_service.py,sha256=GM_AiNVGP2Pgp-3BwGOAIfA8lLXl6ah28ey77KZz_C4,2750
 ai_review/tests/suites/services/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/tests/suites/services/llm/test_factory.py,sha256=_i_UFtG_WGT3jpBDm20Hb0rFTFrfPuiFJhhSrlvUlVQ,1120
+ai_review/tests/suites/services/llm/test_factory.py,sha256=Lp37aXM08fHaLzgwRBw5xZEDZkPDNhJ4qwjtEwGENv8,1394
+ai_review/tests/suites/services/llm/claude/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/llm/claude/test_client.py,sha256=ymIeuIax0Bp_CuXBSApK1RDl1JmbGc97uzXZToQOZO8,761
+ai_review/tests/suites/services/llm/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/llm/gemini/test_client.py,sha256=RjYViMZTgTdbzmDpOvwjuwYVkQV3IyNRhxZ8Y_cfJiQ,764
+ai_review/tests/suites/services/llm/ollama/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/llm/ollama/test_client.py,sha256=Eu4OERB00SJwCKznyOCyqSFTDBp9J2Lw-BcW7sPJQM4,760
+ai_review/tests/suites/services/llm/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/llm/openai/test_client.py,sha256=yzIL8GYHyX9iLKIlaF__87aue9w0cr66feoMaCv5gms,761
 ai_review/tests/suites/services/prompt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/prompt/test_adapter.py,sha256=9KZOFQmZUs3l_cW7Q5LIMPs4i4J-gOCQ6VrlDPR0ImU,2156
 ai_review/tests/suites/services/prompt/test_schema.py,sha256=rm2__LA2_4qQwSmNAZ_Wnpy11T3yYRkYUkRUrqxUQKE,5421
@@ -222,9 +247,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.23.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
-xai_review-0.23.0.dist-info/METADATA,sha256=6GjvVteV4O_MPSSgTB4dNhlbS7iNKlGO5Uei0Y6JAuQ,10872
-xai_review-0.23.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xai_review-0.23.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
-xai_review-0.23.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
-xai_review-0.23.0.dist-info/RECORD,,
+xai_review-0.25.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
+xai_review-0.25.0.dist-info/METADATA,sha256=-Ogp0iZijYiluzi9D_yO21Ur9N9Z4qJclmSeHrddU3A,11132
+xai_review-0.25.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xai_review-0.25.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
+xai_review-0.25.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
+xai_review-0.25.0.dist-info/RECORD,,

ai_review/libs/config/claude.py

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

ai_review/libs/config/gemini.py

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

ai_review/libs/config/openai.py

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