mirror of
https://github.com/BerriAI/litellm.git
synced 2026-08-28 05:25:59 +00:00
32 lines
1 KiB
Python
32 lines
1 KiB
Python
"""
|
|
Team membership table model.
|
|
|
|
Canonical definition for ``litellm_teammembership``. Re-exported from
|
|
``litellm.proxy._types`` for backwards compatibility.
|
|
"""
|
|
|
|
from typing import Optional, Union
|
|
|
|
from litellm.models.budget import LiteLLM_BudgetTable, LiteLLM_BudgetTableFull
|
|
from litellm.types.llms.base import LiteLLMPydanticObjectBase
|
|
|
|
|
|
class LiteLLM_TeamMembership(LiteLLMPydanticObjectBase):
|
|
user_id: str
|
|
team_id: str
|
|
budget_id: Optional[str] = None
|
|
spend: Optional[float] = 0.0
|
|
total_spend: Optional[float] = 0.0
|
|
litellm_budget_table: Optional[
|
|
Union[LiteLLM_BudgetTableFull, LiteLLM_BudgetTable]
|
|
] = None
|
|
|
|
def safe_get_team_member_rpm_limit(self) -> Optional[int]:
|
|
if self.litellm_budget_table is not None:
|
|
return self.litellm_budget_table.rpm_limit
|
|
return None
|
|
|
|
def safe_get_team_member_tpm_limit(self) -> Optional[int]:
|
|
if self.litellm_budget_table is not None:
|
|
return self.litellm_budget_table.tpm_limit
|
|
return None
|