70 lines
3.2 KiB
Python
70 lines
3.2 KiB
Python
"""
|
||
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() # 执行验证函数
|