diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index e1115cbe9e5..123d4ac894b 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -6739,7 +6739,7 @@ def _restamp_streaming_chunk_model( attempted_fallbacks = 0 if attempted_fallbacks > 0: return chunk, model_mismatch_logged - + downstream_model = ( chunk.get("model") if isinstance(chunk, dict) else getattr(chunk, "model", None) ) diff --git a/litellm/router.py b/litellm/router.py index 1a86614f605..a59ae4a9e62 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -2139,7 +2139,7 @@ class Router: chunk_hidden_params["additional_headers"].update(response_headers) setattr(stream_chunk, "_hidden_params", chunk_hidden_params) return stream_chunk - + async def _acompletion_streaming_iterator( self, model_response: CustomStreamWrapper,