From 05210fee6af26be0b45b67d56b1bb34efefa73f9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 9 Sep 2024 14:48:29 -0700 Subject: [PATCH] update test_default_tagged_deployments --- litellm/tests/test_router_tag_routing.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/tests/test_router_tag_routing.py b/litellm/tests/test_router_tag_routing.py index 4da109a13..f71a9b762 100644 --- a/litellm/tests/test_router_tag_routing.py +++ b/litellm/tests/test_router_tag_routing.py @@ -111,6 +111,14 @@ async def test_default_tagged_deployments(): }, "model_info": {"id": "default-model"}, }, + { + "model_name": "gpt-4", + "litellm_params": { + "model": "gpt-4o", + "api_base": "https://exampleopenaiendpoint-production.up.railway.app/", + }, + "model_info": {"id": "default-model-2"}, + }, { "model_name": "gpt-4", "litellm_params": {