update

app.py

@@ -55,8 +55,10 @@ def title_generate_main():
         receipt_handle_list = []
         try:
             # 长轮询消费消息。
+            start_time0 = time.time()
             recv_msgs = consumer.consume_message(batch, wait_seconds)
             for msg in recv_msgs:
+                start_time = time.time()
                 receive_info = {
                     'messageId': msg.message_id,
                     'messageBodyMD5': msg.message_body_md5,
@@ -92,12 +94,12 @@ def title_generate_main():
                     retry_count = redis_helper.get_data_from_redis(key_name=key_name)
                     log_.error({'errorType': 'ConnectionResetError', 'videoId': video_id, 'retryCount': retry_count})
                     if retry_count is not None and retry_count == config_.RETRY_MAX_COUNT:
-                        # 确认消息消费成功
+                        # 添加消息handle到ack_message列表
                         receipt_handle_list.append(msg.receipt_handle)
                         pass
 
                 except Exception:
-                    # 确认消息消费成功
+                    # 添加消息handle到ack_message列表
                     receipt_handle_list.append(msg.receipt_handle)
                     log_.error({'videoId': video_id, 'traceback': traceback.format_exc()})
                 else:
@@ -117,7 +119,8 @@ def title_generate_main():
                                 'status': 'success',
                                 'messageID': re_msg.message_id,
                                 'bodyMD5': re_msg.message_body_md5,
-                                'messageContent': msg_content
+                                'messageContent': msg_content,
+                                'executeTime': (time.time() - start_time) * 1000
                             }
                         })
                     except MQExceptionBase as publish_e:
@@ -131,7 +134,7 @@ def title_generate_main():
                         if publish_e.type == "TopicNotExist":
                             sys.exit(1)
 
-                    # 2. 确认消息消费成功
+                    # 2. 添加消息handle到ack_message列表
                     receipt_handle_list.append(msg.receipt_handle)
 
             # msg.next_consume_time前若不确认消息消费成功，则消息会被重复消费。
@@ -141,7 +144,8 @@ def title_generate_main():
             log_.info({
                 'ackMessage': {
                     'status': 'success',
-                    'receiptHandleList': receipt_handle_list
+                    'receiptHandleList': receipt_handle_list,
+                    'executeTime': (time.time() - start_time0) * 1000
                 }
             })
         # except MQExceptionBase as consume_e:

gpt_process.py

@@ -37,7 +37,7 @@ def request_gpt(prompt):
     # print(response.json()['choices'][0]['message']['content'])
     # print('\n')
     # result_content = response.json()['choices'][0]['message']['content']
-    log_.info(f"response.text: {response.text}")
+    # log_.info(f"response.text: {response.text}")
     res_data = json.loads(response.text)
     result_content = res_data['data']['choices'][0]['message']['content']
     return result_content
@@ -53,11 +53,11 @@ def title_generate(video_id, video_path):
     # 1. 下载视频
     # log_.info(f"debug: title_generate 1")
     video_file_path = download_video(video_path=video_path, video_id=video_id, download_folder='videos')
-    log_.info({'videoId': video_id, 'video_file_path': video_file_path})
+    # log_.info({'videoId': video_id, 'video_file_path': video_file_path})
     # 2. 获取视频中的音频
     # log_.info(f"debug: title_generate 2")
     audio_path = get_wav(video_path=video_file_path)
-    log_.info({'videoId': video_id, 'audio_path': audio_path})
+    # log_.info({'videoId': video_id, 'audio_path': audio_path})
     # 3. asr
     # log_.info(f"debug: title_generate 3")
     dialogue_path, asr_res = call_asr(audio_path=audio_path)