From c717ed4d052dfda88aa4db3cf2f1ae6f0e493452 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 5 Dec 2023 14:28:23 -0800 Subject: [PATCH] (test) router: test async embedding + embedding --- litellm/tests/test_router.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 6572b9cd16..085fcc82a9 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -398,20 +398,26 @@ def test_aembedding_on_router(): "rpm": 10000, }, ] - + router = Router(model_list=model_list) async def embedding_call(): - router = Router(model_list=model_list) response = await router.aembedding( model="text-embedding-ada-002", input=["good morning from litellm", "this is another item"], ) print(response) - router.reset() asyncio.run(embedding_call()) + + print("\n Making sync Embedding call\n") + response = router.embedding( + model="text-embedding-ada-002", + input=["good morning from litellm 2"], + ) + print("sync embedding response: ", response) + router.reset() except Exception as e: traceback.print_exc() pytest.fail(f"Error occurred: {e}") -# test_aembedding_on_router() +test_aembedding_on_router() def test_azure_embedding_on_router():