From b507c54385a5600896f358be40d05b86cb6066f9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 11 Jun 2024 22:03:03 -0700 Subject: [PATCH] fix new team test --- litellm/tests/test_jwt.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/tests/test_jwt.py b/litellm/tests/test_jwt.py index a3768bc23cb..17de5729118 100644 --- a/litellm/tests/test_jwt.py +++ b/litellm/tests/test_jwt.py @@ -16,6 +16,7 @@ from litellm.proxy._types import LiteLLM_JWTAuth, LiteLLMRoutes from litellm.proxy.auth.handle_jwt import JWTHandler from litellm.caching import DualCache from datetime import datetime, timedelta +from fastapi import Request public_key = { "kty": "RSA", @@ -346,6 +347,7 @@ async def test_team_token_output(prisma_client, audience): models=["gpt-3.5-turbo", "gpt-4"], ), user_api_key_dict=result, + http_request=Request(scope={"type": "http"}), ) except Exception as e: pytest.fail(f"This should not fail - {str(e)}") @@ -534,6 +536,7 @@ async def test_user_token_output( models=["gpt-3.5-turbo", "gpt-4"], ), user_api_key_dict=result, + http_request=Request(scope={"type": "http"}), ) if default_team_id: await new_team( @@ -544,6 +547,7 @@ async def test_user_token_output( models=["gpt-3.5-turbo", "gpt-4"], ), user_api_key_dict=result, + http_request=Request(scope={"type": "http"}), ) except Exception as e: pytest.fail(f"This should not fail - {str(e)}")