From 1e0b70141b85638096529ab811fc62152fd7c0e5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 24 Jul 2025 14:37:56 -0700 Subject: [PATCH] fix async_pre_routing_hook --- litellm/router.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index eec6129d1c2..5265c9848d5 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -6650,18 +6650,19 @@ class Router: Used for the litellm auto-router to modify the request before the routing decision is made. """ - custom_logger_callbacks: List[CustomLogger] = litellm.logging_callback_manager.get_custom_loggers_for_type( - callback_type=CustomLogger, - ) - - for callback in custom_logger_callbacks: - await callback.async_pre_routing_hook( + ######################################################### + # Check if any auto-router should be used + ######################################################### + if model in self.auto_routers: + await self.auto_routers[model].async_pre_routing_hook( model=model, request_kwargs=request_kwargs, messages=messages, input=input, specific_deployment=specific_deployment, ) + +