From 21d958e46d61395e72d60d7f420e188b04f388ba Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Thu, 26 Feb 2026 13:32:13 -0300 Subject: [PATCH] fix(mypy): add union-attr suppression to all backend_ws method calls Address Greptile review: apply type: ignore[union-attr] consistently on all backend_ws.send(), .recv(), and .close() calls, not just the three that CI flagged. Co-Authored-By: Claude Opus 4.6 --- litellm/litellm_core_utils/realtime_streaming.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/litellm_core_utils/realtime_streaming.py b/litellm/litellm_core_utils/realtime_streaming.py index 760efa71ab..449a489262 100644 --- a/litellm/litellm_core_utils/realtime_streaming.py +++ b/litellm/litellm_core_utils/realtime_streaming.py @@ -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}")