andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: Update type hints in TaskOutput and CrewOutput to match BaseModel

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 19:33:54 +00:00
parent 47919a60a0
commit 3bf531189f
2 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/crews/crew_output.py
View File

@@ -1,5 +1,5 @@
import json
from typing import Any, Callable, Dict, Optional, Set, Union
from typing import Any, Callable, Dict, Optional, Union, AbstractSet, Mapping
from pydantic import BaseModel, Field
from typing_extensions import Literal
@@ -8,9 +8,6 @@ from crewai.tasks.output_format import OutputFormat
from crewai.tasks.task_output import TaskOutput
from crewai.types.usage_metrics import UsageMetrics
# Type definition for include/exclude parameters
IncEx = Union[Set[int], Set[str], Dict[int, Any], Dict[str, Any]]
class CrewOutput(BaseModel):
"""Class that represents the result of a crew."""
@@ -25,7 +22,7 @@ class CrewOutput(BaseModel):
tasks_output: list[TaskOutput] = Field(
description="Output of each task", default=[]
)
token_usage: UsageMetrics = Field(description="Processed token summary", default={})
token_usage: UsageMetrics = Field(description="Processed token summary", default_factory=dict)
def model_json(self) -> str:
"""Get the JSON representation of the output."""
@@ -39,8 +36,8 @@ class CrewOutput(BaseModel):
self,
*,
indent: Optional[int] = None,
include: Optional[IncEx] = None,
exclude: Optional[IncEx] = None,
include: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
exclude: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
context: Optional[Any] = None,
by_alias: bool = False,
exclude_unset: bool = False,

9
src/crewai/tasks/task_output.py
View File

@@ -1,14 +1,11 @@
import json
from typing import Any, Callable, Dict, Optional, Set, Union
from typing import Any, Callable, Dict, Optional, Union, AbstractSet, Mapping
from pydantic import BaseModel, Field, model_validator
from typing_extensions import Literal
from crewai.tasks.output_format import OutputFormat
# Type definition for include/exclude parameters
IncEx = Union[Set[int], Set[str], Dict[int, Any], Dict[str, Any]]
class TaskOutput(BaseModel):
"""Class that represents the result of a task."""
@@ -54,8 +51,8 @@ class TaskOutput(BaseModel):
self,
*,
indent: Optional[int] = None,
include: Optional[IncEx] = None,
exclude: Optional[IncEx] = None,
include: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
exclude: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
context: Optional[Any] = None,
by_alias: bool = False,
exclude_unset: bool = False,