69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
"""
|
||
Graph 骨架自检：编译 + start_thread（Memory）+ 无 sender_id 拒绝。
|
||
|
||
用法（在 inquiry-agent 目录、已装依赖）：
|
||
set CHECKPOINT_BACKEND=memory
|
||
python -m agent.graph.selftest
|
||
|
||
不连企微、不连主账；正式验收仍以真实企微事件为准。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import os
|
||
import sys
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def main() -> int:
|
||
logging.basicConfig(level=logging.INFO, format="%(levelname)s %(message)s")
|
||
# 自检强制内存，避免误连生产 PG
|
||
os.environ["CHECKPOINT_BACKEND"] = "memory"
|
||
os.environ.setdefault("YTD_ENV", "test")
|
||
|
||
from agent.config import get_settings
|
||
from agent.graph.runtime import GraphIdentityError, GraphRuntime, shutdown_graph_runtime
|
||
from agent.graph.wake import WorkerWakePayload
|
||
|
||
get_settings.cache_clear()
|
||
rt = GraphRuntime.create(get_settings())
|
||
try:
|
||
try:
|
||
rt.start_thread(sender_id="")
|
||
logger.error("预期应拒绝空 sender_id")
|
||
return 1
|
||
except GraphIdentityError:
|
||
logger.info("OK: 空 sender_id 已拒绝")
|
||
|
||
started = rt.start_thread(sender_id="selftest_user")
|
||
tid = started["thread_id"]
|
||
logger.info("OK: start_thread thread_id=%s", tid)
|
||
|
||
# 骨架：wake 需要已有 thread；写入 last_wake 后再 resume 一步
|
||
wake = WorkerWakePayload(
|
||
thread_id=tid,
|
||
inquiry_no="SELFTEST-001",
|
||
quote_version=1,
|
||
wait_version=0,
|
||
kind="recognition_done",
|
||
)
|
||
woken = rt.wake_from_worker(wake)
|
||
logger.info("OK: wake_from_worker kind=%s", woken["wake"]["kind"])
|
||
|
||
resumed = rt.resume_thread(thread_id=tid, sender_id="selftest_user")
|
||
logger.info("OK: resume_thread thread_id=%s", resumed["thread_id"])
|
||
logger.info(
|
||
"Graph 骨架自检通过 checkpoint_backend=%s",
|
||
rt.checkpoint.backend,
|
||
)
|
||
return 0
|
||
finally:
|
||
rt.close()
|
||
shutdown_graph_runtime()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|