test vector store on docker

2023-12-08 22:22:01 +08:00
parent 21bccf69d2
commit 0cb7dd5280
5 changed files with 43 additions and 7 deletions
--- a/crazy_functions/vector_fns/vector_database.py
+++ b/crazy_functions/vector_fns/vector_database.py
@@ -242,7 +242,7 @@ class LocalDocQA:

 def construct_vector_store(vs_id, vs_path, files, sentence_size, history, one_conent, one_content_segmentation, text2vec):
    for file in files:
-        assert os.path.exists(file), "输入文件不存在"
+        assert os.path.exists(file), "输入文件不存在：" + file
    import nltk
    if NLTK_DATA_PATH not in nltk.data.path: nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path
    local_doc_qa = LocalDocQA()