scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 75

fix(proxy/batches): null-guard await on find_first for sync MagicMock test harnesses

Browse source
This commit is contained in:
htourinho-clgx 2026-07-23 16:24:37 +02:00 • committed by Yucheng Zhu
parent 0e9f4ca09c
commit 411b525a83
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/batches_endpoints/endpoints.py
View file

@ -241,14 +241,16 @@ async def create_batch(
from litellm.proxy.proxy_server import prisma_client
if prisma_client is not None:
import inspect
from litellm.repositories.table_repositories import (
ManagedFileRepository,
)
db_file = await ManagedFileRepository(prisma_client).table.find_first(
db_file_res = ManagedFileRepository(prisma_client).table.find_first(
where={"unified_file_id": unified_file_id}
)
if db_file is not None and db_file.storage_url:
db_file = await db_file_res if inspect.isawaitable(db_file_res) else db_file_res
if db_file is not None and getattr(db_file, "storage_url", None):
_create_batch_data["input_file_id"] = db_file.storage_url
response = await llm_router.acreate_batch(**_create_batch_data)