JihddlmZddlZddlmZmZmZddlmZddl m Z m Z m Z ddl mZerddlmZGd d ej"eZy) ) annotationsN) TYPE_CHECKINGAnyAsyncGenerator)WorkflowRuntimeError)EventInternalDispatchEvent StopEvent) RunResultT)ContextceZdZUdZded<ded<ded<ded <d d d  dfd Zedd ZddZddZ ddZ ddZ ddZ d ddZ ddZxZS)WorkflowHandlera Handle a running workflow: await results, stream events, access context, or cancel. Instances are returned by [Workflow.run][workflows.workflow.Workflow.run]. They can be awaited for the final result and support streaming intermediate events via [stream_events][workflows.handler.WorkflowHandler.stream_events]. See Also: - [Context][workflows.context.context.Context] - [StopEvent][workflows.events.StopEvent] r _ctxasyncio.Task[None] | None _run_taskbool_all_events_consumedStopEvent | None _stop_eventN)run_idrun_taskc\t||i|||_||_||_d|_y)NF)super__init__rrrr)selfctxrrargskwargs __class__s c/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/handler.pyrzWorkflowHandler.__init__#s4 $)&)  !$)!c|jS)zGThe workflow [Context][workflows.context.context.Context] for this run.)rrs r!rzWorkflowHandler.ctx1syyr"c|jS)zThe stop event for this run. Always defined once the future is done. In a future major release, this will be removed, and the result will be the stop event itself.)rr$s r!get_stop_eventzWorkflowHandler.get_stop_event6sr"c|K|jd{|jJd|jS7#w)zGet the stop event for this run. Always defined once the future is done. In a future major release, this will be removed, and the result will be the stop event itself.Nz3Stop event must be defined once the future is done.)resultrr$s r!stop_event_resultz!WorkflowHandler.stop_event_result:sDkkm+ A + s <:$<cx||_|jt|tur|jy|yN)r set_resulttyper r()r stop_events r!_set_stop_eventzWorkflowHandler._set_stop_eventBs5% !%j!1Y!>J   DN r"c4t|jSr+)strr(r$s r!__str__zWorkflowHandler.__str__Ks4;;=!!r"c"|jS)z,Return True when the workflow has completed.)doner$s r!is_donezWorkflowHandler.is_doneNsyy{r"cK|jr d}t||jj23d{}t |t r|s|t |t s2d|_y766yw)a{ Stream events from the workflow execution as they occur. This method provides real-time access to events generated during workflow execution, allowing for monitoring and processing of intermediate results. Events are yielded in the order they are generated by the workflow. The stream includes all events written to the context's streaming queue, and terminates when a [StopEvent][workflows.events.StopEvent] is encountered, indicating the workflow has completed. Args: expose_internal (bool): Whether to expose internal events. Returns: AsyncGenerator[Event, None]: An async generator that yields Event objects as they are produced by the workflow. Raises: ValueError: If the context is not set on the handler. WorkflowRuntimeError: If all events have already been consumed by a previous call to `stream_events()` on the same handler instance. Examples: ```python handler = workflow.run() # Stream and process events in real-time async for event in handler.stream_events(): if isinstance(event, StopEvent): print(f"Workflow completed with result: {event.result}") else: print(f"Received event: {event}") # Get final result result = await handler ``` Note: Events can only be streamed once per handler instance. Subsequent calls to `stream_events()` will raise a WorkflowRuntimeError. z3All the streamed events have already been consumed.NT)rrr stream_events isinstancer r )rexpose_internalmsgevs r!r7zWorkflowHandler.stream_eventsRsq^  $ $GC&s+ +..0  ""34_H"i(,0) 0s+4A5A3A1A3)A5( A51A33A5cK|jr=|jj|j |jd{yyy7#t$rYywxYww)aqCancel the running workflow. Signals the underlying context to raise [WorkflowCancelledByUser][workflows.errors.WorkflowCancelledByUser], which will be caught by the workflow and gracefully end the run. Examples: ```python handler = workflow.run() await handler.cancel_run() ``` N)r_workflow_cancel_runr Exceptionr$s r! cancel_runzWorkflowHandler.cancel_runs[ 88 HH ) ) +~~)..((* ) s:3AAA A A A AAAA) rrrr rz str | NonerrrrreturnNone)r@r )r@r)r@r )r.r r@rA)r@r1)r@r)F)r9rr@zAsyncGenerator[Event, None])r@rA)__name__ __module__ __qualname____doc____annotations__rpropertyrr&r)r/r2r5r7r? __classcell__)r s@r!rrs  M((!! ".2 * * * * , *  *  *   " ',:#: $:xr"r) __future__rasynciotypingrrrerrorsreventsr r r typesr contextr Futurerr"r!rRs<#55(;; PgnnZ0Pr"