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

Merge pull request #4071 from BerriAI/litellm_log_db_exceptions_otel

Browse source 
[FEAT] OTEL - LOG DB Exceptions
This commit is contained in:
Ishaan Jaff 2024-06-07 17:13:20 -07:00 • committed by GitHub
commit 57b70cde53
1 changed files with 40 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/proxy/utils.py
View file

@ -99,6 +99,8 @@ def log_to_opentelemetry(func):
@wraps(func) @wraps(func)
async def wrapper(*args, **kwargs): async def wrapper(*args, **kwargs):
start_time = datetime.now() start_time = datetime.now()
try:
result = await func(*args, **kwargs) result = await func(*args, **kwargs)
end_time = datetime.now() end_time = datetime.now()
@ -120,6 +122,24 @@ def log_to_opentelemetry(func):
) )
# end of logging to otel # end of logging to otel
return result return result
except Exception as e:
end_time = datetime.now()
if (
"parent_otel_span" in kwargs
and kwargs["parent_otel_span"] is not None
and "proxy_logging_obj" in kwargs
and kwargs["proxy_logging_obj"] is not None
):
proxy_logging_obj = kwargs["proxy_logging_obj"]
await proxy_logging_obj.service_logging_obj.async_service_failure_hook(
error=e,
service=ServiceTypes.DB,
call_type=func.__name__,
duration=0.0,
start_time=start_time,
end_time=end_time,
)
raise e
return wrapper return wrapper