phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

use async_log_proxy_authentication_errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 19:39:36 -08:00
parent 8d57ccb74e
commit 75dacd8859
2 changed files with 8 additions and 6 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}, extra={"requester_ip": requester_ip},
) )
# Log this exception to OTEL # Log this exception to OTEL, Datadog etc
if open_telemetry_logger is not None: asyncio.create_task(
await open_telemetry_logger.async_post_call_failure_hook( # type: ignore proxy_logging_obj.async_log_proxy_authentication_errors(
original_exception=e, original_exception=e,
request_data={}, request=request,
user_api_key_dict=UserAPIKeyAuth(parent_otel_span=parent_otel_span), parent_otel_span=parent_otel_span,
api_key=api_key,
) )
)
if isinstance(e, litellm.BudgetExceededError): if isinstance(e, litellm.BudgetExceededError):
raise ProxyException( raise ProxyException(

2
litellm/proxy/utils.py
View file

@ -899,7 +899,7 @@ class ProxyLogging:
- `post_call_failure_hook` calls `litellm_logging_obj.async_failure_handler`. This led to the Exception being logged twice - `post_call_failure_hook` calls `litellm_logging_obj.async_failure_handler`. This led to the Exception being logged twice
What does this handler do? What does this handler do?
- Logs Authentication Errors (like invalid API Key passed) to CustomLogger compatible classes - Logs Authentication Errors (like invalid API Key passed) to CustomLogger compatible classes (OTEL, Datadog etc)
- calls CustomLogger.async_post_call_failure_hook - calls CustomLogger.async_post_call_failure_hook
""" """