将属性图流程同步至主流程中

backend/app/engine/vectordb.py

@@ -8,11 +8,12 @@ qclient = None
 
 def get_qdrant_vector_store(docType:str):
     collection_name = docType
+    llm_query = os.getenv('LLM_QUERY_WAY','rag')
     vector_store_path = os.getenv("VECTOR_STORE_PATH")
     host=os.getenv("VECTOR_STORE_HOST", "127.0.0.1"),
     port=int(os.getenv("VECTOR_STORE_PORT", "6333")),
 
-    vector_store_path =os.path.join(vector_store_path,docType)
+    vector_store_path =os.path.join(vector_store_path,llm_query,docType)
     if not vector_store_path or not host:
         raise ValueError(
             "Please provide either VECTOR_STORE_PATH or VECTOR_STORE_HOST and VECTOR_STORE_PORT"
@@ -36,7 +37,8 @@ def get_qdrant_vector_store(docType:str):
 
 def get_chroma_vector_store(docType:str):
     collection_name = docType
-    vector_store_path =os.path.join(os.getenv("VECTOR_STORE_PATH"),docType)
+    llm_query = os.getenv('LLM_QUERY_WAY','rag')
+    vector_store_path =os.path.join(os.getenv("VECTOR_STORE_PATH"),llm_query,docType)
     # if VECTOR_STORE_PATH is set, use a local ChromaVectorStore from the path
     # otherwise, use a remote ChromaVectorStore (ChromaDB Cloud is not supported yet)
     if vector_store_path:
@@ -59,7 +61,6 @@ def get_chroma_vector_store(docType:str):
 
 def get_vector_store(docType:str):
     store_type=os.getenv("VECTOR_STORE_TYPE")
-
     store = None
 
     match store_type:
@@ -72,7 +73,6 @@ def get_vector_store(docType:str):
 
     return store
 
-
 def get_Neo4j_Graph_Store(docType:str):
     neo4jStore = Neo4jPropertyGraphStore(
         username= os.getenv('NEO4J_USERNAME'),