From 259821fbaace8cdd443266930a87e0e677c09e9c Mon Sep 17 00:00:00 2001 From: agustin18 Date: Thu, 24 Sep 2026 04:30:45 +0000 Subject: [PATCH] fix(responses): resolve basedpyright final assignment and mock logging handler stubs --- litellm/responses/streaming_iterator.py | 21 ++++++++++++------- .../test_handler.py | 6 ++++++ 2 files changed, 19 insertions(+), 8 deletions(-) diff --git a/litellm/responses/streaming_iterator.py b/litellm/responses/streaming_iterator.py index d1f76eff501..b44a54b8fca 100644 --- a/litellm/responses/streaming_iterator.py +++ b/litellm/responses/streaming_iterator.py @@ -1132,15 +1132,20 @@ class MockResponsesAPIStreamingIterator(BaseResponsesAPIStreamingIterator): call_type: str | None = None, transformed_response: ResponsesAPIResponse | None = None, ): - if transformed_response is not None: - transformed: Final = transformed_response - elif responses_api_provider_config is not None and response is not None: - transformed: Final = responses_api_provider_config.transform_response_api_response( - model=model, - raw_response=response, - logging_obj=logging_obj, + transformed: Final[ResponsesAPIResponse | None] = ( + transformed_response + if transformed_response is not None + else ( + responses_api_provider_config.transform_response_api_response( + model=model, + raw_response=response, + logging_obj=logging_obj, + ) + if responses_api_provider_config is not None and response is not None + else None ) - else: + ) + if transformed is None: raise ValueError( "Either transformed_response or both responses_api_provider_config and response must be provided to MockResponsesAPIStreamingIterator" ) diff --git a/tests/unit/responses/litellm_completion_transformation/test_handler.py b/tests/unit/responses/litellm_completion_transformation/test_handler.py index 2c9ecb1648e..d8633f59ac0 100644 --- a/tests/unit/responses/litellm_completion_transformation/test_handler.py +++ b/tests/unit/responses/litellm_completion_transformation/test_handler.py @@ -340,6 +340,12 @@ def test_mock_responses_streaming_iterator_validation_and_config_branches(): def __init__(self): self.model_call_details = {"litellm_params": {"api_key": "fake"}} + async def async_success_handler(self, *args, **kwargs): + pass + + def success_handler(self, *args, **kwargs): + pass + logging_obj = _MockLoggingObj() iterator = MockResponsesAPIStreamingIterator( response=httpx.Response(200),