将上面的代码写为自定义函数
时间: 2023-06-14 08:05:53 浏览: 53
好的,可以封装成一个名为 `chat_with_chitgpt()` 的自定义函数,代码如下:
```python
import openai
import re
import time
openai.api_key = "YOUR_API_KEY" # 替换成你自己的 OpenAI API key
def chat_with_chitgpt(prompt):
# 设置请求参数
completions = openai.Completion.create(
engine="text-davinci-002",
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.7,
)
# 获取回复结果
message = completions.choices[0].text.strip()
message = re.sub('[^0-9a-zA-Z\u4e00-\u9fa5\.\?!,,。?!\n]+', '', message)
message = message.replace('\n', ' ')
message = message.replace(' ', ' ')
message = message.replace(' ,', ',')
message = message.replace(' .', '.')
message = message.replace(' ?', '?')
message = message.replace(' !', '!')
message = message.strip()
# 延时防止请求频率过快
time.sleep(1)
return message
```
使用方式:
```python
response = chat_with_chitgpt("你好")
print(response)
```
其中 `YOUR_API_KEY` 需要替换成你自己的 OpenAI API key。