From 59771b60df0363b3edd3756faff1d673db7d0150 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 7 May 2025 13:02:22 +0800 Subject: [PATCH] Optimize relationship title to entity1 and entity2 --- lightrag/operate.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/lightrag/operate.py b/lightrag/operate.py index ce1e76f6..2a4137a0 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -1208,8 +1208,7 @@ async def mix_kg_vector_query( return PROMPTS["fail_response"] if query_param.only_need_context: - context_str = f""" -\r\n\r\n-----Knowledge Graph Context-----\r\n\r\n + context_str = f"""\r\n\r\n-----Knowledge Graph Context-----\r\n\r\n {kg_context if kg_context else "No relevant knowledge graph information found"} \r\n\r\n-----Vector Context-----\r\n\r\n @@ -1554,8 +1553,8 @@ async def _get_node_data( relations_section_list = [ [ "id", - "source", - "target", + "entity1", + "entity2", "description", "keywords", "weight", @@ -1839,8 +1838,8 @@ async def _get_edge_data( relations_section_list = [ [ "id", - "source", - "target", + "entity1", + "entity2", "description", "keywords", "weight",