mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-06 14:24:50 +00:00
fix(router.py): fix trailing slash handling for api base which contains /v1
This commit is contained in:
@@ -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/"
|
||||
|
||||
|
||||
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user