#!/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()