From b6fae75cde256c9c8a84783dbd135a9bcca6ac8d Mon Sep 17 00:00:00 2001 From: Chaos Yu Date: Wed, 13 Aug 2025 17:16:58 +0800 Subject: [PATCH] fix(graph): using assignment expression to avoid repeated function call (#1174) * using assignment expression to avoid repeated function call * format --- rdagent/components/knowledge_management/graph.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/rdagent/components/knowledge_management/graph.py b/rdagent/components/knowledge_management/graph.py index 7b575e49..4c448279 100644 --- a/rdagent/components/knowledge_management/graph.py +++ b/rdagent/components/knowledge_management/graph.py @@ -136,10 +136,10 @@ class UndirectedGraph(Graph): ------- """ - if self.get_node(node.id): - node = self.get_node(node.id) - elif self.find_node(content=node.content, label=node.label): - node = self.find_node(content=node.content, label=node.label) + if tmp_node := self.get_node(node.id): + node = tmp_node + elif tmp_node := self.find_node(content=node.content, label=node.label): + node = tmp_node else: # same_node = self.semantic_search(node=node.content, similarity_threshold=same_node_threshold, topk_k=1) # if len(same_node): @@ -150,10 +150,10 @@ class UndirectedGraph(Graph): self.nodes.update({node.id: node}) if neighbor is not None: - if self.get_node(neighbor.id): - neighbor = self.get_node(neighbor.id) - elif self.find_node(content=neighbor.content, label=node.label): - neighbor = self.find_node(content=neighbor.content, label=node.label) + if tmp_neighbor := self.get_node(neighbor.id): + neighbor = tmp_neighbor + elif tmp_neighbor := self.find_node(content=neighbor.content, label=node.label): + neighbor = tmp_neighbor else: # same_node = self.semantic_search(node=neighbor.content, # similarity_threshold=same_node_threshold, topk_k=1)