llamactl 0.3.24__tar.gz → 0.3.25__tar.gz

{llamactl-0.3.24 → llamactl-0.3.25}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: llamactl
-Version: 0.3.24
+Version: 0.3.25
 Summary: A command-line interface for managing LlamaDeploy projects and deployments
 Author: Adrian Lyjak
 Author-email: Adrian Lyjak <adrianlyjak@gmail.com>
 License: MIT
-Requires-Dist: llama-deploy-core[client]>=0.3.24,<0.4.0
-Requires-Dist: llama-deploy-appserver>=0.3.24,<0.4.0
+Requires-Dist: llama-deploy-core[client]>=0.3.25,<0.4.0
+Requires-Dist: llama-deploy-appserver>=0.3.25,<0.4.0
 Requires-Dist: vibe-llama-core>=0.1.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: questionary>=2.0.0

{llamactl-0.3.24 → llamactl-0.3.25}/pyproject.toml

@@ -14,7 +14,7 @@ dev = [
 
 [project]
 name = "llamactl"
-version = "0.3.24"
+version = "0.3.25"
 description = "A command-line interface for managing LlamaDeploy projects and deployments"
 readme = "README.md"
 license = {text = "MIT"}
@@ -23,8 +23,8 @@ authors = [
 ]
 requires-python = ">=3.10, <4"
 dependencies = [
-  "llama-deploy-core[client]>=0.3.24,<0.4.0",
-  "llama-deploy-appserver>=0.3.24,<0.4.0",
+  "llama-deploy-core[client]>=0.3.25,<0.4.0",
+  "llama-deploy-appserver>=0.3.25,<0.4.0",
   "vibe-llama-core>=0.1.0",
   "rich>=13.0.0",
   "questionary>=2.0.0",

{llamactl-0.3.24 → llamactl-0.3.25}/src/llama_deploy/cli/commands/dev.py

@@ -79,6 +79,69 @@ def validate_command(deployment_file: Path, interactive: bool) -> None:
     rprint(f"[green]Validated workflows in {config_dir} successfully.[/green]")
 
 
+@dev.command(
+    "export-json-graph",
+    help="Produce a JSON graph representation of registered workflows",
+    hidden=True,  # perhaps expose if we have a built in visualization (mermaid, etc.)
+)
+@click.argument(
+    "deployment_file",
+    required=False,
+    default=DEFAULT_DEPLOYMENT_FILE_PATH,
+    type=_ClickPath(dir_okay=True, resolve_path=True, path_type=Path),
+)
+@click.option(
+    "--output",
+    help=(
+        "File where output JSON graph will be saved. "
+        "Defaults to workflows.json in the current directory."
+    ),
+    required=False,
+    default=None,
+    type=_ClickPath(dir_okay=True, resolve_path=True, path_type=Path),
+)
+@interactive_option
+@global_options
+def export_json_graph_command(
+    deployment_file: Path,
+    output: Path | None,
+    interactive: bool,
+) -> None:
+    """Export the configured workflows to a JSON document that may be used for graph visualization."""
+    if not deployment_file.exists():
+        rprint(f"[red]Deployment file '{deployment_file}' does not exist[/red]")
+        raise click.Abort()
+
+    _ensure_project_layout(
+        deployment_file, command_name="llamactl dev export-json-graph"
+    )
+    _maybe_inject_llama_cloud_credentials(
+        deployment_file, interactive, require_cloud=False
+    )
+
+    prepare_server(
+        deployment_file=deployment_file,
+        install=True,
+        build=False,
+        install_ui_deps=False,
+    )
+
+    wd = Path.cwd()
+    if output is None:
+        output = wd / "workflows.json"
+
+    try:
+        start_export_json_graph_in_target_venv(
+            cwd=wd,
+            deployment_file=deployment_file,
+            output=output,
+        )
+    except subprocess.CalledProcessError as exc:
+        rprint("[red]Workflow JSON graph export failed. See errors above.[/red]")
+        raise Exit(exc.returncode)
+    rprint(f"[green]Exported workflow JSON graph to {output}[/green]")
+
+
 @dev.command(
     "run",
     help=(
@@ -199,6 +262,21 @@ def start_preflight_in_target_venv(*, cwd: Path, deployment_file: Path) -> None:
     _start_preflight_in_target_venv(cwd=cwd, deployment_file=deployment_file)
 
 
+def start_export_json_graph_in_target_venv(
+    *, cwd: Path, deployment_file: Path, output: Path
+) -> None:
+    """Thin wrapper so tests can monkeypatch `dev.start_export_json_graph_in_target_venv`."""
+    from llama_deploy.appserver.app import (
+        start_export_json_graph_in_target_venv as _start_export_json_graph_in_target_venv,
+    )
+
+    _start_export_json_graph_in_target_venv(
+        cwd=cwd,
+        deployment_file=deployment_file,
+        output=output,
+    )
+
+
 def parse_environment_variables(
     config: DeploymentConfig, config_parent: Path
 ) -> dict[str, str]:
@@ -210,4 +288,4 @@ def parse_environment_variables(
     return _parse_environment_variables(config, config_parent)
 
 
-__all__ = ["dev", "validate_command", "run_command"]
+__all__ = ["dev", "validate_command", "run_command", "export_json_graph_command"]