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

(chore) linting fix

Browse source
This commit is contained in:
ishaan-jaff 2023-12-05 13:23:35 -08:00
parent e579918dd9
commit 4e3040b357
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

@ -45,7 +45,7 @@ celery_app.conf.update(
@celery_app.task(name='process_job', max_retries=3) @celery_app.task(name='process_job', max_retries=3)
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")) # type: ignore
response = llm_router.completion(*args, **kwargs) # type: ignore 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()