qiLdZddlZddlZddlmZddlmZddlmcm cm Z ddl m Z mZmZmZmZmZmZmZddlmZej.GddZGd d eZy) z)Protobuf frame serialization for Pipecat.N)Optional)logger)FrameInputAudioRawFrameInputTransportMessageFrameOutputAudioRawFrameOutputTransportMessageFrame!OutputTransportMessageUrgentFrame TextFrameTranscriptionFrame)FrameSerializerceZdZUdZeed<y) MessageFramezData class for converting transport messages into Protobuf format. Parameters: data: JSON-encoded message data for transport. dataN)__name__ __module__ __qualname____doc__str__annotations__N/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/serializers/protobuf.pyrrs IrrcXeZdZdZedededediZejDcic]\}}|| c}}}Z ede dedediZ e jDcic]\}}|| c}}}Z ddeej ffd Zd ed eezdzfd Zd eezd edzfd ZxZScc}}}wcc}}}w)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 transcriptionmessageNparamsc$t||y)zoInitialize the Protobuf frame serializer. Args: params: Configuration parameters. N)super__init__)selfr __class__s rr#z ProtobufFrameSerializer.__init__@s  rframereturncrKt|ttfr;|j|ryt t j |j}tj}t||jvr#tjdt|dy|jt|}t||}tj |D]I}t||j"}|st%||j"s3t'||j"|K|j)Sw)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) isinstancer r should_ignore_framerjsondumpsr frame_protosrtypeSERIALIZABLE_TYPESrwarninggetattr dataclassesfieldsnamehasattrsetattrSerializeToString)r$r& proto_frameproto_optional_name proto_attrfieldvalues r serializez!ProtobufFrameSerializer.serializeHs e9;\] ^''. ZZ .E#((* ;d55 5 NN[e 5IJ K#55d5kB[*=>  ''. 7EE5::.EUZZ8 EJJ6 7 ,,..sC4D77D7)D7rcfKtjj|}|jd}||jvrt j dy|j|}t||}i}|jj|jjD]%}t||j||j<'t|dd}t|dd} t|dd} d|vr|d=d|vr|d=d|vr|d=|tk(r> tj|d} t!| } t j"d | n|d i|} |r t'| d|| r t'| d| | r t'| d| | S#t$$r"} t j d | Yd} ~ yd} ~ wwxYww) 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 frameNidr4ptsr)rz+ProtobufFrameSerializer: Transport message z!Error parsing MessageFrame data: r)r-r FromString WhichOneofDESERIALIZABLE_FIELDSrerrorr1 DESCRIPTORfields_by_name message_typer3r4rr+loadsrdebug Exceptionr6)r$rprotowhich class_nameargs args_dictr;r?r4r@msginstancees r deserializez#ProtobufFrameSerializer.deserializehs""--d3  ) 22 2 LL> ?//6 ue$ %%44U;HHOO >E$+D%**$=Iejj ! >T4 &tVT*dE4( 9 $ Y &! I %   % jj6!235cB J8*UV ".I.H  HdB '  Hfd +  HeS )  @DE s0D F1rfs^0 ::   @ uour