"""
智能体模型
"""
from sqlalchemy import Column, String, Text, Integer, DateTime, JSON, ForeignKey, func
from sqlalchemy.dialects.mysql import CHAR
from sqlalchemy.orm import relationship
from app.core.database import Base
import uuid


class Agent(Base):
    """智能体表"""
    __tablename__ = "agents"

    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="描述")
    workflow_config = Column(JSON, nullable=False, comment="工作流配置")
    version = Column(Integer, default=1, comment="版本号")
    status = Column(String(20), default="draft", comment="状态: draft/published/running/stopped")
    user_id = Column(CHAR(36), ForeignKey("users.id"), 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="agents")

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