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

chore: refactor type annotations and protocols in wrappers.py

Browse Source
This commit is contained in:
Greyson Lalonde
2025-10-13 14:38:45 -04:00
parent fecf7e9a83
commit 6492852a0c
1 changed files with 10 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/crewai/project/wrappers.py
View File

@@ -2,24 +2,17 @@
from collections.abc import Callable from collections.abc import Callable
from functools import wraps from functools import wraps
from typing import Any, Generic, ParamSpec, Protocol, TypeVar from typing import Any, Generic, ParamSpec, Protocol, Self, TypeVar
P = ParamSpec("P") P = ParamSpec("P")
R = TypeVar("R") R = TypeVar("R")
T = TypeVar("T")
class TaskResult(Protocol): class TaskResult(Protocol):
"""Protocol for task objects that have a name property.""" """Protocol for task objects that have a name attribute."""
@property name: str | None
def name(self) -> str | None:
"""Get the task name."""
...
@name.setter
def name(self, value: str) -> None:
"""Set the task name."""
...
TaskResultT = TypeVar("TaskResultT", bound=TaskResult) TaskResultT = TypeVar("TaskResultT", bound=TaskResult)
@@ -40,8 +33,8 @@ class TaskInstance(Protocol):
class CrewInstance(Protocol): class CrewInstance(Protocol):
"""Protocol for crew class instances with required attributes.""" """Protocol for crew class instances with required attributes."""
_original_tasks: dict[str, Callable[..., Any]] _original_tasks: dict[str, Callable[[Self], TaskInstance]]
_original_agents: dict[str, Callable[..., Any]] _original_agents: dict[str, Callable[[Self], AgentInstance]]
_before_kickoff: dict[str, Callable[..., Any]] _before_kickoff: dict[str, Callable[..., Any]]
_after_kickoff: dict[str, Callable[..., Any]] _after_kickoff: dict[str, Callable[..., Any]]
agents: list[AgentInstance] agents: list[AgentInstance]
@@ -183,9 +176,6 @@ class CrewMethod(DecoratedMethod[P, R]):
is_crew: bool = True is_crew: bool = True
T = TypeVar("T")
class OutputJsonClass(Generic[T]): class OutputJsonClass(Generic[T]):
"""Wrapper for classes marked as JSON output format.""" """Wrapper for classes marked as JSON output format."""
@@ -204,7 +194,7 @@ class OutputJsonClass(Generic[T]):
self.__module__ = cls.__module__ self.__module__ = cls.__module__
self.__doc__ = cls.__doc__ self.__doc__ = cls.__doc__
def __call__(self, *args, **kwargs) -> T: def __call__(self, *args: Any, **kwargs: Any) -> T:
"""Create an instance of the wrapped class. """Create an instance of the wrapped class.
Args: Args:
@@ -216,7 +206,7 @@ class OutputJsonClass(Generic[T]):
""" """
return self._cls(*args, **kwargs) return self._cls(*args, **kwargs)
def __getattr__(self, name: str): def __getattr__(self, name: str) -> Any:
"""Delegate attribute access to the wrapped class. """Delegate attribute access to the wrapped class.
Args: Args:
@@ -246,7 +236,7 @@ class OutputPydanticClass(Generic[T]):
self.__module__ = cls.__module__ self.__module__ = cls.__module__
self.__doc__ = cls.__doc__ self.__doc__ = cls.__doc__
def __call__(self, *args, **kwargs) -> T: def __call__(self, *args: Any, **kwargs: Any) -> T:
"""Create an instance of the wrapped class. """Create an instance of the wrapped class.
Args: Args:
@@ -258,7 +248,7 @@ class OutputPydanticClass(Generic[T]):
""" """
return self._cls(*args, **kwargs) return self._cls(*args, **kwargs)
def __getattr__(self, name: str): def __getattr__(self, name: str) -> Any:
"""Delegate attribute access to the wrapped class. """Delegate attribute access to the wrapped class.
Args: Args: