andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 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 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 pydantic import BaseModel, Field
from typing_extensions import Literal 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.tasks.task_output import TaskOutput
from crewai.types.usage_metrics import UsageMetrics 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 CrewOutput(BaseModel):
"""Class that represents the result of a crew.""" """Class that represents the result of a crew."""
@@ -25,7 +22,7 @@ class CrewOutput(BaseModel):
tasks_output: list[TaskOutput] = Field( tasks_output: list[TaskOutput] = Field(
description="Output of each task", default=[] 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: def model_json(self) -> str:
"""Get the JSON representation of the output.""" """Get the JSON representation of the output."""
@@ -39,8 +36,8 @@ class CrewOutput(BaseModel):
self, self,
*, *,
indent: Optional[int] = None, indent: Optional[int] = None,
include: Optional[IncEx] = None, include: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
exclude: Optional[IncEx] = None, exclude: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
context: Optional[Any] = None, context: Optional[Any] = None,
by_alias: bool = False, by_alias: bool = False,
exclude_unset: bool = False, exclude_unset: bool = False,

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

@@ -1,14 +1,11 @@
import json 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 pydantic import BaseModel, Field, model_validator
from typing_extensions import Literal from typing_extensions import Literal
from crewai.tasks.output_format import OutputFormat 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 TaskOutput(BaseModel):
"""Class that represents the result of a task.""" """Class that represents the result of a task."""
@@ -54,8 +51,8 @@ class TaskOutput(BaseModel):
self, self,
*, *,
indent: Optional[int] = None, indent: Optional[int] = None,
include: Optional[IncEx] = None, include: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
exclude: Optional[IncEx] = None, exclude: Optional[Union[AbstractSet[int], AbstractSet[str], Mapping[int, Any], Mapping[str, Any]]] = None,
context: Optional[Any] = None, context: Optional[Any] = None,
by_alias: bool = False, by_alias: bool = False,
exclude_unset: bool = False, exclude_unset: bool = False,