| 123456789101112131415161718192021222324252627282930 |
- from enum import Enum
- from sqlalchemy import Column, Integer, Text, BigInteger, String, SmallInteger, Boolean, TIMESTAMP
- from sqlalchemy.ext.declarative import declarative_base
- Base = declarative_base()
- class AgentType(int, Enum):
- REACTIVE = 0 # 响应式
- PLANNING = 1 # 自主规划式
- class AgentConfiguration(Base):
- __tablename__ = "agent_configuration"
- id = Column(BigInteger, primary_key=True, autoincrement=True, comment="主键id")
- name = Column(String(64), nullable=False, comment="唯一名称")
- display_name = Column(String(64), nullable=True, comment="可选，显示名")
- type = Column(SmallInteger, nullable=False, default=0, comment="Agent类型，0-响应式，1-自主规划式")
- execution_model = Column(String(64), nullable=True, comment="执行LLM")
- system_prompt = Column(Text, nullable=True, comment="系统设定prompt模板")
- task_prompt = Column(Text, nullable=True, comment="执行任务prompt模板")
- tools = Column(Text, nullable=True, comment="JSON数组，tool name")
- sub_agents = Column(Text, nullable=True, comment="JSON数组，agent ID")
- extra_params = Column(Text, nullable=True, comment="JSON KV对象")
- is_delete = Column(Boolean, nullable=False, default=False, comment="逻辑删除标识")
- create_user = Column(String(32), nullable=True, comment="创建用户")
- update_user = Column(String(32), nullable=True, comment="更新用户")
- create_time = Column(TIMESTAMP, nullable=True, server_default="CURRENT_TIMESTAMP", comment="创建时间")
- update_time = Column(TIMESTAMP, nullable=True, server_default="CURRENT_TIMESTAMP", onupdate="CURRENT_TIMESTAMP", comment="更新时间")
|
