Przeglądaj źródła

refactor(模型/评估): 更新EvaluateWorkflowParam字段并添加日志记录

将video_url和channel_video_url字段替换为channel_content_id和result
在评估流程中添加日志记录评估数据
max_liu 2 dni temu
rodzic
commit
f680e6012f
2 zmienionych plików z 5 dodań i 3 usunięć
  1. 3 1
      main.py
  2. 2 2
      utils/params.py

+ 3 - 1
main.py

@@ -108,6 +108,8 @@ def get_decode_result(param: TaskStatusParam):
 def evaluate_video(param:EvaluateListParam):
 
     evaluate_list = param.evaluate_list
+    
+    logger.info(f"创建评估数据 = {evaluate_list}")
 
     data_list = []
     for evaluate in evaluate_list:
@@ -117,7 +119,7 @@ def evaluate_video(param:EvaluateListParam):
         EvaluateRecord(
             evaluate_id=evaluate_id,
             task_id=task_id,
-            search_result= json.dumps(evaluate.search_result),
+            search_result= json.dumps(evaluate.result),
             evaluate_result=None,   
             error_reason=None,
             status = 0

+ 2 - 2
utils/params.py

@@ -21,9 +21,9 @@ class DecodeListParam(BaseModel):
 
 class EvaluateWorkflowParam(BaseModel):
     video_id: str
-    video_url: str
+    channel_content_id: str
     task_id: str
     status: int
-    channel_video_url: str
+    result: str
 class EvaluateListParam(BaseModel):
     evaluate_list: List[EvaluateWorkflowParam]