From 22f370322412074174cde20ecfd14ec03657ab63 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期一, 07 七月 2025 16:20:25 +0800
Subject: [PATCH] 生成数据库
---
knowledgebase/doc/entity_helper.py | 18 ++++++++++--------
1 files changed, 10 insertions(+), 8 deletions(-)
diff --git a/knowledgebase/doc/entity_helper.py b/knowledgebase/doc/entity_helper.py
index 659c5ae..afe0cc4 100644
--- a/knowledgebase/doc/entity_helper.py
+++ b/knowledgebase/doc/entity_helper.py
@@ -15,15 +15,11 @@
class EntityHelper:
# 鏂囨。绫诲瀷鍜岃瘑鍒彁绀鸿瘝map
doc_prompt_map: dict
- # 鎵�鏈夊疄浣�
- entities: list[TEntity]
def __init__(self):
Log.info("鍒濆鍖朎ntityHelper")
current_dir = os.path.dirname(__file__)
- self.entities = doc_dbh.get_all_entities()
self.doc_prompt_map = {}
- entity_names = [entity.name for entity in self.entities]
with open(f'{current_dir}/../../tpl/entities.json', 'r', encoding='utf-8') as f:
text = f.read()
obj = json.loads(text)
@@ -33,13 +29,19 @@
prompts = obj2[doc_ty]['prompts']
self.doc_prompt_map[doc_ty] = prompts
for entity in obj2[doc_ty]['entities']:
- if entity in entity_names:
- continue
_entity = TEntity(name=entity, type=ty, doc_type=doc_ty,
prompts=obj2[doc_ty]['entities'][entity])
+ if doc_dbh.get_entity(_entity):
+ continue
doc_dbh.add_entity(_entity)
- self.entities.append(_entity)
Log.info(f"鏂板Entity锛歿entity}锛宨d锛歿_entity.id}")
+_entity_helper:EntityHelper = None
-entity_helper = EntityHelper()
+def get_entity_helper():
+ global _entity_helper
+ return _entity_helper
+
+def init_entity_helper():
+ global _entity_helper
+ _entity_helper = EntityHelper()
--
Gitblit v1.9.1