llamactl 0.3.0a22__py3-none-any.whl → 0.3.0a24__py3-none-any.whl

llama_deploy/cli/commands/auth.py

@@ -5,10 +5,12 @@ import platform
 import subprocess
 import sys
 import webbrowser
+from pathlib import Path
 
 import click
 import httpx
 import questionary
+from dotenv import set_key
 from llama_deploy.cli.auth.client import (
     DeviceAuthorizationRequest,
     OIDCClient,
@@ -28,6 +30,7 @@ from llama_deploy.cli.styles import (
     PRIMARY_COL,
     WARNING,
 )
+from llama_deploy.cli.utils.env_inject import env_vars_from_profile
 from llama_deploy.core.client.manage_client import (
     ControlPlaneClient,
 )
@@ -314,6 +317,56 @@ def change_project(project_id: str | None, interactive: bool) -> None:
         raise click.Abort()
 
 
+@auth.command("inject")
+@global_options
+@click.option(
+    "--env-file",
+    "env_file",
+    default=Path(".env"),
+    type=click.Path(dir_okay=False, resolve_path=True, path_type=Path),
+    help="Path to the .env file to write",
+)
+@interactive_option
+def inject_env_vars(
+    env_file: Path,
+    interactive: bool,
+) -> None:
+    """Inject auth environment variables into a .env file.
+
+    Writes LLAMA_CLOUD_API_KEY, LLAMA_CLOUD_BASE_URL, and LLAMA_DEPLOY_PROJECT_ID
+    based on the current profile. Always overwrites and creates the file if missing.
+    """
+    try:
+        auth_svc = service.current_auth_service()
+        profile = auth_svc.get_current_profile()
+        if not profile:
+            if interactive:
+                profile = validate_authenticated_profile(True)
+            else:
+                raise click.ClickException(
+                    "No profile configured. Run `llamactl auth token` to create a profile."
+                )
+        if not profile.api_key:
+            raise click.ClickException(
+                "Current profile is unauthenticated (missing API key)"
+            )
+
+        vars = env_vars_from_profile(profile)
+        if not vars:
+            rprint(f"[{WARNING}]No variables to inject[/]")
+            return
+        env_file.parent.mkdir(parents=True, exist_ok=True)
+        for key, value in vars.items():
+            set_key(str(env_file), key, value)
+        rel = os.path.relpath(env_file, Path.cwd())
+        rprint(
+            f"[green]Wrote environment variables: {', '.join(vars.keys())} to {rel}[/green]"
+        )
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
 def _auto_device_name() -> str:
     try:
         if sys.platform == "darwin":  # macOS

llama_deploy/cli/commands/serve.py

@@ -1,6 +1,7 @@
 import logging
 import os
 from pathlib import Path
+from typing import Literal
 
 import click
 import questionary
@@ -8,6 +9,7 @@ from llama_deploy.appserver.app import (
     prepare_server,
     start_server_in_target_venv,
 )
+from llama_deploy.appserver.deployment_config_parser import get_deployment_config
 from llama_deploy.appserver.workflow_loader import parse_environment_variables
 from llama_deploy.cli.commands.auth import validate_authenticated_profile
 from llama_deploy.cli.config.env_service import service
@@ -61,6 +63,16 @@ logger = logging.getLogger(__name__)
     type=click.Choice(["console", "json"], case_sensitive=False),
     help="The format to use for logging",
 )
+@click.option(
+    "--persistence",
+    type=click.Choice(["memory", "local", "cloud"]),
+    help="The persistence mode to use for the workflow server",
+)
+@click.option(
+    "--local-persistence-path",
+    type=click.Path(dir_okay=True, resolve_path=True, path_type=Path),
+    help="The path to the sqlite database to use for the workflow server if using local persistence",
+)
 @interactive_option
 def serve(
     deployment_file: Path,
@@ -72,6 +84,8 @@ def serve(
     ui_port: int | None = None,
     log_level: str | None = None,
     log_format: str | None = None,
+    persistence: Literal["memory", "local", "cloud"] | None = None,
+    local_persistence_path: Path | None = None,
     interactive: bool = False,
 ) -> None:
     """Run llama_deploy API Server in the foreground. Reads the deployment configuration from the current directory. Can optionally specify a deployment file path."""
@@ -81,13 +95,16 @@ def serve(
 
     try:
         # Pre-check: if the template requires llama cloud access, ensure credentials
-        _maybe_inject_llama_cloud_credentials(deployment_file, interactive)
+        _maybe_inject_llama_cloud_credentials(
+            deployment_file, interactive, require_cloud=persistence == "cloud"
+        )
 
         prepare_server(
             deployment_file=deployment_file,
             install=not no_install,
             build=preview,
         )
+        deployment_config = get_deployment_config()
         start_server_in_target_venv(
             cwd=Path.cwd(),
             deployment_file=deployment_file,
@@ -98,6 +115,13 @@ def serve(
             ui_port=ui_port,
             log_level=log_level.upper() if log_level else None,
             log_format=log_format.lower() if log_format else None,
+            persistence=persistence if persistence else "local",
+            local_persistence_path=str(local_persistence_path)
+            if local_persistence_path and persistence == "local"
+            else None,
+            cloud_persistence_name=f"_public:serve_workflows_{deployment_config.name}"
+            if persistence == "cloud"
+            else None,
         )
 
     except KeyboardInterrupt:
@@ -144,7 +168,7 @@ def _set_project_id_from_profile(profile: Auth):
 
 
 def _maybe_inject_llama_cloud_credentials(
-    deployment_file: Path, interactive: bool
+    deployment_file: Path, interactive: bool, require_cloud: bool
 ) -> None:
     """If the deployment config indicates Llama Cloud usage, ensure LLAMA_CLOUD_API_KEY is set.
 
@@ -165,7 +189,7 @@ def _maybe_inject_llama_cloud_credentials(
         )
         raise click.Abort()
 
-    if not config.llama_cloud:
+    if not config.llama_cloud and not require_cloud:
         return
 
     vars = parse_environment_variables(

llama_deploy/cli/utils/env_inject.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+from typing import Dict
+
+from llama_deploy.cli.config.schema import Auth
+
+
+def env_vars_from_profile(profile: Auth) -> Dict[str, str]:
+    """Return env var values derived strictly from the given profile.
+
+    Produces the three keys expected by CLI commands:
+    - LLAMA_CLOUD_API_KEY
+    - LLAMA_CLOUD_BASE_URL
+    - LLAMA_DEPLOY_PROJECT_ID
+    """
+    values: Dict[str, str] = {}
+    if profile.api_key:
+        values["LLAMA_CLOUD_API_KEY"] = profile.api_key
+    if profile.api_url:
+        values["LLAMA_CLOUD_BASE_URL"] = profile.api_url
+    if profile.project_id:
+        values["LLAMA_DEPLOY_PROJECT_ID"] = profile.project_id
+    return values

{llamactl-0.3.0a22.dist-info → llamactl-0.3.0a24.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: llamactl
-Version: 0.3.0a22
+Version: 0.3.0a24
 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.0a22,<0.4.0
-Requires-Dist: llama-deploy-appserver>=0.3.0a22,<0.4.0
+Requires-Dist: llama-deploy-core[client]>=0.3.0a24,<0.4.0
+Requires-Dist: llama-deploy-appserver>=0.3.0a24,<0.4.0
 Requires-Dist: httpx>=0.24.0,<1.0.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: questionary>=2.0.0

{llamactl-0.3.0a22.dist-info → llamactl-0.3.0a24.dist-info}/RECORD

@@ -3,11 +3,11 @@ llama_deploy/cli/app.py,sha256=9170e4f506c482522bd745eb1cdb700a198cfcfd7204c168c
 llama_deploy/cli/auth/client.py,sha256=0673e2fdc71e8cf1f6d79a64e39596e608dc0257b7dddfebcc38e13e1545a2a8,11670
 llama_deploy/cli/client.py,sha256=f4053b5183224cff55c1393e78887d1af2597219135379a851b742c676adc154,1727
 llama_deploy/cli/commands/aliased_group.py,sha256=bc41007c97b7b93981217dbd4d4591df2b6c9412a2d9ed045b0ec5655ed285f2,1066
-llama_deploy/cli/commands/auth.py,sha256=d15d9c234b98f2e7a8c1be1adb111fbae64a2f5bc39b6dd610000a287bf1a132,22141
+llama_deploy/cli/commands/auth.py,sha256=1381eee494c3a0c73253322b4a54af1a857d5b89e5f1685b8afa3422eecc5607,23937
 llama_deploy/cli/commands/deployment.py,sha256=46339e09135521c46ff90235ccf765c37b1a161cec11d92e92a54ceac6528b01,9883
 llama_deploy/cli/commands/env.py,sha256=36cb1b0abb9e3d1c5546d3e8a3c4c7839c4d6c2abf75763e39efb08376b3eae9,6808
 llama_deploy/cli/commands/init.py,sha256=46f9fcdc43880edbd2f244b4c80d5cc5d387a6f8d7407c182f5995c46e41c851,10153
-llama_deploy/cli/commands/serve.py,sha256=ce08f0ec9c183d4a6f43ee10468b1bbd10edf4af3f2b3311cc1e54c3f4384528,7327
+llama_deploy/cli/commands/serve.py,sha256=309b416bfc0b527cd8c8e041beb36b64e78342545f767af7fde3bc55dbbce961,8448
 llama_deploy/cli/config/_config.py,sha256=654a4b6d06542e3503edab7023fc1c3148de510b3e3f6194e28cd4bd3e7c029a,14230
 llama_deploy/cli/config/_migrations.py,sha256=37055641970e1ea41abc583f270dc8a9dab03076224a02cd5fb08bbab2b9259f,2333
 llama_deploy/cli/config/auth_service.py,sha256=8a61110e18c752bbec5fbca23cd5d35d4ec232a4371f8c8291ba07ad83d30c6c,5208
@@ -31,7 +31,8 @@ llama_deploy/cli/textual/github_callback_server.py,sha256=3111cc45b3ff2632255a37
 llama_deploy/cli/textual/llama_loader.py,sha256=33cb32a46dd40bcf889c553e44f2672c410e26bd1d4b17aa6cca6d0a5d59c2c4,1468
 llama_deploy/cli/textual/secrets_form.py,sha256=df6699de29d2bc2cbcaddd41ad2495ce0e622cdccaadbc8369a6ee09a9e79d34,7251
 llama_deploy/cli/textual/styles.tcss,sha256=c8fa0eec00a97fa6907d223faaad82c6add1ea3f60009f1630be19282ea77e3b,3271
-llamactl-0.3.0a22.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
-llamactl-0.3.0a22.dist-info/entry_points.txt,sha256=b67e1eb64305058751a651a80f2d2268b5f7046732268421e796f64d4697f83c,52
-llamactl-0.3.0a22.dist-info/METADATA,sha256=6e9ebb747f9e3c0683229ce618fba5dc0c1c577949ef3217c1b932e811ac1f24,3261
-llamactl-0.3.0a22.dist-info/RECORD,,
+llama_deploy/cli/utils/env_inject.py,sha256=01911758bcc3cf22aad0db0d1ade56aece48d6ad6bdb7186ea213337c67f5a89,688
+llamactl-0.3.0a24.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
+llamactl-0.3.0a24.dist-info/entry_points.txt,sha256=b67e1eb64305058751a651a80f2d2268b5f7046732268421e796f64d4697f83c,52
+llamactl-0.3.0a24.dist-info/METADATA,sha256=f8c96a953953939365f3c16a56bf2d0f418fd951023ae8f4fba166207e4cde46,3261
+llamactl-0.3.0a24.dist-info/RECORD,,