python编写调用chatgpt代码

以下是使用Python调用ChatGPT的示例代码：

import openai
import json

openai.api_key = "YOUR_API_KEY"

def ask_gpt(prompt):
    model_engine = "davinci"
    prompt = prompt.strip()
    prompt = prompt[:1024]
    response = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )
    message = response.choices[0].text.strip()
    return message

prompt = "Hello, how are you doing today?"
response = ask_gpt(prompt)
print(response)

在上面的代码中，我们首先设置OpenAI API密钥，并定义一个`ask_gpt`函数，该函数将输入的prompt作为参数，使用OpenAI API调用ChatGPT模型生成一个回复，并返回该回复。

然后，我们定义一个prompt字符串，并调用`ask_gpt`函数，将prompt作为参数传递给该函数。最后，我们打印输出ChatGPT生成的回复。