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

feat(proxy_server.py): new /key/info endpoint to access key information (master key only)

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-24 15:24:50 -08:00
parent 4f22e7de18
commit bc84b38154
1 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/proxy/proxy_server.py
View file

@ -162,8 +162,8 @@ async def user_api_key_auth(request: Request, api_key: str = fastapi.Security(ap
"api_key": master_key
}
if (route == "/key/generate" or route == "/key/delete") and not is_master_key_valid:
raise Exception(f"If master key is set, only master key can be used to generate new keys")
if (route == "/key/generate" or route == "/key/delete" or route == "/key/info") and not is_master_key_valid:
raise Exception(f"If master key is set, only master key can be used to generate, delete or get info for new keys")
if prisma_client:
## check for cache hit (In-Memory Cache)
@ -786,6 +786,24 @@ async def delete_key_fn(request: Request):
detail={"error": str(e)},
)
@router.get("/key/info", dependencies=[Depends(user_api_key_auth)])
async def info_key_fn(key: str = fastapi.Query(..., description="Key in the request parameters")):
global prisma_client
try:
key_info = await prisma_client.litellm_verificationtoken.find_unique(
where={
"token": key
}
)
return {"key": key, "info": key_info}
except Exception as e:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={"error": str(e)},
)
@router.get("/test")
async def test_endpoint(request: Request):
return {"route": request.url.path}