From edc33a1cec62e3c23a3c6ad67554604581736be4 Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Mon, 30 Sep 2024 19:08:19 -0400
Subject: [PATCH] clean up prior to PR

---
 pyproject.toml                           |  6 ------
 src/crewai/__init__.py                   |  5 +++--
 src/crewai/cli/templates/flow/.gitignore |  1 +
 src/crewai/flow/__init__.py              | 10 ++++++++++
 4 files changed, 14 insertions(+), 8 deletions(-)
 create mode 100644 src/crewai/flow/__init__.py

diff --git a/pyproject.toml b/pyproject.toml
index 36d218e23..82418531f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,13 +32,7 @@ json-repair = "^0.25.2"
 auth0-python = "^4.7.1"
 poetry = "^1.8.3"
 litellm = "^1.44.22"
-graphviz = "^0.20.3"
-matplotlib = "^3.9.2"
-networkx = "^3.3"
-ipython = "^8.27.0"
 pyvis = "^0.3.2"
-playwright = "^1.47.0"
-pillow = "^10.4.0"
 
 [tool.poetry.extras]
 tools = ["crewai-tools"]
diff --git a/src/crewai/__init__.py b/src/crewai/__init__.py
index 5a553fef3..b716a8e29 100644
--- a/src/crewai/__init__.py
+++ b/src/crewai/__init__.py
@@ -1,12 +1,13 @@
 import warnings
+
 from crewai.agent import Agent
 from crewai.crew import Crew
+from crewai.flow import Flow
 from crewai.pipeline import Pipeline
 from crewai.process import Process
 from crewai.routers import Router
 from crewai.task import Task
 
-
 warnings.filterwarnings(
     "ignore",
     message="Pydantic serializer warnings:",
@@ -14,4 +15,4 @@ warnings.filterwarnings(
     module="pydantic.main",
 )
 
-__all__ = ["Agent", "Crew", "Process", "Task", "Pipeline", "Router"]
+__all__ = ["Agent", "Crew", "Process", "Task", "Pipeline", "Router", "Flow"]
diff --git a/src/crewai/cli/templates/flow/.gitignore b/src/crewai/cli/templates/flow/.gitignore
index d50a09fc9..02dc677b9 100644
--- a/src/crewai/cli/templates/flow/.gitignore
+++ b/src/crewai/cli/templates/flow/.gitignore
@@ -1,2 +1,3 @@
 .env
 __pycache__/
+lib/
diff --git a/src/crewai/flow/__init__.py b/src/crewai/flow/__init__.py
new file mode 100644
index 000000000..f3aed2496
--- /dev/null
+++ b/src/crewai/flow/__init__.py
@@ -0,0 +1,10 @@
+# from crewai.pipeline.pipeline import Pipeline
+# from crewai.pipeline.pipeline_kickoff_result import PipelineKickoffResult
+# from crewai.pipeline.pipeline_output import PipelineOutput
+
+# __all__ = ["Pipeline", "PipelineKickoffResult", "PipelineOutput"]
+
+
+from crewai.flow import Flow
+
+__all__ = ["Flow"]