Merge pull request #4809 from BerriAI/litellm_refactor_router_prints

router - use verbose logger when using litellm.Router
This commit is contained in:
Ishaan Jaff 2024-07-20 18:17:48 -07:00 committed by GitHub
commit 27e25fb229
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -417,14 +417,15 @@ class Router:
litellm.failure_callback.append(self.deployment_callback_on_failure)
else:
litellm.failure_callback = [self.deployment_callback_on_failure]
print( # noqa
verbose_router_logger.info(
f"Intialized router with Routing strategy: {self.routing_strategy}\n\n"
f"Routing enable_pre_call_checks: {self.enable_pre_call_checks}\n\n"
f"Routing fallbacks: {self.fallbacks}\n\n"
f"Routing content fallbacks: {self.content_policy_fallbacks}\n\n"
f"Routing context window fallbacks: {self.context_window_fallbacks}\n\n"
f"Router Redis Caching={self.cache.redis_cache}\n"
) # noqa
)
self.routing_strategy_args = routing_strategy_args
self.retry_policy: Optional[RetryPolicy] = retry_policy
self.model_group_retry_policy: Optional[Dict[str, RetryPolicy]] = (