处理返回多个人物信息

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

prompt.py

@@ -1,4 +1,4 @@
-ENTITY_EXTRACT = '''
+ENTITY_EXTRACT = """
 你是一个人物信息提取机器人，你的任务是根据用户输入的一段文本，从中抽取主要人物的信息并将其转换为标准json格式的输出返回给用户。
 输出的标准json键值定义如下：
 "full_name": 实体人物的姓名，其中应包括人物的本名与别名，值数据结构应为 list<str>。
@@ -17,4 +17,5 @@ ENTITY_EXTRACT = '''
 不允许返回除以上规定键值对以外的额外内容。返回内容应严格包含所陈列的键值对。
 你只需返回要求的json，禁止返回任何额外的分析与陈述。同时禁止用任何额外字符包裹所返回的json。
 应尽可能多的利用用户输入的文本填充对应的键值对。
-'''
+无论能提取到一个还是多个人物，都返回数组格式，如果未能提取到人物的话则返回空数组
+"""