"""
测试硅基流动API连接
"""
import os
from dotenv import load_dotenv
import requests

# 加载环境变量
load_dotenv()

def test_siliconflow_api():
    """测试硅基流动API连接"""
    print("="*80)
    print("测试硅基流动API连接")
    print("="*80)
    print()
    
    # 读取配置
    api_key = os.getenv('SILICONFLOW_API_KEY')
    model = os.getenv('SILICONFLOW_MODEL', 'deepseek-ai/DeepSeek-V3.2-Exp')
    api_url = "https://api.siliconflow.cn/v1/chat/completions"
    
    # 检查配置
    print("1. 检查配置...")
    if not api_key:
        print("   ✗ 错误: 未找到 SILICONFLOW_API_KEY")
        print("   请在 .env 文件中设置: SILICONFLOW_API_KEY=你的API密钥")
        return False
    
    print(f"   ✓ API密钥: {api_key[:10]}...{api_key[-5:]}")
    print(f"   ✓ 模型: {model}")
    print(f"   ✓ API地址: {api_url}")
    print()
    
    # 测试API调用
    print("2. 测试API调用...")
    try:
        payload = {
            "model": model,
            "messages": [
                {
                    "role": "user",
                    "content": "请回复'测试成功'"
                }
            ],
            "max_tokens": 20
        }
        
        headers = {
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json"
        }
        
        print("   正在发送请求...")
        response = requests.post(
            api_url,
            json=payload,
            headers=headers,
            timeout=30
        )
        
        print(f"   响应状态码: {response.status_code}")
        
        if response.status_code == 200:
            result = response.json()
            if 'choices' in result and len(result['choices']) > 0:
                content = result['choices'][0]['message']['content']
                print(f"   ✓ API调用成功！")
                print(f"   响应内容: {content}")
                return True
            else:
                print(f"   ✗ API响应格式异常: {result}")
                return False
        else:
            print(f"   ✗ API调用失败")
            print(f"   状态码: {response.status_code}")
            print(f"   错误信息: {response.text[:500]}")
            return False
            
    except requests.exceptions.Timeout:
        print(f"   ✗ 请求超时（30秒）")
        return False
    except Exception as e:
        print(f"   ✗ 请求失败: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == '__main__':
    success = test_siliconflow_api()
    print()
    print("="*80)
    if success:
        print("✓ API测试通过，可以正常使用AI分析功能")
    else:
        print("✗ API测试失败，请检查配置后重试")
    print("="*80)