fix img gen test

2026-08-08 22:25:25 +00:00 · 2024-06-29 20:54:22 -07:00
parent 11338c409d
commit d25b079caf
1 changed files with 2 additions and 2 deletions
@@ -1081,7 +1081,7 @@ async def test_aimg_gen_on_router():
            {
                "model_name": "dall-e-3",
                "litellm_params": {
-                    "model": "openai/dall-e-3",
+                    "model": "dall-e-3",
                },
            },
            {
@@ -1137,7 +1137,7 @@ def test_img_gen_on_router():
    try:
        model_list = [
            {
-                "model_name": "openai/dall-e-3",
+                "model_name": "dall-e-3",
                "litellm_params": {
                    "model": "dall-e-3",
                },