add /rerank test

This commit is contained in:
Ishaan Jaff 2024-08-27 17:50:37 -07:00
parent 5f2f7aa754
commit c27640e6e4
3 changed files with 71 additions and 0 deletions

View file

@ -88,6 +88,8 @@ def _get_bearer_token(
api_key = api_key.replace("Bearer ", "") # extract the token
elif api_key.startswith("Basic "):
api_key = api_key.replace("Basic ", "") # handle langfuse input
elif api_key.startswith("bearer "):
api_key = api_key.replace("bearer ", "")
else:
api_key = ""
return api_key