mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-13 01:58:30 +00:00
fix: Add type check for guardrail result and remove unused import
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user