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

feat - use log_to_opentelemetry for _PROXY_track_cost_callback

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-27 11:08:22 -07:00
parent a105348614
commit cde46a4a09
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -203,6 +203,7 @@ from litellm.proxy.utils import (
get_error_message_str, get_error_message_str,
get_instance_fn, get_instance_fn,
hash_token, hash_token,
log_to_opentelemetry,
reset_budget, reset_budget,
send_email, send_email,
update_spend, update_spend,
@ -649,6 +650,7 @@ async def _PROXY_failure_handler(
pass pass
@log_to_opentelemetry
async def _PROXY_track_cost_callback( async def _PROXY_track_cost_callback(
kwargs, # kwargs to completion kwargs, # kwargs to completion
completion_response: litellm.ModelResponse, # response from completion completion_response: litellm.ModelResponse, # response from completion