小鱼开发
bd2f7b9afe
title/topic 职责分离： - topic: 创作主题分类标签 - title: 用户自定义项目名称，独立保存 数据层： - buildTitle 不再截断，优先保留已有 title - MetaOverrides 允许覆盖 title - projectStore 新增 setTitle action - VideoCreation 加载时恢复 title UI： - 草稿列表项 hover 显示编辑按钮，点击可重命名 - 直接修改 meta.json，无需进入项目 后端： - script_handler 返回中文分类名称作为 title
145 lines
5.1 KiB
Python
145 lines
5.1 KiB
Python
"""
|
|
Script 任务处理器
|
|
================
|
|
|
|
管理脚本生成的执行。
|
|
不占用 Volc 槽位，使用独立的 script 槽位池。
|
|
"""
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from app.core.platform_config import get_platform_config_loader
|
|
from app.scheduler.handlers.base import AsyncHandler
|
|
from app.scheduler.models import StateChange
|
|
from app.scheduler.registry import TaskRegistry
|
|
from app.scheduler.slot_manager import SlotManager
|
|
from app.services.script_service import ScriptService
|
|
from app.ai.prompts.loader import _load_system_meta
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _get_category_name(category: str, subcategory: str) -> str:
|
|
"""从 _meta.json 查找分类中文名称，组合为 title"""
|
|
meta = _load_system_meta()
|
|
for cat in meta.get("categories", []):
|
|
if cat.get("code") == category:
|
|
cat_name = cat.get("name", category)
|
|
for sub in cat.get("subcategories", []):
|
|
if sub.get("code") == subcategory:
|
|
sub_name = sub.get("name", subcategory)
|
|
return f"{cat_name} · {sub_name}"
|
|
return cat_name
|
|
return f"{category}/{subcategory}"
|
|
|
|
SLOT_KEY = "script:slots"
|
|
|
|
|
|
def _get_script_max_slots() -> int:
|
|
"""从 platform-config.yaml 读取 rate_limit 配置作为 max_slots"""
|
|
try:
|
|
loader = get_platform_config_loader()
|
|
platform = loader.get_platform("volcengine_ark")
|
|
if platform:
|
|
# LLM 推理是慢请求，max_slots 不应超过服务器承载能力
|
|
return min(int(platform.rate_limit_qps), 10)
|
|
except Exception:
|
|
pass
|
|
return 10
|
|
|
|
|
|
class ScriptHandler(AsyncHandler):
|
|
name = "script"
|
|
slot_key = SLOT_KEY
|
|
max_slots = _get_script_max_slots()
|
|
|
|
def __init__(self, service: ScriptService | None = None):
|
|
self.service = service
|
|
|
|
def _get_service(self) -> ScriptService:
|
|
if self.service is None:
|
|
raise RuntimeError(
|
|
"ScriptHandler 需要通过构造函数传入 ScriptService 实例"
|
|
)
|
|
return self.service
|
|
|
|
async def tick(
|
|
self, tasks: list[Any], registry: TaskRegistry, slots: SlotManager
|
|
) -> list[StateChange]:
|
|
changes: list[StateChange] = []
|
|
|
|
for task in tasks:
|
|
async with slots.acquire_ctx(SLOT_KEY, task.task_id, self.max_slots) as acquired:
|
|
if not acquired:
|
|
continue
|
|
|
|
try:
|
|
changes.extend(await self._process_task(task, registry, slots))
|
|
except Exception as e:
|
|
logger.exception(f"[Script {task.task_id}] failed")
|
|
changes.append(StateChange(task_id=task.task_id, field_path="status", value="failed"))
|
|
changes.append(
|
|
StateChange(task_id=task.task_id, field_path="error", value=str(e)[:500])
|
|
)
|
|
|
|
return changes
|
|
|
|
async def _process_task(
|
|
self, task: Any, registry: TaskRegistry, slots: SlotManager
|
|
) -> list[StateChange]:
|
|
changes: list[StateChange] = []
|
|
params = task.params or {}
|
|
category = params.get("category", "")
|
|
subcategory = params.get("subcategory", "")
|
|
|
|
await registry.update(
|
|
task.task_id,
|
|
status="running",
|
|
progress=10,
|
|
message="分析需求中...",
|
|
completed=0,
|
|
total=1,
|
|
)
|
|
|
|
try:
|
|
await registry.update(
|
|
task.task_id,
|
|
progress=10,
|
|
message="构思脚本中...",
|
|
)
|
|
|
|
service = self._get_service()
|
|
shots = await service.generate_script(
|
|
category=category,
|
|
subcategory=subcategory,
|
|
)
|
|
|
|
# 计算分镜真实总时长
|
|
total_duration = sum(s.duration for s in shots if s.duration)
|
|
result_data = {
|
|
"title": _get_category_name(category, subcategory),
|
|
"scenes": [s.model_dump() for s in shots],
|
|
"total_duration": total_duration,
|
|
"shot_count": len(shots),
|
|
}
|
|
|
|
changes.append(StateChange(task_id=task.task_id, field_path="status", value="completed"))
|
|
changes.append(StateChange(task_id=task.task_id, field_path="progress", value=100))
|
|
changes.append(
|
|
StateChange(task_id=task.task_id, field_path="message", value="脚本生成完成")
|
|
)
|
|
changes.append(StateChange(task_id=task.task_id, field_path="completed", value=1))
|
|
changes.append(StateChange(task_id=task.task_id, field_path="total", value=1))
|
|
changes.append(StateChange(task_id=task.task_id, field_path="result", value=result_data))
|
|
|
|
except Exception as exc:
|
|
logger.exception(f"[ScriptTask {task.task_id}] Failed")
|
|
changes.append(StateChange(task_id=task.task_id, field_path="status", value="failed"))
|
|
changes.append(
|
|
StateChange(task_id=task.task_id, field_path="message", value=str(exc)[:200])
|
|
)
|
|
changes.append(StateChange(task_id=task.task_id, field_path="error", value=str(exc)[:500]))
|
|
|
|
return changes
|