#!/usr/bin/env python3
"""
手动触发旁路由写入 RAG 的测试。

构造一段用户明确要求保存信息的会话，让旁路模型 qwen 决定是否 worth_rag，
并观察它是否会调用 RAG 写入接口。
"""

import sys
import os

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from brain import Brain


def main():
    print("== 初始化 Brain（旁路由）==")
    brain = Brain("config.yaml")

    user_input = (
        "@review 请帮我把以下个人信息保存到 RAG 知识库，方便以后查询：\n"
        "一级注册结构工程师延续注册网址：https://zwfwzjb.mohurd.gov.cn/zjblogincheck/qjd/index.html?checktype=1，密码：Jyq_810302\n"
        "继续教育网址：https://www.zjjsrc.cn/，密码：025610\n"
    )

    print("\n== 模拟用户输入（含 @review 召唤）==")
    print(user_input)
    brain.feed_user_input(user_input)

    print("\n== 模拟主路由 Kimi 回复 ==")
    brain.feed_kimi_output("好的，我已经记录了这些注册和继续教育信息。")

    print("\n== 测试结束 ==")
    print("请查看日志 kimi-copilot.log 确认旁路模型调用和 RAG 写入结果。")


if __name__ == "__main__":
    main()