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 301d7a8962
commit 7d23544f19
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)

View File

@@ -3,7 +3,6 @@
import pytest
from unittest.mock import Mock
from crewai.agents.agent_builder.base_agent import BaseAgent
from crewai.task import Task
from crewai.tasks.task_output import TaskOutput