mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-10 22:41:41 +00:00
b604e2b20c
About 35,000 fixes ruff marks safe across 32 rules (UP006/UP045/UP007 modern annotations, UP032 f-strings, SIM114/SIM118, RET501, and friends), removal of the 1,296 typing imports the rewrite orphaned, and hand fixes for what the fixers could not see: five star-import freeloaders of typing names, two F823 late-import annotations, the /get/config/list introspection crash on types.UnionType, redundant function-local RoleMappings imports in ui_sso.py that shadowed the module-level name once the annotation lost its quotes, and one FURB168 tautology. B009/B010/PIE804/RUF019 are excluded on purpose: their safe fixes rewrite getattr/setattr/**-splat/key-in-dict escape hatches into forms basedpyright then rejects (283 new errors measured), so their budgets stay at base values. ruff-strict-budget.json drops by 39,579 this commit (39,968 across the branch) with 28 rules at an actual 0 and 9 more sharply down. type-discipline-budget.json ratchets LIT002/LIT006/LIT009 down; LIT001 moves to the now-honest total: the checker matches the spelling `set` but not the alias `Set`, so the 160 typing.Set annotations rewritten to set[...] were always mutable-set annotations and only now count.
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""
|
|
End-user table model.
|
|
|
|
Canonical definition for ``litellm_endusertable``. Re-exported from
|
|
``litellm.proxy._types`` for backwards compatibility.
|
|
"""
|
|
|
|
from typing import Literal
|
|
|
|
from pydantic import ConfigDict, model_validator
|
|
|
|
from litellm.models.budget import LiteLLM_BudgetTable
|
|
from litellm.models.object_permission import LiteLLM_ObjectPermissionTable
|
|
from litellm.types.llms.base import LiteLLMPydanticObjectBase
|
|
|
|
|
|
class LiteLLM_EndUserTable(LiteLLMPydanticObjectBase):
|
|
user_id: str
|
|
blocked: bool
|
|
alias: str | None = None
|
|
spend: float = 0.0
|
|
allowed_model_region: Literal["eu", "us"] | None = None
|
|
default_model: str | None = None
|
|
budget_id: str | None = None
|
|
litellm_budget_table: LiteLLM_BudgetTable | None = None
|
|
object_permission_id: str | None = None
|
|
object_permission: LiteLLM_ObjectPermissionTable | None = None
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def set_model_info(cls, values):
|
|
if values.get("spend") is None:
|
|
values.update({"spend": 0.0})
|
|
return values
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|