From 0bda80ddea8c9a1995141c080476ac37fede387a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 29 Jun 2024 17:28:08 -0700 Subject: [PATCH] test- router when using openai prefix --- litellm/tests/test_router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 7c59611d730..5f22e88a17c 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -1081,7 +1081,7 @@ async def test_aimg_gen_on_router(): { "model_name": "dall-e-3", "litellm_params": { - "model": "dall-e-3", + "model": "openai/dall-e-3", }, }, { @@ -1137,7 +1137,7 @@ def test_img_gen_on_router(): try: model_list = [ { - "model_name": "dall-e-3", + "model_name": "openai/dall-e-3", "litellm_params": { "model": "dall-e-3", },