feat: 完成工作流调度与上传下载 API

2026-08-08 21:04:21 +08:00
commit a8d11eafc6
32 changed files with 3675 additions and 0 deletions
@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from wov_sdk.models import NodeManifest, WorkflowDefinition, WorkflowEdge, WorkflowNode
+
+from app.db import Database
+
+
+def seed_nodes(db: Database, workspace_root: Path) -> None:
+    for node_dir in sorted(workspace_root.glob("wov-node-*")):
+        manifest_path = node_dir / "node.manifest.json"
+        if not manifest_path.is_file():
+            continue
+        manifest = NodeManifest.load(str(manifest_path))
+        manifest.repo_dir = node_dir.name
+        db.upsert_node(manifest)
+
+
+def seed_demo_workflow(db: Database) -> None:
+    if db.get_workflow("demo") is not None:
+        return
+    definition = WorkflowDefinition(
+        name="视频字幕生成",
+        version=1,
+        nodes=[
+            WorkflowNode(
+                id="extract",
+                node_type="ffmpeg-extract",
+                params={"sample_rate": 16000, "channels": 1},
+                inputs={"video_uri": "input.video_uri"},
+            ),
+            WorkflowNode(
+                id="asr",
+                node_type="faster-whisper",
+                params={"language": "ja"},
+                inputs={"audio_uri": "extract.audio_uri"},
+            ),
+            WorkflowNode(
+                id="translate",
+                node_type="llm-translate",
+                params={"target_language": "zh-CN"},
+                inputs={"srt_uri": "asr.srt_uri"},
+            ),
+            WorkflowNode(
+                id="ass",
+                node_type="srt-to-dual-eye-ass",
+                params={"resolution": "3840x1920"},
+                inputs={"cn_srt_uri": "translate.cn_srt_uri"},
+            ),
+        ],
+        edges=[
+            WorkflowEdge(from_node="extract", to_node="asr"),
+            WorkflowEdge(from_node="asr", to_node="translate"),
+            WorkflowEdge(from_node="translate", to_node="ass"),
+        ],
+        entry_inputs={"video_uri": "file"},
+        final_outputs={
+            "cn_srt": "translate.cn_srt_uri",
+            "ass": "ass.ass_uri",
+        },
+    )
+    db.upsert_workflow(
+        {
+            "id": "demo",
+            "name": definition.name,
+            "description": "上传视频，自动生成中文字幕和 VR 双眼 ASS。",
+            "published": 1,
+            "latest_version": 1,
+        }
+    )
+    db.create_workflow_version("demo", 1, definition.to_dict())