diff --git a/litellm/main.py b/litellm/main.py index c3d1c2e05b0..b17f4c535f7 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -6274,7 +6274,8 @@ async def aadapter_completion( if isinstance(response, CustomStreamWrapper): translated_response = ( translation_obj.translate_completion_output_params_streaming( - completion_stream=response + completion_stream=response, + model=kwargs.get("model", "") ) ) @@ -6325,7 +6326,8 @@ def adapter_completion( elif isinstance(response, CustomStreamWrapper) or inspect.isgenerator(response): translated_response = ( translation_obj.translate_completion_output_params_streaming( - completion_stream=response + completion_stream=response, + model=kwargs.get("model", "") ) )