mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
feat(scheduler): add scheduled jobs with channel-aware delivery
Introduce persisted cron/interval jobs, gateway scheduler loop, worker turns, and admin CRUD/edit UI. Route proactive reminders via the originating chat channel (WhatsApp vs WeChat), harden Weixin outbound with durable queue and PG-compatible polling, and fix chat UI to show each scheduled reminder separately. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
parent
5336f8835f
commit
47813f6a94
45 changed files with 4956 additions and 43 deletions
77
tests/test_tool_runtime_schedule_validation.py
Normal file
77
tests/test_tool_runtime_schedule_validation.py
Normal file
|
|
@ -0,0 +1,77 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
from runtime.chat.tool_runtime import ToolExecutionContext, ToolExecutor
|
||||
from runtime.tools.experts.productivity.schedule_tools import schedule_create_tool
|
||||
from runtime.tools.base import ToolRegistry
|
||||
from svc.llm.transports.base import LLMToolCall
|
||||
from svc.persistence.assistant_store import reset_assistant_store_singleton
|
||||
from svc.persistence.sqlite_store import SqliteStore
|
||||
|
||||
|
||||
class ToolRuntimeScheduleValidationTests(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self._tmp = tempfile.TemporaryDirectory(ignore_cleanup_errors=True)
|
||||
self.db = Path(self._tmp.name) / "rt.sqlite"
|
||||
os.environ["OPS_ASSISTANT_DB_PATH"] = str(self.db)
|
||||
os.environ["AIA_ASSISTANT_DB_BACKEND"] = "sqlite"
|
||||
reset_assistant_store_singleton()
|
||||
self.store = SqliteStore(str(self.db))
|
||||
t = self.store.create_tenant("Team")
|
||||
self.tenant_id = str(t["id"])
|
||||
user = self.store.create_user_account(
|
||||
tenant_id=self.tenant_id,
|
||||
username="administrator",
|
||||
display_name="Admin",
|
||||
role="owner",
|
||||
password_hash="x",
|
||||
is_active=True,
|
||||
)
|
||||
self.user_id = str(user["id"])
|
||||
sess = self.store.create_session_for_user(
|
||||
title="wx",
|
||||
tenant_id=self.tenant_id,
|
||||
user_id=self.user_id,
|
||||
)
|
||||
self.session_id = str(sess.id)
|
||||
self.store.ensure_ui_session_owner(
|
||||
session_id=self.session_id,
|
||||
tenant_id=self.tenant_id,
|
||||
user_id=self.user_id,
|
||||
)
|
||||
|
||||
def tearDown(self) -> None:
|
||||
reset_assistant_store_singleton()
|
||||
self._tmp.cleanup()
|
||||
|
||||
def test_execute_schedule_create_with_hallucinated_user_id(self) -> None:
|
||||
reg = ToolRegistry([schedule_create_tool()])
|
||||
ex = ToolExecutor()
|
||||
ctx = ToolExecutionContext(
|
||||
store=self.store,
|
||||
tools=reg,
|
||||
session_id=self.session_id,
|
||||
path_policy_tenant_id=self.tenant_id,
|
||||
path_policy_user_id=self.user_id,
|
||||
)
|
||||
tc = LLMToolCall(
|
||||
id="call1",
|
||||
name="schedule_create",
|
||||
arguments={
|
||||
"user_id": "wechat_external_123",
|
||||
"name": "Break",
|
||||
"prompt_text": "Stand up",
|
||||
"schedule_kind": "interval",
|
||||
"schedule_expr": "300",
|
||||
},
|
||||
)
|
||||
result, _dur = ex._execute_tool(ctx, tc)
|
||||
self.assertTrue(result.get("ok"), result)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue