修改大模型生成json逻辑

lyg

2025-05-14 37c40c84aa27ff68f6dc7325fb45c9a8c7b70fe8

 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()