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

(test) custom key/generate

Browse source
This commit is contained in:
ishaan-jaff 2024-01-20 10:13:32 -08:00
parent 6b7fd172a3
commit ccfcc39681
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_key_generate_prisma.py
View file

@ -599,16 +599,21 @@ def test_generate_and_update_key(prisma_client):
def test_key_generate_with_custom_auth(prisma_client): def test_key_generate_with_custom_auth(prisma_client):
# custom - generate key function # custom - generate key function
async def custom_generate_key_fn(data: GenerateKeyRequest): async def custom_generate_key_fn(data: GenerateKeyRequest) -> dict:
""" """
Asynchronously decides if a key should be generated or not based on the provided data. Asynchronous function for generating a key based on the input data.
Args: Args:
data (GenerateKeyRequest): The data to be used for decision making. data (GenerateKeyRequest): The input data for key generation.
Returns: Returns:
bool: True if a key should be generated, False otherwise. dict: A dictionary containing the decision and an optional message.
{
"decision": False,
"message": "This violates LiteLLM Proxy Rules. No team id provided.",
}
""" """
# decide if a key should be generated or not # decide if a key should be generated or not
print("using custom auth function!") print("using custom auth function!")
data_json = data.json() # type: ignore data_json = data.json() # type: ignore