diff --git a/src/pipecat/serializers/abstract_frame_serializer.py b/src/pipecat/serializers/abstract_frame_serializer.py deleted file mode 100644 index 8da0bd11d..000000000 --- a/src/pipecat/serializers/abstract_frame_serializer.py +++ /dev/null @@ -1,16 +0,0 @@ -from abc import abstractmethod - -from pipecat.pipeline.frames import Frame - - -class FrameSerializer: - def __init__(self): - pass - - @abstractmethod - def serialize(self, frame: Frame) -> bytes: - raise NotImplementedError - - @abstractmethod - def deserialize(self, data: bytes) -> Frame: - raise NotImplementedError diff --git a/src/pipecat/serializers/base_serializer.py b/src/pipecat/serializers/base_serializer.py new file mode 100644 index 000000000..c137f873d --- /dev/null +++ b/src/pipecat/serializers/base_serializer.py @@ -0,0 +1,20 @@ +# +# Copyright (c) 2024, Daily +# +# SPDX-License-Identifier: BSD 2-Clause License +# + +from abc import ABC, abstractmethod + +from pipecat.frames.frames import Frame + + +class FrameSerializer(ABC): + + @abstractmethod + def serialize(self, frame: Frame) -> bytes: + pass + + @abstractmethod + def deserialize(self, data: bytes) -> Frame: + pass diff --git a/src/pipecat/serializers/protobuf_serializer.py b/src/pipecat/serializers/protobuf.py similarity index 88% rename from src/pipecat/serializers/protobuf_serializer.py rename to src/pipecat/serializers/protobuf.py index 04b348b86..24c484b7c 100644 --- a/src/pipecat/serializers/protobuf_serializer.py +++ b/src/pipecat/serializers/protobuf.py @@ -1,14 +1,20 @@ +# +# Copyright (c) 2024, Daily +# +# SPDX-License-Identifier: BSD 2-Clause License +# + import dataclasses -from typing import Text -from pipecat.pipeline.frames import AudioFrame, Frame, TextFrame, TranscriptionFrame -import pipecat.pipeline.protobufs.frames_pb2 as frame_protos -from pipecat.serializers.abstract_frame_serializer import FrameSerializer + +import pipecat.frames.protobufs.frames_pb2 as frame_protos +from pipecat.frames.frames import AudioRawFrame, Frame, TextFrame, TranscriptionFrame +from pipecat.serializers.base_serializer import FrameSerializer class ProtobufFrameSerializer(FrameSerializer): SERIALIZABLE_TYPES = { TextFrame: "text", - AudioFrame: "audio", + AudioRawFrame: "audio", TranscriptionFrame: "transcription" }