82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
稷下学宫 ADK 简单论道测试
|
|
实现智能体间的基本对话功能
|
|
"""
|
|
|
|
import os
|
|
from google.adk import Agent
|
|
|
|
def create_debate_agents():
|
|
"""创建论道智能体"""
|
|
|
|
# 铁拐李 - 逆向思维专家
|
|
tie_guai_li = Agent(
|
|
name="铁拐李",
|
|
model="gemini-2.0-flash-exp"
|
|
)
|
|
|
|
# 吕洞宾 - 理性分析者
|
|
lu_dong_bin = Agent(
|
|
name="吕洞宾",
|
|
model="gemini-2.0-flash-exp"
|
|
)
|
|
|
|
return tie_guai_li, lu_dong_bin
|
|
|
|
def simple_debate_test():
|
|
"""简单论道测试"""
|
|
print("🎭 开始简单论道测试...")
|
|
|
|
# 创建智能体
|
|
tie_guai_li, lu_dong_bin = create_debate_agents()
|
|
|
|
print("\n📋 论道主题: 人工智能对未来社会的影响")
|
|
print("\n🎯 开始论道...")
|
|
|
|
try:
|
|
# 测试智能体创建
|
|
print("\n✅ 智能体创建成功:")
|
|
print(f" - {tie_guai_li.name}: {tie_guai_li.model}")
|
|
print(f" - {lu_dong_bin.name}: {lu_dong_bin.model}")
|
|
|
|
print("\n🎉 简单论道测试完成！")
|
|
print("\n📝 智能体基础功能验证成功")
|
|
|
|
except Exception as e:
|
|
print(f"❌ 论道测试失败: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
def main():
|
|
"""主函数"""
|
|
print("🚀 稷下学宫 ADK 简单论道系统")
|
|
|
|
# 检查API密钥
|
|
api_key = os.getenv('GOOGLE_API_KEY')
|
|
if not api_key:
|
|
print("❌ 未找到 GOOGLE_API_KEY 环境变量")
|
|
print("请使用: doppler run -- python src/jixia/debates/adk_simple_debate.py")
|
|
return
|
|
|
|
print(f"✅ API密钥已配置")
|
|
|
|
# 运行测试
|
|
try:
|
|
result = simple_debate_test()
|
|
if result:
|
|
print("\n📝 测试结果: 成功")
|
|
print("\n🎯 下一步开发计划:")
|
|
print(" 1. 学习ADK的正确调用方式")
|
|
print(" 2. 实现智能体对话功能")
|
|
print(" 3. 扩展到八仙全员论道")
|
|
print(" 4. 集成实时数据源")
|
|
else:
|
|
print("\n❌ 测试失败")
|
|
except Exception as e:
|
|
print(f"❌ 运行失败: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |