tongflow 0.0.1__py3-none-any.whl

tongflow/__init__.py

@@ -0,0 +1,3 @@
+"""Tongflow: Openflow plugin convention + deploy scan (no per-repo JSON)."""
+
+__version__ = "0.0.1"

tongflow/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from .scan import main
+
+if __name__ == "__main__":
+    main()

tongflow/abi.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+import json
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any, FrozenSet
+
+
+@dataclass(frozen=True)
+class Abi:
+    version: int
+    node_slots: FrozenSet[str]
+    path: Path
+
+
+def load_abi(path: Path) -> Abi:
+    raw: dict[str, Any] = json.loads(path.read_text(encoding="utf-8"))
+    v = int(raw.get("version", 0))
+    if v != 1:
+        raise ValueError(f"Unsupported ABI version {v} in {path}")
+    nodes = raw.get("nodes")
+    if not isinstance(nodes, list) or not nodes:
+        raise ValueError("ABI 'nodes' must be a non-empty list")
+    slots: set[str] = set()
+    for n in nodes:
+        if not isinstance(n, dict):
+            continue
+        s = n.get("nodeSlot")
+        if isinstance(s, str) and s:
+            slots.add(s)
+    if not slots:
+        raise ValueError("No nodeSlot entries in ABI")
+    return Abi(version=1, node_slots=frozenset(slots), path=path)

tongflow/gen_models.py

@@ -0,0 +1,109 @@
+from __future__ import annotations
+
+import argparse
+import json
+import re
+from pathlib import Path
+from typing import Any
+
+
+def slot_to_ident(slot: str) -> str:
+    s = slot.upper()
+    s = re.sub(r"[^A-Z0-9]+", "_", s).strip("_")
+    if not s:
+        return "UNKNOWN"
+    if s[0].isdigit():
+        s = f"S_{s}"
+    return s
+
+
+def slot_to_module(slot: str) -> str:
+    # python module name
+    s = slot.lower()
+    s = re.sub(r"[^a-z0-9]+", "_", s).strip("_")
+    if not s:
+        return "unknown"
+    if s[0].isdigit():
+        s = f"s_{s}"
+    return s
+
+
+def read_abi_nodes(abi_path: Path) -> list[dict[str, Any]]:
+    raw: dict[str, Any] = json.loads(abi_path.read_text(encoding="utf-8"))
+    nodes = raw.get("nodes")
+    if not isinstance(nodes, list):
+        raise ValueError("ABI missing nodes[]")
+    out: list[dict[str, Any]] = []
+    for n in nodes:
+        if isinstance(n, dict) and isinstance(n.get("nodeSlot"), str):
+            out.append(n)
+    return out
+
+
+def write_models(out_dir: Path, nodes: list[dict[str, Any]]) -> None:
+    out_dir.mkdir(parents=True, exist_ok=True)
+    (out_dir / "__init__.py").write_text(
+        "\"\"\"Generated models from tongflow ABI.\"\"\"\n",
+        encoding="utf-8",
+    )
+
+    asset_py = (
+        "from __future__ import annotations\n\n"
+        "from typing import TypedDict\n\n\n"
+        "class Asset(TypedDict, total=False):\n"
+        "    bytesBase64: str\n"
+        "    filename: str\n"
+        "    mime: str\n"
+    )
+    (out_dir / "asset.py").write_text(asset_py, encoding="utf-8")
+
+    for n in nodes:
+        slot = str(n["nodeSlot"])
+        mod = slot_to_module(slot)
+        cls = slot_to_ident(slot).title().replace("_", "")
+        input_name = f"{cls}Input"
+        output_name = f"{cls}Output"
+
+        # Only transcribe is strongly modeled for now.
+        if slot in ("transcribe", "transcribe_timestamp"):
+            text = (
+                "from __future__ import annotations\n\n"
+                "from typing import TypedDict\n"
+                "from .asset import Asset\n\n\n"
+                f"class {input_name}(TypedDict, total=False):\n"
+                "    audio: Asset\n"
+                "    context: str\n"
+                "    language: str\n"
+                "    max_new_tokens: float\n\n\n"
+                f"class {output_name}(TypedDict, total=False):\n"
+                "    text: str\n"
+                "    language: str\n"
+                "    time_stamps: list[dict[str, float | str]]\n"
+            )
+        else:
+            text = (
+                "from __future__ import annotations\n\n"
+                "from typing import Any, TypedDict\n\n\n"
+                f"class {input_name}(TypedDict, total=False):\n"
+                "    \"\"\"Fallback generic input. Tighten via ABI later.\"\"\"\n"
+                "    _data: dict[str, Any]\n\n\n"
+                f"class {output_name}(TypedDict, total=False):\n"
+                "    \"\"\"Fallback generic output. Tighten via ABI later.\"\"\"\n"
+                "    _data: dict[str, Any]\n"
+            )
+        (out_dir / f"{mod}.py").write_text(text, encoding="utf-8")
+
+
+def main() -> int:
+    ap = argparse.ArgumentParser()
+    ap.add_argument("--abi", type=Path, required=True)
+    ap.add_argument("--out-dir", type=Path, required=True)
+    ns = ap.parse_args()
+    nodes = read_abi_nodes(ns.abi)
+    write_models(ns.out_dir, nodes)
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())
+

tongflow/gen_node_slots.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+import argparse
+import json
+import re
+from pathlib import Path
+from typing import Any
+
+
+def slot_to_ident(slot: str) -> str:
+    s = slot.upper()
+    s = re.sub(r"[^A-Z0-9]+", "_", s).strip("_")
+    if not s:
+        return "UNKNOWN"
+    if s[0].isdigit():
+        s = f"S_{s}"
+    return s
+
+
+def load_slots(abi_path: Path) -> list[str]:
+    raw: dict[str, Any] = json.loads(abi_path.read_text(encoding="utf-8"))
+    nodes = raw.get("nodes")
+    if not isinstance(nodes, list):
+        raise ValueError("ABI missing nodes[]")
+    out: list[str] = []
+    seen: set[str] = set()
+    for n in nodes:
+        if not isinstance(n, dict):
+            continue
+        s = n.get("nodeSlot")
+        if isinstance(s, str) and s and s not in seen:
+            seen.add(s)
+            out.append(s)
+    return out
+
+
+def generate_node_slots_py(slots: list[str]) -> str:
+    lines: list[str] = []
+    lines.append('"""Generated from config/tongflow.abi.json. DO NOT EDIT."""')
+    lines.append("")
+    lines.append("from __future__ import annotations")
+    lines.append("")
+    lines.append("import re")
+    lines.append("from typing import Final")
+    lines.append("")
+    lines.append("")
+    lines.append("def _slot_to_ident(slot: str) -> str:")
+    lines.append("    s = slot.upper()")
+    lines.append("    s = re.sub(r\"[^A-Z0-9]+\", \"_\", s).strip(\"_\")")
+    lines.append("    if not s:")
+    lines.append("        return \"UNKNOWN\"")
+    lines.append("    if s[0].isdigit():")
+    lines.append("        s = f\"S_{s}\"")
+    lines.append("    return s")
+    lines.append("")
+    lines.append("")
+    lines.append("class NodeSlots:")
+    lines.append("    \"\"\"ABI nodeSlot constants. Use these in @node_slot(...)\"\"\"")
+    for slot in slots:
+        ident = slot_to_ident(slot)
+        lines.append(f"    {ident}: Final[str] = {slot!r}")
+    lines.append("")
+    lines.append("")
+    lines.append("ALL_NODE_SLOTS: Final[tuple[str, ...]] = (")
+    for slot in slots:
+        lines.append(f"    {slot!r},")
+    lines.append(")")
+    lines.append("")
+    return "\n".join(lines) + "\n"
+
+
+def main() -> int:
+    ap = argparse.ArgumentParser()
+    ap.add_argument("--abi", type=Path, required=True)
+    ap.add_argument("--out", type=Path, required=True)
+    ns = ap.parse_args()
+
+    slots = load_slots(ns.abi)
+    text = generate_node_slots_py(slots)
+    ns.out.write_text(text, encoding="utf-8")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())
+

tongflow/models/__init__.py

@@ -0,0 +1 @@
+"""Generated models from tongflow ABI."""

tongflow/models/arrange_group.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ArrangeGroupInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ArrangeGroupOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/asset.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+from typing import TypedDict
+
+
+class Asset(TypedDict, total=False):
+    bytesBase64: str
+    filename: str
+    mime: str

tongflow/models/audio_image_gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class AudioImageGenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class AudioImageGenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/combine_text.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class CombineTextInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class CombineTextOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/concat_videos.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ConcatVideosInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ConcatVideosOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/drop_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class DropVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class DropVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/extract_audio.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ExtractAudioInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ExtractAudioOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_music.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenMusicInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenMusicOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_speech.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenSpeechInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenSpeechOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_text.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenTextInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenTextOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_text_gemini.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenTextGeminiInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenTextGeminiOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_text_openai.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenTextOpenaiInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenTextOpenaiOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/get_first_frame.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GetFirstFrameInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GetFirstFrameOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/get_last_frame.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class GetLastFrameInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class GetLastFrameOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_describe.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageDescribeInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageDescribeOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_edit.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageEditInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageEditOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_fusion.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageFusionInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageFusionOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_gen.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageGenInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageGenOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_gen_model.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageGenModelInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageGenModelOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_gen_text.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageGenTextInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageGenTextOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageGenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageGenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_image_gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageImageGenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageImageGenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/image_upscale.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ImageUpscaleInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ImageUpscaleOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/link.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class LinkInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class LinkOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/merge_video_audio.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class MergeVideoAudioInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class MergeVideoAudioOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/parse_document.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class ParseDocumentInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class ParseDocumentOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/speech_image_video_gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class SpeechImageVideoGenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class SpeechImageVideoGenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]

tongflow/models/speech_text_gen_video.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any, TypedDict
+
+
+class SpeechTextGenVideoInput(TypedDict, total=False):
+    """Fallback generic input. Tighten via ABI later."""
+    _data: dict[str, Any]
+
+
+class SpeechTextGenVideoOutput(TypedDict, total=False):
+    """Fallback generic output. Tighten via ABI later."""
+    _data: dict[str, Any]