mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-14 23:21:35 +00:00
fix(user_info): include user-level rpm_limit and tpm_limit in /v2/user/info
This commit is contained in:
parent
69a491e168
commit
6d8b5afc57
4 changed files with 14 additions and 0 deletions
|
|
@ -2708,6 +2708,8 @@ class UserInfoV2Response(LiteLLMPydanticObjectBase):
|
|||
user_role: Optional[str] = None
|
||||
spend: float = 0.0
|
||||
max_budget: Optional[float] = None
|
||||
rpm_limit: Optional[int] = None
|
||||
tpm_limit: Optional[int] = None
|
||||
models: List[str] = []
|
||||
budget_duration: Optional[str] = None
|
||||
budget_reset_at: Optional[datetime] = None
|
||||
|
|
|
|||
|
|
@ -942,6 +942,8 @@ async def user_info_v2(
|
|||
user_role=user_data.get("user_role"),
|
||||
spend=user_data.get("spend", 0.0),
|
||||
max_budget=user_data.get("max_budget"),
|
||||
rpm_limit=user_data.get("rpm_limit"),
|
||||
tpm_limit=user_data.get("tpm_limit"),
|
||||
models=user_data.get("models") or [],
|
||||
budget_duration=user_data.get("budget_duration"),
|
||||
budget_reset_at=user_data.get("budget_reset_at"),
|
||||
|
|
|
|||
|
|
@ -2490,6 +2490,8 @@ async def test_user_info_v2_proxy_admin_can_query_any_user(mocker):
|
|||
"user_role": "internal_user",
|
||||
"spend": 42.5,
|
||||
"max_budget": 100.0,
|
||||
"rpm_limit": 100,
|
||||
"tpm_limit": 200000,
|
||||
"models": ["gpt-4"],
|
||||
"budget_duration": "30d",
|
||||
"budget_reset_at": None,
|
||||
|
|
@ -2530,6 +2532,8 @@ async def test_user_info_v2_proxy_admin_can_query_any_user(mocker):
|
|||
assert response.user_role == "internal_user"
|
||||
assert response.spend == 42.5
|
||||
assert response.max_budget == 100.0
|
||||
assert response.rpm_limit == 100
|
||||
assert response.tpm_limit == 200000
|
||||
assert response.models == ["gpt-4"]
|
||||
assert response.teams == ["team-1", "team-2"]
|
||||
assert response.sso_user_id == "sso-abc"
|
||||
|
|
@ -2885,6 +2889,8 @@ async def test_user_info_v2_response_shape(mocker):
|
|||
"user_role",
|
||||
"spend",
|
||||
"max_budget",
|
||||
"rpm_limit",
|
||||
"tpm_limit",
|
||||
"models",
|
||||
"budget_duration",
|
||||
"budget_reset_at",
|
||||
|
|
|
|||
4
ui/litellm-dashboard/src/lib/http/schema.d.ts
generated
vendored
4
ui/litellm-dashboard/src/lib/http/schema.d.ts
generated
vendored
|
|
@ -33060,6 +33060,8 @@ export interface components {
|
|||
* @default []
|
||||
*/
|
||||
models: string[];
|
||||
/** Rpm Limit */
|
||||
rpm_limit?: number | null;
|
||||
/**
|
||||
* Spend
|
||||
* @default 0
|
||||
|
|
@ -33072,6 +33074,8 @@ export interface components {
|
|||
* @default []
|
||||
*/
|
||||
teams: string[];
|
||||
/** Tpm Limit */
|
||||
tpm_limit?: number | null;
|
||||
/** Updated At */
|
||||
updated_at?: string | null;
|
||||
/** User Alias */
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue