mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 17:18:29 +00:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""Test the markdown attribute in Task class."""
|
|
|
|
import pytest
|
|
from unittest.mock import patch
|
|
|
|
from crewai import Agent, Task
|
|
|
|
|
|
def test_markdown_option_in_task_prompt():
|
|
"""Test that when markdown=True, the task prompt includes markdown formatting instructions."""
|
|
|
|
researcher = Agent(
|
|
role="Researcher",
|
|
goal="Research a topic",
|
|
backstory="You're a researcher specialized in providing well-formatted content.",
|
|
allow_delegation=False,
|
|
)
|
|
|
|
task = Task(
|
|
description="Research advances in AI in 2023",
|
|
expected_output="A summary of key AI advances in 2023",
|
|
markdown=True,
|
|
agent=researcher,
|
|
)
|
|
|
|
prompt = task.prompt()
|
|
|
|
assert "Research advances in AI in 2023" in prompt
|
|
assert "A summary of key AI advances in 2023" in prompt
|
|
assert "Your final answer MUST be formatted in Markdown syntax." in prompt
|
|
|
|
|
|
def test_markdown_option_not_in_task_prompt_by_default():
|
|
"""Test that by default (markdown=False), the task prompt does not include markdown formatting instructions."""
|
|
|
|
researcher = Agent(
|
|
role="Researcher",
|
|
goal="Research a topic",
|
|
backstory="You're a researcher specialized in providing well-formatted content.",
|
|
allow_delegation=False,
|
|
)
|
|
|
|
task = Task(
|
|
description="Research advances in AI in 2023",
|
|
expected_output="A summary of key AI advances in 2023",
|
|
agent=researcher,
|
|
)
|
|
|
|
prompt = task.prompt()
|
|
|
|
assert "Research advances in AI in 2023" in prompt
|
|
assert "A summary of key AI advances in 2023" in prompt
|
|
assert "Your final answer MUST be formatted in Markdown syntax." not in prompt
|