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