Support set max_execution_time to Agent (#2610)

* Fixed fake max_execution_time paramenter
---------

Co-authored-by: Lucas Gomide <lucaslg200@gmail.com>

src/crewai/agent.py

@@ -177,7 +177,7 @@ class Agent(BaseAgent):
         self,
         task: Task,
         context: Optional[str] = None,
-        tools: Optional[List[BaseTool]] = None,
+        tools: Optional[List[BaseTool]] = None
     ) -> str:
         """Execute a task with the agent.
 
@@ -188,6 +188,11 @@ class Agent(BaseAgent):
 
         Returns:
             Output of the agent
+
+        Raises:
+            TimeoutError: If execution exceeds the maximum execution time.
+            ValueError: If the max execution time is not a positive integer.
+            RuntimeError: If the agent execution fails for other reasons.
         """
         if self.tools_handler:
             self.tools_handler.last_used_tool = {}  # type: ignore # Incompatible types in assignment (expression has type "dict[Never, Never]", variable has type "ToolCalling")
@@ -264,14 +269,26 @@ class Agent(BaseAgent):
                     task=task,
                 ),
             )
-            result = self.agent_executor.invoke(
-                {
-                    "input": task_prompt,
-                    "tool_names": self.agent_executor.tools_names,
-                    "tools": self.agent_executor.tools_description,
-                    "ask_for_human_input": task.human_input,
-                }
-            )["output"]
+
+            # Determine execution method based on timeout setting
+            if self.max_execution_time is not None:
+                if not isinstance(self.max_execution_time, int) or self.max_execution_time <= 0:
+                    raise ValueError("Max Execution time must be a positive integer greater than zero")
+                result = self._execute_with_timeout(task_prompt, task, self.max_execution_time)
+            else:
+                result = self._execute_without_timeout(task_prompt, task)
+                
+        except TimeoutError as e:
+            # Propagate TimeoutError without retry
+            crewai_event_bus.emit(
+                self,
+                event=AgentExecutionErrorEvent(
+                    agent=self,
+                    task=task,
+                    error=str(e),
+                ),
+            )
+            raise e
         except Exception as e:
             if e.__class__.__module__.startswith("litellm"):
                 # Do not retry on litellm errors
@@ -312,6 +329,66 @@ class Agent(BaseAgent):
         )
         return result
 
+    def _execute_with_timeout(
+        self,
+        task_prompt: str,
+        task: Task,
+        timeout: int
+    ) -> str:
+        """Execute a task with a timeout.
+        
+        Args:
+            task_prompt: The prompt to send to the agent.
+            task: The task being executed.
+            timeout: Maximum execution time in seconds.
+            
+        Returns:
+            The output of the agent.
+            
+        Raises:
+            TimeoutError: If execution exceeds the timeout.
+            RuntimeError: If execution fails for other reasons.
+        """
+        import concurrent.futures
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            future = executor.submit(
+                self._execute_without_timeout,
+                task_prompt=task_prompt,
+                task=task
+            )
+            
+            try:
+                return future.result(timeout=timeout)
+            except concurrent.futures.TimeoutError:
+                future.cancel()
+                raise TimeoutError(f"Task '{task.description}' execution timed out after {timeout} seconds. Consider increasing max_execution_time or optimizing the task.")
+            except Exception as e:
+                future.cancel()
+                raise RuntimeError(f"Task execution failed: {str(e)}")
+
+    def _execute_without_timeout(
+        self,
+        task_prompt: str,
+        task: Task
+    ) -> str:
+        """Execute a task without a timeout.
+        
+        Args:
+            task_prompt: The prompt to send to the agent.
+            task: The task being executed.
+            
+        Returns:
+            The output of the agent.
+        """
+        return self.agent_executor.invoke(
+            {
+                "input": task_prompt,
+                "tool_names": self.agent_executor.tools_names,
+                "tools": self.agent_executor.tools_description,
+                "ask_for_human_input": task.human_input,
+            }
+        )["output"]
+
     def create_agent_executor(
         self, tools: Optional[List[BaseTool]] = None, task=None
     ) -> None: