bug fix

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)

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")