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

use management_endpoint_wrapper for key endpoints

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-12 11:02:21 -07:00
parent 30a54ef007
commit 47953aaa96
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -26,6 +26,7 @@ import litellm
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
from litellm.proxy._types import * from litellm.proxy._types import *
from litellm.proxy.auth.user_api_key_auth import user_api_key_auth from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
from litellm.proxy.management_helpers.utils import management_endpoint_wrapper
from litellm.proxy.utils import _duration_in_seconds from litellm.proxy.utils import _duration_in_seconds
router = APIRouter() router = APIRouter()
@ -37,6 +38,7 @@ router = APIRouter()
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
response_model=GenerateKeyResponse, response_model=GenerateKeyResponse,
) )
@management_endpoint_wrapper
async def generate_key_fn( async def generate_key_fn(
data: GenerateKeyRequest, data: GenerateKeyRequest,
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
@ -278,6 +280,7 @@ async def generate_key_fn(
@router.post( @router.post(
"/key/update", tags=["key management"], dependencies=[Depends(user_api_key_auth)] "/key/update", tags=["key management"], dependencies=[Depends(user_api_key_auth)]
) )
@management_endpoint_wrapper
async def update_key_fn( async def update_key_fn(
request: Request, request: Request,
data: UpdateKeyRequest, data: UpdateKeyRequest,
@ -398,6 +401,7 @@ async def update_key_fn(
@router.post( @router.post(
"/key/delete", tags=["key management"], dependencies=[Depends(user_api_key_auth)] "/key/delete", tags=["key management"], dependencies=[Depends(user_api_key_auth)]
) )
@management_endpoint_wrapper
async def delete_key_fn( async def delete_key_fn(
data: KeyRequest, data: KeyRequest,
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),