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,
|