phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(test_proxy_server.py): fix import

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-04 16:11:23 +05:30
parent 263d20cdd0
commit c7644915f9
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_proxy_server.py
View file

@ -280,7 +280,7 @@ def test_chat_completion_optional_params(client_no_auth):
# test_chat_completion_optional_params() # test_chat_completion_optional_params()
# Test Reading config.yaml file # Test Reading config.yaml file
from litellm.proxy.proxy_server import load_router_config from litellm.proxy.proxy_server import ProxyConfig
def test_load_router_config(): def test_load_router_config():
@ -288,7 +288,8 @@ def test_load_router_config():
print("testing reading config") print("testing reading config")
# this is a basic config.yaml with only a model # this is a basic config.yaml with only a model
filepath = os.path.dirname(os.path.abspath(__file__)) filepath = os.path.dirname(os.path.abspath(__file__))
result = load_router_config( proxy_config = ProxyConfig()
result = proxy_config.load_config(
router=None, router=None,
config_file_path=f"{filepath}/example_config_yaml/simple_config.yaml", config_file_path=f"{filepath}/example_config_yaml/simple_config.yaml",
) )