olivebot/llm/nlp_ChatGLM3.py

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("你叫什么名字")