Xjg PddlmZddlmZmZddlmZmZmZm Z ddl Z ddl m Z m Z ddlmZddlmZGd d Zed e j&d ed ededeef dZed e j,d ed ededeef dZde j0deefdZde j0deefdZy))AsyncGenerator)asynccontextmanagercontextmanager)Any AsyncIteratorIteratorcastN) SSEDecoderSSELineDecoder)SSEError)ServerSentEventceZdZdejddfdZd dZedejfdZde e fdZ de e dffdZ y) EventSourceresponsereturnNc||_yN _response)selfrs b/mnt/ssd/data/Dropbox/adrian/vault-secondbrain/venv/lib/python3.12/site-packages/httpx_sse/_api.py__init__zEventSource.__init__ s !c|jjjddjdd}d|vrt d|y)Nz content-type;rtext/event-streamzJExpected response header Content-Type to contain 'text/event-stream', got )rheadersget partitionr )r content_types r_check_content_typezEventSource._check_content_typesZ~~--11."EOOPSTUVW l 2#&(  3rc|jSrr)rs rrzEventSource.responses ~~rc#K|jt}t|jD]+}|j d}|j |}|(|-ywN )r#r _iter_sse_linesrrstripdecode)rdecoderlinesses riter_ssezEventSource.iter_ssesW   ",#DNN3 D;;t$D..&C  s AA!A!c|K|jt}tttdft |j } |23d{}|jd}|j|}|.|47/6 |jd{7y#|jd{7wxYwwr&) r#r r rstr_aiter_sse_linesrr)r*aclose)rr+linesr,r-s r aiter_ssezEventSource.aiter_sse%s   ",^CI.0@0PQ !#  d{{4(nnT*?I  e ,,. %,,. s`AB< B BBB%B:BBBB<BB<B92B53B99B<)rN)__name__ __module__ __qualname__httpxResponserr#propertyrrrr.rr4rrrr sd""D"%..(?3 !0E!F !rrclientmethodurlkwargsrc+K|jdi}d|d<d|d<|j||fd|i|5}t|dddy#1swYyxYwwNrrAcceptzno-storez Cache-Controlpopstreamrr<r=r>r?rrs r connect_sserG3sgjjB'G+GH)GO vs >G >v >$((##$$$s3AA  A AAc K|jdi}d|d<d|d<|j||fd|i|4d{}t|dddd{y7#7#1d{7swYyxYwwrArCrFs r aconnect_sserI?sjjB'G+GH)GOv}}VSD'DVD$$(##$$$$$$$sD7A6AA6A! A6AA6A6!A3'A* (A3/A6rcKt}|j23d{}|j|D]}| %7 6|jD]}| ywr)r aiter_textr*flushrr+textr,s rr1r1NshG))+dNN4( DJ +  s%A AAAA AA c#Kt}|jD]}|j|D]}||jD]}|ywr)r iter_textr*rLrMs rr(r(Ws_G""$NN4( DJ   sAA)collections.abcr contextlibrrtypingrrrr r8 _decodersr r _exceptionsr _modelsrrClientr0rG AsyncClientrIr9r1r(r;rrrYs*:55 1!$$!$!N$ LL$"%$,/$;>$ k$$ $    $  $  $ $ ; $ $U^^ c8Jenn#r