mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 09:08:31 +00:00
45704fe16d
* Almost working! * It fully works but not clean enought * Working but not clean engouth * Everything is workign * WIP. Working on adding and & or to flows. In the middle of setting up template for flow as well * template working * Everything is working * More changes and todos * Add more support for @start * Router working now * minor tweak to * minor tweak to conditions and event handling * Update logs * Too trigger happy with cleanup * Added in Thiago fix * Flow passing results again * Working on docs. * made more progress updates on docs * Finished talking about controlling flows * add flow output * fixed flow output section * add crews to flows section is looking good now * more flow doc changes * Update docs and add more examples * drop visualizer * save visualizer * pyvis is beginning to work * pyvis working * it is working * regular methods and triggers working. Need to work on router next. * properly identifying router and router children nodes. Need to fix color * children router working. Need to support loops * curving cycles but need to add curve conditionals * everythin is showing up properly need to fix curves * all working. needs to be cleaned up * adjust padding * drop lib * clean up prior to PR * incorporate joao feedback * final tweaks for joao
95 lines
3.4 KiB
Python
95 lines
3.4 KiB
Python
from pathlib import Path
|
|
|
|
import click
|
|
|
|
|
|
def create_flow(name):
|
|
"""Create a new flow."""
|
|
folder_name = name.replace(" ", "_").replace("-", "_").lower()
|
|
class_name = name.replace("_", " ").replace("-", " ").title().replace(" ", "")
|
|
|
|
click.secho(f"Creating flow {folder_name}...", fg="green", bold=True)
|
|
|
|
project_root = Path(folder_name)
|
|
if project_root.exists():
|
|
click.secho(f"Error: Folder {folder_name} already exists.", fg="red")
|
|
return
|
|
|
|
# Create directory structure
|
|
(project_root / "src" / folder_name).mkdir(parents=True)
|
|
(project_root / "src" / folder_name / "crews").mkdir(parents=True)
|
|
(project_root / "src" / folder_name / "tools").mkdir(parents=True)
|
|
(project_root / "tests").mkdir(exist_ok=True)
|
|
|
|
# Create .env file
|
|
with open(project_root / ".env", "w") as file:
|
|
file.write("OPENAI_API_KEY=YOUR_API_KEY")
|
|
|
|
package_dir = Path(__file__).parent
|
|
templates_dir = package_dir / "templates" / "flow"
|
|
|
|
# List of template files to copy
|
|
root_template_files = [".gitignore", "pyproject.toml", "README.md"]
|
|
src_template_files = ["__init__.py", "main.py"]
|
|
tools_template_files = ["tools/__init__.py", "tools/custom_tool.py"]
|
|
|
|
crew_folders = [
|
|
"poem_crew",
|
|
]
|
|
|
|
def process_file(src_file, dst_file):
|
|
if src_file.suffix in [".pyc", ".pyo", ".pyd"]:
|
|
return
|
|
|
|
try:
|
|
with open(src_file, "r", encoding="utf-8") as file:
|
|
content = file.read()
|
|
except Exception as e:
|
|
click.secho(f"Error processing file {src_file}: {e}", fg="red")
|
|
return
|
|
content = file.read()
|
|
|
|
content = content.replace("{{name}}", name)
|
|
content = content.replace("{{flow_name}}", class_name)
|
|
content = content.replace("{{folder_name}}", folder_name)
|
|
|
|
with open(dst_file, "w") as file:
|
|
file.write(content)
|
|
|
|
# Copy and process root template files
|
|
for file_name in root_template_files:
|
|
src_file = templates_dir / file_name
|
|
dst_file = project_root / file_name
|
|
process_file(src_file, dst_file)
|
|
|
|
# Copy and process src template files
|
|
for file_name in src_template_files:
|
|
src_file = templates_dir / file_name
|
|
dst_file = project_root / "src" / folder_name / file_name
|
|
process_file(src_file, dst_file)
|
|
|
|
# Copy tools files
|
|
for file_name in tools_template_files:
|
|
src_file = templates_dir / file_name
|
|
dst_file = project_root / "src" / folder_name / file_name
|
|
process_file(src_file, dst_file)
|
|
|
|
# Copy crew folders
|
|
for crew_folder in crew_folders:
|
|
src_crew_folder = templates_dir / "crews" / crew_folder
|
|
dst_crew_folder = project_root / "src" / folder_name / "crews" / crew_folder
|
|
if src_crew_folder.exists():
|
|
for src_file in src_crew_folder.rglob("*"):
|
|
if src_file.is_file():
|
|
relative_path = src_file.relative_to(src_crew_folder)
|
|
dst_file = dst_crew_folder / relative_path
|
|
dst_file.parent.mkdir(parents=True, exist_ok=True)
|
|
process_file(src_file, dst_file)
|
|
else:
|
|
click.secho(
|
|
f"Warning: Crew folder {crew_folder} not found in template.",
|
|
fg="yellow",
|
|
)
|
|
|
|
click.secho(f"Flow {name} created successfully!", fg="green", bold=True)
|