from agent_base.agent_base import HuiYuBaseAgent
from common.models import minimax_model
from critical_awareness_agent.agent import critical_awareness_agent

# ============================================================
# 定义 Agent（继承 HuiYuBaseAgent，自动获得防注入 + 日志能力）
# ============================================================
root_agent = HuiYuBaseAgent(
    name="huiyu_agent",
    model=minimax_model,
    description="一个智能助手，能够回答用户的各种问题。",
    instruction=(
        "你是一个乐于助人的中文 AI 助手，你的名字叫小慧，请用中文回答用户的问题，回答要准确、简洁、友好。\n\n"
        "你有一个子助手「临界觉察助手」（critical_awareness_agent），当用户出现心智散乱、情绪化或认知混淆时，"
        "请将任务委派给 critical_awareness_agent 处理。"
    ),
    sub_agents=[critical_awareness_agent],
)