phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) test - prisma key budget tracking

Browse source
This commit is contained in:
ishaan-jaff 2024-01-26 14:31:04 -08:00
parent 2873365f02
commit 0ee8e8f081
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_key_generate_prisma.py
View file

@ -46,7 +46,7 @@ from litellm.proxy.proxy_server import (
spend_key_fn, spend_key_fn,
view_spend_logs, view_spend_logs,
) )
from litellm.proxy.utils import PrismaClient, ProxyLogging from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
verbose_proxy_logger.setLevel(level=logging.DEBUG) verbose_proxy_logger.setLevel(level=logging.DEBUG)
@ -918,7 +918,7 @@ def test_call_with_key_over_budget(prisma_client):
"stream": False, "stream": False,
"litellm_params": { "litellm_params": {
"metadata": { "metadata": {
"user_api_key": generated_key, "user_api_key": hash_token(generated_key),
"user_api_key_user_id": user_id, "user_api_key_user_id": user_id,
} }
}, },
@ -1009,7 +1009,7 @@ async def test_call_with_key_never_over_budget(prisma_client):
"stream": False, "stream": False,
"litellm_params": { "litellm_params": {
"metadata": { "metadata": {
"user_api_key": generated_key, "user_api_key": hash_token(generated_key),
"user_api_key_user_id": user_id, "user_api_key_user_id": user_id,
} }
}, },
@ -1083,7 +1083,7 @@ async def test_call_with_key_over_budget_stream(prisma_client):
"complete_streaming_response": resp, "complete_streaming_response": resp,
"litellm_params": { "litellm_params": {
"metadata": { "metadata": {
"user_api_key": generated_key, "user_api_key": hash_token(generated_key),
"user_api_key_user_id": user_id, "user_api_key_user_id": user_id,
} }
}, },