andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

improved context window exceeding exception class

Browse Source
This commit is contained in:
Lorenze Jay
2024-08-01 13:12:43 -07:00
parent a7b0c7e122
commit c1cb7ee9c5
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/utilities/exceptions/context_window_exceeding_exception.py
View File

@@ -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."
)