Optimize relationship title to entity1 and entity2

This commit is contained in:
yangdx
2025-05-07 13:02:22 +08:00
parent 1e03888cef
commit 59771b60df

View File

@@ -1208,8 +1208,7 @@ async def mix_kg_vector_query(
return PROMPTS["fail_response"] return PROMPTS["fail_response"]
if query_param.only_need_context: if query_param.only_need_context:
context_str = f""" context_str = f"""\r\n\r\n-----Knowledge Graph Context-----\r\n\r\n
\r\n\r\n-----Knowledge Graph Context-----\r\n\r\n
{kg_context if kg_context else "No relevant knowledge graph information found"} {kg_context if kg_context else "No relevant knowledge graph information found"}
\r\n\r\n-----Vector Context-----\r\n\r\n \r\n\r\n-----Vector Context-----\r\n\r\n
@@ -1554,8 +1553,8 @@ async def _get_node_data(
relations_section_list = [ relations_section_list = [
[ [
"id", "id",
"source", "entity1",
"target", "entity2",
"description", "description",
"keywords", "keywords",
"weight", "weight",
@@ -1839,8 +1838,8 @@ async def _get_edge_data(
relations_section_list = [ relations_section_list = [
[ [
"id", "id",
"source", "entity1",
"target", "entity2",
"description", "description",
"keywords", "keywords",
"weight", "weight",