From a108e36f37a07a4a861fb6aafb9231de60dc1abf Mon Sep 17 00:00:00 2001 From: YM <479443481@qq.com> Date: 星期四, 15 五月 2025 18:49:12 +0800 Subject: [PATCH] Merge branch 'master' of http://182.92.203.7:2001/r/KnowledgeBase --- knowledgebase/db/neo4j.py | 19 ++++++++++++++++++- 1 files changed, 18 insertions(+), 1 deletions(-) diff --git a/knowledgebase/db/neo4j.py b/knowledgebase/db/neo4j.py index 5b9f887..b0916f4 100644 --- a/knowledgebase/db/neo4j.py +++ b/knowledgebase/db/neo4j.py @@ -24,7 +24,15 @@ entities=json.dumps(page_info.entities, ensure_ascii=False, indent=2)) self.graph.create(node) return node - + def create_trunk_node(self, trunk, entities): + """ + 鍒涘缓椤甸潰鑺傜偣 + """ + # 鍒涘缓鑺傜偣 + node = Node("Trunk", trunk=trunk, + entities=json.dumps(entities, ensure_ascii=False, indent=2)) + self.graph.create(node) + return node def create_entity_node(self, entity: str): """ 鍒涘缓瀹炰綋鑺傜偣 @@ -36,6 +44,15 @@ self.graph.create(node) return node + + def create_trunk_entity_relationship(self, page_node, entity_node): + """ + 鍒涘缓椤甸潰鍜屽疄浣撹妭鐐圭殑鍏崇郴 + """ + relationship = Relationship(page_node, "trunk_entity", entity_node) + self.graph.create(relationship) + return relationship + def create_page_entity_relationship(self, page_node, entity_node): """ 鍒涘缓椤甸潰鍜屽疄浣撹妭鐐圭殑鍏崇郴 -- Gitblit v1.9.1