ソースを参照

增加返回为空处理

zhangyong 7 ヶ月 前
コミット
e82969bc71
2 ファイル変更6 行追加2 行削除
  1. 2 1
      common/cover_gpt4o.py
  2. 4 1
      video_cover_method/cover_method.py

+ 2 - 1
common/cover_gpt4o.py

@@ -8,7 +8,8 @@ class CoverAnalyzer:
         list_data = []
         for cover in cover_list:
             data_dict = CoverAnalyzer.analyze_covers(cover)
-            list_data.append(data_dict)
+            if data_dict:
+                list_data.append(data_dict)
         return list_data
 
     @staticmethod

+ 4 - 1
video_cover_method/cover_method.py

@@ -46,6 +46,9 @@ class CoverMethod:
             # gpt 评分
             start_time = time.time()
             cover_all_data = CoverAnalyzer.cover_list_data(cover_list)
+            if not cover_all_data:
+                AliyunLogger.logging( video_id, "GPT评分失败", "3002", cover_all_data)
+                insert_pq_data( [video_id] )
             end_time = time.time()
             execution_time = end_time - start_time
             print(f"代码运行时间: {execution_time:.4f} 秒")
@@ -142,4 +145,4 @@ class CoverMethod:
 
 
 if __name__ == '__main__':
-    CoverMethod.cover_method("30479284")
+    CoverMethod.cover_method("30488175")