Merge pull request #950 from crewAIInc/conditional-task-f

conditional task feat
This commit is contained in:
Brandon Hancock (bhancock_ai)
2024-07-17 11:08:06 -04:00
committed by GitHub
12 changed files with 7549 additions and 11 deletions

View File

@@ -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
@@ -696,6 +697,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.",