官方 openai 库中使用

import openai

def query_gpt4(question):
    openai.api_key = "sk-xxx"
    #openai.base_url = url
    openai.base_url = 'https://4.0.wokaai.com/v1/'


    try:
        response = openai.chat.completions.create(
            model="gpt-4",  # 确认使用 GPT-4 模型
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": question}
            ]
        )
        print(response)
        return response['choices'][0].message['content']
    except Exception as e:
        return str(e)

# 问题
question = "为什么太阳那么红？"

# 获取并打印回答
answer = query_gpt4(question)
print(answer)

# 使用方法
若出现module 'openai' has no attribute 'chat'报错，请更新openai库：pip3 install --upgrade openai
若出现module 'attr' has no attribute 'frozen'报错，请更新输入：pip3 install --upgrade attrs