maolin.liu commited on
Commit
475a8bc
·
1 Parent(s): 4aa3849

[bugfix]Bugfix.

Browse files
Files changed (2) hide show
  1. consumer/asr.py +2 -3
  2. consumer/base.py +1 -1
consumer/asr.py CHANGED
@@ -2,14 +2,13 @@ import base64
2
  import io
3
  import logging
4
  import os
5
- import uuid
6
  from pathlib import Path
7
  from typing import Literal
8
 
9
  from faster_whisper import WhisperModel
10
  from pydantic import BaseModel, Field, ValidationError, model_validator
11
 
12
- from .base import BasicMessageReceiver, BasicMessageSender, Headers, Priority
13
 
14
 
15
  class TranscribeInputMessage(BaseModel):
@@ -61,7 +60,7 @@ class TranscribeConsumer(BasicMessageReceiver):
61
 
62
  def send_message(self, message: dict):
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,
 
2
  import io
3
  import logging
4
  import os
 
5
  from pathlib import Path
6
  from typing import Literal
7
 
8
  from faster_whisper import WhisperModel
9
  from pydantic import BaseModel, Field, ValidationError, model_validator
10
 
11
+ from .base import BasicMessageReceiver, BasicMessageSender
12
 
13
 
14
  class TranscribeInputMessage(BaseModel):
 
60
 
61
  def send_message(self, message: dict):
62
  routing_key = 'transcribe-output'
63
+ # headers = Headers(job_id=f'{uuid.uuid4()}', priority=Priority.NORMAL)
64
  self.sender.send_message(
65
  exchange_name=self.exchange_name,
66
  routing_key=routing_key,
consumer/base.py CHANGED
@@ -156,7 +156,7 @@ class BasicMessageSender(BasicPikaClient):
156
  body = self.encode_message(body=body, encoding_type=self.message_encoding_type)
157
 
158
  properties = pika.BasicProperties(delivery_mode=pika.spec.PERSISTENT_DELIVERY_MODE,
159
- priority=headers.priority.value,
160
  headers=headers.model_dump() if headers else None)
161
  self.channel.basic_publish(
162
  exchange=exchange_name,
 
156
  body = self.encode_message(body=body, encoding_type=self.message_encoding_type)
157
 
158
  properties = pika.BasicProperties(delivery_mode=pika.spec.PERSISTENT_DELIVERY_MODE,
159
+ priority=headers.priority.value if headers else None,
160
  headers=headers.model_dump() if headers else None)
161
  self.channel.basic_publish(
162
  exchange=exchange_name,