From 06c48ac277320cbf4c217c921af38af4244e4585 Mon Sep 17 00:00:00 2001
From: million <wanguo.2004.1@163.com>
Date: Fri, 22 Nov 2024 11:23:46 +0800
Subject: [PATCH] bug fix

---
 entity_extraction_process.py | 13 +++++++++++--
 main.py                      |  8 ++++++--
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/entity_extraction_process.py b/entity_extraction_process.py
index 48d6758..1cd1ce0 100644
--- a/entity_extraction_process.py
+++ b/entity_extraction_process.py
@@ -5,7 +5,8 @@ import time
 from openai import OpenAI
 from datetime import datetime
 
-import config, debug, prompt, json
+import config, debug, prompt, json, uuid
+from utils.logger import logger
 
 """
 备注：
@@ -34,6 +35,14 @@ class entityExtractionProcess:
             .choices[0]
             .message.content
         )
-        print(f"本次输出花费时间：{time.time() - timenow} 秒")
+
+        # 补全其他信息
+        response["sys_id"] = uuid.uuid4()
+        response["entity_code"] = uuid.uuid4()
+        response["sys_version"] = 1
+        response["sys_create_time"] = datetime.datetime.now()
+        response["sys_update_time"] = None
+
+        logger.warning(f"本次输出花费时间：{time.time() - timenow} 秒")
 
         return json.loads(response)
diff --git a/main.py b/main.py
index 85f768a..52f8a52 100644
--- a/main.py
+++ b/main.py
@@ -25,8 +25,12 @@ class Main(QueueProcessor):
         inputData = json.loads(input["inputData"])
         match currentEngineId:
             case 3000:  # InfoExtractor 实体信息提取
+                logger.warning(f"get queueing inputData= {inputData}")
                 input_text = inputData["text"]
-                return entityExtractionProcess.entity_extract(input_text)
+                result = entityExtractionProcess.entity_extract(input_text)
+
+                logger.warning(f"process result= {result}")
+                return result
 
 
 if __name__ == "__main__":
@@ -51,7 +55,7 @@ if __name__ == "__main__":
 
             import traceback
 
-            logger.warning(f"Crawler_NewsCN {traceback.format_exc()}")
+            logger.warning(f"Exception: {traceback.format_exc()}")
 
         finally:
             logger.warning("application completed")