andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 07:13:00 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

feat: renaming GuardrailTask to TaskGuardrail

Browse Source
This commit is contained in:
Lucas Gomide
2025-04-23 14:37:46 -03:00
parent 0ed683241d
commit 05e99bdfe5
7 changed files with 42 additions and 244 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/crewai/task.py
View File

@@ -480,22 +480,22 @@ class Task(BaseModel):
raise ValueError("Guardrail is not set")
from crewai.utilities.events import (
GuardrailTaskCompletedEvent,
GuardrailTaskStartedEvent,
TaskGuardrailCompletedEvent,
TaskGuardrailStartedEvent,
)
from crewai.utilities.events.crewai_event_bus import crewai_event_bus
crewai_event_bus.emit(
self,
GuardrailTaskStartedEvent(
TaskGuardrailStartedEvent(
guardrail=self.guardrail, retry_count=self.retry_count
),
)
if isinstance(self.guardrail, str):
from crewai.tasks.guardrail_task import GuardrailTask
from crewai.tasks.task_guardrail import TaskGuardrail
result = GuardrailTask(description=self.guardrail, task=self)(task_output)
result = TaskGuardrail(description=self.guardrail, task=self)(task_output)
else:
result = self.guardrail(task_output)
@@ -503,7 +503,7 @@ class Task(BaseModel):
crewai_event_bus.emit(
self,
GuardrailTaskCompletedEvent(
TaskGuardrailCompletedEvent(
success=guardrail_result.success,
result=guardrail_result.result,
error=guardrail_result.error,