phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) dynamo - use hashed tokens budget tracking

Browse source
This commit is contained in:
ishaan-jaff 2024-01-26 14:32:34 -08:00
parent 2c79585e18
commit 8b5e397abe
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_key_generate_dynamodb.py
View file

@ -33,7 +33,7 @@ from litellm.proxy.proxy_server import (
) )
from litellm.proxy._types import NewUserRequest, DynamoDBArgs, GenerateKeyRequest from litellm.proxy._types import NewUserRequest, DynamoDBArgs, GenerateKeyRequest
from litellm.proxy.utils import DBClient from litellm.proxy.utils import DBClient, hash_token
from starlette.datastructures import URL from starlette.datastructures import URL
@ -232,7 +232,7 @@ def test_call_with_user_over_budget(custom_db_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,
} }
}, },
@ -305,7 +305,7 @@ def test_call_with_user_over_budget_stream(custom_db_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,
} }
}, },
@ -376,7 +376,7 @@ def test_call_with_user_key_budget(custom_db_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,
} }
}, },
@ -449,7 +449,7 @@ def test_call_with_key_over_budget_stream(custom_db_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,
} }
}, },