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

(ci/cd) add print_verbose for /key/generate

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 22:38:46 +05:30
parent dfdd329ddf
commit 40aaac69cc
3 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -896,6 +896,7 @@ async def generate_key_helper_fn(
"max_budget": max_budget, "max_budget": max_budget,
"user_email": user_email, "user_email": user_email,
} }
print_verbose("PrismaClient: Before Insert Data")
new_verification_token = await prisma_client.insert_data( new_verification_token = await prisma_client.insert_data(
data=verification_token_data data=verification_token_data
) )
@ -1769,6 +1770,7 @@ async def generate_key_fn(
- expires: (datetime) Datetime object for when key expires. - expires: (datetime) Datetime object for when key expires.
- user_id: (str) Unique user id - used for tracking spend across multiple keys for same user id. - user_id: (str) Unique user id - used for tracking spend across multiple keys for same user id.
""" """
print_verbose("entered /key/generate")
data_json = data.json() # type: ignore data_json = data.json() # type: ignore
response = await generate_key_helper_fn(**data_json) response = await generate_key_helper_fn(**data_json)
return GenerateKeyResponse( return GenerateKeyResponse(

3
litellm/proxy/utils.py
View file

@ -417,6 +417,9 @@ class PrismaClient:
db_data["token"] = hashed_token db_data["token"] = hashed_token
max_budget = db_data.pop("max_budget", None) max_budget = db_data.pop("max_budget", None)
user_email = db_data.pop("user_email", None) user_email = db_data.pop("user_email", None)
print_verbose(
"PrismaClient: Before upsert into litellm_verificationtoken"
)
new_verification_token = await self.db.litellm_verificationtoken.upsert( # type: ignore new_verification_token = await self.db.litellm_verificationtoken.upsert( # type: ignore
where={ where={
"token": hashed_token, "token": hashed_token,

4
litellm/tests/test_proxy_server_keys.py
View file

@ -64,7 +64,7 @@ async def wrapper_startup_event():
import asyncio import asyncio
@pytest.yield_fixture @pytest.fixture
def event_loop(): def event_loop():
"""Create an instance of the default event loop for each test case.""" """Create an instance of the default event loop for each test case."""
policy = asyncio.WindowsSelectorEventLoopPolicy() policy = asyncio.WindowsSelectorEventLoopPolicy()
@ -99,7 +99,7 @@ def test_add_new_key(client):
"aliases": {"mistral-7b": "gpt-3.5-turbo"}, "aliases": {"mistral-7b": "gpt-3.5-turbo"},
"duration": "20m", "duration": "20m",
} }
print("testing proxy server") print("testing proxy server - test_add_new_key")
# Your bearer token # Your bearer token
token = os.getenv("PROXY_MASTER_KEY") token = os.getenv("PROXY_MASTER_KEY")