vrex-flow-engine 0.3.2__tar.gz → 0.3.3__tar.gz

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vrex-flow-engine
-Version: 0.3.2
+Version: 0.3.3
 Summary: Launcher + supervisor and OpenAI-compatible image/video generation service (Vrex Flow Engine).
 Author: Vrex
 License: Proprietary

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/flow_engine/bridge/flow_client.py

@@ -12,8 +12,11 @@ Control flow:
    separate ``upload_request`` command (resumable upload, not batchexecute).
 4. That HTTP handler resolves the pending future by id.
 5. Inbound WS messages (``extension_ready``, ``session_captured``,
-   ``session_lost``, ``pong``) update per-instance state. A fresh session
-   triggers an account refresh (credits + paygate tier) through the bridge.
+   ``session_lost``, ``rpc_map``, ``pong``) update per-instance state. A fresh
+   session triggers an account refresh (credits + paygate tier) through the
+   bridge. ``rpc_map`` carries the ``method → rpcId`` table the extension
+   rebuilt from Flow's JS bundle; every ``rpc_request`` names its method so
+   the extension can substitute a rotated id.
 """
 from __future__ import annotations
 
@@ -61,6 +64,12 @@ class FlowClient:
         self._paygate_tier: Optional[int] = None
         self._credits: Optional[int] = None
         self._plan_code: Optional[int] = None
+        # RPC-id discovery pushed by the extension: Flow build label, the
+        # method → id map it read from the bundle, and the ids that differ
+        # from flow_rpc's constants (empty = everything still matches).
+        self._flow_build: Optional[str] = None
+        self._rpc_map_size: int = 0
+        self._rpc_ids_rotated: dict[str, dict[str, str]] = {}
         self._request_count = 0
         self._success_count = 0
         self._failed_count = 0
@@ -109,6 +118,18 @@ class FlowClient:
     def session_present(self) -> bool:
         return self._session_present
 
+    @property
+    def flow_build(self) -> Optional[str]:
+        return self._flow_build
+
+    @property
+    def rpc_map_size(self) -> int:
+        return self._rpc_map_size
+
+    @property
+    def rpc_ids_rotated(self) -> dict[str, dict[str, str]]:
+        return dict(self._rpc_ids_rotated)
+
     async def refresh_account(self) -> bool:
         """Resolve credits + paygate tier via the credits RPC. Returns True when
         the tier is cached. Runs through the extension, so it needs a live
@@ -153,6 +174,28 @@ class FlowClient:
             logger.warning("session_lost: %s", data.get("reason"))
             self._clear_session()
             return
+        if t == "rpc_map":
+            self._flow_build = data.get("build") if isinstance(data.get("build"), str) else None
+            discovered = data.get("map") if isinstance(data.get("map"), dict) else {}
+            self._rpc_map_size = len(discovered)
+            rotated: dict[str, dict[str, str]] = {}
+            for method, expected in flow_rpc.RPC_ID_BY_METHOD.items():
+                actual = discovered.get(method)
+                if isinstance(actual, str) and actual and actual != expected:
+                    rotated[method] = {"expected": expected, "actual": actual}
+            self._rpc_ids_rotated = rotated
+            if rotated:
+                logger.warning(
+                    "Flow build %s rotated %d rpc id(s); the extension resolves them by "
+                    "method name, update flow_rpc.py when convenient: %s",
+                    self._flow_build, len(rotated), rotated,
+                )
+            else:
+                logger.info(
+                    "rpc_map from Flow build %s: %d methods, all known ids match",
+                    self._flow_build, self._rpc_map_size,
+                )
+            return
         if t == "pong":
             return
         # Inbound response (legacy path; production flow uses HTTP callback)
@@ -248,6 +291,9 @@ class FlowClient:
         ``captcha_path`` is the single-path shorthand).
         """
         params: dict[str, Any] = {"rpcId": rpc_id, "payload": payload}
+        method = flow_rpc.RPC_METHOD_BY_ID.get(rpc_id)
+        if method:
+            params["method"] = method
         if captcha_action:
             params["captchaAction"] = captcha_action
             params["captchaPaths"] = captcha_paths or ([captcha_path] if captcha_path else [])

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/flow_engine/bridge/flow_rpc.py

@@ -34,6 +34,38 @@ RPC_GENERATE_VIDEO = RPC_GENERATE_VIDEO_T2V  # historical alias
 RPC_CHECK_MEDIA = "jwpduf"
 RPC_GET_MEDIA = "as29s"
 
+# Known ids for methods whose payload has NOT been captured yet (kept so the
+# rotation check covers them and porting them later starts from a known id).
+RPC_GENERATE_VIDEO_EXTEND = "fZytfe"
+RPC_GENERATE_VIDEO_EDIT = "jIps6"
+RPC_GENERATE_VIDEO_UPSAMPLE = "p0UkFb"
+
+# Backend method path behind each id, as declared in Flow's JS bundle
+# (``new _.kx("nzlxg", …, "/VideoFxService.GetCredits")``). The ids are Wiz
+# hashes Google may rotate; the method names are the stable anchor. The
+# extension rebuilds ``method → id`` from the live bundle on every build
+# change and resolves calls by method, so the ids here are the fallback.
+RPC_METHOD_BY_ID: dict[str, str] = {
+    RPC_LIST_PROJECTS: "FlowService.GetProjects",
+    RPC_CREATE_PROJECT: "AiSandbox.CreateProject",
+    RPC_PROJECT_INFO: "AiSandbox.GetProject",
+    RPC_PROJECT_CONTENTS: "FlowService.GetProjectContents",
+    RPC_MODEL_CATALOG: "FlowService.GetModels",
+    RPC_ACCOUNT_CREDITS: "VideoFxService.GetCredits",
+    RPC_UPLOAD_IMAGE: "FlowService.UploadImage",
+    RPC_GENERATE_IMAGE: "FlowService.BatchGenerateImages",
+    RPC_GENERATE_VIDEO_T2V: "VideoFxService.BatchAsyncGenerateVideoText",
+    RPC_GENERATE_VIDEO_I2V: "VideoFxService.BatchAsyncGenerateVideoStartImage",
+    RPC_GENERATE_VIDEO_R2V: "VideoFxService.BatchAsyncGenerateVideoReferenceImages",
+    RPC_GENERATE_VIDEO_INTERPOLATION: "VideoFxService.BatchAsyncGenerateVideoStartAndEndImage",
+    RPC_GENERATE_VIDEO_EXTEND: "VideoFxService.BatchAsyncGenerateVideoExtendVideo",
+    RPC_GENERATE_VIDEO_EDIT: "VideoFxService.BatchAsyncGenerateVideoEditVideo",
+    RPC_GENERATE_VIDEO_UPSAMPLE: "VideoFxService.BatchAsyncGenerateVideoUpsampleVideo",
+    RPC_CHECK_MEDIA: "VideoFxService.BatchCheckAsyncVideoGenerationStatus",
+    RPC_GET_MEDIA: "FlowService.GetMedia",
+}
+RPC_ID_BY_METHOD: dict[str, str] = {m: i for i, m in RPC_METHOD_BY_ID.items()}
+
 # Video upload is NOT a batchexecute call: a cookie-authenticated resumable
 # upload (``x-goog-upload-*`` headers) against this path, project id appended.
 UPLOAD_VIDEO_PATH = "/upload/v1/flow/upload/video/{project_id}"

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/flow_engine/pool.py

@@ -128,6 +128,9 @@ class PoolRegistry:
                 "email": (inst.client.user_info or {}).get("email"),
                 "paygate_tier": inst.client.paygate_tier,
                 "credits": inst.client.credits,
+                "flow_build": inst.client.flow_build,
+                "rpc_map_size": inst.client.rpc_map_size,
+                "rpc_ids_rotated": inst.client.rpc_ids_rotated,
                 "in_flight": inst.in_flight,
                 "capacity": POOL_MAX_CONCURRENCY,
             }

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vrex-flow-engine"
-version = "0.3.2"
+version = "0.3.3"
 description = "Launcher + supervisor and OpenAI-compatible image/video generation service (Vrex Flow Engine)."
 # Neutral public description (the in-repo README.md carries the full internal docs).
 readme = "PYPI_README.md"

{vrex_flow_engine-0.3.2 → vrex_flow_engine-0.3.3}/vrex_flow_engine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vrex-flow-engine
-Version: 0.3.2
+Version: 0.3.3
 Summary: Launcher + supervisor and OpenAI-compatible image/video generation service (Vrex Flow Engine).
 Author: Vrex
 License: Proprietary