#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
直接测试Google Gemini API连接
"""

import os
import google.generativeai as genai

def test_gemini_direct():
    """直接测试Gemini API"""
    print("🔍 测试Gemini API直连...")
    
    # 检查API密钥
    api_key = os.getenv('GOOGLE_API_KEY')
    if not api_key:
        print("❌ 未找到 GOOGLE_API_KEY")
        return False
    
    print(f"✅ API密钥已配置 (长度: {len(api_key)})")
    
    try:
        # 配置API
        genai.configure(api_key=api_key)
        
        # 创建模型
        print("📝 创建Gemini模型...")
        model = genai.GenerativeModel('gemini-2.0-flash-exp')
        
        # 发送测试消息
        print("💬 发送测试消息...")
        response = model.generate_content("请简单说'你好，我是Gemini'")
        
        print(f"✅ 测试成功！回复: {response.text}")
        return True
        
    except Exception as e:
        print(f"❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    print("🚀 Gemini直连测试")
    test_gemini_direct()