[changes] Construct the final return structure directly.

This commit is contained in:
lanceyq
2026-03-27 12:24:52 +08:00
parent 8ba0a74473
commit 9730c5ce0f

View File

@@ -182,7 +182,7 @@ class ExtractionOrchestrator:
list[StatementEntityEdge], list[StatementEntityEdge],
list[EntityEntityEdge], list[EntityEntityEdge],
list[PerceptualEdge], list[PerceptualEdge],
dict list[DialogData]
]: ]:
""" """
运行完整的知识提取流水线(优化版:并行执行) 运行完整的知识提取流水线(优化版:并行执行)
@@ -1571,7 +1571,16 @@ class ExtractionOrchestrator:
final_entity_entity_edges = dedup_entity_entity_edges final_entity_entity_edges = dedup_entity_entity_edges
else: else:
# 正式模式:执行完整的两阶段去重 # 正式模式:执行完整的两阶段去重
dedup_result_tuple = await dedup_layers_and_merge_and_return( (
dialogue_nodes,
chunk_nodes,
statement_nodes,
final_entity_nodes,
statement_chunk_edges,
final_statement_entity_edges,
final_entity_entity_edges,
dedup_details,
) = await dedup_layers_and_merge_and_return(
dialogue_nodes, dialogue_nodes,
chunk_nodes, chunk_nodes,
statement_nodes, statement_nodes,
@@ -1585,18 +1594,6 @@ class ExtractionOrchestrator:
llm_client=self.llm_client, llm_client=self.llm_client,
) )
# 解包返回值
(
dialogue_nodes,
chunk_nodes,
statement_nodes,
final_entity_nodes,
statement_chunk_edges,
final_statement_entity_edges,
final_entity_entity_edges,
dedup_details,
) = dedup_result_tuple
# 保存去重消歧的详细记录到实例变量 # 保存去重消歧的详细记录到实例变量
self._save_dedup_details(dedup_details, entity_nodes, final_entity_nodes) self._save_dedup_details(dedup_details, entity_nodes, final_entity_nodes)