From 3d3e3b7ac704aa353e6e9737d2a5b92643d823f9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 May 2025 20:49:16 -0700 Subject: [PATCH] fix: linting error --- litellm/responses/streaming_iterator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/responses/streaming_iterator.py b/litellm/responses/streaming_iterator.py index a111fbec094..e9e41789f09 100644 --- a/litellm/responses/streaming_iterator.py +++ b/litellm/responses/streaming_iterator.py @@ -44,7 +44,7 @@ class BaseResponsesAPIStreamingIterator: self.responses_api_provider_config = responses_api_provider_config self.completed_response: Optional[ResponsesAPIStreamingResponse] = None self.start_time = datetime.now() - + # set request kwargs self.litellm_metadata = litellm_metadata self.custom_llm_provider = custom_llm_provider @@ -330,7 +330,8 @@ class MockResponsesAPIStreamingIterator(BaseResponsesAPIStreamingIterator): def _collect_text(self, resp: ResponsesAPIResponse) -> str: out = "" for out_item in resp.output: - if out_item.type == "message": + item_type = getattr(out_item, "type", None) + if item_type == "message": for c in getattr(out_item, "content", []): out += c.text return out