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

(fix) linting

Browse source
This commit is contained in:
ishaan-jaff 2023-12-01 14:08:19 -08:00
parent 24386c05aa
commit 73eb5a7718
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/queue/celery_app.py
View file

@ -46,7 +46,7 @@ celery_app.conf.update(
def process_job(*args, **kwargs): def process_job(*args, **kwargs):
try: try:
llm_router: litellm.Router = litellm.Router(model_list=kwargs.pop("llm_model_list")) llm_router: litellm.Router = litellm.Router(model_list=kwargs.pop("llm_model_list"))
response = llm_router.completion(*args, **kwargs) response = llm_router.completion(*args, **kwargs) # type: ignore
if isinstance(response, litellm.ModelResponse): if isinstance(response, litellm.ModelResponse):
response = response.model_dump_json() response = response.model_dump_json()
return json.loads(response) return json.loads(response)