tongflow 0.2.2__tar.gz → 0.2.3__tar.gz

{tongflow-0.2.2/tongflow.egg-info → tongflow-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tongflow
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python SDK for TongFlow — author plugins and run multi-modal GenAI workflows
 Author: TongFlow contributors
 License-Expression: AGPL-3.0-only

{tongflow-0.2.2 → tongflow-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tongflow"
-version = "0.2.2"
+version = "0.2.3"
 description = "Python SDK for TongFlow — author plugins and run multi-modal GenAI workflows"
 readme = "README.md"
 requires-python = ">=3.10"

{tongflow-0.2.2 → tongflow-0.2.3}/tongflow/__init__.py

@@ -11,6 +11,6 @@ from .deploy_marker import deploy
 from .engine import run_workflow
 from .progress import progress
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 
 __all__ = ["deploy", "progress", "run_workflow"]

{tongflow-0.2.2 → tongflow-0.2.3}/tongflow/_data/tongflow.abi.json

@@ -1225,6 +1225,265 @@
                 "additionalProperties": false
             }
         },
+        {
+            "nodeSlot": "music-repaint",
+            "inputs": {
+                "type": "object",
+                "required": ["audio", "start_time", "end_time"],
+                "properties": {
+                    "audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "text": {
+                        "type": "string"
+                    },
+                    "lyrics": {
+                        "type": "string"
+                    },
+                    "start_time": {
+                        "type": "number"
+                    },
+                    "end_time": {
+                        "type": "number"
+                    },
+                    "strength": {
+                        "type": "number"
+                    },
+                    "seed": {
+                        "type": "integer"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "audio": {
+                        "$ref": "#/$defs/AudioRef"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        {
+            "nodeSlot": "music-cover",
+            "inputs": {
+                "type": "object",
+                "required": ["audio"],
+                "properties": {
+                    "audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "ref_audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "text": {
+                        "type": "string"
+                    },
+                    "lyrics": {
+                        "type": "string"
+                    },
+                    "strength": {
+                        "type": "number"
+                    },
+                    "seed": {
+                        "type": "integer"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "audio": {
+                        "$ref": "#/$defs/AudioRef"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        {
+            "nodeSlot": "music-extract",
+            "inputs": {
+                "type": "object",
+                "required": ["audio", "track"],
+                "properties": {
+                    "audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "track": {
+                        "type": "string"
+                    },
+                    "seed": {
+                        "type": "integer"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "audio": {
+                        "$ref": "#/$defs/AudioRef"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        {
+            "nodeSlot": "music-lego",
+            "inputs": {
+                "type": "object",
+                "required": ["audio", "track"],
+                "properties": {
+                    "audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "track": {
+                        "type": "string"
+                    },
+                    "text": {
+                        "type": "string"
+                    },
+                    "lyrics": {
+                        "type": "string"
+                    },
+                    "seed": {
+                        "type": "integer"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "audio": {
+                        "$ref": "#/$defs/AudioRef"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        {
+            "nodeSlot": "music-complete",
+            "inputs": {
+                "type": "object",
+                "required": ["audio"],
+                "properties": {
+                    "audio": {
+                        "$ref": "#/$defs/Asset"
+                    },
+                    "tracks": {
+                        "type": "array",
+                        "items": {
+                            "type": "string"
+                        }
+                    },
+                    "text": {
+                        "type": "string"
+                    },
+                    "seed": {
+                        "type": "integer"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "audio": {
+                        "$ref": "#/$defs/AudioRef"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
+        {
+            "nodeSlot": "music-brief",
+            "inputs": {
+                "type": "object",
+                "required": ["text"],
+                "properties": {
+                    "text": {
+                        "type": "string",
+                        "minLength": 1
+                    },
+                    "instrumental": {
+                        "type": "boolean"
+                    },
+                    "language": {
+                        "type": "string"
+                    }
+                },
+                "additionalProperties": false
+            },
+            "outputs": {
+                "type": "object",
+                "required": ["success"],
+                "properties": {
+                    "success": {
+                        "type": "boolean"
+                    },
+                    "error": {
+                        "type": "string"
+                    },
+                    "lyrics": {
+                        "type": "string"
+                    },
+                    "tags": {
+                        "type": "string"
+                    },
+                    "bpm": {
+                        "type": "number"
+                    },
+                    "keyscale": {
+                        "type": "string"
+                    },
+                    "duration": {
+                        "type": "number"
+                    },
+                    "language": {
+                        "type": "string"
+                    }
+                },
+                "additionalProperties": false
+            }
+        },
         {
             "nodeSlot": "audio-image-gen-video",
             "inputs": {

tongflow-0.2.3/tongflow/models/music_brief.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicBriefInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    text: str
+    instrumental: bool | None = None
+    language: str | None = None
+
+class MusicBriefOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    bpm: float | None = None
+    duration: float | None = None
+    error: str | None = None
+    keyscale: str | None = None
+    language: str | None = None
+    lyrics: str | None = None
+    tags: str | None = None
+

tongflow-0.2.3/tongflow/models/music_complete.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicCompleteInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    audio: Asset
+    seed: int | None = None
+    text: str | None = None
+    tracks: list[str] | None = None
+
+class MusicCompleteOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    audio: Asset | None = None
+    error: str | None = None
+

tongflow-0.2.3/tongflow/models/music_cover.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicCoverInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    audio: Asset
+    lyrics: str | None = None
+    ref_audio: Asset | None = None
+    seed: int | None = None
+    strength: float | None = None
+    text: str | None = None
+
+class MusicCoverOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    audio: Asset | None = None
+    error: str | None = None
+

tongflow-0.2.3/tongflow/models/music_extract.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicExtractInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    audio: Asset
+    track: str
+    seed: int | None = None
+
+class MusicExtractOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    audio: Asset | None = None
+    error: str | None = None
+

tongflow-0.2.3/tongflow/models/music_lego.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicLegoInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    audio: Asset
+    track: str
+    lyrics: str | None = None
+    seed: int | None = None
+    text: str | None = None
+
+class MusicLegoOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    audio: Asset | None = None
+    error: str | None = None
+

tongflow-0.2.3/tongflow/models/music_repaint.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+from .asset import Asset, AudioRef, FileRef, ImageRef, ModelRef, VideoRef
+
+
+class MusicRepaintInput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    audio: Asset
+    end_time: float
+    start_time: float
+    lyrics: str | None = None
+    seed: int | None = None
+    strength: float | None = None
+    text: str | None = None
+
+class MusicRepaintOutput(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    success: bool
+    audio: Asset | None = None
+    error: str | None = None
+

{tongflow-0.2.2 → tongflow-0.2.3}/tongflow/node_slots.py

@@ -47,6 +47,12 @@ class NodeSlots:
     IMAGE_DESCRIBE: Final[str] = 'image-describe'
     VIDEO_DESCRIBE: Final[str] = 'video-describe'
     AUDIO_DESCRIBE: Final[str] = 'audio-describe'
+    MUSIC_REPAINT: Final[str] = 'music-repaint'
+    MUSIC_COVER: Final[str] = 'music-cover'
+    MUSIC_EXTRACT: Final[str] = 'music-extract'
+    MUSIC_LEGO: Final[str] = 'music-lego'
+    MUSIC_COMPLETE: Final[str] = 'music-complete'
+    MUSIC_BRIEF: Final[str] = 'music-brief'
     AUDIO_IMAGE_GEN_VIDEO: Final[str] = 'audio-image-gen-video'
     SPEECH_TEXT_GEN_VIDEO: Final[str] = 'speech-text-gen-video'
     VIDEO_IMAGE_GEN_VIDEO_MIX: Final[str] = 'video-image-gen-video-mix'
@@ -99,6 +105,12 @@ ALL_NODE_SLOTS: Final[tuple[str, ...]] = (
     'image-describe',
     'video-describe',
     'audio-describe',
+    'music-repaint',
+    'music-cover',
+    'music-extract',
+    'music-lego',
+    'music-complete',
+    'music-brief',
     'audio-image-gen-video',
     'speech-text-gen-video',
     'video-image-gen-video-mix',

{tongflow-0.2.2 → tongflow-0.2.3}/tongflow/slots.py

@@ -21,6 +21,7 @@
 
 from __future__ import annotations
 
+import contextvars
 import functools
 import inspect
 import types
@@ -32,6 +33,20 @@ from pydantic import BaseModel
 
 F = TypeVar("F", bound=Callable[..., object])
 
+# Per-request model selection (router-style plugins). The platform's envelope
+# carries `model` top-level; bridges that forward it to a remote backend tuck
+# it into the prompt dict under this reserved key, and the @node_slot wrapper
+# pops it back out before constructing the typed input.
+MODEL_KEY = "_model"
+_current_model: contextvars.ContextVar[str | None] = contextvars.ContextVar(
+    "tongflow_current_model", default=None
+)
+
+
+def current_model() -> str | None:
+    """The model chosen on the node for this request, or None for default."""
+    return _current_model.get()
+
 
 def _basemodel_from_annotation(ann: object) -> type[BaseModel] | None:
     """If `ann` is `BaseModel`/`BaseModel | None`/`Optional[BaseModel]`, return the class.
@@ -150,8 +165,12 @@ def node_slot(*slots: str) -> Callable[[F], F]:
         is_method = _is_method(fn)
 
         def _marshal(input: Any) -> Any:
-            if input_cls is not None and isinstance(input, dict):
-                return _deep_construct(input_cls, input)
+            if isinstance(input, dict):
+                # Reserved routing key, never an ABI field: reset per call so a
+                # previous request's selection can't leak into this one.
+                _current_model.set(input.pop(MODEL_KEY, None))
+                if input_cls is not None:
+                    return _deep_construct(input_cls, input)
             return input
 
         def _finalize(result: Any) -> Any:

{tongflow-0.2.2 → tongflow-0.2.3/tongflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tongflow
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python SDK for TongFlow — author plugins and run multi-modal GenAI workflows
 Author: TongFlow contributors
 License-Expression: AGPL-3.0-only

{tongflow-0.2.2 → tongflow-0.2.3}/tongflow.egg-info/SOURCES.txt

@@ -66,6 +66,12 @@ tongflow/models/image_upscale.py
 tongflow/models/images_gen_video.py
 tongflow/models/link.py
 tongflow/models/merge_video_audio.py
+tongflow/models/music_brief.py
+tongflow/models/music_complete.py
+tongflow/models/music_cover.py
+tongflow/models/music_extract.py
+tongflow/models/music_lego.py
+tongflow/models/music_repaint.py
 tongflow/models/parse_document.py
 tongflow/models/remove_video_audio.py
 tongflow/models/remove_watermark.py