andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-27 17:18:13 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

5
src/crewai/task.py
View File

@@ -280,6 +280,11 @@ class Task(BaseModel):
context = f"Previous attempt failed validation: {guardrail_result.error}\nPlease try again." context = f"Previous attempt failed validation: {guardrail_result.error}\nPlease try again."
return self._execute_core(agent, context, tools) 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 result = guardrail_result.result
pydantic_output, json_output = self._export_output(result) pydantic_output, json_output = self._export_output(result)

1
tests/test_task_guardrails.py
View File

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