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 #4216 from BerriAI/litellm_refactor_logging

Browse source 
refactor(utils.py): Cut down utils.py to <10k lines.
This commit is contained in:
Krish Dholakia 2024-06-15 15:19:42 -07:00 • committed by GitHub
commit fa2d8bc794
26 changed files with 3052 additions and 3087 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/utils.py
View file

@ -12,6 +12,8 @@ import litellm
import backoff
import traceback
from pydantic import BaseModel
import litellm.litellm_core_utils
import litellm.litellm_core_utils.litellm_logging
from litellm.proxy._types import (
UserAPIKeyAuth,
DynamoDBArgs,
@ -266,7 +268,9 @@ class ProxyLogging:
+ litellm.failure_callback
)
)
litellm.utils.set_callbacks(callback_list=callback_list)
litellm.litellm_core_utils.litellm_logging.set_callbacks(
callback_list=callback_list
)
# The actual implementation of the function
async def pre_call_hook(
@ -331,7 +335,9 @@ class ProxyLogging:
return data
except Exception as e:
if "litellm_logging_obj" in data:
logging_obj: litellm.utils.Logging = data["litellm_logging_obj"]
logging_obj: litellm.litellm_core_utils.litellm_logging.Logging = data[
"litellm_logging_obj"
]
## ASYNC FAILURE HANDLER ##
error_message = ""