import requests

# 直接测试DeepSeek API
def test_deepseek_api():
    api_key = "sk-591e36a6b1bd4b34b663b466ff22085e"
    api_base = "https://api.deepseek.com"
    model = "deepseek-chat"
    
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }
    
    payload = {
        "model": model,
        "messages": [
            {
                "role": "system",
                "content": "You are a professional translator. Translate the following text to Chinese. Keep the original meaning and tone. Do not add any additional information."
            },
            {
                "role": "user",
                "content": "Hello, how are you?"
            }
        ],
        "max_tokens": 1000,
        "temperature": 0.1
    }
    
    try:
        response = requests.post(
            f"{api_base}/chat/completions",
            headers=headers,
            json=payload,
            timeout=30
        )
        response.raise_for_status()
        
        result = response.json()
        print("API响应:", result)
        translated_text = result["choices"][0]["message"]["content"].strip()
        print(f"翻译结果: {translated_text}")
        return translated_text
    except requests.exceptions.RequestException as e:
        print(f"翻译失败: {e}")
        return None


if __name__ == "__main__":
    test_deepseek_api()