init

ADKLearning/google-adk-tutorial/code/setup_demo.py

@@ -0,0 +1,69 @@
+"""
+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()                    # 执行验证函数