xai-review 0.33.0__tar.gz → 0.34.0__tar.gz

{xai_review-0.33.0/xai_review.egg-info → xai_review-0.34.0}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.33.0
-Summary: AI-powered code review tool
+Version: 0.34.0
+Summary: AI-powered code review tool for GitHub, GitLab, Bitbucket and Gitea — built with LLMs like OpenAI, Claude, Gemini, Ollama, and OpenRouter
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>
 License: Apache-2.0
 Project-URL: Issues, https://github.com/Nikita-Filonov/ai-review/issues
 Project-URL: Homepage, https://github.com/Nikita-Filonov/ai-review
 Project-URL: Repository, https://github.com/Nikita-Filonov/ai-review
-Keywords: ai,code review,llm,openai,claude,gemini
+Keywords: ai,code review,llm,openai,claude,gemini,ollama,openrouter,ci/cd,gitlab,github,gitea,bitbucket
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -66,7 +66,8 @@ improve code quality, enforce consistency, and speed up the review process.
 
 ✨ Key features:
 
-- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, or **Ollama**, and switch anytime.
+- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, **Ollama**, or **OpenRouter**, and
+  switch anytime.
 - **VCS integration** — works out of the box with **GitLab**, **GitHub**, **Bitbucket**, and **Gitea**.
 - **Customizable prompts** — adapt inline, context, and summary reviews to match your team’s coding guidelines.
 - **Reply modes** — AI can now **participate in existing review threads**, adding follow-up replies in both inline and
@@ -120,7 +121,7 @@ pip install xai-review
 Or run directly via Docker:
 
 ```bash
-docker run --rm -v $(pwd):/app nikitafilonov/ai-review:latest run-summary
+docker run --rm -v $(pwd):/app nikitafilonov/ai-review:latest ai-review run-summary
 ```
 
 🐳 Pull from [DockerHub](https://hub.docker.com/r/nikitafilonov/ai-review)
@@ -146,8 +147,8 @@ vcs:
   provider: GITLAB
 
   pipeline:
-    project_id: 1
-    merge_request_id: 100
+    project_id: "1"
+    merge_request_id: "100"
 
   http_client:
     timeout: 120
@@ -176,7 +177,7 @@ for complete, ready-to-use examples.
 
 Key things you can customize:
 
-- **LLM provider** — OpenAI, Gemini, Claude, or Ollama
+- **LLM provider** — OpenAI, Gemini, Claude, Ollama, or OpenRouter
 - **Model settings** — model name, temperature, max tokens
 - **VCS integration** — works out of the box with **GitLab**, **GitHub**, **Bitbucket**, and **Gitea**
 - **Review policy** — which files to include/exclude, review modes
@@ -220,7 +221,7 @@ jobs:
         with:
           fetch-depth: 0
 
-      - uses: Nikita-Filonov/ai-review@v0.33.0
+      - uses: Nikita-Filonov/ai-review@v0.34.0
         with:
           review-command: ${{ inputs.review-command }}
         env:
@@ -298,7 +299,7 @@ AI Review does **not store**, **log**, or **transmit** your source code to any e
 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.
+Claude, OpenRouter). 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

{xai_review-0.33.0 → xai_review-0.34.0}/README.md

@@ -32,7 +32,8 @@ improve code quality, enforce consistency, and speed up the review process.
 
 ✨ Key features:
 
-- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, or **Ollama**, and switch anytime.
+- **Multiple LLM providers** — choose between **OpenAI**, **Claude**, **Gemini**, **Ollama**, or **OpenRouter**, and
+  switch anytime.
 - **VCS integration** — works out of the box with **GitLab**, **GitHub**, **Bitbucket**, and **Gitea**.
 - **Customizable prompts** — adapt inline, context, and summary reviews to match your team’s coding guidelines.
 - **Reply modes** — AI can now **participate in existing review threads**, adding follow-up replies in both inline and
@@ -86,7 +87,7 @@ pip install xai-review
 Or run directly via Docker:
 
 ```bash
-docker run --rm -v $(pwd):/app nikitafilonov/ai-review:latest run-summary
+docker run --rm -v $(pwd):/app nikitafilonov/ai-review:latest ai-review run-summary
 ```
 
 🐳 Pull from [DockerHub](https://hub.docker.com/r/nikitafilonov/ai-review)
@@ -112,8 +113,8 @@ vcs:
   provider: GITLAB
 
   pipeline:
-    project_id: 1
-    merge_request_id: 100
+    project_id: "1"
+    merge_request_id: "100"
 
   http_client:
     timeout: 120
@@ -142,7 +143,7 @@ for complete, ready-to-use examples.
 
 Key things you can customize:
 
-- **LLM provider** — OpenAI, Gemini, Claude, or Ollama
+- **LLM provider** — OpenAI, Gemini, Claude, Ollama, or OpenRouter
 - **Model settings** — model name, temperature, max tokens
 - **VCS integration** — works out of the box with **GitLab**, **GitHub**, **Bitbucket**, and **Gitea**
 - **Review policy** — which files to include/exclude, review modes
@@ -186,7 +187,7 @@ jobs:
         with:
           fetch-depth: 0
 
-      - uses: Nikita-Filonov/ai-review@v0.33.0
+      - uses: Nikita-Filonov/ai-review@v0.34.0
         with:
           review-command: ${{ inputs.review-command }}
         env:
@@ -264,7 +265,7 @@ AI Review does **not store**, **log**, or **transmit** your source code to any e
 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.
+Claude, OpenRouter). 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

xai_review-0.34.0/ai_review/clients/openrouter/client.py

@@ -0,0 +1,50 @@
+from httpx import AsyncClient, Response, AsyncHTTPTransport
+
+from ai_review.clients.openrouter.schema import OpenRouterChatRequestSchema, OpenRouterChatResponseSchema
+from ai_review.clients.openrouter.types import OpenRouterHTTPClientProtocol
+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 OpenRouterHTTPClientError(HTTPClientError):
+    pass
+
+
+class OpenRouterHTTPClient(HTTPClient, OpenRouterHTTPClientProtocol):
+    @handle_http_error(client="OpenRouterHTTPClient", exception=OpenRouterHTTPClientError)
+    async def chat_api(self, request: OpenRouterChatRequestSchema) -> Response:
+        return await self.post("/chat/completions", json=request.model_dump())
+
+    async def chat(self, request: OpenRouterChatRequestSchema) -> OpenRouterChatResponseSchema:
+        response = await self.chat_api(request)
+        return OpenRouterChatResponseSchema.model_validate_json(response.text)
+
+
+def get_openrouter_http_client() -> OpenRouterHTTPClient:
+    logger = get_logger("OPENROUTER_HTTP_CLIENT")
+    logger_event_hook = LoggerEventHook(logger=logger)
+    retry_transport = RetryTransport(logger=logger, transport=AsyncHTTPTransport())
+
+    headers = {"Authorization": f"Bearer {settings.llm.http_client.api_token_value}"}
+    if settings.llm.meta.title:
+        headers["X-Title"] = settings.llm.meta.title
+
+    if settings.llm.meta.referer:
+        headers["Referer"] = settings.llm.meta.referer
+
+    client = AsyncClient(
+        timeout=settings.llm.http_client.timeout,
+        headers=headers,
+        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 OpenRouterHTTPClient(client=client)

xai_review-0.34.0/ai_review/clients/openrouter/schema.py

@@ -0,0 +1,36 @@
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class OpenRouterUsageSchema(BaseModel):
+    total_tokens: int
+    prompt_tokens: int
+    completion_tokens: int
+
+
+class OpenRouterMessageSchema(BaseModel):
+    role: Literal["system", "user", "assistant"]
+    content: str
+
+
+class OpenRouterChoiceSchema(BaseModel):
+    message: OpenRouterMessageSchema
+
+
+class OpenRouterChatRequestSchema(BaseModel):
+    model: str
+    messages: list[OpenRouterMessageSchema]
+    max_tokens: int
+    temperature: float
+
+
+class OpenRouterChatResponseSchema(BaseModel):
+    usage: OpenRouterUsageSchema
+    choices: list[OpenRouterChoiceSchema]
+
+    @property
+    def first_text(self) -> str:
+        if not self.choices:
+            return ""
+        return (self.choices[0].message.content or "").strip()

xai_review-0.34.0/ai_review/clients/openrouter/types.py

@@ -0,0 +1,11 @@
+from typing import Protocol
+
+from ai_review.clients.openrouter.schema import (
+    OpenRouterChatRequestSchema,
+    OpenRouterChatResponseSchema
+)
+
+
+class OpenRouterHTTPClientProtocol(Protocol):
+    async def chat(self, request: OpenRouterChatRequestSchema) -> OpenRouterChatResponseSchema:
+        ...

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/libs/config/llm/base.py

@@ -8,6 +8,7 @@ from ai_review.libs.config.llm.claude import ClaudeHTTPClientConfig, ClaudeMetaC
 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.config.llm.openrouter import OpenRouterHTTPClientConfig, OpenRouterMetaConfig
 from ai_review.libs.constants.llm_provider import LLMProvider
 from ai_review.libs.resources import load_resource
 
@@ -62,7 +63,13 @@ class OllamaLLMConfig(LLMConfigBase):
     http_client: OllamaHTTPClientConfig
 
 
+class OpenRouterLLMConfig(LLMConfigBase):
+    meta: OpenRouterMetaConfig
+    provider: Literal[LLMProvider.OPENROUTER]
+    http_client: OpenRouterHTTPClientConfig
+
+
 LLMConfig = Annotated[
-    OpenAILLMConfig | GeminiLLMConfig | ClaudeLLMConfig | OllamaLLMConfig,
+    OpenAILLMConfig | GeminiLLMConfig | ClaudeLLMConfig | OllamaLLMConfig | OpenRouterLLMConfig,
     Field(discriminator="provider")
 ]

xai_review-0.34.0/ai_review/libs/config/llm/openrouter.py

@@ -0,0 +1,12 @@
+from ai_review.libs.config.http import HTTPClientWithTokenConfig
+from ai_review.libs.config.llm.meta import LLMMetaConfig
+
+
+class OpenRouterMetaConfig(LLMMetaConfig):
+    model: str = "openai/gpt-4o-mini"
+    title: str | None = None
+    referer: str | None = None
+
+
+class OpenRouterHTTPClientConfig(HTTPClientWithTokenConfig):
+    pass

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/libs/constants/llm_provider.py

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

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/services/llm/factory.py

@@ -4,6 +4,7 @@ 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.openrouter.client import OpenRouterLLMClient
 from ai_review.services.llm.types import LLMClientProtocol
 
 
@@ -17,5 +18,7 @@ def get_llm_client() -> LLMClientProtocol:
             return ClaudeLLMClient()
         case LLMProvider.OLLAMA:
             return OllamaLLMClient()
+        case LLMProvider.OPENROUTER:
+            return OpenRouterLLMClient()
         case _:
             raise ValueError(f"Unsupported LLM provider: {settings.llm.provider}")

xai_review-0.34.0/ai_review/services/llm/openrouter/client.py

@@ -0,0 +1,31 @@
+from ai_review.clients.openrouter.client import get_openrouter_http_client
+from ai_review.clients.openrouter.schema import (
+    OpenRouterMessageSchema,
+    OpenRouterChatRequestSchema,
+)
+from ai_review.config import settings
+from ai_review.services.llm.types import LLMClientProtocol, ChatResultSchema
+
+
+class OpenRouterLLMClient(LLMClientProtocol):
+    def __init__(self):
+        self.http_client = get_openrouter_http_client()
+
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        meta = settings.llm.meta
+        request = OpenRouterChatRequestSchema(
+            model=meta.model,
+            messages=[
+                OpenRouterMessageSchema(role="system", content=prompt_system),
+                OpenRouterMessageSchema(role="user", content=prompt),
+            ],
+            max_tokens=meta.max_tokens,
+            temperature=meta.temperature,
+        )
+        response = await self.http_client.chat(request)
+        return ChatResultSchema(
+            text=response.first_text,
+            total_tokens=response.usage.total_tokens,
+            prompt_tokens=response.usage.prompt_tokens,
+            completion_tokens=response.usage.completion_tokens,
+        )

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/fixtures/clients/openai.py

@@ -39,7 +39,7 @@ class FakeOpenAIHTTPClient(OpenAIHTTPClientProtocol):
 
 
 @pytest.fixture
-def fake_openai_http_client():
+def fake_openai_http_client() -> FakeOpenAIHTTPClient:
     return FakeOpenAIHTTPClient()
 
 

xai_review-0.34.0/ai_review/tests/fixtures/clients/openrouter.py

@@ -0,0 +1,72 @@
+from typing import Any
+
+import pytest
+from pydantic import HttpUrl, SecretStr
+
+from ai_review.clients.openrouter.schema import (
+    OpenRouterUsageSchema,
+    OpenRouterChoiceSchema,
+    OpenRouterMessageSchema,
+    OpenRouterChatRequestSchema,
+    OpenRouterChatResponseSchema,
+)
+from ai_review.clients.openrouter.types import OpenRouterHTTPClientProtocol
+from ai_review.config import settings
+from ai_review.libs.config.llm.base import OpenRouterLLMConfig
+from ai_review.libs.config.llm.openrouter import OpenRouterMetaConfig, OpenRouterHTTPClientConfig
+from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.openrouter.client import OpenRouterLLMClient
+
+
+class FakeOpenRouterHTTPClient(OpenRouterHTTPClientProtocol):
+    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: OpenRouterChatRequestSchema) -> OpenRouterChatResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            OpenRouterChatResponseSchema(
+                usage=OpenRouterUsageSchema(total_tokens=12, prompt_tokens=5, completion_tokens=7),
+                choices=[
+                    OpenRouterChoiceSchema(
+                        message=OpenRouterMessageSchema(
+                            role="assistant",
+                            content="FAKE_OPENROUTER_RESPONSE"
+                        )
+                    )
+                ],
+            ),
+        )
+
+
+@pytest.fixture
+def fake_openrouter_http_client() -> FakeOpenRouterHTTPClient:
+    return FakeOpenRouterHTTPClient()
+
+
+@pytest.fixture
+def openrouter_llm_client(
+        monkeypatch: pytest.MonkeyPatch,
+        fake_openrouter_http_client: FakeOpenRouterHTTPClient
+) -> OpenRouterLLMClient:
+    monkeypatch.setattr(
+        "ai_review.services.llm.openrouter.client.get_openrouter_http_client",
+        lambda: fake_openrouter_http_client,
+    )
+    return OpenRouterLLMClient()
+
+
+@pytest.fixture
+def openrouter_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = OpenRouterLLMConfig(
+        meta=OpenRouterMetaConfig(),
+        provider=LLMProvider.OPENROUTER,
+        http_client=OpenRouterHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://openrouter.ai/api/v1"),
+            api_token=SecretStr("fake-token"),
+        )
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)

xai_review-0.34.0/ai_review/tests/suites/clients/openrouter/test_client.py

@@ -0,0 +1,12 @@
+import pytest
+from httpx import AsyncClient
+
+from ai_review.clients.openrouter.client import get_openrouter_http_client, OpenRouterHTTPClient
+
+
+@pytest.mark.usefixtures('openrouter_http_client_config')
+def test_get_openrouter_http_client_builds_ok():
+    openrouter_http_client = get_openrouter_http_client()
+
+    assert isinstance(openrouter_http_client, OpenRouterHTTPClient)
+    assert isinstance(openrouter_http_client.client, AsyncClient)

xai_review-0.34.0/ai_review/tests/suites/clients/openrouter/test_schema.py

@@ -0,0 +1,57 @@
+from ai_review.clients.openrouter.schema import (
+    OpenRouterUsageSchema,
+    OpenRouterChoiceSchema,
+    OpenRouterMessageSchema,
+    OpenRouterChatRequestSchema,
+    OpenRouterChatResponseSchema,
+)
+
+
+# ---------- OpenRouterChatResponseSchema ----------
+
+def test_first_text_returns_text():
+    resp = OpenRouterChatResponseSchema(
+        usage=OpenRouterUsageSchema(total_tokens=5, prompt_tokens=2, completion_tokens=3),
+        choices=[
+            OpenRouterChoiceSchema(
+                message=OpenRouterMessageSchema(role="assistant", content=" hello world ")
+            )
+        ],
+    )
+    assert resp.first_text == "hello world"
+
+
+def test_first_text_empty_if_no_choices():
+    resp = OpenRouterChatResponseSchema(
+        usage=OpenRouterUsageSchema(total_tokens=1, prompt_tokens=1, completion_tokens=0),
+        choices=[],
+    )
+    assert resp.first_text == ""
+
+
+def test_first_text_strips_and_handles_empty_content():
+    resp = OpenRouterChatResponseSchema(
+        usage=OpenRouterUsageSchema(total_tokens=1, prompt_tokens=1, completion_tokens=0),
+        choices=[
+            OpenRouterChoiceSchema(
+                message=OpenRouterMessageSchema(role="assistant", content="   ")
+            )
+        ],
+    )
+    assert resp.first_text == ""
+
+
+# ---------- OpenRouterChatRequestSchema ----------
+
+def test_chat_request_schema_builds_ok():
+    msg = OpenRouterMessageSchema(role="user", content="hello")
+    req = OpenRouterChatRequestSchema(
+        model="gpt-4o-mini",
+        messages=[msg],
+        max_tokens=100,
+        temperature=0.3,
+    )
+    assert req.model == "gpt-4o-mini"
+    assert req.messages[0].content == "hello"
+    assert req.max_tokens == 100
+    assert req.temperature == 0.3

xai_review-0.34.0/ai_review/tests/suites/services/llm/openrouter/test_client.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ai_review.services.llm.openrouter.client import OpenRouterLLMClient
+from ai_review.services.llm.types import ChatResultSchema
+from ai_review.tests.fixtures.clients.openrouter import FakeOpenRouterHTTPClient
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("openrouter_http_client_config")
+async def test_openrouter_llm_chat(
+        openrouter_llm_client: OpenRouterLLMClient,
+        fake_openrouter_http_client: FakeOpenRouterHTTPClient
+):
+    result = await openrouter_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_OPENROUTER_RESPONSE"
+    assert result.total_tokens == 12
+    assert result.prompt_tokens == 5
+    assert result.completion_tokens == 7
+
+    assert fake_openrouter_http_client.calls[0][0] == "chat"

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/suites/services/llm/test_factory.py

@@ -5,6 +5,7 @@ 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
+from ai_review.services.llm.openrouter.client import OpenRouterLLMClient
 
 
 @pytest.mark.usefixtures("openai_http_client_config")
@@ -31,6 +32,12 @@ def test_get_llm_client_returns_ollama(monkeypatch: pytest.MonkeyPatch):
     assert isinstance(client, OllamaLLMClient)
 
 
+@pytest.mark.usefixtures("openrouter_http_client_config")
+def test_get_llm_client_returns_openrouter(monkeypatch: pytest.MonkeyPatch):
+    client = get_llm_client()
+    assert isinstance(client, OpenRouterLLMClient)
+
+
 def test_get_llm_client_unsupported_provider(monkeypatch: pytest.MonkeyPatch):
     monkeypatch.setattr("ai_review.services.llm.factory.settings.llm.provider", "UNSUPPORTED")
     with pytest.raises(ValueError):

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/suites/services/review/gateway/test_review_comment_gateway.py

@@ -72,7 +72,7 @@ async def test_get_summary_threads_filters_by_tag(
 
 @pytest.mark.asyncio
 async def test_has_existing_inline_comments_true(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_comment_gateway: ReviewCommentGateway,
 ):
@@ -118,7 +118,7 @@ async def test_process_inline_reply_happy_path(
 
 @pytest.mark.asyncio
 async def test_process_inline_reply_error(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_comment_gateway: ReviewCommentGateway,
 ):
@@ -151,7 +151,7 @@ async def test_process_summary_reply_success(
 
 @pytest.mark.asyncio
 async def test_process_summary_reply_error(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_comment_gateway: ReviewCommentGateway,
 ):
@@ -184,7 +184,7 @@ async def test_process_inline_comment_happy_path(
 
 @pytest.mark.asyncio
 async def test_process_inline_comment_error_fallback(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_comment_gateway: ReviewCommentGateway,
 ):
@@ -218,7 +218,7 @@ async def test_process_summary_comment_happy_path(
 
 @pytest.mark.asyncio
 async def test_process_summary_comment_error(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_comment_gateway: ReviewCommentGateway,
 ):

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/suites/services/review/gateway/test_review_dry_run_comment_gateway.py

@@ -10,7 +10,7 @@ from ai_review.tests.fixtures.services.vcs import FakeVCSClient
 
 @pytest.mark.asyncio
 async def test_process_inline_reply_dry_run_logs_and_no_vcs_calls(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_dry_run_comment_gateway: ReviewDryRunCommentGateway
 ):
@@ -26,7 +26,7 @@ async def test_process_inline_reply_dry_run_logs_and_no_vcs_calls(
 
 @pytest.mark.asyncio
 async def test_process_summary_reply_dry_run_logs_and_no_vcs_calls(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_dry_run_comment_gateway: ReviewDryRunCommentGateway
 ):
@@ -42,7 +42,7 @@ async def test_process_summary_reply_dry_run_logs_and_no_vcs_calls(
 
 @pytest.mark.asyncio
 async def test_process_inline_comment_dry_run_logs_and_no_vcs_calls(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_dry_run_comment_gateway: ReviewDryRunCommentGateway
 ):
@@ -59,7 +59,7 @@ async def test_process_inline_comment_dry_run_logs_and_no_vcs_calls(
 
 @pytest.mark.asyncio
 async def test_process_summary_comment_dry_run_logs_and_no_vcs_calls(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_dry_run_comment_gateway: ReviewDryRunCommentGateway
 ):
@@ -75,7 +75,7 @@ async def test_process_summary_comment_dry_run_logs_and_no_vcs_calls(
 
 @pytest.mark.asyncio
 async def test_process_inline_comments_iterates_all(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_vcs_client: FakeVCSClient,
         review_dry_run_comment_gateway: ReviewDryRunCommentGateway
 ):

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/suites/services/review/gateway/test_review_llm_gateway.py

@@ -27,7 +27,7 @@ async def test_ask_happy_path(
 
 @pytest.mark.asyncio
 async def test_ask_warns_on_empty_response(
-        capsys,
+        capsys: pytest.CaptureFixture,
         review_llm_gateway: ReviewLLMGateway,
         fake_llm_client: FakeLLMClient,
         fake_cost_service: FakeCostService,
@@ -49,7 +49,7 @@ async def test_ask_warns_on_empty_response(
 
 @pytest.mark.asyncio
 async def test_ask_handles_llm_error(
-        capsys,
+        capsys: pytest.CaptureFixture,
         fake_llm_client: FakeLLMClient,
         review_llm_gateway: ReviewLLMGateway,
 ):

{xai_review-0.33.0 → xai_review-0.34.0}/ai_review/tests/suites/services/review/test_service.py

@@ -63,7 +63,7 @@ async def test_run_summary_reply_review_invokes_runner(
 
 
 def test_report_total_cost_with_data(
-        capsys,
+        capsys: pytest.CaptureFixture,
         review_service: ReviewService,
         fake_cost_service: FakeCostService
 ):
@@ -87,7 +87,7 @@ def test_report_total_cost_with_data(
     assert "0.006" in output
 
 
-def test_report_total_cost_no_data(capsys, review_service: ReviewService):
+def test_report_total_cost_no_data(capsys: pytest.CaptureFixture, review_service: ReviewService):
     """Should log message when no cost data is available."""
     review_service.report_total_cost()
     output = capsys.readouterr().out