agent_task.py 1.3 KB

12345678910111213141516171819202122
  1. from sqlalchemy import Column, Integer, Text, BigInteger, String, SmallInteger, Boolean, TIMESTAMP
  2. from sqlalchemy.ext.declarative import declarative_base
  3. Base = declarative_base()
  4. class AgentTask(Base):
  5. __tablename__ = "agent_task"
  6. id = Column(BigInteger, primary_key=True, autoincrement=True, comment="主键id")
  7. agent_id = Column(BigInteger, nullable=False, comment="agent主键")
  8. status = Column(Integer, nullable=False, default=0, comment="状态(0:未开始, 1:进行中, 2:已完成, 3:失败)")
  9. start_time = Column(TIMESTAMP, nullable=True, comment="任务开始执行时间")
  10. end_time = Column(TIMESTAMP, nullable=True, comment="任务结束执行时间")
  11. create_user = Column(String(32), nullable=True, comment="创建用户")
  12. input = Column(Text, nullable=True, comment="任务执行输入")
  13. tools = Column(Text, nullable=True, comment="任务使用的工具")
  14. output = Column(Text, nullable=True, comment="任务执行输出")
  15. error_message = Column(Text, nullable=True, comment="错误详情(失败时记录)")
  16. create_time = Column(TIMESTAMP, nullable=False, server_default="CURRENT_TIMESTAMP", comment="创建时间")
  17. update_time = Column(TIMESTAMP, nullable=False, server_default="CURRENT_TIMESTAMP", onupdate="CURRENT_TIMESTAMP",
  18. comment="更新时间")