From c6986c66cccbd5957526a34b9ca4c4f38242f249 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 15 Jun 2024 13:24:13 -0700 Subject: [PATCH] fix new team test --- litellm/tests/test_jwt.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_jwt.py b/litellm/tests/test_jwt.py index 17de5729118..376b2bbe879 100644 --- a/litellm/tests/test_jwt.py +++ b/litellm/tests/test_jwt.py @@ -14,6 +14,7 @@ sys.path.insert( import pytest from litellm.proxy._types import LiteLLM_JWTAuth, LiteLLMRoutes from litellm.proxy.auth.handle_jwt import JWTHandler +from litellm.proxy.team_endpoints import new_team from litellm.caching import DualCache from datetime import datetime, timedelta from fastapi import Request @@ -218,7 +219,7 @@ async def test_team_token_output(prisma_client, audience): from cryptography.hazmat.backends import default_backend from fastapi import Request from starlette.datastructures import URL - from litellm.proxy.proxy_server import user_api_key_auth, new_team + from litellm.proxy.proxy_server import user_api_key_auth from litellm.proxy._types import NewTeamRequest, UserAPIKeyAuth import litellm import uuid @@ -399,7 +400,6 @@ async def test_user_token_output( from starlette.datastructures import URL from litellm.proxy.proxy_server import ( user_api_key_auth, - new_team, new_user, user_info, ) @@ -623,7 +623,7 @@ async def test_allowed_routes_admin(prisma_client, audience): from cryptography.hazmat.backends import default_backend from fastapi import Request from starlette.datastructures import URL - from litellm.proxy.proxy_server import user_api_key_auth, new_team + from litellm.proxy.proxy_server import user_api_key_auth from litellm.proxy._types import NewTeamRequest, UserAPIKeyAuth import litellm import uuid