phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: improve debugging for test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 19:40:49 -07:00
parent dce7c459f8
commit 4ba576724c
4 changed files with 19 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/auth/auth_utils.py
View file

@ -1,5 +1,6 @@
import re import re
import sys import sys
import traceback
from fastapi import Request from fastapi import Request
@ -87,14 +88,16 @@ def get_request_route(request: Request) -> str:
remove base url from path if set e.g. `/genai/chat/completions` -> `/chat/completions remove base url from path if set e.g. `/genai/chat/completions` -> `/chat/completions
""" """
try: try:
if request.url.path.startswith(request.base_url.path): if hasattr(request, "base_url") and request.url.path.startswith(
request.base_url.path
):
# remove base_url from path # remove base_url from path
return request.url.path[len(request.base_url.path) - 1 :] return request.url.path[len(request.base_url.path) - 1 :]
else: else:
return request.url.path return request.url.path
except Exception as e: except Exception as e:
verbose_proxy_logger.warning( verbose_proxy_logger.debug(
f"error on get_request_route: {str(e)}, defaulting to request.url.path" f"error on get_request_route: {str(e)}, defaulting to request.url.path={request.url.path}"
) )
return request.url.path return request.url.path

6
litellm/proxy/auth/user_api_key_auth.py
View file

@ -532,6 +532,12 @@ async def user_api_key_auth(
api_key, str api_key, str
): # if generated token, make sure it starts with sk-. ): # if generated token, make sure it starts with sk-.
assert api_key.startswith("sk-") # prevent token hashes from being used assert api_key.startswith("sk-") # prevent token hashes from being used
else:
verbose_logger.warning(
"litellm.proxy.proxy_server.user_api_key_auth(): Warning - Key={} is not a string.".format(
api_key
)
)
if ( if (
prisma_client is None and custom_db_client is None prisma_client is None and custom_db_client is None

1
litellm/proxy/utils.py
View file

@ -1389,6 +1389,7 @@ class PrismaClient:
WHERE v.token = '{token}' WHERE v.token = '{token}'
""" """
print_verbose("sql_query being made={}".format(sql_query))
response = await self.db.query_first(query=sql_query) response = await self.db.query_first(query=sql_query)
if response is not None: if response is not None:

6
litellm/tests/test_key_generate_prisma.py
View file

@ -303,6 +303,7 @@ def test_call_with_invalid_key(prisma_client):
def test_call_with_invalid_model(prisma_client): def test_call_with_invalid_model(prisma_client):
litellm.set_verbose = True
# 3. Make a call to a key with an invalid model - expect to fail # 3. Make a call to a key with an invalid model - expect to fail
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client) setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234") setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
@ -326,6 +327,11 @@ def test_call_with_invalid_model(prisma_client):
request.body = return_body request.body = return_body
# use generated key to auth in # use generated key to auth in
print(
"Bearer token being sent to user_api_key_auth() - {}".format(
bearer_token
)
)
result = await user_api_key_auth(request=request, api_key=bearer_token) result = await user_api_key_auth(request=request, api_key=bearer_token)
pytest.fail(f"This should have failed!. IT's an invalid model") pytest.fail(f"This should have failed!. IT's an invalid model")