xai-review 0.33.0__py3-none-any.whl → 0.35.0__py3-none-any.whl

ai_review/tests/fixtures/clients/openai.py

@@ -3,14 +3,22 @@ from typing import Any
 import pytest
 from pydantic import HttpUrl, SecretStr
 
-from ai_review.clients.openai.schema import (
+from ai_review.clients.openai.v1.schema import (
     OpenAIUsageSchema,
     OpenAIChoiceSchema,
     OpenAIMessageSchema,
     OpenAIChatRequestSchema,
     OpenAIChatResponseSchema,
 )
-from ai_review.clients.openai.types import OpenAIHTTPClientProtocol
+from ai_review.clients.openai.v1.types import OpenAIV1HTTPClientProtocol
+from ai_review.clients.openai.v2.schema import (
+    OpenAIResponsesRequestSchema,
+    OpenAIResponsesResponseSchema,
+    OpenAIResponseUsageSchema,
+    OpenAIResponseOutputSchema,
+    OpenAIResponseContentSchema,
+)
+from ai_review.clients.openai.v2.types import OpenAIV2HTTPClientProtocol
 from ai_review.config import settings
 from ai_review.libs.config.llm.base import OpenAILLMConfig
 from ai_review.libs.config.llm.openai import OpenAIMetaConfig, OpenAIHTTPClientConfig
@@ -18,7 +26,7 @@ from ai_review.libs.constants.llm_provider import LLMProvider
 from ai_review.services.llm.openai.client import OpenAILLMClient
 
 
-class FakeOpenAIHTTPClient(OpenAIHTTPClientProtocol):
+class FakeOpenAIV1HTTPClient(OpenAIV1HTTPClientProtocol):
     def __init__(self, responses: dict[str, Any] | None = None) -> None:
         self.calls: list[tuple[str, dict]] = []
         self.responses = responses or {}
@@ -31,7 +39,39 @@ class FakeOpenAIHTTPClient(OpenAIHTTPClientProtocol):
                 usage=OpenAIUsageSchema(total_tokens=12, prompt_tokens=5, completion_tokens=7),
                 choices=[
                     OpenAIChoiceSchema(
-                        message=OpenAIMessageSchema(role="assistant", content="FAKE_OPENAI_RESPONSE")
+                        message=OpenAIMessageSchema(
+                            role="assistant",
+                            content="FAKE_OPENAI_V1_RESPONSE"
+                        )
+                    )
+                ],
+            ),
+        )
+
+
+class FakeOpenAIV2HTTPClient(OpenAIV2HTTPClientProtocol):
+    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: OpenAIResponsesRequestSchema) -> OpenAIResponsesResponseSchema:
+        self.calls.append(("chat", {"request": request}))
+        return self.responses.get(
+            "chat",
+            OpenAIResponsesResponseSchema(
+                usage=OpenAIResponseUsageSchema(
+                    total_tokens=20, input_tokens=10, output_tokens=10
+                ),
+                output=[
+                    OpenAIResponseOutputSchema(
+                        type="message",
+                        role="assistant",
+                        content=[
+                            OpenAIResponseContentSchema(
+                                type="output_text",
+                                text="FAKE_OPENAI_V2_RESPONSE"
+                            )
+                        ],
                     )
                 ],
             ),
@@ -39,31 +79,63 @@ class FakeOpenAIHTTPClient(OpenAIHTTPClientProtocol):
 
 
 @pytest.fixture
-def fake_openai_http_client():
-    return FakeOpenAIHTTPClient()
+def fake_openai_v1_http_client() -> FakeOpenAIV1HTTPClient:
+    return FakeOpenAIV1HTTPClient()
+
+
+@pytest.fixture
+def fake_openai_v2_http_client() -> FakeOpenAIV2HTTPClient:
+    return FakeOpenAIV2HTTPClient()
 
 
 @pytest.fixture
 def openai_llm_client(
         monkeypatch: pytest.MonkeyPatch,
-        fake_openai_http_client: FakeOpenAIHTTPClient
+        fake_openai_v1_http_client: FakeOpenAIV1HTTPClient,
+        fake_openai_v2_http_client: FakeOpenAIV2HTTPClient,
 ) -> OpenAILLMClient:
     monkeypatch.setattr(
-        "ai_review.services.llm.openai.client.get_openai_http_client",
-        lambda: fake_openai_http_client,
+        "ai_review.services.llm.openai.client.get_openai_v1_http_client",
+        lambda: fake_openai_v1_http_client,
+    )
+    monkeypatch.setattr(
+        "ai_review.services.llm.openai.client.get_openai_v2_http_client",
+        lambda: fake_openai_v2_http_client,
     )
     return OpenAILLMClient()
 
 
 @pytest.fixture
-def openai_http_client_config(monkeypatch: pytest.MonkeyPatch):
+def openai_v1_http_client_config(monkeypatch: pytest.MonkeyPatch):
     fake_config = OpenAILLMConfig(
-        meta=OpenAIMetaConfig(),
+        meta=OpenAIMetaConfig(
+            model="gpt-4o-mini",
+            max_tokens=1200,
+            temperature=0.3
+        ),
         provider=LLMProvider.OPENAI,
         http_client=OpenAIHTTPClientConfig(
             timeout=10,
             api_url=HttpUrl("https://api.openai.com/v1"),
             api_token=SecretStr("fake-token"),
-        )
+        ),
+    )
+    monkeypatch.setattr(settings, "llm", fake_config)
+
+
+@pytest.fixture
+def openai_v2_http_client_config(monkeypatch: pytest.MonkeyPatch):
+    fake_config = OpenAILLMConfig(
+        meta=OpenAIMetaConfig(
+            model="gpt-5",
+            max_tokens=2000,
+            temperature=0.2
+        ),
+        provider=LLMProvider.OPENAI,
+        http_client=OpenAIHTTPClientConfig(
+            timeout=10,
+            api_url=HttpUrl("https://api.openai.com/v1"),
+            api_token=SecretStr("fake-token"),
+        ),
     )
     monkeypatch.setattr(settings, "llm", fake_config)

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)

ai_review/tests/suites/clients/openai/v1/test_client.py

@@ -0,0 +1,12 @@
+import pytest
+from httpx import AsyncClient
+
+from ai_review.clients.openai.v1.client import get_openai_v1_http_client, OpenAIV1HTTPClient
+
+
+@pytest.mark.usefixtures('openai_v1_http_client_config')
+def test_get_openai_v1_http_client_builds_ok():
+    openai_http_client = get_openai_v1_http_client()
+
+    assert isinstance(openai_http_client, OpenAIV1HTTPClient)
+    assert isinstance(openai_http_client.client, AsyncClient)

ai_review/tests/suites/clients/openai/{test_schema.py → v1/test_schema.py}

@@ -1,4 +1,4 @@
-from ai_review.clients.openai.schema import (
+from ai_review.clients.openai.v1.schema import (
     OpenAIUsageSchema,
     OpenAIMessageSchema,
     OpenAIChoiceSchema,

ai_review/tests/suites/clients/openai/v2/test_client.py

@@ -0,0 +1,12 @@
+import pytest
+from httpx import AsyncClient
+
+from ai_review.clients.openai.v2.client import get_openai_v2_http_client, OpenAIV2HTTPClient
+
+
+@pytest.mark.usefixtures('openai_v2_http_client_config')
+def test_get_openai_v2_http_client_builds_ok():
+    openai_http_client = get_openai_v2_http_client()
+
+    assert isinstance(openai_http_client, OpenAIV2HTTPClient)
+    assert isinstance(openai_http_client.client, AsyncClient)

ai_review/tests/suites/clients/openai/v2/test_schema.py

@@ -0,0 +1,80 @@
+from ai_review.clients.openai.v2.schema import (
+    OpenAIResponseUsageSchema,
+    OpenAIInputMessageSchema,
+    OpenAIResponseContentSchema,
+    OpenAIResponseOutputSchema,
+    OpenAIResponsesRequestSchema,
+    OpenAIResponsesResponseSchema,
+)
+
+
+# ---------- OpenAIResponsesResponseSchema ----------
+
+def test_first_text_returns_combined_text():
+    resp = OpenAIResponsesResponseSchema(
+        usage=OpenAIResponseUsageSchema(total_tokens=42, input_tokens=21, output_tokens=21),
+        output=[
+            OpenAIResponseOutputSchema(
+                type="message",
+                role="assistant",
+                content=[
+                    OpenAIResponseContentSchema(type="output_text", text="Hello"),
+                    OpenAIResponseContentSchema(type="output_text", text=" World"),
+                ],
+            )
+        ],
+    )
+
+    assert resp.first_text == "Hello World"
+
+
+def test_first_text_empty_if_no_output():
+    resp = OpenAIResponsesResponseSchema(
+        usage=OpenAIResponseUsageSchema(total_tokens=0, input_tokens=0, output_tokens=0),
+        output=[],
+    )
+    assert resp.first_text == ""
+
+
+def test_first_text_ignores_non_message_blocks():
+    resp = OpenAIResponsesResponseSchema(
+        usage=OpenAIResponseUsageSchema(total_tokens=5, input_tokens=2, output_tokens=3),
+        output=[
+            OpenAIResponseOutputSchema(
+                type="reasoning",  # игнорируется
+                role=None,
+                content=None,
+            )
+        ],
+    )
+    assert resp.first_text == ""
+
+
+# ---------- OpenAIResponsesRequestSchema ----------
+
+def test_responses_request_schema_builds_ok():
+    msg = OpenAIInputMessageSchema(role="user", content="hello")
+    req = OpenAIResponsesRequestSchema(
+        model="gpt-5",
+        input=[msg],
+        temperature=0.2,
+        max_output_tokens=512,
+        instructions="You are a helpful assistant.",
+    )
+
+    assert req.model == "gpt-5"
+    assert req.input[0].role == "user"
+    assert req.input[0].content == "hello"
+    assert req.temperature == 0.2
+    assert req.max_output_tokens == 512
+    assert req.instructions == "You are a helpful assistant."
+
+
+def test_responses_request_schema_allows_none_tokens():
+    req = OpenAIResponsesRequestSchema(
+        model="gpt-5",
+        input=[OpenAIInputMessageSchema(role="user", content="test")],
+    )
+
+    dumped = req.model_dump(exclude_none=True)
+    assert "max_output_tokens" not in dumped

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)

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

ai_review/tests/suites/libs/config/llm/test_openai.py

@@ -0,0 +1,28 @@
+import pytest
+
+from ai_review.libs.config.llm.openai import OpenAIMetaConfig
+
+
+@pytest.mark.parametrize(
+    "model, expected",
+    [
+        ("gpt-5", True),
+        ("gpt-5-preview", True),
+        ("gpt-4.1", True),
+        ("gpt-4.1-mini", True),
+        ("gpt-4o", False),
+        ("gpt-4o-mini", False),
+        ("gpt-3.5-turbo", False),
+        ("text-davinci-003", False),
+    ],
+)
+def test_is_v2_model_detection(model: str, expected: bool):
+    meta = OpenAIMetaConfig(model=model)
+    assert meta.is_v2_model is expected, f"Model {model} expected {expected} but got {meta.is_v2_model}"
+
+
+def test_is_v2_model_default_false():
+    meta = OpenAIMetaConfig()
+    assert meta.model == "gpt-4o-mini"
+    assert meta.is_v2_model is False
+    assert meta.max_tokens is None

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

@@ -2,21 +2,38 @@ 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
+from ai_review.tests.fixtures.clients.openai import FakeOpenAIV1HTTPClient, FakeOpenAIV2HTTPClient
 
 
 @pytest.mark.asyncio
-@pytest.mark.usefixtures("openai_http_client_config")
-async def test_openai_llm_chat(
+@pytest.mark.usefixtures("openai_v1_http_client_config")
+async def test_openai_llm_chat_v1(
         openai_llm_client: OpenAILLMClient,
-        fake_openai_http_client: FakeOpenAIHTTPClient
+        fake_openai_v1_http_client: FakeOpenAIV1HTTPClient
 ):
     result = await openai_llm_client.chat("prompt", "prompt_system")
 
     assert isinstance(result, ChatResultSchema)
-    assert result.text == "FAKE_OPENAI_RESPONSE"
+    assert result.text == "FAKE_OPENAI_V1_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"
+    assert fake_openai_v1_http_client.calls[0][0] == "chat"
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("openai_v2_http_client_config")
+async def test_openai_llm_chat_v2(
+        openai_llm_client: OpenAILLMClient,
+        fake_openai_v2_http_client: FakeOpenAIV2HTTPClient
+):
+    result = await openai_llm_client.chat("prompt", "prompt_system")
+
+    assert isinstance(result, ChatResultSchema)
+    assert result.text == "FAKE_OPENAI_V2_RESPONSE"
+    assert result.total_tokens == 20
+    assert result.prompt_tokens == 10
+    assert result.completion_tokens == 10
+
+    assert fake_openai_v2_http_client.calls[0][0] == "chat"

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"

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

@@ -5,9 +5,10 @@ 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")
+@pytest.mark.usefixtures("openai_v1_http_client_config")
 def test_get_llm_client_returns_openai(monkeypatch: pytest.MonkeyPatch):
     client = get_llm_client()
     assert isinstance(client, OpenAILLMClient)
@@ -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):

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

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
 ):

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

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