scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

uv black reformat

Browse source
This commit is contained in:
Oles Lomako 2026-04-23 15:32:08 +03:00
parent 8ccde74413
commit 7987649c3c
1 changed files with 15 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/router.py
View file

@ -871,9 +871,11 @@ class Router:
not hasattr(self, "lowesttpm_logger_v2")
or self.lowesttpm_logger_v2 is None
):
self.lowesttpm_logger_v2: LowestTPMLoggingHandler_v2 = LowestTPMLoggingHandler_v2(
router_cache=self.cache,
routing_args=routing_strategy_args,
self.lowesttpm_logger_v2: LowestTPMLoggingHandler_v2 = (
LowestTPMLoggingHandler_v2(
router_cache=self.cache,
routing_args=routing_strategy_args,
)
)
if isinstance(litellm.callbacks, list):
litellm.logging_callback_manager.add_litellm_callback(self.lowesttpm_logger_v2) # type: ignore
@ -885,9 +887,11 @@ class Router:
not hasattr(self, "lowestlatency_logger")
or self.lowestlatency_logger is None
):
self.lowestlatency_logger: LowestLatencyLoggingHandler = LowestLatencyLoggingHandler(
router_cache=self.cache,
routing_args=routing_strategy_args,
self.lowestlatency_logger: LowestLatencyLoggingHandler = (
LowestLatencyLoggingHandler(
router_cache=self.cache,
routing_args=routing_strategy_args,
)
)
if isinstance(litellm.callbacks, list):
litellm.logging_callback_manager.add_litellm_callback(self.lowestlatency_logger) # type: ignore
@ -896,9 +900,11 @@ class Router:
or routing_strategy == RoutingStrategy.COST_BASED
):
if not hasattr(self, "lowestcost_logger") or self.lowestcost_logger is None:
self.lowestcost_logger: LowestCostLoggingHandler = LowestCostLoggingHandler(
router_cache=self.cache,
routing_args={},
self.lowestcost_logger: LowestCostLoggingHandler = (
LowestCostLoggingHandler(
router_cache=self.cache,
routing_args={},
)
)
if isinstance(litellm.callbacks, list):
litellm.logging_callback_manager.add_litellm_callback(self.lowestcost_logger) # type: ignore