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_recognition.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/knowledgebase/doc/entity_recognition.py b/knowledgebase/doc/entity_recognition.py
index 1144aab..6366f8f 100644
--- a/knowledgebase/doc/entity_recognition.py
+++ b/knowledgebase/doc/entity_recognition.py
@@ -11,7 +11,7 @@
import json
from knowledgebase import utils
-from knowledgebase.doc.entity_helper import entity_helper
+from knowledgebase.db.doc_db_helper import doc_dbh
from knowledgebase.log import Log
llm = ChatOpenAI(temperature=0,
@@ -31,7 +31,7 @@
def __init__(self, doc_type: str):
# 瀹炰綋璇嶅垪琛�
- entities = list(filter(lambda x: x.doc_type == doc_type, entity_helper.entities))
+ entities = doc_dbh.get_entities_by_doc_type(doc_type)
entity_list = '锛�'.join([entity.name for entity in entities]) + "銆�"
entity_rules = "锛沑n".join([f"- {entity.name}锛歿entity.prompts}" for entity in entities]) + "銆�"
tpl = """
@@ -74,7 +74,7 @@
淇濆瓨缂撳瓨銆�
"""
text = json.dumps(self.cache)
- utils.save_to_file(text, self.cache_file)
+ utils.save_text_to_file(text, self.cache_file)
def run(self, in_text: str) -> list[str]:
"""
--
Gitblit v1.9.1