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

(feat) write team_id to User Table

Browse source
This commit is contained in:
ishaan-jaff 2024-01-18 14:42:46 -08:00
parent 5beef6dbcd
commit 2b6972111e
3 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -1081,12 +1081,15 @@ async def generate_key_helper_fn(
config_json = json.dumps(config) config_json = json.dumps(config)
metadata_json = json.dumps(metadata) metadata_json = json.dumps(metadata)
user_id = user_id or str(uuid.uuid4()) user_id = user_id or str(uuid.uuid4())
if type(team_id) is not str:
team_id = str(team_id)
try: try:
# Create a new verification token (you may want to enhance this logic based on your needs) # Create a new verification token (you may want to enhance this logic based on your needs)
user_data = { user_data = {
"max_budget": max_budget, "max_budget": max_budget,
"user_email": user_email, "user_email": user_email,
"user_id": user_id, "user_id": user_id,
"team_id": team_id,
"spend": spend, "spend": spend,
"models": models, "models": models,
} }

1
litellm/proxy/schema.prisma
View file

@ -9,6 +9,7 @@ generator client {
model LiteLLM_UserTable { model LiteLLM_UserTable {
user_id String @unique user_id String @unique
team_id String?
max_budget Float? max_budget Float?
spend Float @default(0.0) spend Float @default(0.0)
user_email String? user_email String?

1
schema.prisma
View file

@ -9,6 +9,7 @@ generator client {
model LiteLLM_UserTable { model LiteLLM_UserTable {
user_id String @unique user_id String @unique
team_id String?
max_budget Float? max_budget Float?
spend Float @default(0.0) spend Float @default(0.0)
user_email String? user_email String?