导库: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'