mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
Update endpoints.py
resolved_model gets the real model ID from the router config via prepare_data_with_credentials
This commit is contained in:
parent
ce1b7c900d
commit
1f2f24e018
1 changed files with 2 additions and 1 deletions
|
|
@ -172,8 +172,9 @@ async def create_batch(
|
|||
)
|
||||
|
||||
# Create batch using model credentials
|
||||
resolved_model = _create_batch_data.pop("model", None) or model_from_file_id
|
||||
response = await litellm.acreate_batch(
|
||||
model=model_from_file_id,
|
||||
model=resolved_model,
|
||||
custom_llm_provider=credentials["custom_llm_provider"],
|
||||
**_create_batch_data, # type: ignore
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue