diff --git a/litellm/batches/main.py b/litellm/batches/main.py index db220e15faa..79aefa5f518 100644 --- a/litellm/batches/main.py +++ b/litellm/batches/main.py @@ -50,7 +50,7 @@ async def acreate_batch( extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, -) -> Coroutine[Any, Any, Batch]: +) -> Batch: """ Async: Creates and executes a batch from an uploaded file of request @@ -189,7 +189,7 @@ async def aretrieve_batch( extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, -) -> Coroutine[Any, Any, Batch]: +) -> Batch: """ Async: Retrieves a batch.