tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-09 16:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

(chore) util: remove_model_id

Browse Source
This commit is contained in:
ishaan-jaff
2023-11-29 17:30:33 -08:00
parent 7a38a45d62
commit 2c74dbed17
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+10 -1
View File
@@ -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
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