0bbf68d5bb
- 新增 AgentSchedule 模型、CRUD API、调度服务 - 集成 Celery Beat 每分钟检查到期任务并自动触发执行 - 支持手动触发、cron 表达式解析、执行状态跟踪 - 依赖: croniter (cron 表达式解析) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
29 lines
1.7 KiB
Python
29 lines
1.7 KiB
Python
"""Agent 定时任务表：按 cron 表达式周期执行 Agent"""
|
|
import uuid
|
|
from datetime import datetime
|
|
from sqlalchemy import Column, String, Text, Integer, DateTime, ForeignKey, Boolean
|
|
from sqlalchemy.dialects.mysql import CHAR
|
|
from app.core.database import Base
|
|
|
|
|
|
class AgentSchedule(Base):
|
|
"""Agent 定时任务 — 按 cron 表达式周期执行指定 Agent"""
|
|
__tablename__ = "agent_schedules"
|
|
|
|
id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
agent_id = Column(CHAR(36), ForeignKey("agents.id"), nullable=False, index=True, comment="关联 Agent ID")
|
|
name = Column(String(100), nullable=False, comment="任务名称")
|
|
cron_expression = Column(String(100), nullable=False, comment="cron 表达式，如 0 9 * * *")
|
|
input_message = Column(Text, nullable=False, comment="定时执行时发送的消息内容")
|
|
timezone = Column(String(64), default="Asia/Shanghai", comment="时区")
|
|
enabled = Column(Boolean, default=True, comment="是否启用")
|
|
last_run_at = Column(DateTime, nullable=True, comment="上次执行时间")
|
|
last_run_status = Column(String(32), nullable=True, comment="上次执行状态: success/failed")
|
|
next_run_at = Column(DateTime, nullable=False, comment="下次执行时间")
|
|
user_id = Column(CHAR(36), ForeignKey("users.id"), nullable=False, index=True, comment="创建者 ID")
|
|
created_at = Column(DateTime, default=datetime.utcnow, comment="创建时间")
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, comment="更新时间")
|
|
|
|
def __repr__(self):
|
|
return f"<AgentSchedule(id={self.id}, name={self.name}, cron={self.cron_expression})>"
|