From 8467cda1fab3825d001a371ac308134ab4e08083 Mon Sep 17 00:00:00 2001
From: Eduardo Chiarotti <dudumelgaco@hotmail.com>
Date: Tue, 8 Oct 2024 17:49:17 -0300
Subject: [PATCH] feat: fix type checking

---
 src/crewai/flow/flow.py           | 2 +-
 src/crewai/project/annotations.py | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/crewai/flow/flow.py b/src/crewai/flow/flow.py
index 34e4955fc..9b0b34f23 100644
--- a/src/crewai/flow/flow.py
+++ b/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__}]"
diff --git a/src/crewai/project/annotations.py b/src/crewai/project/annotations.py
index e95909827..3d1bea567 100644
--- a/src/crewai/project/annotations.py
+++ b/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 = []