From 74c1f1f7e13be6dec5a772aed84bd435df2521a0 Mon Sep 17 00:00:00 2001 From: aries-ckt <916701291@qq.com> Date: Tue, 23 May 2023 22:09:22 +0800 Subject: [PATCH] update:vector store config --- pilot/server/webserver.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py index 270eff67f..44c027bdf 100644 --- a/pilot/server/webserver.py +++ b/pilot/server/webserver.py @@ -19,8 +19,7 @@ from langchain import PromptTemplate ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(ROOT_PATH) -from pilot.configs.model_config import DB_SETTINGS, KNOWLEDGE_UPLOAD_ROOT_PATH, LLM_MODEL_CONFIG, VECTOR_SEARCH_TOP_K, \ - VECTOR_STORE_CONFIG +from pilot.configs.model_config import KNOWLEDGE_UPLOAD_ROOT_PATH, LLM_MODEL_CONFIG, VECTOR_SEARCH_TOP_K from pilot.server.vectordb_qa import KnownLedgeBaseQA from pilot.connections.mysql import MySQLOperator from pilot.source_embedding.knowledge_embedding import KnowledgeEmbedding @@ -268,10 +267,8 @@ def http_bot(state, mode, sql_mode, db_selector, temperature, max_new_tokens, re skip_echo_len = len(prompt.replace("", " ")) + 1 if mode == conversation_types["custome"] and not db_selector: - # persist_dir = os.path.join(KNOWLEDGE_UPLOAD_ROOT_PATH, vector_store_name["vs_name"]) - print("vector store type: ", VECTOR_STORE_CONFIG) print("vector store name: ", vector_store_name["vs_name"]) - vector_store_config = VECTOR_STORE_CONFIG + vector_store_config = [] vector_store_config["vector_store_name"] = vector_store_name["vs_name"] vector_store_config["text_field"] = "content" vector_store_config["vector_store_path"] = KNOWLEDGE_UPLOAD_ROOT_PATH