forked from phoenix/litellm-mirror
feat - rename end_user -> customer
This commit is contained in:
parent
54e4a2f7ac
commit
9b8c143601
1 changed files with 61 additions and 22 deletions
|
@ -7131,13 +7131,15 @@ async def global_predict_spend_logs(request: Request):
|
||||||
#### INTERNAL USER MANAGEMENT ####
|
#### INTERNAL USER MANAGEMENT ####
|
||||||
@router.post(
|
@router.post(
|
||||||
"/user/new",
|
"/user/new",
|
||||||
tags=["user management"],
|
tags=["Internal User management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
response_model=NewUserResponse,
|
response_model=NewUserResponse,
|
||||||
)
|
)
|
||||||
async def new_user(data: NewUserRequest):
|
async def new_user(data: NewUserRequest):
|
||||||
"""
|
"""
|
||||||
Use this to create a new user with a budget. This creates a new user and generates a new api key for the new user. The new api key is returned.
|
Use this to create a new INTERNAL user with a budget.
|
||||||
|
Internal Users can access LiteLLM Admin UI to make keys, request access to models.
|
||||||
|
This creates a new user and generates a new api key for the new user. The new api key is returned.
|
||||||
|
|
||||||
Returns user id, budget + new key.
|
Returns user id, budget + new key.
|
||||||
|
|
||||||
|
@ -7208,7 +7210,9 @@ async def new_user(data: NewUserRequest):
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/user/auth", tags=["user management"], dependencies=[Depends(user_api_key_auth)]
|
"/user/auth",
|
||||||
|
tags=["Internal User management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def user_auth(request: Request):
|
async def user_auth(request: Request):
|
||||||
"""
|
"""
|
||||||
|
@ -7274,7 +7278,9 @@ async def user_auth(request: Request):
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/user/info", tags=["user management"], dependencies=[Depends(user_api_key_auth)]
|
"/user/info",
|
||||||
|
tags=["Internal User management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def user_info(
|
async def user_info(
|
||||||
user_id: Optional[str] = fastapi.Query(
|
user_id: Optional[str] = fastapi.Query(
|
||||||
|
@ -7446,7 +7452,9 @@ async def user_info(
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/user/update", tags=["user management"], dependencies=[Depends(user_api_key_auth)]
|
"/user/update",
|
||||||
|
tags=["Internal User management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def user_update(data: UpdateUserRequest):
|
async def user_update(data: UpdateUserRequest):
|
||||||
"""
|
"""
|
||||||
|
@ -7540,7 +7548,7 @@ async def user_update(data: UpdateUserRequest):
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/user/request_model",
|
"/user/request_model",
|
||||||
tags=["user management"],
|
tags=["Internal User management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def user_request_model(request: Request):
|
async def user_request_model(request: Request):
|
||||||
|
@ -7593,7 +7601,7 @@ async def user_request_model(request: Request):
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/user/get_requests",
|
"/user/get_requests",
|
||||||
tags=["user management"],
|
tags=["Internal User management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def user_get_requests():
|
async def user_get_requests():
|
||||||
|
@ -7635,7 +7643,7 @@ async def user_get_requests():
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/user/get_users",
|
"/user/get_users",
|
||||||
tags=["user management"],
|
tags=["Internal User management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def get_users(
|
async def get_users(
|
||||||
|
@ -7672,7 +7680,13 @@ async def get_users(
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/end_user/block",
|
"/end_user/block",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
include_in_schema=False,
|
||||||
|
)
|
||||||
|
@router.post(
|
||||||
|
"/customer/block",
|
||||||
|
tags=["Customer Management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def block_user(data: BlockUsers):
|
async def block_user(data: BlockUsers):
|
||||||
|
@ -7715,9 +7729,15 @@ async def block_user(data: BlockUsers):
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/end_user/unblock",
|
"/end_user/unblock",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
|
@router.post(
|
||||||
|
"/customer/unblock",
|
||||||
|
tags=["Customer Management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
include_in_schema=False,
|
||||||
|
)
|
||||||
async def unblock_user(data: BlockUsers):
|
async def unblock_user(data: BlockUsers):
|
||||||
"""
|
"""
|
||||||
[BETA] Unblock calls with this user id
|
[BETA] Unblock calls with this user id
|
||||||
|
@ -7762,7 +7782,13 @@ async def unblock_user(data: BlockUsers):
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
"/end_user/new",
|
"/end_user/new",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
|
include_in_schema=False,
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
)
|
||||||
|
@router.post(
|
||||||
|
"/customer/new",
|
||||||
|
tags=["Customer Management"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def new_end_user(
|
async def new_end_user(
|
||||||
|
@ -7779,18 +7805,13 @@ async def new_end_user(
|
||||||
|
|
||||||
Example curl:
|
Example curl:
|
||||||
```
|
```
|
||||||
curl --location 'http://0.0.0.0:4000/end_user/new' \
|
curl --location 'http://0.0.0.0:4000/customer/new' \
|
||||||
--header 'Authorization: Bearer sk-1234' \
|
--header 'Authorization: Bearer sk-1234' \
|
||||||
--header 'Content-Type: application/json' \
|
--header 'Content-Type: application/json' \
|
||||||
--data '{
|
--data '{
|
||||||
"end_user_id" : "ishaan-jaff-3", <- specific customer
|
"user_id" : "ishaan-jaff-3",
|
||||||
|
"allowed_region": "eu"
|
||||||
"allowed_region": "eu" <- set region for models
|
|
||||||
|
|
||||||
+
|
|
||||||
|
|
||||||
"default_model": "azure/gpt-3.5-turbo-eu" <- all calls from this user, use this model?
|
"default_model": "azure/gpt-3.5-turbo-eu" <- all calls from this user, use this model?
|
||||||
|
|
||||||
}'
|
}'
|
||||||
|
|
||||||
# return end-user object
|
# return end-user object
|
||||||
|
@ -7860,9 +7881,15 @@ async def new_end_user(
|
||||||
return end_user_record
|
return end_user_record
|
||||||
|
|
||||||
|
|
||||||
|
@router.get(
|
||||||
|
"/customer/info",
|
||||||
|
tags=["Customer Management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
)
|
||||||
@router.get(
|
@router.get(
|
||||||
"/end_user/info",
|
"/end_user/info",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
|
include_in_schema=False,
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def end_user_info(
|
async def end_user_info(
|
||||||
|
@ -7885,9 +7912,15 @@ async def end_user_info(
|
||||||
return user_info
|
return user_info
|
||||||
|
|
||||||
|
|
||||||
|
@router.post(
|
||||||
|
"/customer/update",
|
||||||
|
tags=["Customer Management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
)
|
||||||
@router.post(
|
@router.post(
|
||||||
"/end_user/update",
|
"/end_user/update",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
|
include_in_schema=False,
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def update_end_user():
|
async def update_end_user():
|
||||||
|
@ -7897,9 +7930,15 @@ async def update_end_user():
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@router.post(
|
||||||
|
"/customer/delete",
|
||||||
|
tags=["Customer Management"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
)
|
||||||
@router.post(
|
@router.post(
|
||||||
"/end_user/delete",
|
"/end_user/delete",
|
||||||
tags=["End User Management"],
|
tags=["Customer Management"],
|
||||||
|
include_in_schema=False,
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def delete_end_user():
|
async def delete_end_user():
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue