diff --git a/litellm/proxy/db/model_usage_rollup.py b/litellm/proxy/db/model_usage_rollup.py index d0679df6822..50b2b181bf8 100644 --- a/litellm/proxy/db/model_usage_rollup.py +++ b/litellm/proxy/db/model_usage_rollup.py @@ -103,7 +103,7 @@ def build_model_usage_transaction(payload: SpendLogsPayload) -> ModelUsageTransa return None model: Final = payload["model"] or "unknown" successful: Final = payload["status"] == "success" - duration: Final = payload["request_duration_ms"] if "request_duration_ms" in payload else None + duration: Final = payload["request_duration_ms"] return ModelUsageTransaction( key=ModelUsageKey( date=date, diff --git a/litellm/repositories/alert_repository.py b/litellm/repositories/alert_repository.py index 9d9606b5d98..a62a9924a0b 100644 --- a/litellm/repositories/alert_repository.py +++ b/litellm/repositories/alert_repository.py @@ -1,11 +1,11 @@ import hashlib import json from collections.abc import Sequence -from typing import ( # noqa: TID251 # Prisma exposes generated table actions dynamically +from typing import ( TYPE_CHECKING, Final, Protocol, - cast, + cast, # noqa: TID251 # Prisma exposes generated table actions dynamically ) from litellm.models.alert import Alert, AlertDetection, AlertPage, AlertStatus diff --git a/tests/unit/proxy/db/test_model_usage_rollup.py b/tests/unit/proxy/db/test_model_usage_rollup.py index 6e60c575e34..2657d3fd2e3 100644 --- a/tests/unit/proxy/db/test_model_usage_rollup.py +++ b/tests/unit/proxy/db/test_model_usage_rollup.py @@ -51,6 +51,7 @@ def _payload(**overrides: Any) -> dict[str, Any]: "request_tags": "[]", "custom_llm_provider": "openai", "status": "success", + "request_duration_ms": None, **overrides, }