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

Merge pull request #3661 from BerriAI/litellm_enable_alerts_completions_endpoint

Browse source 
[FIX]  raise alerts for exceptions on `/completions` endpoint
This commit is contained in:
Ishaan Jaff 2024-05-15 20:46:02 -07:00 • committed by GitHub
commit 370049c549
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -3935,6 +3935,7 @@ async def completion(
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
): ):
global user_temperature, user_request_timeout, user_max_tokens, user_api_base global user_temperature, user_request_timeout, user_max_tokens, user_api_base
data = {}
check_request_disconnected = None check_request_disconnected = None
try: try:
body = await request.body() body = await request.body()
@ -4080,6 +4081,9 @@ async def completion(
return response return response
except Exception as e: except Exception as e:
data["litellm_status"] = "fail" # used for alerting data["litellm_status"] = "fail" # used for alerting
await proxy_logging_obj.post_call_failure_hook(
user_api_key_dict=user_api_key_dict, original_exception=e, request_data=data
)
verbose_proxy_logger.debug("EXCEPTION RAISED IN PROXY MAIN.PY") verbose_proxy_logger.debug("EXCEPTION RAISED IN PROXY MAIN.PY")
litellm_debug_info = getattr(e, "litellm_debug_info", "") litellm_debug_info = getattr(e, "litellm_debug_info", "")
verbose_proxy_logger.debug( verbose_proxy_logger.debug(