Better agent execution error handling (crewAIInc#54)

A few quality of life improvements around cache handling and repeated tool usage

crewai/agents/__init__.py

@@ -1,4 +1,4 @@
-from .cache_handler import CacheHandler
+from .cache.cache_handler import CacheHandler
 from .executor import CrewAgentExecutor
 from .output_parser import CrewAgentOutputParser
 from .tools_handler import ToolsHandler

crewai/agents/cache/__init__.py

@@ -0,0 +1,2 @@
+from .cache_handler import CacheHandler
+from .cache_hit import CacheHit

crewai/agents/exceptions.py

@@ -0,0 +1,20 @@
+from langchain_core.exceptions import OutputParserException
+
+
+class TaskRepeatedUsageException(OutputParserException):
+    """Exception raised when a task is used twice in a roll."""
+
+    error: str = "TaskRepeatedUsageException"
+    message: str = "\nI just used the {tool} tool with input {tool_input}. So I already know the result of that.\n"
+
+    def __init__(self, tool: str, tool_input: str):
+        self.tool = tool
+        self.tool_input = tool_input
+        self.message = self.message.format(tool=tool, tool_input=tool_input)
+
+        super().__init__(
+            error=self.error, observation=self.message, send_to_llm=True, llm_output=""
+        )
+
+    def __str__(self):
+        return self.message

crewai/agents/executor.py

@@ -9,7 +9,7 @@
 from langchain_core.tools import BaseTool
 
 from ..tools.cache_tools import CacheTools
-from .cache_hit import CacheHit
+from .cache.cache_hit import CacheHit
 
 
 class CrewAgentExecutor(AgentExecutor):

crewai/agents/output_parser.py

@@ -3,10 +3,9 @@
 
 from langchain.agents.output_parsers import ReActSingleInputOutputParser
 from langchain_core.agents import AgentAction, AgentFinish
-from langchain_core.exceptions import OutputParserException
 
-from .cache_handler import CacheHandler
-from .cache_hit import CacheHit
+from .cache import CacheHandler, CacheHit
+from .exceptions import TaskRepeatedUsageException
 from .tools_handler import ToolsHandler
 
 FINAL_ANSWER_ACTION = "Final Answer:"
@@ -67,9 +66,7 @@ def parse(self, text: str) -> Union[AgentAction, AgentFinish, CacheHit]:
                     "input": tool_input,
                 }
                 if usage == last_tool_usage:
-                    raise OutputParserException(
-                        f"""\nI just used the {action} tool with input {tool_input}. So I already know the result of that."""
-                    )
+                    raise TaskRepeatedUsageException(tool=action, tool_input=tool_input)
 
             result = self.cache.read(action, tool_input)
             if result:

crewai/agents/tools_handler.py

@@ -3,7 +3,7 @@
 from langchain.callbacks.base import BaseCallbackHandler
 
 from ..tools.cache_tools import CacheTools
-from .cache_handler import CacheHandler
+from .cache.cache_handler import CacheHandler
 
 
 class ToolsHandler(BaseCallbackHandler):

crewai/crew.py

@@ -15,7 +15,7 @@
 from pydantic_core import PydanticCustomError
 
 from crewai.agent import Agent
-from crewai.agents import CacheHandler
+from crewai.agents.cache import CacheHandler
 from crewai.process import Process
 from crewai.task import Task
 from crewai.tools.agent_tools import AgentTools

crewai/tools/agent_tools.py

@@ -53,10 +53,10 @@ def __execute(self, command):
         try:
             agent, task, information = command.split("|")
         except ValueError:
-            return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|information`."
+            return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|information`.\n"
 
         if not agent or not task or not information:
-            return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|question|information`."
+            return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|question|information`.\n"
 
         agent = [
             available_agent
@@ -65,7 +65,7 @@ def __execute(self, command):
         ]
 
         if len(agent) == 0:
-            return f"\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: {', '.join([agent.role for agent in self.agents])}."
+            return f"\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: {', '.join([agent.role for agent in self.agents])}.\n"
 
         agent = agent[0]
         result = agent.execute_task(task, information)

crewai/tools/cache_tools.py

@@ -1,7 +1,7 @@
 from langchain.tools import Tool
 from pydantic import BaseModel, ConfigDict, Field
 
-from crewai.agents import CacheHandler
+from crewai.agents.cache import CacheHandler
 
 
 class CacheTools(BaseModel):

tests/agent_test.py

@@ -4,7 +4,7 @@
 from langchain.chat_models import ChatOpenAI as OpenAI
 
 from crewai.agent import Agent
-from crewai.agents import CacheHandler
+from crewai.agents.cache import CacheHandler
 
 
 def test_agent_creation():

tests/agent_tools/agent_tools_test.py

@@ -48,7 +48,7 @@ def test_delegate_work_with_wrong_input():
 
     assert (
         result
-        == "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|information`."
+        == "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|information`.\n"
     )
 
 
@@ -59,7 +59,7 @@ def test_delegate_work_to_wrong_agent():
 
     assert (
         result
-        == "\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: researcher."
+        == "\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: researcher.\n"
     )
 
 
@@ -70,5 +70,5 @@ def test_ask_question_to_wrong_agent():
 
     assert (
         result
-        == "\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: researcher."
+        == "\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: researcher.\n"
     )

tests/crew_test.py

@@ -5,7 +5,7 @@
 import pytest
 
 from crewai.agent import Agent
-from crewai.agents import CacheHandler
+from crewai.agents.cache import CacheHandler
 from crewai.crew import Crew
 from crewai.process import Process
 from crewai.task import Task