ei0@ddlZddlZddlZddlZddlmZmZddlmZmZddl m Z m Z m Z m Z mZmZmZmZmZmZddlZddlmZddlmZddlmZddlmZdd lmZmZm Z m!Z!dd l"m#Z#m$Z$ej%e&Z'eGd d Z(ej)Z*d e(fdZ+dZ,ddZ-ddZ.Gdde/Z0GddZ1 ddl2m3Z3e3j4e1_5e1j4e3_4n #e6$re'7dYnwxYwee8e9e:e#e fZ;ee;ZGddeZ?dS)N) dataclassfield)datetimetimezone) Any AsyncIterable AwaitableCallable CoroutineIteratorMappingOptionalSetUnion)BackgroundTask)iterate_in_threadpool)MutableHeaders)Response)ReceiveScopeSendMessage)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__rboolU/home/div/.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_stater21s1 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 getsignalSIGTERMhasattrr4 Exception)handlerservers r+_get_uvicorn_serverr=:sp "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) r2r= AppStatusr5enable_automatic_graceful_drainr%sleeplistrr$r)r1uvicorn_serverevents r+_shutdown_watcherrEQs ! !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)r2rasyncioget_running_loop create_taskrE RuntimeError)r1loops r+$_ensure_watcher_started_on_this_looprLus  ! !E  * $ *+--D   .00 1 1 1 1 1 * * *$)E ! ! ! ! * **s4AA)(A)ceZdZdS)SendTimeoutErrorN)r r!r"r)r*r+rNrNsDr*rNcxeZdZUdZdZdZdZeee 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?r@r)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(). TNrRr)r*r+$enable_automatic_graceful_drain_modez.AppStatus.enable_automatic_graceful_drain_modes59 111r*cttjr dt_tjtj|i|dSdS)NT)r?r@r5rP)argskwargss r+ handle_exitzAppStatus.handle_exitsB  4 )$(I !  % 1  & 7 7 7 7 7 7 2 1r*)r r!r"r#r5r@rPrr r' staticmethodrSrUrYr)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)Nr_  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_SEPARATORrh isinstancer body_iteratorrrcrerfrjrkrupdate setdefault init_headersDEFAULT_PING_INTERVAL ping_intervalrirl_shutdown_event_shutdown_grace_periodactiver%Lock _send_lock)selfrbrcrdrerfrgrhrirjrkrlrmrn_headerss r+__init__zEventSourceResponse.__init__se* 0 0 0P3PPQQ Q0$0 g} - - @!(D  !6w!?!?D &-7-?$//Z$$8!("##   OOG $ $ $ OZ888!-(,$% (###;?.cancel_on_finishs9dff '..00000r*c.Sr)rrrsr+z.EventSourceResponse.__call__..sD.sDJJt.s$*E*Eg*N*Nr*)r%create_task_groupr r start_soonrrjrf)rrrrrrs` `` @r+__call__zEventSourceResponse.__call__s*,,         1Xb)D/6I-J 1 1 1 1 1 1  ! !"24W4W4W4W4W X X X  ! !"24L4L4L4L4L M M M  ! ! $"I   ( A%%d&?@@@  ! ! "N"N"N"N"N   !                           ( ? &//## # # # # # # # # # ' &sB+C C'*C') r`NraNNNNNNNNr)Fr,N)%r r!r"r#r{ru ContentStreamrrr strrr rr rrr r%r&rpropertyrr|setterr(rrrrrrZrrrrrr)r*r+r]r]s. /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$$$$$SSSSSSSX4XDXXXX8 G     (((\(&++++(6$E$G$4$D$$$$$$r*r]r)@rGloggingr6 threading dataclassesrrrrtypingrrr r r r r rrrr%starlette.backgroundrstarlette.concurrencyrstarlette.datastructuresrstarlette.responsesrstarlette.typesrrrrsse_starlette.eventrr getLoggerr rrlocalr0r2r=rErL TimeoutErrorrNr? uvicorn.mainr[rYrP ImportErrorrrbytesdictContentSyncContentStreamAsyncContentStreamrr]r)r*r+rsL ((((((((''''''''                         //////777777333333((((((999999999999========  8 $ $ """"""" " !! ^.!&!&!&!&H * * * *     |   #8#8#8#8#8#8#8#8L######!'!3I".F LLR UD/36 7W%"7+(*;;< y$y$y$y$y$(y$y$y$y$y$s C))DD