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

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-23 12:35:55 +05:30
parent b5c5f87e25
commit 88db948d29
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_rerank.py
View file

@ -215,7 +215,10 @@ async def test_rerank_custom_api_base():
args_to_api = kwargs["json"] args_to_api = kwargs["json"]
print("Arguments passed to API=", args_to_api) print("Arguments passed to API=", args_to_api)
print("url = ", _url) print("url = ", _url)
assert _url[0] == "https://exampleopenaiendpoint-production.up.railway.app/" assert (
_url[0]
== "https://exampleopenaiendpoint-production.up.railway.app/v1/rerank"
)
assert args_to_api == expected_payload assert args_to_api == expected_payload
assert response.id is not None assert response.id is not None
assert response.results is not None assert response.results is not None