#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
API测试模块

功能：
- 测试OpenRouter AI API的调用
- 验证API响应是否正常
- 提供简单的测试接口

依赖：
- requests
"""

import requests
import json

def test_openrouter_api():
    """测试OpenRouter AI API"""
    # API配置
    api_key = "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"
    url = "https://openrouter.ai/api/v1/chat/completions"
    model = "arcee-ai/trinity-large-preview:free"
    
    # 请求体
    payload = {
        "model": model,
        "messages": [
            {
                "role": "system",
                "content": "中文对话"
            },
            {
                "role": "user",
                "content": "你好"
            }
        ],
        "temperature": 0.7
    }
    
    # 请求头
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }
    
    try:
        # 发送请求
        response = requests.post(url, json=payload, headers=headers)
        
        # 检查响应状态码
        if response.status_code == 200:
            # 解析响应
            result = response.json()
            print("API调用成功！")
            print(f"响应时间: {response.elapsed.total_seconds():.2f}秒")
            print(f"生成的内容: {result['choices'][0]['message']['content']}")
            return True
        else:
            print(f"API调用失败，状态码: {response.status_code}")
            print(f"错误信息: {response.text}")
            return False
    except Exception as e:
        print(f"请求发生错误: {str(e)}")
        return False

if __name__ == "__main__":
    test_openrouter_api()