From 4c8228350bf0d3c3c8fc1e51dc97fde6063dfbc5 Mon Sep 17 00:00:00 2001
From: million <wanguo.2004.1@163.com>
Date: Fri, 22 Nov 2024 16:31:05 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=84=E7=90=86=E8=BF=94=E5=9B=9E=E5=A4=9A?=
 =?UTF-8?q?=E4=B8=AA=E4=BA=BA=E7=89=A9=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 entity_extraction_process.py | 12 +++++++-----
 prompt.py                    |  5 +++--
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/entity_extraction_process.py b/entity_extraction_process.py
index 99b8a89..8113565 100644
--- a/entity_extraction_process.py
+++ b/entity_extraction_process.py
@@ -41,10 +41,12 @@ class entityExtractionProcess:
         result = json.loads(response)
 
         # 补全其他信息
-        result["sys_id"] = str(uuid.uuid4())
-        result["entity_code"] = str(uuid.uuid4())
-        result["sys_version"] = 1
-        result["sys_create_time"] = str(datetime.now())
-        result["sys_update_time"] = None
+        if len(result) > 0:
+            for r in result:
+                r["sys_id"] = str(uuid.uuid4())
+                r["entity_code"] = str(uuid.uuid4())
+                r["sys_version"] = 1
+                r["sys_create_time"] = str(datetime.now())
+                r["sys_update_time"] = None
 
         return result
diff --git a/prompt.py b/prompt.py
index f2b76fb..f6ae485 100644
--- a/prompt.py
+++ b/prompt.py
@@ -1,4 +1,4 @@
-ENTITY_EXTRACT = '''
+ENTITY_EXTRACT = """
 你是一个人物信息提取机器人，你的任务是根据用户输入的一段文本，从中抽取主要人物的信息并将其转换为标准json格式的输出返回给用户。
 输出的标准json键值定义如下：
 "full_name": 实体人物的姓名，其中应包括人物的本名与别名，值数据结构应为 list<str>。
@@ -17,4 +17,5 @@ ENTITY_EXTRACT = '''
 不允许返回除以上规定键值对以外的额外内容。返回内容应严格包含所陈列的键值对。
 你只需返回要求的json，禁止返回任何额外的分析与陈述。同时禁止用任何额外字符包裹所返回的json。
 应尽可能多的利用用户输入的文本填充对应的键值对。
-'''
\ No newline at end of file
+无论能提取到一个还是多个人物，都返回数组格式，如果未能提取到人物的话则返回空数组
+"""