scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

fix: keep health check writes on the repository .table path

Browse source 
The health_check_table property resolved through self, so tests that bind
PrismaClient methods onto a MagicMock via types.MethodType got a bare
MagicMock instead of the mocked prisma table and the awaits failed.

Resolves HealthCheckRepository(self).table at each call site instead, which
runs the same code as before the typing pass, and drops the unused property.
This commit is contained in:
mateo-berri 2026-07-24 20:24:42 -07:00
parent 6afef76268
commit 9958e4e0b3
1 changed files with 21 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
litellm/proxy/utils.py
View file

@ -3272,13 +3272,6 @@ class PrismaClient:
UserNotificationsRepository(self).table,
)
@property
def health_check_table(self) -> "_PrismaTable[LiteLLM_HealthCheckTable]":
return cast( # cast-ok: repository .table is declared Any; typed table protocol pins the real shape
"_PrismaTable[LiteLLM_HealthCheckTable]",
HealthCheckRepository(self).table,
)
def get_request_status(self, payload: Union[dict, SpendLogsPayload]) -> Literal["success", "failure"]:
"""
Determine if a request was successful or failed based on payload metadata.
@ -5272,7 +5265,13 @@ class PrismaClient:
health_check_data.update({k: v for k, v in optional_fields.items() if v is not None})
verbose_proxy_logger.debug(f"Saving health check data: {health_check_data}")
return await self.health_check_table.create(data=health_check_data)
health_check_table = (
cast( # cast-ok: repository .table is declared Any; typed table protocol pins the real shape
"_PrismaTable[LiteLLM_HealthCheckTable]",
HealthCheckRepository(self).table,
)
)
return await health_check_table.create(data=health_check_data)
except Exception as e:
verbose_proxy_logger.error(f"Error saving health check result for model {model_name}: {e}")
@ -5295,7 +5294,13 @@ class PrismaClient:
if status_filter:
where_clause["status"] = status_filter
results = await self.health_check_table.find_many(
health_check_table = (
cast( # cast-ok: repository .table is declared Any; typed table protocol pins the real shape
"_PrismaTable[LiteLLM_HealthCheckTable]",
HealthCheckRepository(self).table,
)
)
results = await health_check_table.find_many(
where=where_clause,
order={"checked_at": "desc"},
take=limit,
@ -5314,7 +5319,13 @@ class PrismaClient:
(via Prisma ``distinct`` + ``order``) so we never load the full history into memory.
"""
try:
return await self.health_check_table.find_many(
health_check_table = (
cast( # cast-ok: repository .table is declared Any; typed table protocol pins the real shape
"_PrismaTable[LiteLLM_HealthCheckTable]",
HealthCheckRepository(self).table,
)
)
return await health_check_table.find_many(
distinct=["model_id", "model_name"],
order=[
{"model_id": "asc"},