fix: resolve type checking errors

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-20 16:58:01 +00:00
parent 9c1f24ee26
commit 8f3936eb09

View File

@@ -232,7 +232,7 @@ class Task(BaseModel):
# Check if the normalized annotation matches any valid pattern
is_valid = (
normalized_annotation == 'tuple[bool,any]' or
any(normalized_annotation == pattern for pattern in VALID_RETURN_TYPES)
normalized_annotation in VALID_RETURN_TYPES
)
if not is_valid:
@@ -478,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 = str(guardrail_result.result)
task_output.raw = json.dumps(guardrail_result.result)
self.output = task_output
self.end_time = datetime.datetime.now()