scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 174

Merge commit 'f52fe547fba8c290ae377fc962fce24092515ff1' of /Users/mfkhalil/Code/litellm-daily-alert-detector into HEAD

Browse source
This commit is contained in:
moe-berri 2026-11-08 15:37:25 +08:00
commit 91cb056d8d
2 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/repositories/usage_alerts.py
View file

@ -649,7 +649,7 @@ class UsageAlertsRepository:
rows: Final = TypeAdapter(tuple[_ModelIdentity, ...]).validate_python(
await self.database.query_raw(
"""SELECT DISTINCT model FROM "LiteLLM_DailyUserSpend"
WHERE date >= $1 AND date <= $2 AND model IS NOT NULL AND model <> ''
WHERE date >= $1 AND date <= $2 AND model IS NOT NULL AND length(model) BETWEEN 1 AND 1024
AND ($3::text IS NULL OR strpos(lower(model),lower($3))>0)
AND ($4::text IS NULL OR model>$4)
ORDER BY model LIMIT $5""",

13
tests/integration/database/test_usage_alerts.py
View file

@ -682,9 +682,14 @@ async def test_reported_model_search_preserves_exact_identity_and_cursor(
prefix: Final = f"reported-{uuid4().hex}"
initial: Final = await repository.list_reported_models(prefix, None, 1)
assert initial.data == ()
models: Final = (f"{prefix}/first", f"{prefix}/literal%quote'", f"{prefix}/second")
models: Final = (
f"{prefix}/first",
f"{prefix}/literal%quote'",
f"{prefix}/second",
prefix + "/z" + "x" * (1022 - len(prefix)),
)
try:
for index, model in enumerate((*models, models[0])):
for index, model in enumerate((*models, models[0], prefix + "/aaa" + "x" * 1025)):
await db.prisma.writer_db.execute_raw(
'INSERT INTO "LiteLLM_DailyUserSpend" (id,date,api_key,model,updated_at) VALUES ($1,$2,$3,$4,CURRENT_TIMESTAMP)',
f"{prefix}-{index}",
@ -706,7 +711,9 @@ async def test_reported_model_search_preserves_exact_identity_and_cursor(
second: Final = await repository.list_reported_models(prefix, first.next_cursor, 1)
third: Final = await repository.list_reported_models(prefix, second.next_cursor, 1)
assert second.data[0].scope_id == models[1] and third.data[0].scope_id == models[2]
assert third.next_cursor is None
fourth: Final = await repository.list_reported_models(prefix, third.next_cursor, 1)
assert fourth.data[0].scope_id == models[3] and len(fourth.data[0].scope_id) == 1024
assert fourth.next_cursor is None
literal: Final = await repository.list_reported_models(f"{prefix}/literal%quote'", None, 20)
assert tuple(option.scope_id for option in literal.data) == (models[1],)
assert literal.data[0].label == models[1]