feat: Enhance crew creation script

* quick fixes

* add generic name

---------

Co-authored-by: João Moura <joaomdmoura@gmail.com>

src/crewai/cli/create_crew.py

@@ -21,6 +21,7 @@ def create_crew(name, parent_folder=None):
         bold=True,
     )
 
+    # Create necessary directories
     if not folder_path.exists():
         folder_path.mkdir(parents=True)
         (folder_path / "tests").mkdir(exist_ok=True)
@@ -28,14 +29,47 @@ def create_crew(name, parent_folder=None):
             (folder_path / "src" / folder_name).mkdir(parents=True)
             (folder_path / "src" / folder_name / "tools").mkdir(parents=True)
             (folder_path / "src" / folder_name / "config").mkdir(parents=True)
-            with open(folder_path / ".env", "w") as file:
-                file.write("OPENAI_API_KEY=YOUR_API_KEY")
     else:
         click.secho(
-            f"\tFolder {folder_name} already exists. Please choose a different name.",
-            fg="red",
+            f"\tFolder {folder_name} already exists. Updating .env file...",
+            fg="yellow",
         )
-        return
+
+    # Path to the .env file
+    env_file_path = folder_path / ".env"
+
+    # Load existing environment variables if .env exists
+    env_vars = {}
+    if env_file_path.exists():
+        with open(env_file_path, "r") as file:
+            for line in file:
+                key_value = line.strip().split('=', 1)
+                if len(key_value) == 2:
+                    env_vars[key_value[0]] = key_value[1]
+
+    # Prompt for keys/variables/LLM settings only if not already set
+    if 'OPENAI_API_KEY' not in env_vars:
+        if click.confirm("Do you want to enter your OPENAI_API_KEY?", default=True):
+            env_vars['OPENAI_API_KEY'] = click.prompt("Enter your OPENAI_API_KEY", type=str)
+
+    if 'ANTHROPIC_API_KEY' not in env_vars:
+        if click.confirm("Do you want to enter your ANTHROPIC_API_KEY?", default=False):
+            env_vars['ANTHROPIC_API_KEY'] = click.prompt("Enter your ANTHROPIC_API_KEY", type=str)
+
+    if 'GEMINI_API_KEY' not in env_vars:
+        if click.confirm("Do you want to specify your GEMINI_API_KEY?", default=True):
+            env_vars['GEMINI_API_KEY'] = click.prompt("Enter your GEMINI_API_KEY", type=str)
+
+    # Loop to add other environment variables
+    while click.confirm("Do you want to specify another environment variable?", default=False):
+        var_name = click.prompt("Enter the variable name", type=str)
+        var_value = click.prompt(f"Enter the value for {var_name}", type=str)
+        env_vars[var_name] = var_value
+
+    # Write the environment variables to .env file
+    with open(env_file_path, "w") as file:
+        for key, value in env_vars.items():
+            file.write(f"{key}={value}\n")
 
     package_dir = Path(__file__).parent
     templates_dir = package_dir / "templates" / "crew"

src/crewai/flow/flow.py

@@ -149,15 +149,16 @@ class Flow(Generic[T], metaclass=FlowMeta):
     _router_paths: Dict[str, List[str]] = {}
     initial_state: Union[Type[T], T, None] = None
 
-    def __class_getitem__(cls, item):
+    def __class_getitem__(cls, item: Type[T]) -> Type["Flow"]:
         class _FlowGeneric(cls):
-            _initial_state_T = item
+            _initial_state_T: Type[T] = item
 
+        _FlowGeneric.__name__ = f"{cls.__name__}[{item.__name__}]"
         return _FlowGeneric
 
-    def __init__(self):
+    def __init__(self) -> None:
         self._methods: Dict[str, Callable] = {}
-        self._state = self._create_initial_state()
+        self._state: T = self._create_initial_state()
         self._completed_methods: Set[str] = set()
         self._pending_and_listeners: Dict[str, Set[str]] = {}
         self._method_outputs: List[Any] = []  # List to store all method outputs
@@ -212,11 +213,11 @@ class Flow(Generic[T], metaclass=FlowMeta):
         else:
             return None  # Or raise an exception if no methods were executed
 
-    async def _execute_start_method(self, start_method: str):
+    async def _execute_start_method(self, start_method: str) -> None:
         result = await self._execute_method(self._methods[start_method])
         await self._execute_listeners(start_method, result)
 
-    async def _execute_method(self, method: Callable, *args, **kwargs):
+    async def _execute_method(self, method: Callable, *args: Any, **kwargs: Any) -> Any:
         result = (
             await method(*args, **kwargs)
             if asyncio.iscoroutinefunction(method)
@@ -225,7 +226,7 @@ class Flow(Generic[T], metaclass=FlowMeta):
         self._method_outputs.append(result)  # Store the output
         return result
 
-    async def _execute_listeners(self, trigger_method: str, result: Any):
+    async def _execute_listeners(self, trigger_method: str, result: Any) -> None:
         listener_tasks = []
 
         if trigger_method in self._routers:
@@ -253,7 +254,7 @@ class Flow(Generic[T], metaclass=FlowMeta):
         # Run all listener tasks concurrently and wait for them to complete
         await asyncio.gather(*listener_tasks)
 
-    async def _execute_single_listener(self, listener: str, result: Any):
+    async def _execute_single_listener(self, listener: str, result: Any) -> None:
         try:
             method = self._methods[listener]
             sig = inspect.signature(method)
@@ -277,7 +278,7 @@ class Flow(Generic[T], metaclass=FlowMeta):
 
             traceback.print_exc()
 
-    def plot(self, filename: str = "crewai_flow"):
+    def plot(self, filename: str = "crewai_flow") -> None:
         self._telemetry.flow_plotting_span(
             self.__class__.__name__, list(self._methods.keys())
         )