simplify flow

src/crewai/cli/templates/flow/main.py

@@ -37,29 +37,14 @@ class PoemFlow(Flow[PoemState]):
             f.write(self.state.poem)
         print(f"State after save_poem: {self.state}")
 
-async def run_flow():
+async def kickoff():
-    """
-    Run the flow.
-    """
     poem_flow = PoemFlow()
     await poem_flow.kickoff()
 
-async def plot_flow():
+def plot():
-    """
-    Plot the flow.
-    """
     poem_flow = PoemFlow()
     poem_flow.plot()
 
 
-def main():
-    asyncio.run(run_flow())
-
-
-def plot():
-    asyncio.run(plot_flow())
-
-
-
 if __name__ == "__main__":
-    main()
+    asyncio.run(kickoff())

src/crewai/cli/templates/flow/pyproject.toml

@@ -10,8 +10,8 @@ dependencies = [
 ]
 
 [project.scripts]
-{{folder_name}} = "{{folder_name}}.main:main"
+{{folder_name}} = "{{folder_name}}.main:kickoff"
-run_flow = "{{folder_name}}.main:main"
+run_flow = "{{folder_name}}.main:kickoff"
 plot_flow = "{{folder_name}}.main:plot"
 
 [build-system]