Skip to content

Commit

Permalink
Update nlp_ChatGLM2.py
Browse files Browse the repository at this point in the history
修正查库顺序
  • Loading branch information
xszyou committed Aug 24, 2023
1 parent c01256e commit 8a4beb9
Showing 1 changed file with 15 additions and 15 deletions.
30 changes: 15 additions & 15 deletions ai_module/nlp_ChatGLM2.py
Original file line number Diff line number Diff line change
@@ -1,26 +1,26 @@
import json
import requests
from core.content_db import Content_Db
content_db = Content_Db()
list = content_db.get_list('all','desc',10)
answer_info = dict()
chat_list = []
for val in list:
answer_info = dict()
if val[0] == "member":
answer_info["role"] = "user"
answer_info["content"] = val[2]
elif val[0] == "fay":
answer_info["role"] = "bot"
answer_info["content"] = val[2]
chat_list.append(answer_info)


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

0 comments on commit 8a4beb9

Please sign in to comment.