帮我封装成一个python函数
时间: 2023-07-07 13:39:53 浏览: 115
当然可以,可以封装成一个Python函数,如下所示:
```python
import openai
import re
from typing import List
openai.api_key = "YOUR_API_KEY"
def call_chatgpt(prompt: str, temperature: float = 0.5, max_tokens: int = 50, n: int = 1, stop: str = None) -> List[str]:
completions = openai.Completion.create(
engine="text-davinci-002",
prompt=prompt,
temperature=temperature,
max_tokens=max_tokens,
n=n,
stop=stop
)
message = completions.choices[0].text.strip()
message = re.sub('[^0-9a-zA-Z\u4e00-\u9fa5.,,。?“”]+', '', message)
return message
```
其中,`YOUR_API_KEY`需要替换为你自己的OpenAI API key。这个函数会使用OpenAI的API接口,调用ChatGPT进行对话,并返回ChatGPT的回答。你可以将问题输入到这个函数中,然后获得ChatGPT的回答。
阅读全文