aiagent/backend/app/models/orchestration_template.py

"""
编排模板模型 — 保存可视化 Agent 编排画布配置
"""
from sqlalchemy import Column, String, Text, JSON, DateTime, ForeignKey, Index, func
from sqlalchemy.dialects.mysql import CHAR
from sqlalchemy.orm import relationship
from app.core.database import Base
import uuid


class OrchestrationTemplate(Base):
    """Agent 编排模板表"""
    __tablename__ = "orchestration_templates"

    id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()), comment="模板ID")
    name = Column(String(100), nullable=False, comment="模板名称")
    description = Column(Text, comment="模板描述")
    nodes = Column(JSON, nullable=False, comment="编排节点（含Agent配置）")
    edges = Column(JSON, nullable=False, comment="编排连线")
    user_id = Column(CHAR(36), ForeignKey("users.id"), comment="创建者ID")
    workspace_id = Column(CHAR(36), ForeignKey("workspaces.id"), nullable=True, comment="所属工作区ID")
    created_at = Column(DateTime, default=func.now(), comment="创建时间")
    updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间")

    user = relationship("User", backref="orchestration_templates")

    __table_args__ = (
        Index("ix_orchestration_templates_workspace", "workspace_id"),
    )

    def __repr__(self):
        return f"<OrchestrationTemplate(id={self.id}, name={self.name})>"