Update endpoints.py

This commit is contained in:
Anya Ishmukh 2026-07-16 10:35:57 +10:00 committed by GitHub
parent 1f2f24e018
commit d4344fb85e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -172,9 +172,9 @@ async def create_batch(
)
# Create batch using model credentials
resolved_model = _create_batch_data.pop("model", None) or model_from_file_id
_create_batch_data.pop("model", None) # avoid duplicate kwarg if data has model
response = await litellm.acreate_batch(
model=resolved_model,
model=model_from_file_id,
custom_llm_provider=credentials["custom_llm_provider"],
**_create_batch_data, # type: ignore
)