From a176feeacc5fdf504747978d82056eb84679c4be Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 17 Jul 2024 12:09:08 -0700 Subject: [PATCH] fix(utils.py): return optional params from groq --- litellm/proxy/_new_secret_config.yaml | 5 +++-- litellm/utils.py | 5 +++++ 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/_new_secret_config.yaml b/litellm/proxy/_new_secret_config.yaml index b3888548db8..039a36c7e8f 100644 --- a/litellm/proxy/_new_secret_config.yaml +++ b/litellm/proxy/_new_secret_config.yaml @@ -1,4 +1,5 @@ model_list: - - model_name: ollama-mistral + - model_name: groq-whisper litellm_params: - model: ollama/mistral \ No newline at end of file + model: groq/whisper-large-v3 + \ No newline at end of file diff --git a/litellm/utils.py b/litellm/utils.py index 48fdf80c59f..b9c3f983cac 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5813,6 +5813,11 @@ def convert_to_model_response_object( if "text" in response_object: model_response_object.text = response_object["text"] + optional_keys = ["language", "task", "duration", "words", "segments"] + for key in optional_keys: # not guaranteed to be in response + if key in response_object: + setattr(model_response_object, key, response_object[key]) + if hidden_params is not None: model_response_object._hidden_params = hidden_params return model_response_object