From 4bafdacd886c9569be937f25a419b138569e995f Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 20 Feb 2025 17:35:47 +0000 Subject: [PATCH] fix: allow both string and dict types for TaskOutput.raw Co-Authored-By: Joe Moura --- src/crewai/tasks/task_output.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/crewai/tasks/task_output.py b/src/crewai/tasks/task_output.py index b0e8aecd4..e772cba97 100644 --- a/src/crewai/tasks/task_output.py +++ b/src/crewai/tasks/task_output.py @@ -1,5 +1,5 @@ import json -from typing import Any, Dict, Optional +from typing import Any, Dict, Optional, Union from pydantic import BaseModel, Field, model_validator @@ -15,7 +15,7 @@ class TaskOutput(BaseModel): description="Expected output of the task", default=None ) summary: Optional[str] = Field(description="Summary of the task", default=None) - raw: str = Field(description="Raw output of the task", default="") + raw: Union[str, Dict[str, Any]] = Field(description="Raw output of the task", default="") pydantic: Optional[BaseModel] = Field( description="Pydantic output of task", default=None )