class TestMe(TestCase):
def setUp(self):
self.database = Database(password="******", userid="A0001")
# self.db2 = Database(password="******", userid="A0002")
# self.db3 = Database(password="******", userid="A0003")
def test_add_userdata(self):
"""Add userdata from usb.
"""
# path = "D:/新知识库"
# walker = WalkUserData(db=self.database)
# fnamelist = walker.dir_process(1, path, style="fnamelist")
pass
def test_delete(self):
pass
def test_reset(self):
self.database.delete(pattern='n', label='Config') # 删除知识库配置
self.database.reset(pattern="n",
label='NluCell',
filename="chat_context.xls")
# self.database.reset(pattern="n", label='NluCell', filename="chat.xls")
# self.db2.handle_excel("chat_bank.xls")
# self.db3.handle_excel("chat_hospital.xls")
pass
def test_reset_ts(self):
"""Reset data of label 'TestStandard' in database.
"""
# self.database.reset_ts(pattern="n", label="TestStandard", filename="C:/nlu/data/kb/ts.xls")
pass
def test_add_ts(self):
# self.database.handle_ts("C:/nlu/data/kb/ts.xls")
pass
# @time_me(format_string="ms")
def test_add_qa(self):
# 1.Add qa with excel
# self.db2.handle_excel("chat_bank.xls")
# self.db3.handle_excel("chat_hospital.xls")
# 2.Add qa with txt
# self.database.handle_txt("C:/nlu/data/kb/bank.txt")
pass
def test_download(self):
# akbs = self.database.get_available_kb()
# self.database.download(filename="全部.xls", names=akbs)
# self.database.download(filename="银行业务.xls", names=["银行业务"])
# self.database.download_scene(filename="理财产品.xls", topic="理财产品")
pass
def test_generate_test_cases(self):
# self.database.generate_test_cases(
# filename="chat.xls",
# custom_sheets=["银行业务"],
# savedir="."
# )
pass
def setUp(self):
self.database = Database(password="******", userid="A0001")
self.db2 = Database(password="******", userid="A0002")
self.db3 = Database(password="******", userid="A0003")
The socketserver module simplifies the task of writing network servers.
"""
import os
import json
import socketserver
from chat.config import getConfig
from chat.mytools import Walk, get_current_time
from chat.graph import Database
# from .sql import Database
# from .qa_sql import Robot
from chat.ianswer import answer2xml
from chat.qa import Robot
kb = Database(password=getConfig("neo4j", "password"))
class WalkUserData(Walk):
def handle_file(self, filepath, pattern=None):
kb.handle_excel(filepath)
def add_qa(path=None, names=None):
"""Add subgraph from excel data.
"""
walker = WalkUserData()
print(path)
fnamelist = walker.dir_process(1, path, style="fnamelist")
print("知识库更新内容:", fnamelist)
# 开机从U盘自动导入知识库
add_qa(path=getConfig("path", "usbkb"))
def init_km():
"""初始化知识库
"""
database = Database(password="******", userid="A0001")
def setUp(self):
self.database = Database(userid="A0001")
self.db2 = Database(userid="A0002")
self.db3 = Database(userid="A0003")
def setUp(self):
self.database = Database(password="******", userid="neo4j")