maolin.liu
commited on
Commit
·
475a8bc
1
Parent(s):
4aa3849
[bugfix]Bugfix.
Browse files- consumer/asr.py +2 -3
- 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
|
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,
|