feat: 优化知识库问答功能

modules/models/base_model.py

@@ -373,8 +373,9 @@ class BaseLLMModel:
             msg = "索引获取成功，生成回答中……"
             logging.info(msg)
             with retrieve_proxy():
-                retriever = VectorStoreRetriever(vectorstore=index, search_type="similarity_score_threshold", search_kwargs={
-                                                 "k": 6, "score_threshold": 0.5})
+                retriever = VectorStoreRetriever(vectorstore=index, search_type="similarity", search_kwargs={"k": 6})
+                # retriever = VectorStoreRetriever(vectorstore=index, search_type="similarity_score_threshold", search_kwargs={
+                #                                  "k": 6, "score_threshold": 0.2})
                 try:
                     relevant_documents = retriever.get_relevant_documents(
                         fake_inputs)

requirements.txt

@@ -7,7 +7,7 @@ tqdm
 colorama
 googlesearch-python
 Pygments
-langchain==0.0.276
+langchain==0.0.316
 markdown
 PyPDF2
 pdfplumber
@@ -21,7 +21,7 @@ duckduckgo-search>=3.9.5
 arxiv
 wikipedia
 google.generativeai
-openai>=0.27.9
+openai==0.28.1
 unstructured
 google-api-python-client
 tabulate