罗俊辉 1 anno fa
parent
commit
23016e8e0a

+ 2 - 0
applications/functions/kimi.py

@@ -56,6 +56,8 @@ class KimiServer(object):
                     prompt_type="text"
                 )
             )
+            kimi_info = kimi_info.replace('```json', '').replace('```', '')
+            kimi_info = json.loads(kimi_info)
         kimi_info['k_title'] = kimi_title
         kimi_info['ori_title'] = title
         logging(

+ 8 - 3
applications/schedule/search_schedule.py

@@ -243,7 +243,7 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
     kimi_info = await K.search_kimi_schedule(params=params)
     print(json.dumps(kimi_info, ensure_ascii=False, indent=4))
     kimi_title = kimi_info['k_title']
-    content_title = kimi_info['content_title']
+    content_title = kimi_info['content_title'].replace("'", "").replace('"', "")
     content_keys = json.dumps(kimi_info['content_keys'], ensure_ascii=False)
     update_kimi_sql = f"""
                     UPDATE long_articles_video SET
@@ -265,7 +265,8 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
     if len(un_empty_list) < 3:
         await asyncio.sleep(3)
         recall_obj_4 = await SearchAB.ab_4()
-        un_empty_list.append(recall_obj_4)
+        if recall_obj_4:
+            un_empty_list.append(recall_obj_4)
 
     # 逐条下载,逐条写表
     if un_empty_list:
@@ -294,6 +295,10 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
                     """
                 await mysql_client.async_insert(update_id_sql)
     else:
-        logging(code="7003", info="视频搜索失败", trace_id=trace_id)
+        logging(
+            code="7003",
+            info="视频搜索失败, 被敏感词过滤",
+            trace_id=trace_id
+        )