fix

Author: seratch

tests/realtime/test_ga_session_update_normalization.py

@@ -1,19 +1,21 @@
 from __future__ import annotations
 
 import json
-from typing import Any
+from typing import Any, cast
 
 import pytest
 
 from agents.realtime.model_inputs import RealtimeModelSendSessionUpdate
 from agents.realtime.openai_realtime import OpenAIRealtimeWebSocketModel
+from agents.realtime.config import RealtimeSessionModelSettings
+from websockets.asyncio.client import ClientConnection
 
 
 class _DummyWS:
     def __init__(self) -> None:
         self.sent: list[str] = []
 
-    async def send(self, data: str) -> None:  # type: ignore[override]
+    async def send(self, data: str) -> None:
         self.sent.append(data)
 
 
@@ -22,9 +24,9 @@ async def test_session_update_flattens_audio_and_modalities() -> None:
     model = OpenAIRealtimeWebSocketModel()
     # Inject a dummy websocket so send() works without a network
     dummy = _DummyWS()
-    model._websocket = dummy  # type: ignore[attr-defined]
+    model._websocket = cast(ClientConnection, dummy)
 
-    settings = {
+    settings: dict[str, object] = {
         "model_name": "gpt-realtime",
         "modalities": ["text", "audio"],
         "input_audio_format": "pcm16",
@@ -36,7 +38,11 @@ async def test_session_update_flattens_audio_and_modalities() -> None:
         "max_output_tokens": 2048,
     }
 
-    await model.send_event(RealtimeModelSendSessionUpdate(session_settings=settings))
+    await model.send_event(
+        RealtimeModelSendSessionUpdate(
+            session_settings=cast(RealtimeSessionModelSettings, settings)
+        )
+    )
 
     # One session.update should have been sent
     assert dummy.sent, "no websocket messages were sent"
@@ -68,7 +74,7 @@ async def _fake_interrupt(event: Any) -> None:
         called["interrupt"] = True
 
     # Prevent network use; _websocket only needed for other paths
-    model._websocket = _DummyWS()  # type: ignore[attr-defined]
+    model._websocket = cast(ClientConnection, _DummyWS())
     monkeypatch.setattr(model, "_send_interrupt", _fake_interrupt)
 
     # This event previously triggered an interrupt; now it should be ignored