forked from phoenix/litellm-mirror
(test) proxy: reading config.yaml
This commit is contained in:
parent
f20bdc9b79
commit
de4a7b719d
2 changed files with 18 additions and 1 deletions
4
litellm/proxy/example_config_yaml/simple_config.yaml
Normal file
4
litellm/proxy/example_config_yaml/simple_config.yaml
Normal file
|
@ -0,0 +1,4 @@
|
|||
model_list:
|
||||
- model_name: gpt-3.5-turbo
|
||||
litellm_params:
|
||||
model: gpt-3.5-turbo
|
|
@ -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()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue