1、首先定义一个空列表用来存放历史信息
message = []
2、将用户消息加入列表末尾
message.append({
"role": "user",
"content": str
})
3、将返回的消息加入列表末尾
response = requests.request("POST", url, headers=headers, data=payload).json()
message.append({
"role": "assistant",
"content": response['result']
})
完整代码:
import requests
import json
API_KEY = "xxx" # 填上你自己的API_KEY
SECRET_KEY = "xxx" # 填上你自己的SECRET_KEY
message = []
def chat(str):
url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + get_access_token()
message.append({
"role": "user",
"content": str
})
payload = json.dumps({