iRLdZddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlZddlmZddlmZddlmZdd lmZdd lmZm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&m'Z'ej(e)Z*eGd dZ+ej,Z-de+fdZ.dZ/ddZ0ddZ1de!de!fdZ2Gdde3Z4GddZ5 ddl6m7Z7e7j8e5_9e5j8e7_8n #e:$re*;dYnwxYweee&e fZ?ee?Z@e e?ZAeeAe@fZBGddeZCdS)uServer-Sent Events response for Starlette / FastAPI. Intentional divergence from ``starlette.responses.StreamingResponse`` -------------------------------------------------------------------- ``EventSourceResponse`` is modelled on Starlette's ``StreamingResponse`` and re-syncs most of its behaviour (WebSocket denial, ``collapse_excgroups()`` around the task group, ``memoryview`` chunk handling). The following points are deliberate divergences — DO NOT "fix" them without reading the rationale: 1. ASGI ``spec_version >= 2.4`` fast path is NOT adopted. Upstream short-circuits to ``await stream_response(send)`` and converts ``OSError`` into ``ClientDisconnect``, skipping ``listen_for_disconnect``. We keep ``_listen_for_disconnect`` running because it (a) invokes ``client_close_handler_callable`` on disconnect, (b) flips ``self.active = False`` so ``_ping`` and the cooperative shutdown grace loop exit promptly. Adopting the upstream fast path would regress both features. 2. ``_wrap_websocket_denial_send`` is inlined in this module rather than inherited from ``starlette.responses.Response``. The helper landed on Starlette ``main`` after our minimum pin (``starlette>=0.41.3``); inline until the floor moves past the release that contains it. 3. ``collapse_excgroups()`` is vendored in ``sse_starlette._utils`` rather than imported from ``starlette._utils`` (private module). N) dataclassfield)datetimetimezone) Any AsyncIterable AwaitableCallable CoroutineIteratorMappingOptionalSetUnion)BackgroundTask)iterate_in_threadpool)MutableHeaders)Response)ReceiveScopeSendMessage)collapse_excgroups)ServerSentEvent ensure_bytescZeZdZUdZeeZeej e d<dZ e e d<dS)_ShutdownStatezPer-thread state for shutdown coordination. Issue #152 fix: Uses threading.local() instead of ContextVar to ensure one watcher per thread rather than one per async context. )default_factoryeventsFwatcher_startedN) __name__ __module__ __qualname____doc__rsetrranyioEvent__annotations__r boolX/home/piyush/.hermes/hermes-agent/venv/lib/python3.11/site-packages/sse_starlette/sse.pyrr>sR %uS999FC 999!OT!!!!!r+rreturncjttdd}|t}|t_|S)z4Get or create shutdown state for the current thread.shutdown_stateN)getattr _thread_staterr/)states r,_get_shutdown_stater3Ns1 M#3T : :E }  ', $ Lr+c tjtj}t|dr|j}t|dr|Sn#t $rYnwxYwdS)am Try to get uvicorn Server instance via signal handler introspection. When uvicorn registers signal handlers, they're bound methods on the Server instance. We can retrieve the Server from the handler's __self__ attribute. Returns None if: - Not running under uvicorn - Signal handler isn't a bound method - Any introspection fails __self__ should_exitN)signal getsignalSIGTERMhasattrr5 Exception)handlerservers r,_get_uvicorn_serverr>Wsp "6>22 7J ' ' %Fv}--       4sAA AAcXKt}t} tjrn=tjr||jr dt_nt jdd{VJt|jD]}|  d|_ dS#d|_ wxYw)ag Poll for shutdown and broadcast to all events in this context. One watcher runs per thread (event loop). Checks two shutdown sources: 1. AppStatus.should_exit - set when our monkey-patch works 2. uvicorn Server.should_exit - via signal handler introspection (Issue #132 fix) When either becomes True, signals all registered events. TNg?F) r3r> AppStatusr6enable_automatic_graceful_drainr&sleeplistrr%r )r2uvicorn_serverevents r,_shutdown_watcherrFns ! !E(**N& #$ 9 "."./)- %+c"" " " " " " " " #%,''  E IIKKKK !&%%%%s A6B B)ct}|jsVd|_ tj}|t dS#t $r d|_YdSwxYwdS)zDEnsure the shutdown watcher is running for this thread (event loop).TFN)r3r asyncioget_running_loop create_taskrF RuntimeError)r2loops r,$_ensure_watcher_started_on_this_looprMs  ! !E  * $ *+--D   .00 1 1 1 1 1 * * *$)E ! ! ! ! * **s4AA)(A)sendc&dtddffd }|S)a;Mirror of ``starlette.responses.Response._wrap_websocket_denial_send``. Divergence #2 (see module docstring): inlined because the helper landed on Starlette ``main`` (commit 9ee9519) after our minimum pin ``starlette>=0.41.3``. Drop this once the floor moves past the release that contains it. messager-Nc\K|d}|dvr i|dd|zi}|d{VdS)Ntype>http.response.bodyhttp.response.startz websocket.r*)rP message_typerNs r,wrappedz,_wrap_websocket_denial_send..wrappeds[v H H HFF&,*EFFGd7mmr+)r)rNrVs` r,_wrap_websocket_denial_sendrWs6w4 Nr+ceZdZdS)SendTimeoutErrorN)r!r"r#r*r+r,rYrYsDr+rYcxeZdZUdZdZdZdZeee d<e dZ e dZ e dZ dS) r@z\Helper to capture a shutdown signal from Uvicorn so we can gracefully terminate SSE streams.FTNoriginal_handlercdt_dS)aJ Prevent automatic SSE stream termination on server shutdown. WARNING: When disabled, you MUST set AppStatus.should_exit = True at some point during shutdown, or streams will never close and the server will hang indefinitely (or until uvicorn's graceful shutdown timeout expires). FNr@rAr*r+r, disable_automatic_graceful_drainz*AppStatus.disable_automatic_graceful_drains5: 111r+cdt_dS)a Re-enable automatic SSE stream termination on server shutdown. This restores the default behavior where SIGTERM triggers immediate stream draining. Call this to undo a previous call to disable_automatic_graceful_drain(). TNr]r*r+r,$enable_automatic_graceful_drain_modez.AppStatus.enable_automatic_graceful_drain_modes59 111r+cttjr dt_tjtj|i|dSdS)NT)r@rAr6r[)argskwargss r, handle_exitzAppStatus.handle_exitsB  4 )$(I !  % 1  & 7 7 7 7 7 7 2 1r+)r!r"r#r$r6rAr[rr r( staticmethodr^r`rdr*r+r,r@r@sffK&*#+/hx(/// : :\ :99\988\888r+r@)ServerzHUvicorn not installed. Graceful shutdown on server termination disabled.c$eZdZdZdZdZ d'ded ed ee e e fd e d ee d eedee dee ge fdee gedfdeedee egedfdeejdeddfdZedeeeffdZejdeeefddfdZd(deddfdZdeddfdZd eddfd!Zed)d"Z d)d#Z!deddfd$Z"d%e#d ededdfd&Z$dS)*EventSourceResponseagStreaming response implementing the SSE (Server-Sent Events) specification. Args: content: Async iterable or sync iterator yielding SSE event data. status_code: HTTP status code. Default: 200. headers: Additional HTTP headers. media_type: Response media type. Default: "text/event-stream". background: Background task to run after response completes. ping: Ping interval in seconds (0 to disable). Default: 15. sep: Line separator for SSE messages ("\r\n", "\r", or "\n"). ping_message_factory: Callable returning custom ping ServerSentEvent. data_sender_callable: Async callable for push-based data sending. send_timeout: Timeout in seconds for individual send operations. client_close_handler_callable: Async callback on client disconnect. shutdown_event: Optional ``anyio.Event`` set by the library when server shutdown is detected. Generators can watch this event to send farewell messages and exit cooperatively instead of receiving CancelledError. shutdown_grace_period: Seconds to wait after setting ``shutdown_event`` before force-cancelling the generator. Must be >= 0. Should be less than your ASGI server's graceful shutdown timeout. Default: 0 (immediate cancel, identical to pre-v3.3.0 behavior).  Ntext/event-streamrcontent status_codeheaders media_type backgroundpingsepping_message_factorydata_sender_callable)NNN send_timeoutclient_close_handler_callableshutdown_eventshutdown_grace_periodr-c||dvrtd||p|j|_t|tr||_nt ||_||_||jn||_||_ | |_ | |_ t}|| ||ddd|d<d|d<||||jn||_||_| |_| d krtd | |_| |_d |_t/j|_dS) N)Nrj  z'sep must be one of: \r\n, \r, \n, got: z Cache-Controlzno-storez keep-alive ConnectionnozX-Accel-Bufferingrz"shutdown_grace_period must be >= 0T) ValueErrorDEFAULT_SEPARATORrs isinstancer body_iteratorrrnrprqrurvrupdate setdefault init_headersDEFAULT_PING_INTERVAL ping_intervalrtrw_shutdown_event_shutdown_grace_periodactiver&Lock _send_lock)selfrmrnrorprqrrrsrtrurvrwrxry_headerss r,__init__zEventSourceResponse.__init__se* 0 0 0P3PPQQ Q0$0 g} - - @!(D  !6w!?!?D &-7-?$//Z$$8!("##   OOG $ $ $ OZ888!-(,$% (###;?.cancel_on_finishs9$&&LLLLLLL+2244444r+c.Sr)rrrNsr,z.EventSourceResponse.__call__..sd.C.CD.I.Ir+c.Sr)rrsr,rz.EventSourceResponse.__call__..s 4@P@Pr+c.Sr)r)rrsr,rz.EventSourceResponse.__call__..sd.I.I'.R.Rr+) rWrr&create_task_groupr r start_soonrrurq)rrrrNrrs` `` @r,__call__zEventSourceResponse.__call__sv =K ' '.t44D  ! !  .00       J5"io:M1N555555%%$&I&I&I&I&I%%&68P8P8P8P8PQQQ%%$d&M,E))$*CDDD%%$&R&R&R&R&R%                                          . ? &//## # # # # # # # # # ' &s6D$ B+D5D$ D D$D D$$D(+D() rkNrlNNNNNNNNr)Fr-N)%r!r"r#r$rr ContentStreamrrr strrr rr rrr r&r'rpropertyrrsetterr)rrrrrrerrrrrr*r+r,rhrhs. /3-/3"!HL (, 04'(%G'G'G'G''#s(+, G'  G' ^, G'smG'c]G''xO0C'DEG'' R#344 5 G'uoG'(0 gY $/ 0( G'"!-#G'$ %%G'& 'G'G'G'G'R#uS%Z0###X#$5e#4$$$$$SSSSSSSX4XDXXXX8G"(((\(&++++(6($E($G($4($D($($($($($($r+rhr)Dr$rHloggingr7 threading dataclassesrrrrtypingrrr r r r r rrrr&starlette.backgroundrstarlette.concurrencyrstarlette.datastructuresrstarlette.responsesrstarlette.typesrrrrsse_starlette._utilsrsse_starlette.eventrr getLoggerr!rrlocalr1r3r>rFrMrW TimeoutErrorrYr@ uvicorn.mainrfrdr[ ImportErrorrrbytesdictContentSyncContentStreamAsyncContentStreamrrhr*r+r,rs8 ((((((((''''''''                         //////777777333333((((((999999999999333333========  8 $ $ """"""" " !! ^.!&!&!&!&H * * * *dt$     |   #8#8#8#8#8#8#8#8L######!'!3I".F LLR UD/36 7W%"7+(*;;< K$K$K$K$K$(K$K$K$K$K$sC99DD