tanyar09
d9919828c5
- Add auto-start functionality to cron service when jobs are added if service is not running and event loop is available - Add 'reminder' field to CronPayload to distinguish between simple reminders (send message directly) and tasks (execute via agent) - Update cron tool to accept 'reminder' parameter - Fix callback logic to check reminder field: reminders send directly, tasks are processed through agent - Ensures both 'remind me to X' and 'schedule a task to do X' work correctly
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
"""Cron types."""
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Literal
|
|
|
|
|
|
@dataclass
|
|
class CronSchedule:
|
|
"""Schedule definition for a cron job."""
|
|
kind: Literal["at", "every", "cron"]
|
|
# For "at": timestamp in ms
|
|
at_ms: int | None = None
|
|
# For "every": interval in ms
|
|
every_ms: int | None = None
|
|
# For "cron": cron expression (e.g. "0 9 * * *")
|
|
expr: str | None = None
|
|
# Timezone for cron expressions
|
|
tz: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class CronPayload:
|
|
"""What to do when the job runs."""
|
|
kind: Literal["system_event", "agent_turn"] = "agent_turn"
|
|
message: str = ""
|
|
# Deliver response to channel
|
|
deliver: bool = False
|
|
channel: str | None = None # e.g. "whatsapp"
|
|
to: str | None = None # e.g. phone number
|
|
# If True, this is a simple reminder (send message directly)
|
|
# If False, this is a task (agent executes the message)
|
|
reminder: bool = False
|
|
|
|
|
|
@dataclass
|
|
class CronJobState:
|
|
"""Runtime state of a job."""
|
|
next_run_at_ms: int | None = None
|
|
last_run_at_ms: int | None = None
|
|
last_status: Literal["ok", "error", "skipped"] | None = None
|
|
last_error: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class CronJob:
|
|
"""A scheduled job."""
|
|
id: str
|
|
name: str
|
|
enabled: bool = True
|
|
schedule: CronSchedule = field(default_factory=lambda: CronSchedule(kind="every"))
|
|
payload: CronPayload = field(default_factory=CronPayload)
|
|
state: CronJobState = field(default_factory=CronJobState)
|
|
created_at_ms: int = 0
|
|
updated_at_ms: int = 0
|
|
delete_after_run: bool = False
|
|
|
|
|
|
@dataclass
|
|
class CronStore:
|
|
"""Persistent store for cron jobs."""
|
|
version: int = 1
|
|
jobs: list[CronJob] = field(default_factory=list)
|