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

(fix) logging Auth errors on datadog (#6995)

Browse source 
* fix get_standard_logging_object_payload

* fix async_post_call_failure_hook

* fix post_call_failure_hook

* fix change

* fix _is_proxy_only_error

* fix async_post_call_failure_hook

* fix getting request body

* remove redundant code

* use a well named original function name for auth errors

* fix logging auth fails on DD

* fix using request body

* use helper for _handle_logging_proxy_only_error
This commit is contained in:
Ishaan Jaff 2024-12-02 23:01:21 -08:00 • committed by GitHub
parent 05aa740a7d
commit 9617e7433d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 121 additions and 413 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/auth/user_api_key_auth.py
View file

@ -270,8 +270,8 @@ async def user_api_key_auth( # noqa: PLR0915
parent_otel_span: Optional[Span] = None
start_time = datetime.now()
route: str = get_request_route(request=request)
try:
route: str = get_request_route(request=request)
# get the request body
request_data = await _read_request_body(request=request)
await pre_db_read_auth_checks(
@ -1251,12 +1251,18 @@ async def user_api_key_auth( # noqa: PLR0915
)
# Log this exception to OTEL, Datadog etc
user_api_key_dict = UserAPIKeyAuth(
parent_otel_span=parent_otel_span,
api_key=api_key,
)
request_data = await _read_request_body(request=request)
asyncio.create_task(
proxy_logging_obj.async_log_proxy_authentication_errors(
proxy_logging_obj.post_call_failure_hook(
request_data=request_data,
original_exception=e,
request=request,
parent_otel_span=parent_otel_span,
api_key=api_key,
user_api_key_dict=user_api_key_dict,
error_type=ProxyErrorTypes.auth_error,
route=route,
)
)