修改属性图节点的层级结构,新增子父级关系
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user