import os
import requests

# 配置 API Key 和 URL
# API_KEY = "473ac711-84a6-4899-b35b-672e4333e269" # 从环境变量读取 API Key
API_KEY = "e0430cc0-a75f-4398-bcb2-c455344fd837"
BASE_URL = "https://ark.cn-beijing.volces.com/api/v3/chat/completions"

# 构造请求头和数据
def ask_question(question):
    headers = {
        "Authorization": f"Bearer {API_KEY}",  # 使用 Bearer Token 授权方式 [[1]]
        "Content-Type": "application/json"
    }
    
    data = {
        # "model": "doubao-lite-32k-240828",  # 模型名称，推荐使用 32k 版本 [[4]]
        "model": "doubao-1-5-pro-32k-250115",
        "messages": [
            {"role": "user", "content": question}
        ],
        "max_tokens": 4096,  # 控制返回的最大 token 数
        "top_p": 0.7,
        # "temperature": 0.7,  # 控制随机性
        "stream": False      # 禁用流式传输，确保一次性返回完整结果 [[10]]
    }

    # 发起 POST 请求
    response = requests.post(BASE_URL, headers=headers, json=data)
    
    # 检查响应状态码
    if response.status_code == 200:
        result = response.json()
        return result.get("choices", [{}])[0].get("message", {}).get("content", "")
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None

# 测试提问
if __name__ == "__main__":
    question = "Python如何实现多线程？"
    answer = ask_question(question)
    if answer:
        print(f"问题: {question}")
        print(f"回答: {answer}")