diff --git a/litellm/responses/streaming_iterator.py b/litellm/responses/streaming_iterator.py index 33d09efadf3..32a36ffe4e8 100644 --- a/litellm/responses/streaming_iterator.py +++ b/litellm/responses/streaming_iterator.py @@ -2329,7 +2329,6 @@ class ResponsesWebSocketStreaming: verbose_logger.debug("Responses WS client_to_backend ended: %s", e) async def bidirectional_forward(self) -> Exception | None: - """Run both forwarding directions concurrently and return the provider failure that ended the connection.""" forward_task: Final = asyncio.create_task(self.backend_to_client()) try: await self.client_to_backend() diff --git a/tests/test_litellm/litellm_core_utils/test_litellm_logging.py b/tests/test_litellm/litellm_core_utils/test_litellm_logging.py index 0d2d600a7fc..91c334692ee 100644 --- a/tests/test_litellm/litellm_core_utils/test_litellm_logging.py +++ b/tests/test_litellm/litellm_core_utils/test_litellm_logging.py @@ -1070,9 +1070,6 @@ async def test_arealtime_marks_litellm_params_async(monkeypatch): @pytest.mark.asyncio async def test_aresponses_websocket_hands_back_the_provider_failure_without_a_success_log(monkeypatch): - """A native Responses WebSocket connection the provider rejected comes back from the ``@client`` - wrapper as the mapped failure, and the wrapper books no success for it: the relay's own dispatch - is the connection's single log, so the proxy can record the connection as a failed request.""" from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER from litellm.responses.main import base_llm_http_handler diff --git a/tests/test_litellm/responses/test_responses_websocket_all_providers.py b/tests/test_litellm/responses/test_responses_websocket_all_providers.py index 6bd137be788..b6d4d9e93a6 100644 --- a/tests/test_litellm/responses/test_responses_websocket_all_providers.py +++ b/tests/test_litellm/responses/test_responses_websocket_all_providers.py @@ -2973,7 +2973,6 @@ def _wrapped_reasoning_item(): class TestNativeWebSocketEncryptedContentAffinity: - """The native relay must restore and wrap ids the same way the HTTP /v1/responses path does.""" @pytest.mark.asyncio @pytest.mark.parametrize("nested", [False, True])