test: fix test

This commit is contained in:
Krrish Dholakia 2024-11-23 12:35:55 +05:30
parent b5c5f87e25
commit 88db948d29

View file

@ -215,7 +215,10 @@ async def test_rerank_custom_api_base():
args_to_api = kwargs["json"]
print("Arguments passed to API=", args_to_api)
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 response.id is not None
assert response.results is not None