diff --git a/litellm/tests/test_router_init.py b/litellm/tests/test_router_init.py new file mode 100644 index 00000000000..8068940817f --- /dev/null +++ b/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()