add /rerank test

This commit is contained in:
Ishaan Jaff 2024-08-27 17:50:37 -07:00
parent df4f1458e6
commit d4db993130
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