update local_doc_qa.py

chains/local_doc_qa.py

@@ -10,6 +10,7 @@ from langchain.docstore.document import Document
 import numpy as np
 from utils import torch_gc
 from tqdm import tqdm
+from pypinyin import lazy_pinyin
 
 
 DEVICE_ = EMBEDDING_DEVICE
@@ -187,7 +188,7 @@ class LocalDocQA:
             else:
                 if not vs_path:
                     vs_path = os.path.join(VS_ROOT_PATH,
-                                           f"""{os.path.splitext(file)[0]}_FAISS_{datetime.datetime.now().strftime("%Y%m%d_%H%M%S")}""")
+                                           f"""{"".join(lazy_pinyin(os.path.splitext(file)[0]))}_FAISS_{datetime.datetime.now().strftime("%Y%m%d_%H%M%S")}""")
                 vector_store = FAISS.from_documents(docs, self.embeddings)
                 torch_gc()
 

requirements.txt

@@ -13,4 +13,5 @@ gradio==3.28.3
 fastapi
 uvicorn
 peft
+pypinyin
 #detectron2@git+https://github.com/facebookresearch/detectron2.git@v0.6#egg=detectron2