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

(feat) log proxy auth errors on datadog (#6931)

Browse source 
* add new dd type for auth errors

* add async_log_proxy_authentication_errors

* fix comment

* use async_log_proxy_authentication_errors

* test_datadog_post_call_failure_hook

* test_async_log_proxy_authentication_errors
This commit is contained in:
Ishaan Jaff 2024-11-26 20:26:57 -08:00 • committed by GitHub
parent b08b37b11d
commit c3ac98f992
7 changed files with 241 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1197,13 +1197,15 @@ async def user_api_key_auth( # noqa: PLR0915
extra={"requester_ip": requester_ip},
)
# Log this exception to OTEL
if open_telemetry_logger is not None:
await open_telemetry_logger.async_post_call_failure_hook( # type: ignore
# Log this exception to OTEL, Datadog etc
asyncio.create_task(
proxy_logging_obj.async_log_proxy_authentication_errors(
original_exception=e,
request_data={},
user_api_key_dict=UserAPIKeyAuth(parent_otel_span=parent_otel_span),
request=request,
parent_otel_span=parent_otel_span,
api_key=api_key,
)
)
if isinstance(e, litellm.BudgetExceededError):
raise ProxyException(