From d28c6b390ca19d50ba52985988b4e005383ae281 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 11 Oct 2024 12:12:57 +0530 Subject: [PATCH] fix pattern match router --- litellm/router_utils/pattern_match_deployments.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/litellm/router_utils/pattern_match_deployments.py b/litellm/router_utils/pattern_match_deployments.py index c9ea8c1a7..f735c965a 100644 --- a/litellm/router_utils/pattern_match_deployments.py +++ b/litellm/router_utils/pattern_match_deployments.py @@ -5,6 +5,8 @@ Class to handle llm wildcard routing and regex pattern matching import re from typing import Dict, List, Optional +from litellm._logging import verbose_router_logger + class PatternMatchRouter: """ @@ -72,9 +74,12 @@ class PatternMatchRouter: Returns: Optional[List[Deployment]]: llm deployments """ - for pattern, llm_deployments in self.patterns.items(): - if re.match(pattern, request): - return llm_deployments + try: + for pattern, llm_deployments in self.patterns.items(): + if re.match(pattern, request): + return llm_deployments + except Exception as e: + verbose_router_logger.error(f"Error in PatternMatchRouter.route: {str(e)}") return None # No matching pattern found