fix(router.py): fix trailing slash handling for api base which contains /v1

This commit is contained in:
Krrish Dholakia 2024-04-27 17:36:05 -07:00
parent de8f928bdd
commit 280148543f
2 changed files with 4 additions and 0 deletions

View file

@ -1994,6 +1994,8 @@ class Router:
# check if it ends with a trailing slash
if api_base.endswith("/"):
api_base += "v1/"
elif api_base.endswith("/v1"):
api_base += "/"
else:
api_base += "/v1/"

View file

@ -57,6 +57,7 @@ def test_router_num_retries_init(num_retries, max_retries):
else:
assert getattr(model_client, "max_retries") == 0
@pytest.mark.parametrize(
"timeout", [10, 1.0, httpx.Timeout(timeout=300.0, connect=20.0)]
)
@ -137,6 +138,7 @@ def test_router_azure_ai_studio_init(mistral_api_base):
print(f"uri_reference: {uri_reference}")
assert "/v1/" in uri_reference
assert uri_reference.count("v1") == 1
def test_exception_raising():