From 698cac066f388e0365e5f64eb6ba25d61f127b01 Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Mon, 21 Oct 2024 10:51:59 -0400
Subject: [PATCH] simplify flow

---
 src/crewai/cli/templates/flow/main.py        | 21 +++-----------------
 src/crewai/cli/templates/flow/pyproject.toml |  4 ++--
 2 files changed, 5 insertions(+), 20 deletions(-)

diff --git a/src/crewai/cli/templates/flow/main.py b/src/crewai/cli/templates/flow/main.py
index 38d2d8736..f6cbc2f35 100644
--- a/src/crewai/cli/templates/flow/main.py
+++ b/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():
-    """
-    Run the flow.
-    """
+async def kickoff():
     poem_flow = PoemFlow()
     await poem_flow.kickoff()
 
-async def plot_flow():
-    """
-    Plot the flow.
-    """
+def plot():
     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())
diff --git a/src/crewai/cli/templates/flow/pyproject.toml b/src/crewai/cli/templates/flow/pyproject.toml
index 490d66346..a9c0dd916 100644
--- a/src/crewai/cli/templates/flow/pyproject.toml
+++ b/src/crewai/cli/templates/flow/pyproject.toml
@@ -10,8 +10,8 @@ dependencies = [
 ]
 
 [project.scripts]
-{{folder_name}} = "{{folder_name}}.main:main"
-run_flow = "{{folder_name}}.main:main"
+{{folder_name}} = "{{folder_name}}.main:kickoff"
+run_flow = "{{folder_name}}.main:kickoff"
 plot_flow = "{{folder_name}}.main:plot"
 
 [build-system]