andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: improve type validation logic

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-20 16:57:10 +00:00
parent 24da7be540
commit 9c1f24ee26
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/task.py
View File

@@ -230,11 +230,10 @@ class Task(BaseModel):
}
# Check if the normalized annotation matches any valid pattern
is_valid = False
for pattern in VALID_RETURN_TYPES:
if pattern == normalized_annotation or pattern == 'tuple[bool,any]':
is_valid = True
break
is_valid = (
normalized_annotation == 'tuple[bool,any]' or
any(normalized_annotation == pattern for pattern in VALID_RETURN_TYPES)
)
if not is_valid:
raise GuardrailValidationError(
@@ -479,7 +478,7 @@ class Task(BaseModel):
elif isinstance(guardrail_result.result, TaskOutput):
task_output = guardrail_result.result
elif isinstance(guardrail_result.result, dict):
task_output.raw = guardrail_result.result
task_output.raw = str(guardrail_result.result)
self.output = task_output
self.end_time = datetime.datetime.now()