From b75a49c22e7d2b9aa8d3dc4975df8801c52b4d5b Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期三, 14 五月 2025 16:04:28 +0800
Subject: [PATCH] 修改文档拆分和实体词提取逻辑,增加实体词文本抽取

---
 knowledgebase/doc/entity_helper.py |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/knowledgebase/doc/entity_helper.py b/knowledgebase/doc/entity_helper.py
index e354449..659c5ae 100644
--- a/knowledgebase/doc/entity_helper.py
+++ b/knowledgebase/doc/entity_helper.py
@@ -38,6 +38,7 @@
                         _entity = TEntity(name=entity, type=ty, doc_type=doc_ty,
                                           prompts=obj2[doc_ty]['entities'][entity])
                         doc_dbh.add_entity(_entity)
+                        self.entities.append(_entity)
                         Log.info(f"鏂板Entity锛歿entity}锛宨d锛歿_entity.id}")
 
 

--
Gitblit v1.9.1