maolin.liu
commited on
Commit
·
fa21f8b
1
Parent(s):
854bf8d
[bugfix]Fix rabbitmq message sender lost connection.
Browse files- consumer/asr.py +2 -5
consumer/asr.py
CHANGED
@@ -45,7 +45,6 @@ class TranscribeConsumer(BasicMessageReceiver):
|
|
45 |
self.routing_key = 'transcribe-input'
|
46 |
|
47 |
self.setup_consume_parameters()
|
48 |
-
self.setup_message_sender()
|
49 |
|
50 |
model_size = os.environ.get('WHISPER-MODEL-SIZE', 'large-v3')
|
51 |
# Run on GPU with FP16
|
@@ -56,13 +55,11 @@ class TranscribeConsumer(BasicMessageReceiver):
|
|
56 |
self.declare_queue(self.queue_name, max_priority=-1)
|
57 |
self.bind_queue(self.exchange_name, self.queue_name, self.routing_key)
|
58 |
|
59 |
-
def setup_message_sender(self):
|
60 |
-
self.sender = BasicMessageSender()
|
61 |
-
|
62 |
def send_message(self, message: Union[dict, str]):
|
63 |
routing_key = 'transcribe-output'
|
64 |
# headers = Headers(job_id=f'{uuid.uuid4()}', priority=Priority.NORMAL)
|
65 |
-
|
|
|
66 |
exchange_name=self.exchange_name,
|
67 |
routing_key=routing_key,
|
68 |
body=message,
|
|
|
45 |
self.routing_key = 'transcribe-input'
|
46 |
|
47 |
self.setup_consume_parameters()
|
|
|
48 |
|
49 |
model_size = os.environ.get('WHISPER-MODEL-SIZE', 'large-v3')
|
50 |
# Run on GPU with FP16
|
|
|
55 |
self.declare_queue(self.queue_name, max_priority=-1)
|
56 |
self.bind_queue(self.exchange_name, self.queue_name, self.routing_key)
|
57 |
|
|
|
|
|
|
|
58 |
def send_message(self, message: Union[dict, str]):
|
59 |
routing_key = 'transcribe-output'
|
60 |
# headers = Headers(job_id=f'{uuid.uuid4()}', priority=Priority.NORMAL)
|
61 |
+
sender = BasicMessageSender()
|
62 |
+
sender.send_message(
|
63 |
exchange_name=self.exchange_name,
|
64 |
routing_key=routing_key,
|
65 |
body=message,
|