mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-07-02 05:38:12 +00:00
9e220b1617
Currently, declarative flows can be loaded by the runtime, but the CLI still treats them as an experimental definition file instead of a first-class Flow project shape. With this PR, `crewai create flow --declarative` scaffolds a YAML-backed Flow project, and `crewai run`, `crewai flow kickoff`, and `crewai flow plot` can run against the configured definition. This also lets crew actions reference reusable crew definition files or folders and override their inputs from the Flow definition, so declarative flows can compose existing declarative crews without inlining everything.
140 lines
4.0 KiB
Python
140 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import crewai_cli.kickoff_flow as kickoff_flow_module
|
|
import crewai_cli.plot_flow as plot_flow_module
|
|
|
|
|
|
def test_kickoff_flow_uses_configured_definition(monkeypatch):
|
|
calls = []
|
|
subprocess_calls = []
|
|
monkeypatch.delenv("UV_RUN_RECURSION_DEPTH", raising=False)
|
|
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.configured_project_declarative_flow",
|
|
lambda: "flow.yaml",
|
|
)
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.run_declarative_flow",
|
|
lambda **kwargs: calls.append({"in_process": kwargs}),
|
|
)
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.run_declarative_flow_in_project_env",
|
|
lambda **kwargs: calls.append(kwargs),
|
|
)
|
|
monkeypatch.setattr(
|
|
kickoff_flow_module.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: subprocess_calls.append((args, kwargs)),
|
|
)
|
|
|
|
kickoff_flow_module.kickoff_flow()
|
|
|
|
assert calls == [{"definition": "flow.yaml"}]
|
|
assert subprocess_calls == []
|
|
|
|
|
|
def test_kickoff_flow_keeps_python_entrypoint_without_definition(monkeypatch):
|
|
subprocess_calls = []
|
|
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.configured_project_declarative_flow",
|
|
lambda: None,
|
|
)
|
|
monkeypatch.setattr(
|
|
kickoff_flow_module.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: subprocess_calls.append((args, kwargs)),
|
|
)
|
|
|
|
kickoff_flow_module.kickoff_flow()
|
|
|
|
assert subprocess_calls == [
|
|
(
|
|
(["uv", "run", "kickoff"],),
|
|
{"capture_output": False, "text": True, "check": True},
|
|
)
|
|
]
|
|
|
|
|
|
def test_plot_flow_uses_configured_definition(monkeypatch):
|
|
calls = []
|
|
subprocess_calls = []
|
|
monkeypatch.delenv("UV_RUN_RECURSION_DEPTH", raising=False)
|
|
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.configured_project_declarative_flow",
|
|
lambda: "flow.yaml",
|
|
)
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.plot_declarative_flow",
|
|
lambda definition: calls.append({"in_process": definition}),
|
|
)
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.plot_declarative_flow_in_project_env",
|
|
lambda definition: calls.append(definition),
|
|
)
|
|
monkeypatch.setattr(
|
|
plot_flow_module.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: subprocess_calls.append((args, kwargs)),
|
|
)
|
|
|
|
plot_flow_module.plot_flow()
|
|
|
|
assert calls == ["flow.yaml"]
|
|
assert subprocess_calls == []
|
|
|
|
|
|
def test_plot_flow_delegates_project_env_detection_to_runner(monkeypatch):
|
|
calls = []
|
|
monkeypatch.setenv("UV_RUN_RECURSION_DEPTH", "1")
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.configured_project_declarative_flow",
|
|
lambda: "flow.yaml",
|
|
)
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.plot_declarative_flow_in_project_env",
|
|
lambda definition: calls.append({"project_env": definition}),
|
|
)
|
|
|
|
plot_flow_module.plot_flow()
|
|
|
|
assert calls == [{"project_env": "flow.yaml"}]
|
|
|
|
|
|
def test_plot_flow_keeps_python_entrypoint_without_definition(monkeypatch):
|
|
subprocess_calls = []
|
|
|
|
monkeypatch.setattr(
|
|
"crewai_cli.run_declarative_flow.configured_project_declarative_flow",
|
|
lambda: None,
|
|
)
|
|
monkeypatch.setattr(
|
|
plot_flow_module.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: subprocess_calls.append((args, kwargs)),
|
|
)
|
|
|
|
plot_flow_module.plot_flow()
|
|
|
|
assert subprocess_calls == [
|
|
(
|
|
(["uv", "run", "plot"],),
|
|
{"capture_output": False, "text": True, "check": True},
|
|
)
|
|
]
|
|
|
|
|
|
def test_configured_project_declarative_flow(monkeypatch, tmp_path: Path):
|
|
monkeypatch.chdir(tmp_path)
|
|
(tmp_path / "pyproject.toml").write_text(
|
|
'[tool.crewai]\ntype = "flow"\ndefinition = "flow.yaml"\n',
|
|
encoding="utf-8",
|
|
)
|
|
|
|
from crewai_cli.run_declarative_flow import configured_project_declarative_flow
|
|
|
|
assert configured_project_declarative_flow() == "flow.yaml"
|