feat: fix type checking

src/crewai/flow/flow.py

@@ -150,7 +150,7 @@ class Flow(Generic[T], metaclass=FlowMeta):
     initial_state: Union[Type[T], T, None] = None
 
     def __class_getitem__(cls, item: Type[T]) -> Type["Flow"]:
-        class _FlowGeneric(cls):
+        class _FlowGeneric(cls):  # type: ignore # Variable "cls" is not valid as a type
             _initial_state_T: Type[T] = item
 
         _FlowGeneric.__name__ = f"{cls.__name__}[{item.__name__}]"

src/crewai/project/annotations.py

@@ -1,7 +1,8 @@
 from functools import wraps
+from typing import Any, Callable
 
-from crewai.project.utils import memoize
 from crewai import Crew
+from crewai.project.utils import memoize
 
 
 def task(func):
@@ -73,8 +74,8 @@ def pipeline(func):
     return memoize(func)
 
 
-def crew(func) -> "Crew":
-    def wrapper(self, *args, **kwargs):
+def crew(func) -> Callable[..., "Crew"]:
+    def wrapper(self, *args: Any, **kwargs: Any) -> "Crew":
         instantiated_tasks = []
         instantiated_agents = []