From 316243283315d9c39950ddf9555e30c8f3686d54 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 12 Dec 2024 04:36:22 +0000 Subject: [PATCH] style: Fix import sorting in task guardrails files Co-Authored-By: Joe Moura --- src/crewai/task.py | 2 +- src/crewai/tasks/guardrail_result.py | 1 + tests/test_task_guardrails.py | 3 ++- 3 files changed, 4 insertions(+), 2 deletions(-) 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