o V۷iY@s8ddlmZddZddZGdddeZdd Zd S) wrapscstfdd}|S)z Wraps a generator which is intended to be used as a pure coroutine by .send()ing it values. The only thing that the wrapper does is calling .next() for the first time which is required by Python generator protocol. cs|i|}t||S)N)next)argskwargsgfuncA/home/ubuntu/vllm_env/lib/python3.10/site-packages/ijson/utils.pywrapper szcoroutine..wrapperr)r r r rr coroutinesr cGs.|}|D]\}}}||g|Ri|}q|S)aT Chains together a sink and a number of coroutines to form a coroutine pipeline. The pipeline works by calling send() on the coroutine created with the information in `coro_pipeline[-1]`, which sends its results to the coroutine created from `coro_pipeline[-2]`, and so on, until the final result is sent to `sink`. r )sink coro_pipelinef coro_func coro_args coro_kwargsr r r chainsrc@seZdZdZejZdS) sendable_lista[ A list that mimics a coroutine receiving values. Coroutine are sent values via their send() method. This class defines such a method so that values sent into it are appended into the list, which can be inspected later. As such, this type can be used as an "accumulating sink" in a pipeline consisting on many coroutines. N)__name__ __module__ __qualname____doc__listappendsendr r r r r!s rc gst}t|g|R}z@|D]:}z||Wn#ty;}z|D]}|Vq"t|tr6WYd}~WdSd}~ww|D]}|Vq>|dd=qWdStycz|WYdSYYdSw)z A utility function that returns a generator yielding values dispatched by a coroutine pipeline after *it* has received values coming from `source`. N)rrr Exception isinstance StopIteration GeneratorExitclose)sourcereventsrvalueexeventr r r coros2gen-s2   r'N) functoolsrr rrrr'r r r r s