phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) utils - delete remove_model_id

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 07:07:48 +05:30
parent fff148a835
commit ad18ab2144
1 changed files with 0 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -6381,12 +6381,3 @@ def transform_logprobs(hf_response):
transformed_logprobs = token_info transformed_logprobs = token_info
return transformed_logprobs 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