bakaEC/olivebot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update nlp_ChatGLM2.py

Browse Source 
修正查库顺序
This commit is contained in:
xszyou 2023-08-24 09:59:10 +08:00
parent c01256e8dc
commit 8a4beb9d7f
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ai_module/nlp_ChatGLM2.py
View File

@ -1,11 +1,14 @@
import json import json
import requests import requests
from core.content_db import Content_Db from core.content_db import Content_Db
content_db = Content_Db()
list = content_db.get_list('all','desc',10)
answer_info = dict() def question(cont):
chat_list = [] content_db = Content_Db()
for val in list: list = content_db.get_list('all','desc',10)
answer_info = dict()
chat_list = []
for val in list:
answer_info = dict() answer_info = dict()
if val[0] == "member": if val[0] == "member":
answer_info["role"] = "user" answer_info["role"] = "user"
@ -14,13 +17,10 @@ for val in list:
answer_info["role"] = "bot" answer_info["role"] = "bot"
answer_info["content"] = val[2] answer_info["content"] = val[2]
chat_list.append(answer_info) chat_list.append(answer_info)
def question(cont):
print(chat_list)
content = { content = {
"prompt":"请简单回复我。" + cont, "prompt":"请简单回复我。" + cont,
"history":chat_list} "history":chat_list}
url = "http://192.168.1.23:8000" url = "http://127.0.0.1:8000"
req = json.dumps(content) req = json.dumps(content)
headers = {'content-type': 'application/json'} headers = {'content-type': 'application/json'}
r = requests.post(url, headers=headers, data=req) r = requests.post(url, headers=headers, data=req)