[bugfix]Do not serialize string.

consumer/asr.py

@@ -3,7 +3,7 @@ import io
 import logging
 import os
 from pathlib import Path
-from typing import Literal
+from typing import Literal, Union
 
 from faster_whisper import WhisperModel
 from pydantic import BaseModel, Field, ValidationError, model_validator
@@ -58,7 +58,7 @@ class TranscribeConsumer(BasicMessageReceiver):
     def setup_message_sender(self):
         self.sender = BasicMessageSender()
 
-    def send_message(self, message: dict):
+    def send_message(self, message: Union[dict, str]):
         routing_key = 'transcribe-output'
         # headers = Headers(job_id=f'{uuid.uuid4()}', priority=Priority.NORMAL)
         self.sender.send_message(
@@ -71,11 +71,11 @@ class TranscribeConsumer(BasicMessageReceiver):
     def send_success_message(self, uuid: str, transcribed_text):
         message = TranscribeOutputMessage(uuid=uuid, if_success=True, msg='Transcribe finished.',
                                           transcribed_text=transcribed_text)
-        self.send_message(message.model_dump())
+        self.send_message(message.model_dump_json())
 
     def send_fail_message(self, uuid: str, error: str):
         message = TranscribeOutputMessage(uuid=uuid, if_success=False, msg=error)
-        self.send_message(message.model_dump())
+        self.send_message(message.model_dump_json())
 
     def consume(self, channel, method, properties, message):
         body = self.decode_message(message)

consumer/base.py

@@ -6,7 +6,7 @@ import os
 import ssl
 import time
 from enum import Enum
-from typing import Dict, Optional, Literal
+from typing import Dict, Optional, Literal, Union
 
 import msgpack
 import pika
@@ -140,11 +140,13 @@ class BasicPikaClient:
 class BasicMessageSender(BasicPikaClient):
     message_encoding_type: Literal['bytes', 'json'] = 'json'
 
-    def encode_message(self, body: Dict, encoding_type: str = "bytes"):
+    def encode_message(self, body: Union[Dict, str], encoding_type: str = "bytes"):
         if encoding_type == "bytes":
             return msgpack.packb(body)
         elif encoding_type == "json":
-            return json.dumps(body)
+            if isinstance(body, dict):
+                return json.dumps(body)
+            return body
         else:
             raise NotImplementedError
 
@@ -152,7 +154,7 @@ class BasicMessageSender(BasicPikaClient):
             self,
             exchange_name: str,
             routing_key: str,
-            body: Dict,
+            body: Union[Dict, str],
             headers: Optional[Headers],
     ):
         body = self.encode_message(body=body, encoding_type=self.message_encoding_type)