qiO fdZddlZddlmZmZddlmZmZmZddl m Z m Z ddl m Z Gdde Zy) z@Async generator processor for frame serialization and streaming.N)AnyAsyncGenerator) CancelFrameEndFrameFrame)FrameDirectionFrameProcessor)FrameSerializercReZdZdZdeffd Zdedeffd Zde e dffd Z xZ S) AsyncGeneratorProcessorawA frame processor that serializes frames and provides them via async generator. This processor passes frames through unchanged while simultaneously serializing them and making the serialized data available through an async generator interface. Useful for streaming frame data to external consumers while maintaining the normal frame processing pipeline. serializerc dt|di|||_tj|_y)zInitialize the async generator processor. Args: serializer: The frame serializer to use for converting frames to data. **kwargs: Additional arguments passed to the parent FrameProcessor. N)super__init__ _serializerasyncioQueue _data_queue)selfr kwargs __class__s T/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/processors/async_generator.pyrz AsyncGeneratorProcessor.__init__s+ "6"%"==?frame directioncKt|||d{|j||d{t|tt fr$|j jdd{y|jj|d{}|r$|j j|d{yy777S717w)zProcess frames by passing them through and queuing serialized data. Args: frame: The frame to process. direction: The direction of frame flow in the pipeline. N) r process_frame push_frame isinstancerrrputr serialize)rrrdatars rrz%AsyncGeneratorProcessor.process_frame)sg#E9555ooeY/// ek84 5""&&t, , ,))33E::D&&**4000 6/ -:0sUCB:CB<8C*B>+#CC$C3C4C<C>CCCreturnNczKd}|r1|jjd{}|du}|r||r0yy7w)zGenerate serialized frame data asynchronously. Yields: Serialized frame data from the internal queue until a termination signal (None) is received. TN)rget)rrunningr#s r generatorz!AsyncGeneratorProcessor.generator;sA))--//D$&G /s";9;;) __name__ __module__ __qualname____doc__r rrrrrrr( __classcell__)rs@rr r s< +o +11>1$ T !: rr )r,rtypingrrpipecat.frames.framesrrr"pipecat.processors.frame_processorrr #pipecat.serializers.base_serializerr r rrrr2s0G& N?2n2r