1、安装openai依赖
pip install openai
2、编写聊天代码
# coding: utf-8
# author: liangshiqiang
# date : 2023年04月26日import openaiopenai.proxy = 'http://127.0.0.1:10809' # 翻墙代理
openai.api_key = 'xxxxxxxxxxxxx' # openai的keymessages = []def add_message(content, role='user'):"""添加消息到上下文。每次发送消息给openai都要携带上下文以便openai理解:param content: 上下文内容:param role: 上下文的角色。一般是三个值:user, assistant, system:return: None"""global messagesmessages.append({'role': role,'content': content})print('您好啊,我是AI机器人,你现在可以开始提问了')
while True:question = input()add_message(question) # 询问的内容加入上下文response = openai.ChatCompletion.create(model="gpt-3.5-turbo",messages=messages)choices = response.get('choices') or []if not choices:print('没有找到你想要的内容,请重新描述')continueanswer = choices[0].get('message', {}).get('content')if not answer:print('没有找到你想要的内容,请重新描述')continueadd_message(answer, 'assistant') # 回答的内容加入上下文print(answer)
3、运行: