"""
    response = ask("写一篇高中生的文章")
    print("豆包回答:", response)
"""


import requests
import time
from requests.exceptions import RequestException

# 配置API信息
ARK_API_KEY = "you_key"  # 替换为实际API密钥
API_URL = "https://ark.cn-beijing.volces.com/api/v3/chat/completions"

def ask(quest):
    messages = [{"role": "user", "content": quest}]
    """
    调用豆包API的通用函数
    """
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {ARK_API_KEY}"
    }

    data = {
        "model": "doubao-1-5-pro-32k-250115",
        "messages": messages
    }
    max_retries=1
    for attempt in range(max_retries):
        try:
            response = requests.post(API_URL, headers=headers, json=data)
            response.raise_for_status()  # 检查状态码
            return response.json()["choices"][0]["message"]["content"]

        except RequestException as e:
            print(f"Request failed (attempt {attempt + 1}/{max_retries}): {e}")
            if attempt < max_retries - 1:
                time.sleep(2 ** attempt)  # 指数退避重试

    raise Exception("Max retries exceeded")