scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

Add starting status for replicate

Browse source
This commit is contained in:
Sameer Kankute 2026-01-12 15:20:38 +05:30
parent 3d38ec4756
commit 028cdaf77b
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/replicate/chat/handler.py
View file

@ -213,7 +213,7 @@ def completion(
response = httpx_client.get(url=prediction_url, headers=headers)
if (
response.status_code == 200
and response.json().get("status") == "processing"
and response.json().get("status") in ["processing", "starting"]
):
continue
return litellm.ReplicateConfig().transform_response(
@ -284,7 +284,7 @@ async def async_completion(
response = await async_handler.get(url=prediction_url, headers=headers)
if (
response.status_code == 200
and response.json().get("status") == "processing"
and response.json().get("status") in ["processing", "starting"]
):
continue
return litellm.ReplicateConfig().transform_response(