(test) proxy: reading config.yaml

This commit is contained in:
ishaan-jaff 2023-12-04 13:16:19 -08:00
parent f337b6569e
commit 6f73a3e682
2 changed files with 18 additions and 1 deletions

View file

@ -164,4 +164,17 @@ def test_chat_completion_optional_params():
pytest.fail("LiteLLM Proxy test failed. Exception", e)
# Run the test
test_chat_completion_optional_params()
# test_chat_completion_optional_params()
# Test Reading config.yaml file
from litellm.proxy.proxy_server import load_router_config
def test_load_router_config():
try:
print("testing reading config")
result = load_router_config(router=None, config_file_path="../proxy/example_config_yaml/simple_config.yaml")
print(result)
assert len(result[1]) == 1
except Exception as e:
pytest.fail("Proxy: Got exception reading config", e)
# test_load_router_config()