fix: Add type check for guardrail result and remove unused import

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2024-12-11 05:06:11 +00:00
parent 8bfadec4bf
commit 3e240f269c
2 changed files with 5 additions and 1 deletions

View File

@@ -280,6 +280,11 @@ class Task(BaseModel):
context = f"Previous attempt failed validation: {guardrail_result.error}\nPlease try again."
return self._execute_core(agent, context, tools)
# Ensure result is not None before assignment
if guardrail_result.result is None:
raise Exception(
f"Task guardrail returned None as result. This is not allowed."
)
result = guardrail_result.result
pydantic_output, json_output = self._export_output(result)