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

fix bug when updating team

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-13 15:18:09 -07:00
parent d23e941bac
commit 676fdfb1ca
2 changed files with 35 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/common_utils/management_endpoint_utils.py
View file

@ -32,7 +32,7 @@ def management_endpoint_wrapper(func):
if open_telemetry_logger is not None: if open_telemetry_logger is not None:
_http_request: Request = kwargs.get("http_request") _http_request: Request = kwargs.get("http_request")
if _http_request:
_route = _http_request.url.path _route = _http_request.url.path
_request_body: dict = await _read_request_body( _request_body: dict = await _read_request_body(
request=_http_request request=_http_request
@ -67,6 +67,7 @@ def management_endpoint_wrapper(func):
if open_telemetry_logger is not None: if open_telemetry_logger is not None:
_http_request: Request = kwargs.get("http_request") _http_request: Request = kwargs.get("http_request")
if _http_request:
_route = _http_request.url.path _route = _http_request.url.path
_request_body: dict = await _read_request_body( _request_body: dict = await _read_request_body(
request=_http_request request=_http_request

1
litellm/proxy/proxy_server.py
View file

@ -10181,6 +10181,7 @@ async def create_audit_log_for_update(request_data: LiteLLM_AuditLogs):
@management_endpoint_wrapper @management_endpoint_wrapper
async def update_team( async def update_team(
data: UpdateTeamRequest, data: UpdateTeamRequest,
http_request: Request,
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
litellm_changed_by: Optional[str] = Header( litellm_changed_by: Optional[str] = Header(
None, None,