From 698b2e4c7026fcb783d4db7b891a690fa7eff19b Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Thu, 23 Jan 2025 11:57:43 -0500
Subject: [PATCH] Fix litellm issues to be more broad

---
 docs/installation.mdx                    |  1 -
 src/crewai/agents/crew_agent_executor.py | 24 +++++-------------------
 2 files changed, 5 insertions(+), 20 deletions(-)

diff --git a/docs/installation.mdx b/docs/installation.mdx
index d629c4c80..8abba152a 100644
--- a/docs/installation.mdx
+++ b/docs/installation.mdx
@@ -139,7 +139,6 @@ Now let's get you set up! 🚀
             │   └── __init__.py
             └── config/
                 ├── agents.yaml
-                ├── config.yaml
                 └── tasks.yaml
     ```
     </Frame>
diff --git a/src/crewai/agents/crew_agent_executor.py b/src/crewai/agents/crew_agent_executor.py
index ee5b9c582..d7bf97795 100644
--- a/src/crewai/agents/crew_agent_executor.py
+++ b/src/crewai/agents/crew_agent_executor.py
@@ -149,33 +149,19 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
                 if self._is_context_length_exceeded(e):
                     self._handle_context_length()
                     continue
-                elif self._is_litellm_authentication_error(e):
-                    self._handle_litellm_auth_error(e)
-                    raise e
                 else:
-                    self._printer.print(
-                        content=f"Unhandled exception: {e}",
-                        color="red",
-                    )
+                    self._handle_unknown_error(e)
+                    raise e
             finally:
                 self.iterations += 1
 
         self._show_logs(formatted_answer)
         return formatted_answer
 
-    def _is_litellm_authentication_error(self, exception: Exception) -> bool:
-        """Check if the exception is a litellm authentication error."""
-        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."""
+    def _handle_unknown_error(self, exception: Exception) -> None:
+        """Handle unknown errors by informing the user."""
         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",
         )
         self._printer.print(