phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

build: adding debug logs to gitignore

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-27 22:05:02 -08:00
parent c52861906b
commit c4aea7432f
3 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/router.py
View file

@ -176,13 +176,13 @@ class Router:
########## remove -ModelID-XXXX from model ##############
original_model_string = data["model"]
# Find the index of "ModelID" in the string
self.print_verbose(f"completion model: {original_model_string}"}
index_of_model_id = original_model_string.find("-ModelID")
# Remove everything after "-ModelID" if it exists
if index_of_model_id != -1:
data["model"] = original_model_string[:index_of_model_id]
else:
data["model"] = original_model_string
self.print_verbose(f"completion model: {data['model']}")
data["model"] = original_model_string)
return litellm.completion(**{**data, "messages": messages, "caching": self.cache_responses, **kwargs})
except Exception as e:
raise e