improving guardrails
Some checks failed
Mark stale issues and pull requests / stale (push) Has been cancelled

This commit is contained in:
João Moura
2025-01-04 16:30:20 -03:00
parent d3da73136c
commit 440883e9e8
3 changed files with 1101 additions and 0 deletions

View File

@@ -41,6 +41,7 @@ from crewai.tools.base_tool import BaseTool
from crewai.utilities.config import process_config
from crewai.utilities.converter import Converter, convert_to_model
from crewai.utilities.i18n import I18N
from crewai.utilities.printer import Printer
class Task(BaseModel):
@@ -394,6 +395,11 @@ class Task(BaseModel):
guardrail_result_error=guardrail_result.error,
task_output=task_output.raw
)
printer = Printer()
printer.print(
content=f"Guardrail blocked, retrying, due to:{guardrail_result.error}\n",
color="yellow",
)
return self._execute_core(agent, context, tools)
if guardrail_result.result is None: