diff --git a/src/crewai/utilities/exceptions/context_window_exceeding_exception.py b/src/crewai/utilities/exceptions/context_window_exceeding_exception.py index 786a0aa9b..c02c40cc7 100644 --- a/src/crewai/utilities/exceptions/context_window_exceeding_exception.py +++ b/src/crewai/utilities/exceptions/context_window_exceeding_exception.py @@ -11,12 +11,7 @@ class LLMContextLengthExceededException(Exception): def __init__(self, error_message: str): self.original_error_message = error_message - if self._is_context_limit_error(error_message): - super().__init__(self._get_error_message()) - else: - raise ValueError( - "The provided error message is not related to context length limits." - ) + super().__init__(self._get_error_message(error_message)) def _is_context_limit_error(self, error_message: str) -> bool: return any( @@ -24,8 +19,8 @@ class LLMContextLengthExceededException(Exception): for phrase in self.CONTEXT_LIMIT_ERRORS ) - def _get_error_message(self): + def _get_error_message(self, error_message: str): return ( - f"LLM context length exceeded. Original error: {self.original_error_message}\n" + f"LLM context length exceeded. Original error: {error_message}\n" "Consider using a smaller input or implementing a text splitting strategy." )