Spaces:
Running
Running
File size: 791 Bytes
5e9cd1d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from server.knowledge_base.kb_service.faiss_kb_service import FaissKBService
from server.knowledge_base.kb_service.pg_kb_service import PGKBService
from server.knowledge_base.migrate import create_tables
from server.knowledge_base.utils import KnowledgeFile
kbService = PGKBService("test")
test_kb_name = "test"
test_file_name = "README.md"
testKnowledgeFile = KnowledgeFile(test_file_name, test_kb_name)
search_content = "如何启动api服务"
def test_init():
create_tables()
def test_create_db():
assert kbService.create_kb()
def test_add_doc():
assert kbService.add_doc(testKnowledgeFile)
def test_search_db():
result = kbService.search_docs(search_content)
assert len(result) > 0
def test_delete_doc():
assert kbService.delete_doc(testKnowledgeFile)
|