mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
29 lines
1 KiB
Python
29 lines
1 KiB
Python
from typing import TYPE_CHECKING, Any
|
|
|
|
import litellm
|
|
from litellm._logging import verbose_router_logger
|
|
from litellm.integrations.custom_logger import CustomLogger
|
|
|
|
|
|
async def log_success_fallback_event(original_model_group: str, kwargs: dict):
|
|
for _callback in litellm.callbacks:
|
|
if isinstance(_callback, CustomLogger):
|
|
try:
|
|
await _callback.log_success_fallback_event()
|
|
except Exception as e:
|
|
verbose_router_logger.error(
|
|
f"Error in log_success_fallback_event: {(str(e))}"
|
|
)
|
|
pass
|
|
|
|
|
|
async def log_failure_fallback_event(original_model_group: str, kwargs: dict):
|
|
for _callback in litellm.callbacks:
|
|
if isinstance(_callback, CustomLogger):
|
|
try:
|
|
await _callback.log_failure_fallback_event()
|
|
except Exception as e:
|
|
verbose_router_logger.error(
|
|
f"Error in log_failure_fallback_event: {(str(e))}"
|
|
)
|
|
pass
|