导库:openai
import openai
openai.api_key = '输入你的API地址'
def welcome_message():
print("欢迎使用聊天机器人!")
print("我是一个智能聊天机器人,可以回答你的问题和进行对话。")
print("请随意提问,我会尽力提供有用的回答。\n")
def generate_reply(user_input):
messages = [
{"role": "system", "content": "你是一个聪明的机器助手"},
{"role": "user", "content": user_input},
]
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo-16k-0613',
messages=messages,
max_tokens=1000,
n=1,
stop=None,
temperature=0.7,
top_p=1.0,
frequency_penalty=0.5,
presence_penalty=0.3
)
reply = response['choices'][0]['message']['content']
return reply
# 在启动聊天机器人之前调用欢迎消息函数
welcome_message()
# 聊天机器人的主循环
def chat_loop():
while True:
user_input = input("用户输入(quit结束聊天,不满意请输入'不满意'): ")
if user_input.lower() == 'quit'
发布评论