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

Merge pull request #5059 from BerriAI/litelm_log_otel_args

Browse source 
OTEL - Log DB queries / functions on OTEL
This commit is contained in:
Ishaan Jaff 2024-08-05 20:51:11 -07:00 • committed by GitHub
commit da848696e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 37 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/utils.py
View file

@ -128,6 +128,11 @@ def log_to_opentelemetry(func):
duration=0.0,
start_time=start_time,
end_time=end_time,
event_metadata={
"function_name": func.__name__,
"function_kwargs": kwargs,
"function_args": args,
},
)
elif (
# in litellm custom callbacks kwargs is passed as arg[0]
@ -167,9 +172,15 @@ def log_to_opentelemetry(func):
error=e,
service=ServiceTypes.DB,
call_type=func.__name__,
parent_otel_span=kwargs["parent_otel_span"],
duration=0.0,
start_time=start_time,
end_time=end_time,
event_metadata={
"function_name": func.__name__,
"function_kwargs": kwargs,
"function_args": args,
},
)
raise e