From dc52c83b8880c2d41c0b96abfbb6b073bb0a3bc7 Mon Sep 17 00:00:00 2001 From: Marc Abramowitz Date: Thu, 16 May 2024 10:05:35 -0700 Subject: [PATCH] Add more routes to test_generate_and_call_with_valid_key --- litellm/tests/test_key_generate_prisma.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 75379a58a..34913ba20 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -150,8 +150,10 @@ async def test_new_user_response(prisma_client): @pytest.mark.parametrize( "api_route", [ + APIRoute(path="/engines/{model}/chat/completions", endpoint=chat_completion), + APIRoute(path="/openai/deployments/{model}/chat/completions", endpoint=chat_completion), APIRoute(path="/chat/completions", endpoint=chat_completion), - APIRoute(path="/engines/gpt-35-turbo-0125/chat/completions", endpoint=chat_completion), + APIRoute(path="/v1/chat/completions", endpoint=chat_completion), ], ) def test_generate_and_call_with_valid_key(prisma_client, api_route):