From 2c74dbed17c6f77a53c22b047cfb3d6a4dd4262e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 29 Nov 2023 17:30:33 -0800 Subject: [PATCH] (chore) util: remove_model_id --- litellm/utils.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 489809199..b80732d49 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5846,4 +5846,13 @@ def transform_logprobs(hf_response): # Add the 'token_info' list to the 'transformed_logprobs' list transformed_logprobs = token_info - return transformed_logprobs \ No newline at end of file + return transformed_logprobs + +# used in LiteLLM Router +def remove_model_id(original_model_string): + # Find the index of "ModelID" in the string + index_of_model_id = original_model_string.find("-ModelID") + # Remove everything after "-ModelID" if it exists + if index_of_model_id != -1: + return original_model_string[:index_of_model_id] + return original_model_string \ No newline at end of file