From e60d75228fb161e464ca59fa2526bf0765f4d902 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期四, 22 五月 2025 12:35:55 +0800
Subject: [PATCH] 修改指令json生成，加入fastapi
---
knowledgebase/doc/entity_helper.py | 9 ++-------
1 files changed, 2 insertions(+), 7 deletions(-)
diff --git a/knowledgebase/doc/entity_helper.py b/knowledgebase/doc/entity_helper.py
index e354449..219eed2 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,12 +29,11 @@
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)
Log.info(f"鏂板Entity锛歿entity}锛宨d锛歿_entity.id}")
-
entity_helper = EntityHelper()
--
Gitblit v1.9.1