main.py

"""Mini Agent CLI 入口：交互式对话循环。"""

import logging
import sys
from pathlib import Path
from uuid import uuid4


def _configure_logging() -> None:
    root = Path(__file__).resolve().parent
    log_dir = root / "logs"
    log_dir.mkdir(parents=True, exist_ok=True)
    log_path = log_dir / "mini_agent.log"

    console_handler = logging.StreamHandler(sys.stderr)
    console_handler.setLevel(logging.WARNING)

    file_handler = logging.FileHandler(log_path, encoding="utf-8")
    file_handler.setLevel(logging.INFO)

    logging.basicConfig(
        level=logging.INFO,
        format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
        handlers=[console_handler, file_handler],
        force=True,
    )


_configure_logging()

WELCOME_BANNER = """
╔══════════════════════════════════════════════════╗
║           Mini Agent - 基于 LangGraph            ║
║  输入你的问题开始对话，输入 'exit' 退出          ║
║  输入 'new' 开始新对话（清除历史）               ║
║  输入 'help' 查看帮助                            ║
╚══════════════════════════════════════════════════╝
"""

HELP_TEXT = """
可用命令：
  exit / quit / 退出  — 退出程序
  new / 新对话        — 开始新的对话（新的 thread_id）
  help / 帮助         — 显示此帮助
  其他输入            — 发送消息给 Agent

运行时特性：
  - 显式 user_id / thread_id
  - OpenViking session 维护可恢复会话历史
  - OpenViking user memories 维护长期记忆
"""


def _new_thread_id() -> str:
    return str(uuid4())


def _get_last_ai_message(messages: list) -> str:
    for msg in reversed(messages):
        if getattr(msg, "type", None) == "ai":
            content = getattr(msg, "content", "")
            if isinstance(content, list):
                parts = [part.get("text", "") for part in content if isinstance(part, dict)]
                return "\n".join(parts)
            return str(content)
    return "（无回复）"


def main() -> None:
    print(WELCOME_BANNER)
    user_id = input("请输入 user_id（默认 local-user）: ").strip() or "local-user"

    print("正在初始化 Agent...")
    try:
        from backend.runtime import MiniAgentRuntime
        runtime = MiniAgentRuntime()
    except Exception as exc:
        print(f"\n❌ 初始化失败：{exc}")
        print("请检查 config.yaml 和环境变量配置是否正确")
        sys.exit(1)

    print("✅ Agent 已就绪\n")
    thread_id = _new_thread_id()
    print(f"当前用户: {user_id}")
    print(f"当前会话 ID: {thread_id[:8]}...\n")

    while True:
        try:
            user_input = input("你: ").strip()
        except (EOFError, KeyboardInterrupt):
            print("\n\n再见！")
            break

        if not user_input:
            continue
        if user_input.lower() in ("exit", "quit", "退出", "q"):
            print("再见！")
            break
        if user_input.lower() in ("new", "新对话"):
            thread_id = _new_thread_id()
            print(f"✅ 已开始新对话，会话 ID: {thread_id[:8]}...\n")
            continue
        if user_input.lower() in ("help", "帮助", "?"):
            print(HELP_TEXT)
            continue

        try:
            response = runtime.invoke(user_input, user_id=user_id, thread_id=thread_id)
            answer = _get_last_ai_message(response.get("messages", []))
            print(f"\n助手: {answer}\n")
        except KeyboardInterrupt:
            print("\n（已中断）\n")
        except Exception as exc:
            print(f"\n❌ 出错了：{exc}\n")
            logging.getLogger(__name__).exception("Agent 调用失败")


if __name__ == "__main__":
    main()