修改属性图节点的层级结构,新增子父级关系

This commit is contained in:
wanyaokun
2024-09-24 17:11:20 +08:00
parent e0fc5381d8
commit aace9ce292
7 changed files with 494 additions and 188 deletions
+3 -3
View File
@@ -14,6 +14,7 @@ from app.engine.response.treeSummResponse import CustomTreeResponse
from llama_index.core.settings import Settings
from llama_index.core.indices.property_graph import LLMSynonymRetriever,VectorContextRetriever
from llama_index.core import PropertyGraphIndex
from app.engine.retriever.graphKeyWordRetriever import GraphKeyWordRetriever
ModelPlateCategory = '模型平台'
@@ -122,8 +123,7 @@ def create_query_engine(index,top_k=3, use_reranker=False, filters=None, respons
llm_query = os.getenv('LLM_QUERY_WAY','rag')
if llm_query == 'graph':
graphIndex:PropertyGraphIndex = index
synonym_retriver = LLMSynonymRetriever(graphIndex.property_graph_store,
llm=Settings.llm,
keyWord_retriver = GraphKeyWordRetriever(graphIndex.property_graph_store,
include_text=False
)
if graphIndex.property_graph_store.supports_vector_queries:
@@ -137,7 +137,7 @@ def create_query_engine(index,top_k=3, use_reranker=False, filters=None, respons
include_text=False
)
retriever = graphIndex.as_retriever(sub_retrievers=[synonym_retriver,vector_retriver])
retriever = graphIndex.as_retriever(sub_retrievers=[keyWord_retriver,vector_retriver])
else:
retriever = get_Retriever(index,