"""
Google ADK 环境搭建验证脚本
验证 ADK 是否正确安装，并打印版本信息

对应教程：第01章 - ADK 简介与环境搭建
"""

# 导入 ADK 核心模块，验证安装是否成功
from google.adk.agents import Agent          # 智能体模块
from google.adk.runners import Runner        # 运行器模块
from google.adk.sessions import InMemorySessionService  # 内存会话服务


def verify_installation():
    """验证 ADK 安装是否成功"""

    # 打印分隔线
    print("=" * 50)                          # 打印分隔线
    print("Google ADK 环境验证")              # 打印标题
    print("=" * 50)                          # 打印分隔线

    # 验证 Agent 类是否可用
    try:                                     # 尝试导入
        agent = Agent(                       # 创建一个简单的测试 Agent
            name="test_agent",               # 设置 Agent 名称
            model="gemini-2.0-flash",        # 使用 Gemini 模型
            instruction="你是一个测试助手。", # 设置系统指令
        )
        print(f"✅ Agent 模块可用")           # 确认 Agent 可用
        print(f"   Agent 名称: {agent.name}")  # 打印 Agent 名称
    except ImportError as e:                 # 如果导入失败
        print(f"❌ Agent 模块导入失败: {e}")  # 打印错误信息
        return False                         # 返回失败

    # 验证 Runner 类是否可用
    try:                                     # 尝试导入
        print(f"✅ Runner 模块可用")          # 确认 Runner 可用
    except ImportError as e:                 # 如果导入失败
        print(f"❌ Runner 模块导入失败: {e}") # 打印错误信息
        return False                         # 返回失败

    # 验证 SessionService 是否可用
    try:                                     # 尝试创建
        session_service = InMemorySessionService()  # 创建内存会话服务
        print(f"✅ SessionService 可用")      # 确认会话服务可用
    except Exception as e:                   # 如果创建失败
        print(f"❌ SessionService 创建失败: {e}")  # 打印错误信息
        return False                         # 返回失败

    # 验证 ADK 版本
    try:                                     # 尝试获取版本
        import importlib.metadata            # 导入元数据模块
        version = importlib.metadata.version("google-adk")  # 获取版本号
        print(f"✅ ADK 版本: {version}")      # 打印版本号
    except Exception:                        # 如果获取失败
        print("⚠️ 无法获取 ADK 版本")         # 打印警告

    # 打印成功信息
    print("=" * 50)                          # 打印分隔线
    print("🎉 所有模块验证通过！")            # 打印成功信息
    print("可以开始使用 Google ADK 了！")     # 提示用户
    print("=" * 50)                          # 打印分隔线

    return True                              # 返回成功


# 当脚本被直接运行时执行验证
if __name__ == "__main__":                  # 判断是否直接运行
    verify_installation()                    # 执行验证函数