#!/usr/bin/env python3
"""Qwen API Key 连通性测试脚本"""

import os
import sys

# 设置环境变量
os.environ['QWEN_API_KEY'] = 'sk-426efxxxxxxxxxxxxxxxxxxxxxxxxxaba'

# 添加项目根目录到 Python 路径
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from svc.qwen import validate_qwen_api_key
from consts.consts import Qwen

def main():
    print("=" * 60)
    print("Qwen API Key 连通性测试")
    print("=" * 60)

    api_key = os.environ.get('QWEN_API_KEY')
    print(f"\nAPI Key: {api_key[:20]}... (已隐藏)")
    print(f"测试模型: {Qwen.TURBO}\n")

    try:
        result = validate_qwen_api_key(api_key, Qwen.TURBO)

        if result:
            print("✅ API Key 测试通过！")
            print("   Qwen API 连接正常，可以进行后续测试")
            return 0
        else:
            print("❌ API Key 测试失败")
            print("   请检查 API Key 是否有效")
            return 1

    except Exception as e:
        print(f"❌ 测试过程中发生错误: {e}")
        import traceback
        traceback.print_exc()
        return 1

if __name__ == "__main__":
    sys.exit(main())