From c099e6662b8a6e320ac314d31eda9b40455e5aa7 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期四, 22 五月 2025 09:27:37 +0800
Subject: [PATCH] 修改指令json生成相关提示词和代码逻辑
---
knowledgebase/doc/entity_recognition.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/knowledgebase/doc/entity_recognition.py b/knowledgebase/doc/entity_recognition.py
index 1144aab..f95fb8f 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 = """
--
Gitblit v1.9.1