|
@@ -77,16 +77,22 @@ class AskForInfo:
|
|
|
trace_id = res["trace_id"]
|
|
|
kimi_title = res['kimi_title']
|
|
|
video_id = res['search_video_id']
|
|
|
+ video_url = res['video_url']
|
|
|
res_obj = {
|
|
|
"search_videos": "success" if video_id else "fail",
|
|
|
"trace_id": trace_id,
|
|
|
- "video_list": [video_id] if video_id else []
|
|
|
+ "video_list": [video_id] if video_id else [],
|
|
|
+ "video_url": [video_url] if video_url else ""
|
|
|
}
|
|
|
time.sleep(2)
|
|
|
final_obj = await self.ask_for_info(res_obj=res_obj, kt=kimi_title)
|
|
|
e = time.time()
|
|
|
print("Rank and Return Time:\t", e - b - 2)
|
|
|
- return final_obj
|
|
|
+ if final_obj['videoUrl']:
|
|
|
+ return final_obj
|
|
|
+ else:
|
|
|
+ final_obj["videoUrl"] = video_url
|
|
|
+ return final_obj
|
|
|
elif res['code'] == 1:
|
|
|
return {
|
|
|
"Error": res['message']
|