maolin.liu commited on
Commit
fa21f8b
·
1 Parent(s): 854bf8d

[bugfix]Fix rabbitmq message sender lost connection.

Browse files
Files changed (1) hide show
  1. 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
- self.sender.send_message(
 
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,