From 17a29bfbfdd6850eedf0609bd35287d915d48977 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 10 Mar 2025 21:59:15 -0700
Subject: [PATCH] test: add direct test - fix code qa check

---
 tests/local_testing/test_router_utils.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/tests/local_testing/test_router_utils.py b/tests/local_testing/test_router_utils.py
index 7c2bbdc2a1..a94f5ceca9 100644
--- a/tests/local_testing/test_router_utils.py
+++ b/tests/local_testing/test_router_utils.py
@@ -418,3 +418,21 @@ def test_router_handle_clientside_credential():
 
     assert new_deployment.litellm_params.api_key == "123"
     assert len(router.get_model_list()) == 2
+
+
+def test_router_get_async_openai_model_client():
+    router = Router(
+        model_list=[
+            {
+                "model_name": "gemini/*",
+                "litellm_params": {
+                    "model": "gemini/*",
+                    "api_base": "https://api.gemini.com",
+                },
+            }
+        ]
+    )
+    model_client = router._get_async_openai_model_client(
+        deployment=MagicMock(), kwargs={}
+    )
+    assert model_client is None