封装成一个Function

好的，可以封装成一个Python函数，如下所示：

import openai
import json

def chitgpt(query):
    # 将你的API Key替换成你自己的
    openai.api_key = "YOUR_API_KEY"

    # 请求参数
    prompt = f"User: {query}\nChitGPT:"

    # GPT-3模型参数
    model_engine = "text-davinci-002"
    temperature = 0.7
    max_tokens = 100

    # 发送请求
    response = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        temperature=temperature,
        max_tokens=max_tokens
    )

    # 获取回复文本
    answer = response.choices[0].text.strip()

    return answer

使用时，只需要调用 `chitgpt(query)` 函数并传入用户的问题，即可获取AI的回答。例如：

answer = chitgpt("你会说中文吗？")
print(answer)

输出：

ChitGPT: 我会说中文呢，你有什么问题需要我解答吗？