from openai import OpenAI


# =============================godgpt提问====================================

def get_god_chat_completion(prompt,model="gpt-4o"):
    """
    gotAPI
    :param prompt:
    :return:
    """
    api_key = "sk-YYHnMAyHysBx7EOLBe6303Ab5d324903905e7c1b3f6e8e58"
    api_base = "https://api.wlai.vip/v1"
    client = OpenAI(api_key=api_key, base_url=api_base)

    completion = client.chat.completions.create(
        model=model,
        messages=[
            {"role": "system", "content": "你是一个十万粉丝的头条爆款文章创作者。"},
            {"role": "user", "content": prompt}
        ]
    )

    print(completion.choices[0].message.content)
    return completion.choices[0].message.content



def get_deepseek_chat_completion(prompt, model="deepseek-reasoner"):
    """
    deepseek问答API
    :param prompt: 提示词
    :param model: 使用的模型
    :return: 返回结果
    deepseek-reasoner 思考模型
    """
    client = OpenAI(api_key="sk-b37c17c924044f73996a77b3ef74904c", base_url="https://api.deepseek.com")

    response = client.chat.completions.create(
        model=model,
        messages=[
            {"role": "system", "content": "你是一个十万粉丝的爆款文章创作者！"},
            {"role": "user", "content": prompt},
        ],
        stream=False
    )

    print(response.choices[0].message.content)
    return response.choices[0].message.content

# =========================读取文档链接，并开始程序================================