steven_li
30ab74ffb2
- 集成MCP连接管理器，支持MCP服务器连接 - 添加多种内置工具：ClarifyTool、CronTool、DelegateTool、ExecuteCodeTool、 PatchFileTool、ProcessTool、SendMessageTool、SpawnTool、TerminalTool、 TodoTool、WebFetchTool、WebSearchTool、WriteFileTool等 - 实现工具注册和装配功能 - 添加技能选择上下文参数 - 支持思考模式控制参数thinking_enabled feat(coordinator): 重构任务执行计划器参数命名 - 将learning_candidate_enabled重命名为allow_candidate_generation - 更新TeamGraphScheduler中的参数传递 - 修改LocalAgentRunner中的相关参数处理 - 更新README文档中的相应描述 refactor(context): 标准化工具调用参数格式 - 添加_json导入用于参数序列化 - 实现_provider_tool_calls方法标准化OpenAI兼容的工具调用载荷 - 修复工具调用中参数非字符串类型的序列化问题 refactor(session): 优化消息历史记录过滤逻辑 - 修改get_messages_as_conversation为基于运行状态过滤消息 - 排除未完成、失败或错误结束的运行记录 - 改进对话历史的可见性控制机制 fix(store): 修复FTS索引重建逻辑 - 添加异常处理防止FTS索引创建失败 - 实现_rebuild_fts_index方法重新构建全文搜索索引 - 优化索引触发器和表的维护流程
127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
import asyncio
|
|
|
|
from beaver.foundation.models import CronExecutionResult, CronRunRecord, CronSchedule
|
|
from beaver.tools.base import ToolContext
|
|
from beaver.tools.builtins import CronTool
|
|
from beaver.services.cron_service import CronService, compute_next_run, parse_schedule, schedule_from_api
|
|
|
|
|
|
def test_parse_hermes_style_schedules() -> None:
|
|
interval = parse_schedule("every 15m")
|
|
assert interval.kind == "every"
|
|
assert interval.every_ms == 15 * 60 * 1000
|
|
|
|
one_shot = parse_schedule("30s")
|
|
assert one_shot.kind == "at"
|
|
assert one_shot.at_ms is not None
|
|
|
|
cron = parse_schedule("0 9 * * *")
|
|
assert cron.kind == "cron"
|
|
assert cron.expr == "0 9 * * *"
|
|
|
|
|
|
def test_schedule_from_frontend_payload() -> None:
|
|
every = schedule_from_api({"every_seconds": 60})
|
|
assert every.kind == "every"
|
|
assert every.every_ms == 60_000
|
|
|
|
cron = schedule_from_api({"cron_expr": "0 10 * * *"})
|
|
assert cron.kind == "cron"
|
|
|
|
|
|
def test_compute_next_run_skips_missed_interval() -> None:
|
|
schedule = CronSchedule(kind="every", every_ms=60_000)
|
|
assert compute_next_run(schedule, now_ms=1_000_000, last_run_at_ms=0) > 1_000_000
|
|
|
|
|
|
def test_manual_run_records_task_history(tmp_path) -> None:
|
|
async def on_job(job):
|
|
return CronExecutionResult(response="done", task_id=f"task-{job.id}", run_id="run-1")
|
|
|
|
service = CronService(tmp_path / "jobs.json", on_job=on_job)
|
|
job = service.add_job(
|
|
name="Daily check",
|
|
message="Check the project",
|
|
schedule=CronSchedule(kind="every", every_ms=3600_000),
|
|
session_key="web:default",
|
|
)
|
|
|
|
assert asyncio.run(service.run_job(job.id, force=True)) is True
|
|
updated = service.get_job(job.id)
|
|
assert updated is not None
|
|
assert updated.last_status == "ok"
|
|
assert updated.history[-1].task_id == f"task-{job.id}"
|
|
assert updated.to_api_dict()["last_task_id"] == f"task-{job.id}"
|
|
|
|
|
|
def test_manual_run_records_scheduled_run_output(tmp_path) -> None:
|
|
async def on_job(job, run):
|
|
return CronExecutionResult(
|
|
response=f"notification for {run.scheduled_run_id}",
|
|
run_id="run-notify",
|
|
notification_session_id="notify:default:scheduled",
|
|
mode="notification",
|
|
)
|
|
|
|
service = CronService(tmp_path / "jobs.json", on_job=on_job)
|
|
job = service.add_job(
|
|
name="Daily news",
|
|
message="Summarize news",
|
|
schedule=CronSchedule(kind="every", every_ms=3600_000),
|
|
)
|
|
|
|
assert asyncio.run(service.run_job(job.id, force=True)) is True
|
|
updated = service.get_job(job.id)
|
|
assert updated is not None
|
|
run = updated.history[-1]
|
|
assert run.scheduled_run_id
|
|
assert run.output == f"notification for {run.scheduled_run_id}"
|
|
assert run.notification_session_id == "notify:default:scheduled"
|
|
assert updated.to_api_dict()["last_scheduled_run_id"] == run.scheduled_run_id
|
|
|
|
|
|
def test_cron_tool_uses_runtime_service(tmp_path) -> None:
|
|
service = CronService(tmp_path / "jobs.json")
|
|
tool = CronTool()
|
|
result = asyncio.run(
|
|
tool.invoke(
|
|
{
|
|
"action": "add",
|
|
"name": "Tool-created task",
|
|
"message": "Check the queue",
|
|
"every_seconds": 300,
|
|
},
|
|
ToolContext(session_id="session-1", services={"cron_service": service}),
|
|
)
|
|
)
|
|
|
|
assert result.success is True
|
|
jobs = service.list_jobs(include_disabled=True)
|
|
assert len(jobs) == 1
|
|
assert jobs[0].payload.session_key == "session-1"
|
|
|
|
|
|
def test_mark_run_engaged_links_task(tmp_path) -> None:
|
|
service = CronService(tmp_path / "jobs.json")
|
|
job = service.add_job(
|
|
name="Daily news",
|
|
message="Summarize news",
|
|
schedule=CronSchedule(kind="every", every_ms=3600_000),
|
|
)
|
|
run = CronRunRecord(
|
|
started_at_ms=1,
|
|
status="ok",
|
|
output="news summary",
|
|
notification_session_id="notify:default:scheduled",
|
|
)
|
|
job.history.append(run)
|
|
service._save_jobs()
|
|
|
|
linked = service.mark_run_engaged(run.scheduled_run_id, task_id="task-1", intent="revise_once")
|
|
|
|
assert linked is not None
|
|
updated = service.get_run(run.scheduled_run_id)
|
|
assert updated is not None
|
|
assert updated[1].engaged is True
|
|
assert updated[1].task_id == "task-1"
|