From 5e065ebb8f9993aee934719665ae360534b52c8b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 5 Dec 2023 21:56:56 -0800 Subject: [PATCH] (test) router - explcitly call one deployment --- litellm/tests/test_router.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index ca84f144c81..47b4881b1f8 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -184,7 +184,7 @@ def test_call_one_endpoint(): "rpm": 10000, }, ] - + litellm.set_verbose=True router = Router(model_list=model_list, routing_strategy="simple-shuffle", set_verbose=True, @@ -200,6 +200,7 @@ def test_call_one_endpoint(): "content": "hello this request will pass" } ], + specific_deployment=True ) print("\n response", response) @@ -212,6 +213,7 @@ def test_call_one_endpoint(): "content": "hello this request will pass" } ], + specific_deployment=True ) print("\n response", response) @@ -219,7 +221,8 @@ def test_call_one_endpoint(): async def call_azure_embedding(): response = await router.aembedding( model="azure/azure-embedding-model", - input = ["good morning from litellm"] + input = ["good morning from litellm"], + specific_deployment=True ) print("\n response", response)