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

test: fix linting testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-19 18:40:33 -07:00
parent 4198901a2d
commit 04f8840a92
3 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/llm.py
View file

@ -130,12 +130,12 @@ def litellm_completion(data: Dict,
if user_headers:
data["headers"] = user_headers
if type == "completion":
if data["model"] in model_router.get_model_names():
if model_router and data["model"] in model_router.get_model_names():
model_router.text_completion(**data)
else:
response = litellm.text_completion(**data)
elif type == "chat_completion":
if data["model"] in model_router.get_model_names():
if model_router and data["model"] in model_router.get_model_names():
model_router.completion(**data)
else:
response = litellm.completion(**data)