o iL@sdZddlZddlZddlmZddlmZddlmm m Z ddl m Z mZmZmZmZmZmZmZddlmZejGdddZGd d d eZdS) z)Protobuf frame serialization for Pipecat.N)Optional)logger)FrameInputAudioRawFrameInputTransportMessageFrameOutputAudioRawFrameOutputTransportMessageFrame!OutputTransportMessageUrgentFrame TextFrameTranscriptionFrame)FrameSerializerc@seZdZUdZeed<dS) MessageFramezData class for converting transport messages into Protobuf format. Parameters: data: JSON-encoded message data for transport. dataN)__name__ __module__ __qualname____doc__str__annotations__rrP/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/serializers/protobuf.pyr s  r cseZdZdZedededediZdde DZ ede dedediZ dde DZ dd eejffd d Zd edeeBd BfddZdeeBded BfddZZS)ProtobufFrameSerializera#Serializer for converting Pipecat frames to/from Protocol Buffer format. Provides efficient binary serialization for frame transport over network connections. Supports text, audio, transcription, and message frames with automatic conversion between transport message types. textaudio transcriptionmessagecCi|]\}}||qSrr.0kvrrr 6z"ProtobufFrameSerializer.cCrrrrrrrr!>r"Nparamscst|dS)zoInitialize the Protobuf frame serializer. Args: params: Configuration parameters. N)super__init__)selfr# __class__rrr%@sz ProtobufFrameSerializer.__init__framereturncst|ttfr||rdStt|jd}t }t ||j vr0t dt |ddS|j t |}t||}t|D]}t||j}|rXt||jrXt||j|qA|S)zSerialize a frame to Protocol Buffer binary format. Args: frame: The frame to serialize. Returns: Serialized frame as bytes, or None if frame type is not serializable. N)rz Frame type z is not serializable) isinstancerr should_ignore_framer jsondumpsr frame_protosrtypeSERIALIZABLE_TYPESrwarninggetattr dataclassesfieldsnamehasattrsetattrSerializeToString)r&r) proto_frameproto_optional_name proto_attrfieldvaluerrr serializeHs&    z!ProtobufFrameSerializer.serializerc sltj|}|d}||jvrtddS|j|}t||}i}|jj |j j D] }t||j ||j <q,t|dd}t|dd} t|dd} d|vrQ|d=d|vrX|d=d|vr_|d=|t krzt|d} t| d} td | Wn"ty} ztd | WYd} ~ dSd} ~ ww|d i|} |rt| d|| rt| d| | rt| d| | S) zDeserialize Protocol Buffer binary data to a frame. Args: data: Binary protobuf data to deserialize. Returns: Deserialized frame instance, or None if deserialization fails. r)z#Unable to deserialize a valid frameNidr6ptsr)rz+ProtobufFrameSerializer: Transport message z!Error parsing MessageFrame data: r)r/r FromString WhichOneofDESERIALIZABLE_FIELDSrerrorr3 DESCRIPTORfields_by_name message_typer5r6r r-loadsrdebug Exceptionr8)r&rprotowhich class_nameargs args_dictr=r@r6rAmsginstanceerrr deserializehsL            z#ProtobufFrameSerializer.deserialize)N)rrrrr rr r r1itemsSERIALIZABLE_FIELDSrDESERIALIZABLE_TYPESrDrr InputParamsr%rrbytesr?rT __classcell__rrr'rr(s"" r)rr4r-typingrlogurur#pipecat.frames.protobufs.frames_pb2frames protobufs frames_pb2r/pipecat.frames.framesrrrrrr r r #pipecat.serializers.base_serializerr dataclassr rrrrrs  (