maolin.liu commited on
Commit
a829e96
·
1 Parent(s): 7ec99bc

[bugfix]Improve code robust.

Browse files
Files changed (1) hide show
  1. consumer/asr.py +9 -3
consumer/asr.py CHANGED
@@ -1,5 +1,6 @@
1
  import base64
2
  import io
 
3
  import logging
4
  import os
5
  from pathlib import Path
@@ -78,7 +79,12 @@ class TranscribeConsumer(BasicMessageReceiver):
78
  self.send_message(message.model_dump_json())
79
 
80
  def consume(self, channel, method, properties, message):
81
- body = self.decode_message(message)
 
 
 
 
 
82
 
83
  try:
84
  validated_message = TranscribeInputMessage.model_validate(body)
@@ -94,10 +100,10 @@ class TranscribeConsumer(BasicMessageReceiver):
94
  transcribed_text = segment.text
95
  break
96
  except ValidationError as exc:
97
- logging.exception('Consume message failed: \n message: %s\n\n exception info: %s', message, exc)
98
  self.send_fail_message(body.get('uuid'), f'{exc}')
99
  except Exception as exc:
100
- logging.exception('Consume message failed: \n message: %s\n\n exception info: %s', message, exc)
101
  self.send_fail_message(body.get('uuid'), f'{exc}')
102
  else:
103
  self.send_success_message(validated_message.uuid, transcribed_text)
 
1
  import base64
2
  import io
3
+ import json
4
  import logging
5
  import os
6
  from pathlib import Path
 
79
  self.send_message(message.model_dump_json())
80
 
81
  def consume(self, channel, method, properties, message):
82
+ try:
83
+ body = self.decode_message(message)
84
+ except json.JSONDecodeError as exc:
85
+ logging.exception('Message decode failed: \n message:\n %s\n\n exception info:\n %s', message, exc)
86
+ self.send_fail_message('', f'Message decode failed, message: \n {message}')
87
+ return
88
 
89
  try:
90
  validated_message = TranscribeInputMessage.model_validate(body)
 
100
  transcribed_text = segment.text
101
  break
102
  except ValidationError as exc:
103
+ logging.exception('Message validated failed: \n message:\n %s\n\n exception info:\n %s', message, exc)
104
  self.send_fail_message(body.get('uuid'), f'{exc}')
105
  except Exception as exc:
106
+ logging.exception('Consume message failed: \n message:\n %s\n\n exception info:\n %s', message, exc)
107
  self.send_fail_message(body.get('uuid'), f'{exc}')
108
  else:
109
  self.send_success_message(validated_message.uuid, transcribed_text)