base_agent/main.py

"""
main.py
项目入口 —— 启动 Agent 交互式对话 或 MCP Server stdio 模式

用法:
    python main.py              # 启动 Agent 交互式对话（默认）
    python main.py --mode agent # 同上
    python main.py --mode mcp   # 启动本地 MCP Server（stdio 模式）
    python main.py --mode check # 检查配置和依赖
"""

import argparse
import atexit
import sys


def run_agent() -> None:
    """启动 Agent 交互式对话"""
    from agent.agent import create_agent
    from config.settings import settings

    print(settings.display())

    agent, registry = create_agent()
    atexit.register(registry.close)

    print(agent.show_tools())
    print("─" * 60)
    print("💡 命令: exit=退出  reset=清空历史  tools=查看工具列表")
    print("─" * 60)

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

        if not user_input:
            continue

        match user_input.lower():
            case "exit" | "quit":
                print("👋 再见！")
                break
            case "reset":
                agent.reset()
                print("🔄 对话历史已清空")
            case "tools":
                print(agent.show_tools())
            case _:
                reply = agent.chat(user_input)
                print(f"\n🤖 Agent: {reply}")


def run_mcp_server() -> None:
    """启动本地 MCP Server（stdio 模式）"""
    from mcp.mcp_server import MCPServer
    with MCPServer() as server:
        server.run_stdio()


# def run_check() -> None:
#     """检查配置和依赖完整性"""
#     print("=" * 60)
#     print("  🔍 项目依赖检查")
#     print("=" * 60)
#
#     checks = [
#         ("pyyaml",    "yaml",      "pip install pyyaml"),
#         ("openai",    "openai",    "pip install openai>=1.0.0"),
#         ("httpx",     "httpx",     "pip install httpx>=0.27.0"),
#         ("httpx-sse", "httpx_sse", "pip install httpx-sse>=0.4.0"),
#         ("paramiko",  "paramiko",  "pip install paramiko>=3.0.0"),
#     ]
#
#     all_ok = True
#     for pkg_name, import_name, install_cmd in checks:
#         try:
#             __import__(import_name)
#             print(f"  ✅ {pkg_name:<15} 已安装")
#         except ImportError:
#             print(f"  ❌ {pkg_name:<15} 未安装  →  {install_cmd}")
#             all_ok = False
#
#     print()
#
#     # 配置检查
#     try:
#         from config.settings import settings
#         print("  ✅ config/settings.py  加载成功")
#         print(f"     LLM     : {settings.llm.provider} / {settings.llm.model_name}")
#         print(f"     本地工具: {settings.mcp.enabled_tools}")
#         skills = settings.enabled_mcp_skills
#         if skills:
#             print(f"     在线Skill: {[s.name for s in skills]}")
#         else:
#             print("     在线Skill: (未配置)")
#     except Exception as e:
#         print(f"  ❌ 配置加载失败: {e}")
#         all_ok = False
#
#     print()
#
#     # 工具注册检查
#     try:
#         from mcp.skill_registry import SkillRegistry
#         from tools.calculator      import CalculatorTool
#         from tools.code_executor   import CodeExecutorTool
#         from tools.file_reader     import FileReaderTool
#         from tools.ssh_docker      import SSHDockerTool
#         from tools.static_analyzer import StaticAnalyzerTool
#         from tools.web_search      import WebSearchTool
#
#         registry = SkillRegistry()
#         registry.register_local_many(
#             CalculatorTool(), WebSearchTool(), FileReaderTool(),
#             CodeExecutorTool(), StaticAnalyzerTool(), SSHDockerTool(),
#         )
#         tools = registry.list_all_tools()
#         print(f"  ✅ 本地工具注册  共 {len(tools)} 个:")
#         for t in tools:
#             print(f"     🔵 {t['name']}: {t['description'][:50]}")
#     except Exception as e:
#         print(f"  ❌ 工具注册失败: {e}")
#         all_ok = False
#
#     print()
#     print("=" * 60)
#     if all_ok:
#         print("  ✅ 所有检查通过，项目可正常运行")
#     else:
#         print("  ⚠️  存在问题，请按提示安装缺失依赖")
#     print("=" * 60)


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Agent Demo —— 支持本地工具 + 在线 MCP Skill"
    )
    parser.add_argument(
        "--mode",
        choices=["agent", "mcp", "check"],
        default="agent",
        help="运行模式: agent（交互对话）| mcp（MCP Server）| check（依赖检查）",
    )
    args = parser.parse_args()

    match args.mode:
        case "agent":
            run_agent()
        case "mcp":
            run_mcp_server()
        # case "check":
        #     run_check()


if __name__ == "__main__":
    main()