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

fix(proxy_server.py): support tracking org spend

Browse source 
currently works when org set for jwt auth
This commit is contained in:
Krrish Dholakia 2024-04-11 23:01:21 -07:00
parent de9258e700
commit aa5da4346a
6 changed files with 136 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/auth/handle_jwt.py
View file

@ -84,6 +84,16 @@ class JWTHandler:
user_id = default_value
return user_id
def get_org_id(self, token: dict, default_value: Optional[str]) -> Optional[str]:
try:
if self.litellm_jwtauth.org_id_jwt_field is not None:
org_id = token[self.litellm_jwtauth.org_id_jwt_field]
else:
org_id = None
except KeyError:
org_id = default_value
return org_id
def get_scopes(self, token: dict) -> list:
try:
if isinstance(token["scope"], str):