added docs and tests

docs/how-to/Conditional-Tasks.md

@@ -0,0 +1,88 @@
+---
+title: Conditional Tasks
+description: Learn how to use conditional tasks in a crewAI kickoff
+---
+
+## Introduction
+Conditional Tasks in crewAI allow for dynamic workflow adaptation based on the outcomes of previous tasks. This powerful feature enables crews to make decisions and execute tasks selectively, enhancing the flexibility and efficiency of your AI-driven processes.
+
+
+```python
+from typing import List
+
+from pydantic import BaseModel
+from crewai import Agent, Crew
+from crewai.tasks.conditional_task import ConditionalTask
+from crewai.tasks.task_output import TaskOutput
+from crewai.task import Task
+from crewai_tools import SerperDevTool
+
+
+# Define a condition function for the conditional task - if false task will be skipped, true, then execute task
+def is_data_fetched(output: TaskOutput) -> bool:
+    if len(output.pydantic.events) >= 10: # this will skip this task
+        return False
+    return True
+
+
+# Define the agents
+data_fetcher_agent = Agent(
+    role="Data Fetcher",
+    goal="Fetch data online using Serper tool",
+    backstory="Backstory 1",
+    verbose=True,
+    tools=[SerperDevTool()],
+)
+
+data_processor_agent = Agent(
+    role="Data Processor",
+    goal="Process fetched data",
+    backstory="Backstory 2",
+    verbose=True,
+)
+
+summary_generator_agent = Agent(
+    role="Summary Generator",
+    goal="Generate summary from fetched data",
+    backstory="Backstory 3",
+    verbose=True,
+)
+
+
+class EventOutput(BaseModel):
+    events: List[str]
+
+
+task1 = Task(
+    name="Data Fetching Task",
+    description="Fetch data about events in San Francisco using Serper tool",
+    expected_output="List of 10 things to do in SF this week",
+    agent=data_fetcher_agent,
+    output_pydantic=EventOutput,
+)
+
+conditional_task = ConditionalTask(
+    name="Data Processing Task",
+    description="Process data if data fetching is successful",
+    expected_output="List of 11 Things to do in SF this week ",
+    condition=is_data_fetched,
+    agent=data_processor_agent,
+)
+
+task3 = Task(
+    name="Summary Generation Task",
+    description="Generate summary of events in San Francisco from fetched data",
+    expected_output="summary_generated",
+    agent=summary_generator_agent,
+)
+
+# Create a crew with the tasks
+crew = Crew(
+    agents=[data_fetcher_agent, data_processor_agent, summary_generator_agent],
+    tasks=[task1, conditional_task, task3],
+    verbose=2,
+)
+
+result = crew.kickoff()
+print("results", result)
+```

docs/index.md

@@ -118,6 +118,11 @@ Cutting-edge framework for orchestrating role-playing, autonomous AI agents. By
                     Replay from a Task
                 </a>
             </li>
+            <li>
+                <a href="./how-to/Conditional-Tasks">
+                    Conditional Tasks
+                </a>
+            </li>
             <li>
                 <a href="./how-to/AgentOps-Observability">
                     Agent Monitoring with AgentOps

mkdocs.yml

@@ -146,6 +146,7 @@ nav:
     - Kickoff a Crew Asynchronously: 'how-to/Kickoff-async.md'
     - Kickoff a Crew for a List: 'how-to/Kickoff-for-each.md'
     - Replay from a specific task from a kickoff: 'how-to/Replay-tasks-from-latest-Crew-Kickoff.md'
+    - Conditional Tasks: 'how-to/Conditional-Tasks.md'
     - Agent Monitoring with AgentOps: 'how-to/AgentOps-Observability.md'
     - Agent Monitoring with LangTrace: 'how-to/Langtrace-Observability.md'
   - Tools Docs:

src/crewai/crew.py

@@ -660,7 +660,7 @@ class Crew(BaseModel):
                     previous_output
                 ):
                     self._logger.log(
-                        "info",
+                        "debug",
                         f"Skipping conditional task: {task.description}",
                         color="yellow",
                     )
@@ -670,6 +670,7 @@ class Crew(BaseModel):
                         agent=task.agent.role if task.agent else "",
                         output_format=OutputFormat.RAW,
                     )
+
                     if not was_replayed:
                         self._store_execution_log(
                             task,

src/crewai/tasks/conditional_task.py

@@ -18,11 +18,10 @@ class ConditionalTask(Task):
 
     def __init__(
         self,
-        *args,
         condition: Callable[[Any], bool],
         **kwargs,
     ):
-        super().__init__(*args, **kwargs)
+        super().__init__(**kwargs)
         self.condition = condition
 
     def should_execute(self, context: TaskOutput) -> bool:

tests/task_test.py

@@ -6,6 +6,7 @@ from unittest.mock import MagicMock, patch
 
 import pytest
 from crewai import Agent, Crew, Process, Task
+from crewai.tasks.conditional_task import ConditionalTask
 from crewai.tasks.task_output import TaskOutput
 from crewai.utilities.converter import Converter
 from pydantic import BaseModel
@@ -695,6 +696,19 @@ def test_task_definition_based_on_dict():
     assert task.agent is None
 
 
+def test_conditional_task_definition_based_on_dict():
+    config = {
+        "description": "Give me an integer score between 1-5 for the following title: 'The impact of AI in the future of work', check examples to based your evaluation.",
+        "expected_output": "The score of the title.",
+    }
+
+    task = ConditionalTask(config=config, condition=lambda x: True)
+
+    assert task.description == config["description"]
+    assert task.expected_output == config["expected_output"]
+    assert task.agent is None
+
+
 def test_interpolate_inputs():
     task = Task(
         description="Give me a list of 5 interesting ideas about {topic} to explore for an article, what makes them unique and interesting.",