fix use spec=textblock in mock

Author: Mohit-Raj-Singh

pyproject.toml

@@ -12,7 +12,7 @@ requires-python = ">=3.11"
 dependencies = [
     "anthropic>=0.40.0",
     "openai>=1.50.0",
-    "google-generativeai>=0.8.0",
+    "google-genai>=1.0.0",
     "pygithub>=2.4.0",
     "httpx>=0.27.0",
     "pydantic>=2.9.0",

src/ai_reviewer/providers/gemini.py

@@ -1,6 +1,7 @@
 import json
 
-import google.generativeai as genai
+import google.genai as genai
+from google.genai import types as genai_types
 
 from ..models import ReviewResult
 from .base import LLMProvider
@@ -9,18 +10,18 @@
 
 class GeminiProvider(LLMProvider):
     def __init__(self, api_key: str, model: str):
-        genai.configure(api_key=api_key)  # type: ignore[attr-defined]
-        self.model = genai.GenerativeModel(  # type: ignore[attr-defined]
-            model_name=model,
+        self.client = genai.Client(api_key=api_key)
+        self.model = model
+        self.config = genai_types.GenerateContentConfig(
             system_instruction=SYSTEM_PROMPT,
-            generation_config=genai.GenerationConfig(  # type: ignore[attr-defined]
-                response_mime_type="application/json",
-                max_output_tokens=4096,
-            ),
+            response_mime_type="application/json",
+            max_output_tokens=4096,
         )
 
     async def review(self, diff: str, review_level: str, security_only: bool) -> ReviewResult:
-        response = await self.model.generate_content_async(
-            build_user_prompt(diff, review_level, security_only)
+        response = await self.client.aio.models.generate_content(
+            model=self.model,
+            contents=build_user_prompt(diff, review_level, security_only),
+            config=self.config,
         )
         return ReviewResult.model_validate(json.loads(response.text))

tests/test_providers.py

@@ -2,6 +2,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+from anthropic.types import TextBlock
 
 from ai_reviewer.models import ReviewResult, Severity
 from ai_reviewer.providers.anthropic import AnthropicProvider
@@ -32,7 +33,7 @@ async def test_anthropic_provider_parses_result():
         mock_client = MagicMock()
         mock_cls.return_value = mock_client
         mock_client.messages.create = AsyncMock(
-            return_value=MagicMock(content=[MagicMock(text=json.dumps(MOCK_RESULT))])
+            return_value=MagicMock(content=[MagicMock(spec=TextBlock, text=json.dumps(MOCK_RESULT))])
         )
 
         provider = AnthropicProvider(api_key="test-key", model="claude-sonnet-4-20250514")