将这个程序改成一个函数,可以运用到其他程序上
时间: 2024-03-10 14:45:09 浏览: 21
好的,将这个程序改成函数形式如下:
```python
def chitgpt_chatbot():
import openai
import re
# set up OpenAI API key and model ID
openai.api_key = "YOUR_API_KEY"
model_id = "YOUR_MODEL_ID"
# welcome message
print("Hi! I am ChitGPT, a chatbot. How can I assist you today?")
while True:
# get user input
user_input = input("You: ")
# check if user wants to exit
if re.search(r'exit|quit', user_input, re.IGNORECASE):
print("ChitGPT: Goodbye!")
break
# generate response from OpenAI model
prompt = f"Conversation with ChitGPT:\nUser: {user_input}\nChitGPT:"
response = openai.Completion.create(
engine=model_id,
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
# print response
chitgpt_response = response.choices[0].text.strip()
print("ChitGPT:", chitgpt_response)
```
你可以在其他程序中调用这个函数以启动 ChitGPT 聊天机器人。请注意,在使用这个函数之前,你需要将 OpenAI API key 和模型 ID 替换为自己的。