(test) init router with 4 clients

litellm/tests/test_router_init.py

@@ -0,0 +1,51 @@
+# this tests if the router is intiaized correctly
+import sys, os, time
+import traceback, asyncio
+import pytest
+sys.path.insert(
+    0, os.path.abspath("../..")
+)  # Adds the parent directory to the system path
+import litellm
+from litellm import Router
+from concurrent.futures import ThreadPoolExecutor
+from collections import defaultdict
+from dotenv import load_dotenv
+load_dotenv()
+
+
+# everytime we load the router we should have 4 clients:
+# Async
+# Sync
+# Async + Stream
+# Sync + Stream
+
+
+def test_init_clients():
+	litellm.set_verbose = True
+	try:
+		model_list = [
+			{ 
+				"model_name": "gpt-3.5-turbo", 
+				"litellm_params": { 
+					"model": "azure/chatgpt-v-2", 
+					"api_key": os.getenv("AZURE_API_KEY"),
+					"api_version": os.getenv("AZURE_API_VERSION"),
+					"api_base": os.getenv("AZURE_API_BASE")
+				},
+			},
+		]
+
+
+		router = Router(model_list=model_list)
+		print(router.model_list)
+		for elem in router.model_list:
+			print(elem)
+			assert elem["client"] is not None
+			assert elem["async_client"] is not None
+			assert elem["stream_client"] is not None
+			assert elem["stream_async_client"] is not None
+
+	except Exception as e:
+		traceback.print_exc()
+		pytest.fail(f"Error occurred: {e}")
+# test_init_clients()