From f4926a981098e736b3679f2b8a49bfc2b756d33e Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Tue, 20 May 2025 05:51:08 +0000 Subject: [PATCH] Fix lint error and validation error in test_markdown_task.py Co-Authored-By: Joe Moura --- tests/test_markdown_task.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/test_markdown_task.py b/tests/test_markdown_task.py index e8678f874..873d18550 100644 --- a/tests/test_markdown_task.py +++ b/tests/test_markdown_task.py @@ -1,7 +1,7 @@ """Test the markdown attribute in Task class.""" import pytest -from unittest.mock import patch +from pydantic import BaseModel from crewai import Agent, Task @@ -70,6 +70,10 @@ def test_markdown_with_empty_description(): def test_markdown_with_complex_output_format(): """Test markdown with JSON output format to ensure compatibility.""" + class ResearchOutput(BaseModel): + title: str + findings: list[str] + researcher = Agent( role="Researcher", goal="Research a topic", @@ -81,7 +85,7 @@ def test_markdown_with_complex_output_format(): description="Research topic", expected_output="Research results", markdown=True, - output_json=True, + output_json=ResearchOutput, agent=researcher, )