phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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

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

@ -32,25 +32,25 @@ 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
_request_body: dict = await _read_request_body(
request=_http_request
)
_response = dict(result) if result is not None else None
_route = _http_request.url.path logging_payload = ManagementEndpointLoggingPayload(
_request_body: dict = await _read_request_body( route=_route,
request=_http_request request_data=_request_body,
) response=_response,
_response = dict(result) if result is not None else None start_time=start_time,
end_time=end_time,
)
logging_payload = ManagementEndpointLoggingPayload( await open_telemetry_logger.async_management_endpoint_success_hook(
route=_route, logging_payload=logging_payload,
request_data=_request_body, parent_otel_span=parent_otel_span,
response=_response, )
start_time=start_time,
end_time=end_time,
)
await open_telemetry_logger.async_management_endpoint_success_hook(
logging_payload=logging_payload,
parent_otel_span=parent_otel_span,
)
return result return result
except Exception as e: except Exception as e:
@ -67,23 +67,24 @@ 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")
_route = _http_request.url.path if _http_request:
_request_body: dict = await _read_request_body( _route = _http_request.url.path
request=_http_request _request_body: dict = await _read_request_body(
) request=_http_request
logging_payload = ManagementEndpointLoggingPayload( )
route=_route, logging_payload = ManagementEndpointLoggingPayload(
request_data=_request_body, route=_route,
response=None, request_data=_request_body,
start_time=start_time, response=None,
end_time=end_time, start_time=start_time,
exception=e, end_time=end_time,
) exception=e,
)
await open_telemetry_logger.async_management_endpoint_failure_hook( await open_telemetry_logger.async_management_endpoint_failure_hook(
logging_payload=logging_payload, logging_payload=logging_payload,
parent_otel_span=parent_otel_span, parent_otel_span=parent_otel_span,
) )
raise e raise e

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,