From 3fafd0062af38be86ad95a5426e3edd65d8bbff6 Mon Sep 17 00:00:00 2001 From: nate-berri Date: Sat, 10 Oct 2026 12:05:47 -0700 Subject: [PATCH] refactor(types): import team and user table models from litellm.models (#45740) Part of LIT-9423 --- litellm/integrations/prometheus.py | 10 ++++------ .../types/proxy/management_endpoints/team_endpoints.py | 6 ++---- 2 files changed, 6 insertions(+), 10 deletions(-) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index 2d1ef62fde3..8f4d1df2662 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -49,12 +49,10 @@ from litellm.litellm_core_utils.service_tier_utils import ( get_service_tier_from_standard_logging_payload, ) from litellm.models.end_user import LiteLLM_EndUserTable -from litellm.proxy._types import ( - LiteLLM_DeletedVerificationToken, - LiteLLM_TeamTable, - LiteLLM_UserTable, - UserAPIKeyAuth, -) +from litellm.models.team import LiteLLM_TeamTable +from litellm.models.user import LiteLLM_UserTable +from litellm.models.verification_token import LiteLLM_DeletedVerificationToken +from litellm.proxy._types import UserAPIKeyAuth from litellm.proxy.common_utils.user_api_key_cache import AUTH_OBJECTS_TARGET from litellm.repositories.base_repository import BaseRepository from litellm.repositories.budget_repository import BudgetRepository diff --git a/litellm/types/proxy/management_endpoints/team_endpoints.py b/litellm/types/proxy/management_endpoints/team_endpoints.py index 6731b826ebd..796bcae37bc 100644 --- a/litellm/types/proxy/management_endpoints/team_endpoints.py +++ b/litellm/types/proxy/management_endpoints/team_endpoints.py @@ -2,12 +2,10 @@ from typing import Any, Final, Literal from pydantic import ConfigDict, Field, field_validator, model_validator +from litellm.models.team import LiteLLM_DeletedTeamTable, LiteLLM_TeamTable, Member +from litellm.models.team_membership import LiteLLM_TeamMembership from litellm.proxy._types import ( KeyManagementRoutes, - LiteLLM_DeletedTeamTable, - LiteLLM_TeamMembership, - LiteLLM_TeamTable, - Member, MemberDeleteRequest, ) from litellm.proxy.common_utils.timezone_utils import budget_duration_error