llama-deploy-appserver 0.3.24__tar.gz → 0.3.26__tar.gz

{llama_deploy_appserver-0.3.24 → llama_deploy_appserver-0.3.26}/PKG-INFO

@@ -1,15 +1,15 @@
 Metadata-Version: 2.3
 Name: llama-deploy-appserver
-Version: 0.3.24
+Version: 0.3.26
 Summary: Application server components for LlamaDeploy
 Author: Massimiliano Pippi, Adrian Lyjak
 Author-email: Massimiliano Pippi <mpippi@gmail.com>, Adrian Lyjak <adrianlyjak@gmail.com>
 License: MIT
-Requires-Dist: llama-index-workflows[server]>=2.9.1
+Requires-Dist: llama-index-workflows[server]>=2.11.3
 Requires-Dist: pydantic-settings>=2.10.1
 Requires-Dist: fastapi>=0.100.0
 Requires-Dist: websockets>=12.0
-Requires-Dist: llama-deploy-core>=0.3.24,<0.4.0
+Requires-Dist: llama-deploy-core>=0.3.26,<0.4.0
 Requires-Dist: httpx>=0.24.0,<1.0.0
 Requires-Dist: prometheus-fastapi-instrumentator>=7.1.0
 Requires-Dist: packaging>=25.0

{llama_deploy_appserver-0.3.24 → llama_deploy_appserver-0.3.26}/pyproject.toml

@@ -14,7 +14,7 @@ dev = [
 
 [project]
 name = "llama-deploy-appserver"
-version = "0.3.24"
+version = "0.3.26"
 description = "Application server components for LlamaDeploy"
 readme = "README.md"
 license = {text = "MIT"}
@@ -24,11 +24,11 @@ authors = [
 ]
 requires-python = ">=3.10, <4"
 dependencies = [
-  "llama-index-workflows[server]>=2.9.1",
+  "llama-index-workflows[server]>=2.11.3",
   "pydantic-settings>=2.10.1",
   "fastapi>=0.100.0",
   "websockets>=12.0",
-  "llama-deploy-core>=0.3.24,<0.4.0",
+  "llama-deploy-core>=0.3.26,<0.4.0",
   "httpx>=0.24.0,<1.0.0",
   "prometheus-fastapi-instrumentator>=7.1.0",
   "packaging>=25.0",

{llama_deploy_appserver-0.3.24 → llama_deploy_appserver-0.3.26}/src/llama_deploy/appserver/app.py

@@ -1,4 +1,5 @@
 import argparse
+import json
 import logging
 import os
 import threading
@@ -329,6 +330,44 @@ def start_preflight_in_target_venv(
     # Note: run_process doesn't return exit code; process runs to completion or raises
 
 
+def start_export_json_graph_in_target_venv(
+    cwd: Path | None = None,
+    deployment_file: Path | None = None,
+    output: Path | None = None,
+) -> None:
+    """
+    Run workflow graph export inside the target project's virtual environment using uv.
+    Mirrors the venv targeting and invocation strategy used by start_preflight_in_target_venv.
+    """
+
+    configure_settings(
+        app_root=cwd,
+        deployment_file_path=deployment_file or Path(DEFAULT_DEPLOYMENT_FILE_PATH),
+    )
+    base_dir = cwd or Path.cwd()
+    path = settings.resolved_config_parent.relative_to(base_dir)
+    args = [
+        "uv",
+        "run",
+        "--no-progress",
+        "python",
+        "-m",
+        "llama_deploy.appserver.app",
+        "--export-json-graph",
+    ]
+    if deployment_file:
+        args.extend(["--deployment-file", str(deployment_file)])
+    if output is not None:
+        args.extend(["--export-output", str(output)])
+
+    run_process(
+        args,
+        cwd=path,
+        env=os.environ.copy(),
+        line_transform=_exclude_venv_warning,
+    )
+
+
 class PreflightValidationError(Exception):
     """Raised when workflow validations fail during preflight.
 
@@ -376,6 +415,40 @@ def preflight_validate(
         raise PreflightValidationError(errors)
 
 
+def export_json_graph(
+    cwd: Path | None = None,
+    deployment_file: Path | None = None,
+    output: Path | None = None,
+) -> None:
+    """
+    Export a JSON representation of the registered workflows' graph.
+
+    This follows the same initialization path as preflight validation and writes
+    a workflows.json-style structure compatible with the CLI expectations.
+    """
+    from workflows.representation_utils import extract_workflow_structure
+
+    configure_settings(
+        app_root=cwd,
+        deployment_file_path=deployment_file or Path(DEFAULT_DEPLOYMENT_FILE_PATH),
+    )
+    cfg = get_deployment_config()
+    load_environment_variables(cfg, settings.resolved_config_parent)
+
+    workflows = load_workflows(cfg)
+
+    graph: dict[str, dict[str, Any]] = {}
+    for name, workflow in workflows.items():
+        wf_repr_dict = (
+            extract_workflow_structure(workflow).to_response_model().model_dump()
+        )
+        graph[name] = wf_repr_dict
+
+    output_path = output or (Path.cwd() / "workflows.json")
+    with output_path.open("w", encoding="utf-8") as f:
+        json.dump(graph, f, indent=2)
+
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--proxy-ui", action="store_true")
@@ -383,10 +456,18 @@ if __name__ == "__main__":
     parser.add_argument("--deployment-file", type=Path)
     parser.add_argument("--open-browser", action="store_true")
     parser.add_argument("--preflight", action="store_true")
+    parser.add_argument("--export-json-graph", action="store_true")
+    parser.add_argument("--export-output", type=Path)
 
     args = parser.parse_args()
     if args.preflight:
         preflight_validate(cwd=Path.cwd(), deployment_file=args.deployment_file)
+    elif args.export_json_graph:
+        export_json_graph(
+            cwd=Path.cwd(),
+            deployment_file=args.deployment_file,
+            output=args.export_output,
+        )
     else:
         start_server(
             proxy_ui=args.proxy_ui,

{llama_deploy_appserver-0.3.24 → llama_deploy_appserver-0.3.26}/src/llama_deploy/appserver/workflow_loader.py

@@ -12,11 +12,9 @@ from pathlib import Path
 from textwrap import dedent
 
 from dotenv import dotenv_values
-from llama_deploy.appserver.deployment_config_parser import (
-    DeploymentConfig,
-)
 from llama_deploy.appserver.process_utils import run_process, spawn_process
 from llama_deploy.appserver.settings import ApiserverSettings, settings
+from llama_deploy.core.deployment_config import DeploymentConfig
 from llama_deploy.core.ui_build import ui_build_output_path
 from packaging.version import InvalidVersion, Version
 from workflows import Workflow
@@ -191,6 +189,34 @@ def _is_missing_or_outdated(path: Path) -> Version | None:
     return None
 
 
+def run_uv(
+    source_root: Path,
+    path: Path,
+    cmd: str,
+    args: list[str] = [],
+    extra_env: dict[str, str] | None = None,
+) -> None:
+    env = os.environ.copy()
+    if extra_env:
+        env.update(extra_env)
+    run_process(
+        ["uv", cmd] + args,
+        cwd=source_root / path,
+        prefix=f"[uv {cmd}]",
+        color_code="36",
+        use_tty=False,
+        line_transform=_exclude_venv_warning,
+        env=env,
+    )
+
+
+def ensure_venv(source_root: Path, path: Path, force: bool = False) -> Path:
+    venv_path = source_root / path / ".venv"
+    if force or not venv_path.exists():
+        run_uv(source_root, path, "venv", [str(venv_path)])
+    return venv_path
+
+
 def _install_and_add_appserver_if_missing(
     path: Path,
     source_root: Path,
@@ -207,31 +233,11 @@ def _install_and_add_appserver_if_missing(
         )
         return
 
-    def run_uv(
-        cmd: str, args: list[str] = [], extra_env: dict[str, str] | None = None
-    ) -> None:
-        env = os.environ.copy()
-        if extra_env:
-            env.update(extra_env)
-        run_process(
-            ["uv", cmd] + args,
-            cwd=source_root / path,
-            prefix=f"[uv {cmd}]",
-            color_code="36",
-            use_tty=False,
-            line_transform=_exclude_venv_warning,
-            env=env,
-        )
-
-    def ensure_venv(path: Path, force: bool = False) -> Path:
-        venv_path = source_root / path / ".venv"
-        if force or not venv_path.exists():
-            run_uv("venv", [str(venv_path)])
-        return venv_path
-
     editable = are_we_editable_mode()
-    venv_path = ensure_venv(path, force=editable)
+    venv_path = ensure_venv(source_root, path, force=editable)
     run_uv(
+        source_root,
+        path,
         "sync",
         ["--no-dev", "--inexact"],
         extra_env={"UV_PROJECT_ENVIRONMENT": str(venv_path)},
@@ -239,6 +245,8 @@ def _install_and_add_appserver_if_missing(
 
     if sdists:
         run_uv(
+            source_root,
+            path,
             "pip",
             ["install"]
             + [str(s.absolute()) for s in sdists]
@@ -261,6 +269,8 @@ def _install_and_add_appserver_if_missing(
         target = f"file://{str(rel)}"
 
         run_uv(
+            source_root,
+            path,
             "pip",
             [
                 "install",
@@ -276,9 +286,11 @@ def _install_and_add_appserver_if_missing(
         version = _is_missing_or_outdated(path)
         if version is not None:
             if save_version:
-                run_uv("add", [f"llama-deploy-appserver>={version}"])
+                run_uv(source_root, path, "add", [f"llama-deploy-appserver>={version}"])
             else:
                 run_uv(
+                    source_root,
+                    path,
                     "pip",
                     [
                         "install",