Fix litellm issues to be more broad

This commit is contained in:
Brandon Hancock
2025-01-23 11:57:43 -05:00
parent e27a15023c
commit 698b2e4c70
2 changed files with 5 additions and 20 deletions

View File

@@ -139,7 +139,6 @@ Now let's get you set up! 🚀
│ └── __init__.py │ └── __init__.py
└── config/ └── config/
├── agents.yaml ├── agents.yaml
├── config.yaml
└── tasks.yaml └── tasks.yaml
``` ```
</Frame> </Frame>

View File

@@ -149,33 +149,19 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
if self._is_context_length_exceeded(e): if self._is_context_length_exceeded(e):
self._handle_context_length() self._handle_context_length()
continue continue
elif self._is_litellm_authentication_error(e):
self._handle_litellm_auth_error(e)
raise e
else: else:
self._printer.print( self._handle_unknown_error(e)
content=f"Unhandled exception: {e}", raise e
color="red",
)
finally: finally:
self.iterations += 1 self.iterations += 1
self._show_logs(formatted_answer) self._show_logs(formatted_answer)
return formatted_answer return formatted_answer
def _is_litellm_authentication_error(self, exception: Exception) -> bool: def _handle_unknown_error(self, exception: Exception) -> None:
"""Check if the exception is a litellm authentication error.""" """Handle unknown errors by informing the user."""
if LiteLLMAuthenticationError and isinstance(
exception, LiteLLMAuthenticationError
):
return True
return False
def _handle_litellm_auth_error(self, exception: Exception) -> None:
"""Handle litellm authentication error by informing the user and exiting."""
self._printer.print( self._printer.print(
content="Authentication error with litellm occurred. Please check your API key and configuration.", content="An unknown error occurred. Please check the details below.",
color="red", color="red",
) )
self._printer.print( self._printer.print(