mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
Fix litellm issues to be more broad
This commit is contained in:
@@ -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>
|
||||||
|
|||||||
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user