diff --git a/src/crewai/task.py b/src/crewai/task.py index 3f543888c..1880cff89 100644 --- a/src/crewai/task.py +++ b/src/crewai/task.py @@ -21,9 +21,9 @@ from pydantic import ( from pydantic_core import PydanticCustomError from crewai.agents.agent_builder.base_agent import BaseAgent +from crewai.tasks.guardrail_result import GuardrailResult from crewai.tasks.output_format import OutputFormat from crewai.tasks.task_output import TaskOutput -from crewai.tasks.guardrail_result import GuardrailResult from crewai.telemetry.telemetry import Telemetry from crewai.tools.base_tool import BaseTool from crewai.utilities.config import process_config diff --git a/src/crewai/tasks/guardrail_result.py b/src/crewai/tasks/guardrail_result.py index 24cfbca80..ba8ebc552 100644 --- a/src/crewai/tasks/guardrail_result.py +++ b/src/crewai/tasks/guardrail_result.py @@ -6,6 +6,7 @@ the way task guardrails return their validation results. """ from typing import Any, Optional, Tuple, Union + from pydantic import BaseModel, field_validator diff --git a/tests/test_task_guardrails.py b/tests/test_task_guardrails.py index 338b771b8..dc96cb878 100644 --- a/tests/test_task_guardrails.py +++ b/tests/test_task_guardrails.py @@ -1,8 +1,9 @@ """Tests for task guardrails functionality.""" -import pytest from unittest.mock import Mock +import pytest + from crewai.task import Task from crewai.tasks.task_output import TaskOutput