supervaizer 0.10.4__tar.gz → 0.10.5__tar.gz

{supervaizer-0.10.4 → supervaizer-0.10.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: supervaizer
-Version: 0.10.4
+Version: 0.10.5
 Summary: Controller system for Supervaize
 Project-URL: Homepage, https://supervaize.com
 Project-URL: Repository, https://github.com/supervaize/supervaizer

{supervaizer-0.10.4 → supervaizer-0.10.5}/pyproject.toml

@@ -121,7 +121,7 @@ mypy_path = "src"
 disallow_any_expr = false
 
 [tool.bumpversion]
-current_version = "0.10.4"
+current_version = "0.10.5"
 commit = true
 tag = true
 tag_name = "v{new_version}"

{supervaizer-0.10.4 → supervaizer-0.10.5}/src/supervaizer/__version__.py

@@ -5,6 +5,6 @@
 # https://mozilla.org/MPL/2.0/.
 
 
-VERSION = "0.10.4"
+VERSION = "0.10.5"
 API_VERSION = "v1"
 TELEMETRY_VERSION = "v1"

{supervaizer-0.10.4 → supervaizer-0.10.5}/src/supervaizer/cli.py

@@ -95,7 +95,9 @@ def main_callback(
 
 # Add deploy subcommand
 app.add_typer(
-    deploy_app, name="deploy", help="Deploy Supervaizer agents to cloud platforms"
+    deploy_app,
+    name="deploy",
+    help="Deploy Supervaizer agents to cloud platforms (requires deploy extras: pip install supervaizer[deploy])",
 )
 
 # Create scaffold subcommand group
@@ -131,6 +133,12 @@ def start(
     environment: str = typer.Option(
         os.environ.get("SUPERVAIZER_ENVIRONMENT", "dev"), help="Environment name"
     ),
+    persist: bool = typer.Option(
+        (os.environ.get("SUPERVAIZER_PERSISTENCE") or "false").lower()
+        in ("true", "1", "yes"),
+        "--persist/--no-persist",
+        help="Persist data to file (default: off; set for self-hosted, off for Vercel/serverless)",
+    ),
     script_path: Optional[str] = typer.Argument(
         None,
         help="Path to the supervaizer_control.py script",
@@ -152,6 +160,7 @@ def start(
     os.environ["SUPERVAIZER_HOST"] = host
     os.environ["SUPERVAIZER_PORT"] = str(port)
     os.environ["SUPERVAIZER_ENVIRONMENT"] = environment
+    os.environ["SUPERVAIZER_PERSISTENCE"] = str(persist).lower()
     os.environ["SUPERVAIZER_LOG_LEVEL"] = log_level
     os.environ["SUPERVAIZER_DEBUG"] = str(debug)
     os.environ["SUPERVAIZER_RELOAD"] = str(reload)

{supervaizer-0.10.4 → supervaizer-0.10.5}/src/supervaizer/deploy/cli.py

@@ -167,6 +167,15 @@ def plan(
     verbose: bool = verbose_option,
 ) -> None:
     """Plan deployment changes without applying them."""
+    # Check if deploy extras are installed (e.g., docker, cloud SDKs)
+    try:
+        import docker
+    except ImportError:
+        console.print(
+            "[bold red]Error:[/] 'deploy' extra requirements are not installed. "
+            "Install them with: [bold]pip install supervaizer[deploy][/]"
+        )
+        raise typer.Exit(1)
     _check_platform_required(platform, "plan")
     source_dir = _check_pyproject_toml()
     plan_deployment(platform, name, env, region, project_id, verbose, source_dir)

{supervaizer-0.10.4 → supervaizer-0.10.5}/src/supervaizer/storage.py

@@ -16,19 +16,37 @@ from pathlib import Path
 from typing import TYPE_CHECKING, Any, Dict, Generic, List, Optional, TypeVar
 
 from tinydb import Query, TinyDB
+from tinydb.storages import MemoryStorage
 
 from supervaizer.common import log, singleton
 from supervaizer.lifecycle import WorkflowEntity
 
+
+class _MemoryStorage(MemoryStorage):
+    """MemoryStorage that accepts TinyDB's sort_keys/indent kwargs (ignored)."""
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args)
+
+
 if TYPE_CHECKING:
     from supervaizer.case import Case
     from supervaizer.job import Job
     from supervaizer.lifecycle import EntityEvents, EntityStatus
 
+
 T = TypeVar("T", bound=WorkflowEntity)
 
 DATA_STORAGE_PATH = os.getenv("DATA_STORAGE_PATH", "./data")
 
+# When False (default), use in-memory storage only (e.g. Vercel, serverless).
+# Set SUPERVAIZER_PERSISTENCE=true to persist to file.
+PERSISTENCE_ENABLED = os.getenv("SUPERVAIZER_PERSISTENCE", "false").lower() in (
+    "true",
+    "1",
+    "yes",
+)
+
 
 @singleton
 class StorageManager:
@@ -37,28 +55,36 @@ class StorageManager:
 
     Stores entities in separate tables by type, with foreign key relationships
     represented as ID references (Job.case_ids, Case.job_id).
+
+    When SUPERVAIZER_PERSISTENCE is false (default), uses in-memory storage only.
     """
 
-    def __init__(self, db_path: str = f"{DATA_STORAGE_PATH}/entities.json"):
+    def __init__(self, db_path: Optional[str] = None):
         """
         Initialize the storage manager.
 
         Args:
-            db_path: Path to the TinyDB JSON file
+            db_path: Path to the TinyDB JSON file, or None to use env-based
+                     persistence (file if SUPERVAIZER_PERSISTENCE=true, else memory).
         """
-        self.db_path: Path = Path(db_path)
         self._lock = threading.Lock()
-
-        # Ensure data directory exists
-        Path(db_path).parent.mkdir(parents=True, exist_ok=True)
-
-        # Initialize TinyDB without caching middleware for immediate persistence
-        # Note: Thread safety is handled by our own threading lock
-        self._db = TinyDB(
-            db_path,
-            sort_keys=True,
-            indent=2,
+        # Explicit file path (e.g. tests) uses file; else file only if persistence enabled
+        use_file = (db_path is not None and db_path != ":memory:") or (
+            db_path is None and PERSISTENCE_ENABLED
         )
+        if use_file:
+            path = (
+                db_path
+                if (db_path and db_path != ":memory:")
+                else f"{DATA_STORAGE_PATH}/entities.json"
+            )
+            self.db_path = Path(path)
+            Path(path).parent.mkdir(parents=True, exist_ok=True)
+            self._db = TinyDB(path, sort_keys=True, indent=2)
+        else:
+            # In-memory only (default for Vercel/serverless)
+            self.db_path = Path(":memory:")
+            self._db = TinyDB(storage=_MemoryStorage, sort_keys=True, indent=2)
 
         # log.debug(
         #    f"[StorageManager] 🗃️ Local DB initialized at {self.db_path.absolute()}"