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

test: testing fixes

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-26 19:10:05 -08:00
parent d0c205fcd1
commit 7942539f9b
1 changed files with 2 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/tests/test_key_generate_prisma.py
View file

@ -124,25 +124,12 @@ def test_generate_and_call_with_valid_key(prisma_client):
bearer_token = "Bearer " + generated_key bearer_token = "Bearer " + generated_key
assert generated_key not in user_api_key_cache.in_memory_cache.cache_dict assert generated_key not in user_api_key_cache.in_memory_cache.cache_dict
assert (
hash_token(generated_key)
in user_api_key_cache.in_memory_cache.cache_dict
)
cached_value = user_api_key_cache.in_memory_cache.cache_dict[ value_from_prisma = await prisma_client.get_data(
hash_token(generated_key)
]
print("cached value=", cached_value)
print("cached token", cached_value.token)
value_from_prisma = valid_token = await prisma_client.get_data(
token=generated_key, token=generated_key,
) )
print("token from prisma", value_from_prisma) print("token from prisma", value_from_prisma)
assert value_from_prisma.token == cached_value.token
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})
request._url = URL(url="/chat/completions") request._url = URL(url="/chat/completions")
@ -1312,7 +1299,7 @@ async def test_call_with_key_over_budget_stream(prisma_client):
generated_key = key.key generated_key = key.key
user_id = key.user_id user_id = key.user_id
bearer_token = "Bearer " + generated_key bearer_token = "Bearer " + generated_key
print(f"generated_key: {generated_key}")
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})
request._url = URL(url="/chat/completions") request._url = URL(url="/chat/completions")