36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
|
import json
|
||
|
import requests
|
||
|
from core import content_db
|
||
|
|
||
|
|
||
|
def question(cont, uid=0):
|
||
|
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("你叫什么名字")
|