feat: support to define a task guardrail using YAML config

This commit is contained in:
Lucas Gomide
2025-04-24 12:11:38 -03:00
parent e5d0cc8ac1
commit 50b603d3d2
8 changed files with 2350 additions and 4 deletions

View File

@@ -248,6 +248,9 @@ def CrewBase(cls: T) -> T:
callback_functions[callback]() for callback in callbacks
]
if guardrail := task_info.get("guardrail"):
self.tasks_config[task_name]["guardrail"] = guardrail
# Include base class (qual)name in the wrapper class (qual)name.
WrappedClass.__name__ = CrewBase.__name__ + "(" + cls.__name__ + ")"
WrappedClass.__qualname__ = CrewBase.__qualname__ + "(" + cls.__name__ + ")"