updated fixes for conditional tasks

src/crewai/__init__.py

@@ -2,6 +2,5 @@ from crewai.agent import Agent
 from crewai.crew import Crew
 from crewai.process import Process
 from crewai.task import Task
-from crewai.conditional_task import ConditionalTask
 
-__all__ = ["Agent", "Crew", "Process", "Task", "ConditionalTask"]
+__all__ = ["Agent", "Crew", "Process", "Task"]

src/crewai/crew.py

@@ -28,7 +28,7 @@ from crewai.memory.long_term.long_term_memory import LongTermMemory
 from crewai.memory.short_term.short_term_memory import ShortTermMemory
 from crewai.process import Process
 from crewai.task import Task
-from crewai.conditional_task import ConditionalTask
+from crewai.tasks.conditional_task import ConditionalTask
 from crewai.tasks.output_format import OutputFormat
 from crewai.tasks.task_output import TaskOutput
 from crewai.telemetry import Telemetry

src/crewai/tasks/conditional_task.py

@@ -1,4 +1,6 @@
-from typing import Callable, Optional, Any
+from typing import Callable, Any
+
+from pydantic import Field
 from crewai.task import Task
 from crewai.tasks.task_output import TaskOutput
 
@@ -9,18 +11,21 @@ class ConditionalTask(Task):
     Note: This cannot be the only task you have in your crew and cannot be the first since its needs context from the previous task.
     """
 
-    condition: Optional[Callable[[TaskOutput], bool]] = None
+    condition: Callable[[TaskOutput], bool] = Field(
+        default=None,
+        description="Maximum number of retries for an agent to execute a task when an error occurs.",
+    )
 
     def __init__(
         self,
         *args,
-        condition: Optional[Callable[[TaskOutput], bool]] = None,
+        condition: Callable[[Any], bool],
         **kwargs,
     ):
         super().__init__(*args, **kwargs)
         self.condition = condition
 
-    def should_execute(self, context: Any) -> bool:
+    def should_execute(self, context: TaskOutput) -> bool:
         """
         Determines whether the conditional task should be executed based on the provided context.