diff --git a/litellm/litellm_core_utils/realtime_streaming.py b/litellm/litellm_core_utils/realtime_streaming.py index 8df41aea4a..449a489262 100644 --- a/litellm/litellm_core_utils/realtime_streaming.py +++ b/litellm/litellm_core_utils/realtime_streaming.py @@ -230,9 +230,9 @@ class RealTimeStreaming: message, self.model, self.session_configuration_request ) for msg in transformed: - await self.backend_ws.send(msg) + await self.backend_ws.send(msg) # type: ignore[union-attr] else: - await self.backend_ws.send(message) + await self.backend_ws.send(message) # type: ignore[union-attr] def _has_realtime_guardrails(self) -> bool: """Return True if any callback is registered for realtime guardrail event types.""" @@ -362,7 +362,7 @@ class RealTimeStreaming: "[realtime guardrail] ending session after violation %d", self._violation_count, ) - await self.backend_ws.close() + await self.backend_ws.close() # type: ignore[union-attr] verbose_logger.warning( "[realtime guardrail] BLOCKED transcript (violation %d): %r", @@ -502,11 +502,11 @@ class RealTimeStreaming: try: while True: try: - raw_response = await self.backend_ws.recv( + raw_response = await self.backend_ws.recv( # type: ignore[union-attr] decode=False ) # improves performance except TypeError: - raw_response = await self.backend_ws.recv() # type: ignore[assignment] + raw_response = await self.backend_ws.recv() # type: ignore[union-attr, assignment] if self.provider_config: try: @@ -573,9 +573,9 @@ class RealTimeStreaming: ) for msg in message: - await self.backend_ws.send(msg) + await self.backend_ws.send(msg) # type: ignore[union-attr] else: - await self.backend_ws.send(message) + await self.backend_ws.send(message) # type: ignore[union-attr] except Exception as e: verbose_logger.debug(f"Error in client ack messages: {e}")