36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import json
|
|
import requests
|
|
from core import content_db
|
|
|
|
|
|
def question(cont, uid=0, observation=""):
|
|
contentdb = content_db.new_instance()
|
|
if uid == 0:
|
|
list = contentdb.get_list('all','desc', 11)
|
|
else:
|
|
list = contentdb.get_list('all','desc', 11, uid)
|
|
answer_info = dict()
|
|
chat_list = []
|
|
i = len(list)-1
|
|
while i >= 0:
|
|
answer_info = dict()
|
|
if list[i][0] == "member":
|
|
answer_info["role"] = "user"
|
|
answer_info["content"] = list[i][2]
|
|
elif list[i][0] == "fay":
|
|
answer_info["role"] = "bot"
|
|
answer_info["content"] = list[i][2]
|
|
chat_list.append(answer_info)
|
|
i -= 1
|
|
content = {
|
|
"prompt":"请简单回复我。" + cont,
|
|
"history":chat_list}
|
|
url = "http://127.0.0.1:8000/v1/completions"
|
|
req = json.dumps(content)
|
|
headers = {'content-type': 'application/json'}
|
|
r = requests.post(url, headers=headers, data=req)
|
|
res = json.loads(r.text).get('response')
|
|
return req
|
|
|
|
if __name__ == "__main__":
|
|
question("你叫什么名字") |