115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
|
"""Cron tool for scheduling reminders and tasks."""
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from nanobot.agent.tools.base import Tool
|
||
|
|
from nanobot.cron.service import CronService
|
||
|
|
from nanobot.cron.types import CronSchedule
|
||
|
|
|
||
|
|
|
||
|
|
class CronTool(Tool):
|
||
|
|
"""Tool to schedule reminders and recurring tasks."""
|
||
|
|
|
||
|
|
def __init__(self, cron_service: CronService):
|
||
|
|
self._cron = cron_service
|
||
|
|
self._channel = ""
|
||
|
|
self._chat_id = ""
|
||
|
|
|
||
|
|
def set_context(self, channel: str, chat_id: str) -> None:
|
||
|
|
"""Set the current session context for delivery."""
|
||
|
|
self._channel = channel
|
||
|
|
self._chat_id = chat_id
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self) -> str:
|
||
|
|
return "cron"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def description(self) -> str:
|
||
|
|
return "Schedule reminders and recurring tasks. Actions: add, list, remove."
|
||
|
|
|
||
|
|
@property
|
||
|
|
def parameters(self) -> dict[str, Any]:
|
||
|
|
return {
|
||
|
|
"type": "object",
|
||
|
|
"properties": {
|
||
|
|
"action": {
|
||
|
|
"type": "string",
|
||
|
|
"enum": ["add", "list", "remove"],
|
||
|
|
"description": "Action to perform"
|
||
|
|
},
|
||
|
|
"message": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "Reminder message (for add)"
|
||
|
|
},
|
||
|
|
"every_seconds": {
|
||
|
|
"type": "integer",
|
||
|
|
"description": "Interval in seconds (for recurring tasks)"
|
||
|
|
},
|
||
|
|
"cron_expr": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "Cron expression like '0 9 * * *' (for scheduled tasks)"
|
||
|
|
},
|
||
|
|
"job_id": {
|
||
|
|
"type": "string",
|
||
|
|
"description": "Job ID (for remove)"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"required": ["action"]
|
||
|
|
}
|
||
|
|
|
||
|
|
async def execute(
|
||
|
|
self,
|
||
|
|
action: str,
|
||
|
|
message: str = "",
|
||
|
|
every_seconds: int | None = None,
|
||
|
|
cron_expr: str | None = None,
|
||
|
|
job_id: str | None = None,
|
||
|
|
**kwargs: Any
|
||
|
|
) -> str:
|
||
|
|
if action == "add":
|
||
|
|
return self._add_job(message, every_seconds, cron_expr)
|
||
|
|
elif action == "list":
|
||
|
|
return self._list_jobs()
|
||
|
|
elif action == "remove":
|
||
|
|
return self._remove_job(job_id)
|
||
|
|
return f"Unknown action: {action}"
|
||
|
|
|
||
|
|
def _add_job(self, message: str, every_seconds: int | None, cron_expr: str | None) -> str:
|
||
|
|
if not message:
|
||
|
|
return "Error: message is required for add"
|
||
|
|
if not self._channel or not self._chat_id:
|
||
|
|
return "Error: no session context (channel/chat_id)"
|
||
|
|
|
||
|
|
# Build schedule
|
||
|
|
if every_seconds:
|
||
|
|
schedule = CronSchedule(kind="every", every_ms=every_seconds * 1000)
|
||
|
|
elif cron_expr:
|
||
|
|
schedule = CronSchedule(kind="cron", expr=cron_expr)
|
||
|
|
else:
|
||
|
|
return "Error: either every_seconds or cron_expr is required"
|
||
|
|
|
||
|
|
job = self._cron.add_job(
|
||
|
|
name=message[:30],
|
||
|
|
schedule=schedule,
|
||
|
|
message=message,
|
||
|
|
deliver=True,
|
||
|
|
channel=self._channel,
|
||
|
|
to=self._chat_id,
|
||
|
|
)
|
||
|
|
return f"Created job '{job.name}' (id: {job.id})"
|
||
|
|
|
||
|
|
def _list_jobs(self) -> str:
|
||
|
|
jobs = self._cron.list_jobs()
|
||
|
|
if not jobs:
|
||
|
|
return "No scheduled jobs."
|
||
|
|
lines = [f"- {j.name} (id: {j.id}, {j.schedule.kind})" for j in jobs]
|
||
|
|
return "Scheduled jobs:\n" + "\n".join(lines)
|
||
|
|
|
||
|
|
def _remove_job(self, job_id: str | None) -> str:
|
||
|
|
if not job_id:
|
||
|
|
return "Error: job_id is required for remove"
|
||
|
|
if self._cron.remove_job(job_id):
|
||
|
|
return f"Removed job {job_id}"
|
||
|
|
return f"Job {job_id} not found"
|