maolin.liu
commited on
Commit
·
a829e96
1
Parent(s):
7ec99bc
[bugfix]Improve code robust.
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
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('
|
98 |
self.send_fail_message(body.get('uuid'), f'{exc}')
|
99 |
except Exception as exc:
|
100 |
-
logging.exception('Consume message failed: \n message
|
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)
|