Adding support to force a tool return to be the final answer. (#867)

* Adding support to force a tool return to be the final answer.
This will at the end of the execution return the tool output.
It will return the output of the latest tool with the flag

* Update src/crewai/agent.py

Co-authored-by: Gui Vieira <guilherme_vieira@me.com>

* Update tests/agent_test.py

Co-authored-by: Gui Vieira <guilherme_vieira@me.com>

---------

Co-authored-by: Gui Vieira <guilherme_vieira@me.com>

src/crewai/agent.py

@@ -90,6 +90,9 @@ class Agent(BaseAgent):
     response_template: Optional[str] = Field(
         default=None, description="Response format for the agent."
     )
+    tools_results: Optional[List[Any]] = Field(
+        default=[], description="Results of the tools used by the agent."
+    )
     allow_code_execution: Optional[bool] = Field(
         default=False, description="Enable code execution for the agent."
     )
@@ -116,7 +119,8 @@ class Agent(BaseAgent):
                 self.llm.callbacks.append(token_handler)
 
             if agentops and not any(
-                isinstance(handler, agentops.LangchainCallbackHandler) for handler in self.llm.callbacks
+                isinstance(handler, agentops.LangchainCallbackHandler)
+                for handler in self.llm.callbacks
             ):
                 agentops.stop_instrumenting()
                 self.llm.callbacks.append(agentops.LangchainCallbackHandler())
@@ -188,6 +192,14 @@ class Agent(BaseAgent):
         )["output"]
         if self.max_rpm:
             self._rpm_controller.stop_rpm_counter()
+
+        # If there was any tool in self.tools_results that had result_as_answer
+        # set to True, return the results of the last tool that had
+        # result_as_answer set to True
+        for tool_result in self.tools_results:
+            if tool_result.get("result_as_answer", False):
+                result = tool_result["result"]
+
         return result
 
     def format_log_to_str(

src/crewai/agents/executor.py

@@ -243,6 +243,7 @@ class CrewAgentExecutor(AgentExecutor, CrewAgentExecutorMixin):
                 tools_names=self.tools_names,
                 function_calling_llm=self.function_calling_llm,
                 task=self.task,
+                agent=self.crew_agent,
                 action=agent_action,
             )
             tool_calling = tool_usage.parse(agent_action.log)

src/crewai/tools/tool_usage.py

@@ -51,6 +51,7 @@ class ToolUsage:
         tools_names: str,
         task: Any,
         function_calling_llm: Any,
+        agent: Any,
         action: Any,
     ) -> None:
         self._i18n: I18N = I18N()
@@ -59,6 +60,7 @@ class ToolUsage:
         self._run_attempts: int = 1
         self._max_parsing_attempts: int = 3
         self._remember_format_after_usages: int = 3
+        self.agent = agent
         self.tools_description = tools_description
         self.tools_names = tools_names
         self.tools_handler = tools_handler
@@ -130,6 +132,10 @@ class ToolUsage:
                 tool=calling.tool_name, input=calling.arguments
             )
 
+        original_tool = next(
+            (ot for ot in self.original_tools if ot.name == tool.name), None
+        )
+
         if result is None:  #! finecwg: if not result --> if result is None
             try:
                 if calling.tool_name in [
@@ -179,9 +185,6 @@ class ToolUsage:
 
             if self.tools_handler:
                 should_cache = True
-                original_tool = next(
-                    (ot for ot in self.original_tools if ot.name == tool.name), None
-                )
                 if (
                     hasattr(original_tool, "cache_function")
                     and original_tool.cache_function  # type: ignore # Item "None" of "Any | None" has no attribute "cache_function"
@@ -203,6 +206,21 @@ class ToolUsage:
             attempts=self._run_attempts,
         )
         result = self._format_result(result=result)  # type: ignore # "_format_result" of "ToolUsage" does not return a value (it only ever returns None)
+        data = {
+            "result": result,
+            "tool_name": tool.name,
+            "tool_args": calling.arguments,
+        }
+
+        if (
+            hasattr(original_tool, "result_as_answer")
+            and original_tool.result_as_answer  # type: ignore # Item "None" of "Any | None" has no attribute "cache_function"
+        ):
+            result_as_answer = original_tool.result_as_answer
+            data["result_as_answer"] = result_as_answer
+
+        self.agent.tools_results.append(data)
+
         return result  # type: ignore # No return value expected
 
     def _format_result(self, result: Any) -> None: