From 37c40c84aa27ff68f6dc7325fb45c9a8c7b70fe8 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期三, 14 五月 2025 14:21:40 +0800
Subject: [PATCH] 修改大模型生成json逻辑
---
knowledgebase/db/doc_db_helper.py | 12 ++++++++++++
1 files changed, 12 insertions(+), 0 deletions(-)
diff --git a/knowledgebase/db/doc_db_helper.py b/knowledgebase/db/doc_db_helper.py
index 5089e30..d1231cc 100644
--- a/knowledgebase/db/doc_db_helper.py
+++ b/knowledgebase/db/doc_db_helper.py
@@ -17,6 +17,7 @@
"""
鏂囨。鏁版嵁搴撳姪鎵�
"""
+
def __init__(self):
self.session = init_doc_db()
@@ -105,6 +106,17 @@
def get_docs(self) -> list[TDoc]:
return self.session.query(TDoc).all()
+ def get_text_with_entities(self, entity_names: list[str]) -> str:
+ """
+ 鏍规嵁瀹炰綋璇嶈幏鍙栨枃鏈唴瀹�
+ :param entity_names: list[str] - 瀹炰綋璇�
+ :return: str - 鏂囨湰
+ """
+ if not entity_names:
+ return ""
+
+ return '\n'.join([entity.name for entity in self.get_all_entities() if entity.name in entity_names])
+
def commit(self):
self.session.commit()
--
Gitblit v1.9.1