From 88db948d29683dba0fc324fa4f7470f8967cc398 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 23 Nov 2024 12:35:55 +0530 Subject: [PATCH] test: fix test --- tests/local_testing/test_rerank.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/local_testing/test_rerank.py b/tests/local_testing/test_rerank.py index f9160991c..5fca6f135 100644 --- a/tests/local_testing/test_rerank.py +++ b/tests/local_testing/test_rerank.py @@ -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