(test) proxy: reading config.yaml

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

View file

@ -0,0 +1,4 @@
model_list:
- model_name: gpt-3.5-turbo
litellm_params:
model: gpt-3.5-turbo

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()