vrex-flow-engine 0.2.0__py3-none-any.whl

flow_engine/video_context.py

@@ -0,0 +1,77 @@
+"""Per-video generation context, captured for extend/edit.
+
+Extend and edit need more than a media id. Extend wants the source clip's
+*scene id* and its *operation/generation id* (``video.operation.name`` — which
+differs from the media name); edit wants the source's *workflow id* and *frame
+count*. None of that is recoverable from a bare media id, and there's no
+media→scene reverse lookup, so we capture it as each video job completes and key
+it by the media id flow_engine hands back to the client.
+
+In-memory, single process — same model as the media registry. A restart forgets
+it; callers can still drive extend/edit by passing the context explicitly
+(``scene_id`` / ``workflow_id`` on the request). Context is only available for
+videos polled via the workflow schema (low-priority / extend / edit) — that poll
+hits ``/v1/media/<id>`` which carries the workflow/operation/scene fields;
+operation-mode (Lite/Fast/Quality) polls go through ``batchCheckAsync`` and
+don't expose them.
+"""
+from __future__ import annotations
+
+import threading
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class VideoContext:
+    media_id: str
+    workflow_id: Optional[str] = None
+    operation_id: Optional[str] = None  # video.operation.name — extend's source ref
+    scene_id: Optional[str] = None
+    duration_sec: Optional[float] = None
+
+
+_registry: dict[str, VideoContext] = {}
+_lock = threading.Lock()
+
+
+def remember(
+    media_id: Optional[str],
+    *,
+    workflow_id: Optional[str] = None,
+    operation_id: Optional[str] = None,
+    scene_id: Optional[str] = None,
+    duration_sec: Optional[float] = None,
+) -> None:
+    """Record (or merge) what we learned about a generated video. Only non-None
+    fields overwrite, so a later, richer poll can fill gaps left by an earlier
+    one."""
+    if not isinstance(media_id, str) or not media_id:
+        return
+    with _lock:
+        ctx = _registry.get(media_id) or VideoContext(media_id=media_id)
+        if workflow_id:
+            ctx.workflow_id = workflow_id
+        if operation_id:
+            ctx.operation_id = operation_id
+        if scene_id:
+            ctx.scene_id = scene_id
+        if duration_sec:
+            ctx.duration_sec = duration_sec
+        _registry[media_id] = ctx
+
+
+def get(media_id: str) -> Optional[VideoContext]:
+    with _lock:
+        return _registry.get(media_id)
+
+
+def parse_seconds(length: Optional[str]) -> Optional[float]:
+    """Parse a Flow duration string like ``"8s"`` into seconds."""
+    if not isinstance(length, str):
+        return None
+    s = length.strip().rstrip("s")
+    try:
+        return float(s)
+    except ValueError:
+        return None

flow_engine/video_context_store.py

@@ -0,0 +1,49 @@
+"""Per-instance video context store.
+
+Extracted from the module-level ``_registry`` dict in video_context.py so
+each pool instance owns an isolated namespace — video context from account A
+is never accessible to account B.
+"""
+from __future__ import annotations
+
+import threading
+from typing import Optional
+
+from flow_engine.video_context import VideoContext
+
+
+class VideoContextStore:
+    """Thread-safe dict-backed video context registry, one per PoolInstance."""
+
+    def __init__(self) -> None:
+        self._registry: dict[str, VideoContext] = {}
+        self._lock = threading.Lock()
+
+    def remember(
+        self,
+        media_id: Optional[str],
+        *,
+        workflow_id: Optional[str] = None,
+        operation_id: Optional[str] = None,
+        scene_id: Optional[str] = None,
+        duration_sec: Optional[float] = None,
+    ) -> None:
+        """Record (or merge) what we learned about a generated video. Only
+        non-None fields overwrite, so a later, richer poll can fill gaps."""
+        if not isinstance(media_id, str) or not media_id:
+            return
+        with self._lock:
+            ctx = self._registry.get(media_id) or VideoContext(media_id=media_id)
+            if workflow_id:
+                ctx.workflow_id = workflow_id
+            if operation_id:
+                ctx.operation_id = operation_id
+            if scene_id:
+                ctx.scene_id = scene_id
+            if duration_sec:
+                ctx.duration_sec = duration_sec
+            self._registry[media_id] = ctx
+
+    def get(self, media_id: str) -> Optional[VideoContext]:
+        with self._lock:
+            return self._registry.get(media_id)

vrex_flow_engine-0.2.0.dist-info/METADATA

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: vrex-flow-engine
+Version: 0.2.0
+Summary: Launcher + supervisor and OpenAI-compatible image/video generation service (Vrex Flow Engine).
+Author: Vrex
+License: Proprietary
+Project-URL: Homepage, https://getvrex.com
+Keywords: vrex,flow-engine,cli,supervisor,image-generation,video-generation
+Classifier: Programming Language :: Python :: 3
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: POSIX
+Classifier: Topic :: Multimedia :: Video
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.110
+Requires-Dist: uvicorn[standard]>=0.27
+Requires-Dist: httpx>=0.27
+Requires-Dist: websockets>=12.0
+Requires-Dist: pydantic>=2.5
+Requires-Dist: python-multipart>=0.0.9
+Requires-Dist: posthog>=3.8
+Requires-Dist: rich>=13.7
+
+# vrex-flow-engine
+
+One-command launcher + supervisor for the **Vrex Flow Engine** — a self-hosted
+service that exposes OpenAI-compatible image/video generation endpoints.
+
+`vrex-flow-engine start` stores your engine key + Cloudflare tunnel token,
+launches the engine and the tunnel as supervised child processes (with
+crash-restart), and renders a live status dashboard.
+
+## Quickstart
+
+```bash
+uvx vrex-flow-engine start
+```
+
+First run prompts for the engine API key and the Cloudflare tunnel token (saved
+to `~/.vrex-flow/config.json`, chmod 600), then brings up the engine + tunnel.
+Ctrl-C stops both cleanly.
+
+## Commands
+
+| Command | What it does |
+|---|---|
+| `vrex-flow-engine start` | Run engine + tunnel with a live dashboard. `--no-tunnel` runs the engine only. |
+| `vrex-flow-engine setup` | Store the engine key + tunnel token. |
+| `vrex-flow-engine doctor` | Check prerequisites and probe a running engine. |
+| `vrex-flow-engine serve` | Run just the engine in the foreground. |
+
+The short alias `flow-engine` is installed for the same commands.
+
+## Requirements
+
+- Python 3.10+
+- [`cloudflared`](https://developers.cloudflare.com/cloudflare-one/connections/connect-networks/downloads/)
+  (for the tunnel; `--no-tunnel` skips it)
+- Access to a Vrex Flow Engine deployment.
+
+Internal Vrex tooling.

vrex_flow_engine-0.2.0.dist-info/RECORD

@@ -0,0 +1,36 @@
+flow_engine/__init__.py,sha256=bJWwmRl8Qu9j0osEFNf3NBfLC0LRP8EdT-nYE6MwTQA,169
+flow_engine/catalog.py,sha256=_Q4FCgeLtxIcf5Udcsi8XWD_LfkKoWaONNotNmXn_ds,12978
+flow_engine/config.py,sha256=QbvUksexrzFt-JNdzfc2poXK_JPk6BzasKu831l3hYE,2092
+flow_engine/ingest.py,sha256=Xf3drEF9gyTHWBUSa1pJOlu8Wx_gyYe8sguGtRkU3DY,6421
+flow_engine/job_store.py,sha256=0Au6I_q7einhhFCO5Oc4Jjc2dtAAdSfWKWSX4A0WWhA,722
+flow_engine/jobs.py,sha256=uy4U5On7tIO8ml1MZlHjdx1Kd3-MvpNGfp_eOevfkX0,8571
+flow_engine/main.py,sha256=MjOcWFflDePiChUVLCBodrDLARZRFxIq0jxSCHeG7VA,6457
+flow_engine/media.py,sha256=Rm_JH2GEEVJCPgZRbq3vtcjTwiDXq9YADsKfW8p-YQY,7650
+flow_engine/media_store.py,sha256=tgDYyCVpWvxJBkJ1kX-1atT_SahMAsCyDWnI2Irwcro,8028
+flow_engine/pool.py,sha256=wwmv9h7S3mOkJdLtCVaGFklRTyoKgLDh99v7FDwmLQs,5680
+flow_engine/posthog_client.py,sha256=8FPH74rxOgm6vYKyO3RwLBeNW6UxfYZnxkk1DcxQfb4,3672
+flow_engine/session.py,sha256=bv59PsVPPFZl9ygRflVAQCCMyQPr_38pBj6LK2K2qhk,3456
+flow_engine/video_context.py,sha256=u7HNEdsn95EtCitt23stNtie4FFGs8bCUzSW-EM9_Pw,2682
+flow_engine/video_context_store.py,sha256=-5_Sr-lcTSFwjLcNxObwk2PECQE7PbeXuoVqFJH8Pjw,1664
+flow_engine/bridge/__init__.py,sha256=VjD9zKm_UAX0rmHHtNi4lDlZCIb4Lg6o6iEo5rlYtXQ,300
+flow_engine/bridge/flow_client.py,sha256=2mpJY6DXCPz7NGvc70ddl19eq-PRjEWTT5FUQdPEAqc,18889
+flow_engine/bridge/flow_sdk.py,sha256=ynHjY48ms75n2QTnzBn5j2H7ewtOrOYPS9v36GhF7l8,76278
+flow_engine/bridge/ws_server.py,sha256=JFq4Iv5ReVau78z2bxEZWN4XC3GGWkePg3Gq63nDh28,2879
+flow_engine/cli/__init__.py,sha256=xETN-IChMvfTFjcW9vYyZ56UOsS6eIAMh3rNYxywU6w,76
+flow_engine/cli/app.py,sha256=elV35tWQE7kkbHqA4BdLgeeVXLdpkqhgihMwag3yxmI,5694
+flow_engine/cli/config.py,sha256=engG_4HC0E2gDd-pzIiifZAOsYoBbJOfVwL1jfNmdmA,2833
+flow_engine/cli/dashboard.py,sha256=EwIRNaTlIdjy2hwwfoCM4wGybD0hiNnWjgKr7olMRoc,4851
+flow_engine/cli/health.py,sha256=75ObJ-Y7uwYKt1FwZcWrTLSp4Y_Q6EUO5T7oqdm6cQI,1596
+flow_engine/cli/processes.py,sha256=Nem-Z2yhwuYct1nRVvo5yXD2wFTYwil33iXenkFgk8k,3924
+flow_engine/cli/supervisor.py,sha256=WrSbEwac2I1ZxNSkyIOtccNvSk9UcZ5uhluJp7QgjV8,7163
+flow_engine/openai/__init__.py,sha256=KWlJNwXmYcxUPlOoSzwN-FcaWci3_7rKsuA7wMP1ePc,73
+flow_engine/openai/_util.py,sha256=IbB4v8JgaOzvTRU45EJgRqsdD24vfQnVAvEuXLouJ1k,2175
+flow_engine/openai/images.py,sha256=hP1vP_7ysXU-YfavsWzO_PWOJKFeazzbq4XrBOUWvvI,4874
+flow_engine/openai/models.py,sha256=7S0pZ-IAZ4gwrX7ui3DUUilHseEjC2umEOzAdpUGzEY,1240
+flow_engine/openai/uploads.py,sha256=ZFmX0J5BItys4JVbpVk1mc6kgdJPnzZ0x-9BOVzlNHg,2720
+flow_engine/openai/videos.py,sha256=45DkN5gUmCtYqASpbF_MG-gQ5C_dNB7Xd_pjxkQMitg,10505
+vrex_flow_engine-0.2.0.dist-info/METADATA,sha256=_r23P_tWo85zGcvuD_q4UCDXj9tdLwsA5WO9u7uzurw,2151
+vrex_flow_engine-0.2.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+vrex_flow_engine-0.2.0.dist-info/entry_points.txt,sha256=H022OlR20lq-INQQo1Dp15XJgCJ6H2KfplwEJXayjV0,101
+vrex_flow_engine-0.2.0.dist-info/top_level.txt,sha256=h2-ndJR-ki0pQZt_P2X9s58wpkRdAA-IKPqn4x3R6eA,12
+vrex_flow_engine-0.2.0.dist-info/RECORD,,

vrex_flow_engine-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

vrex_flow_engine-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+flow-engine = flow_engine.cli.app:main
+vrex-flow-engine = flow_engine.cli.app:main

vrex_flow_engine-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+flow_engine