JiddlmZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddl mZddlmZmZmZmZmZddlZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4ddl5m6Z6ddl7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@ddlAmBZBmCZCmDZDddlEmFZFddlGmHZHmIZImJZJmKZKddlLmMZMddlNmOZOddlPmQZRejZTGddZUeGdd ZVGd!d"eWZXeGd#d$ZYeZd%k(rddl[Z[e[jd&'Z]e]jd(e_d)d*+e]jZaeUZbebjZdeeeajd,5Zgejedd-eg.dddeid/eajyy#1swY xYw)0) annotationsN)asynccontextmanager) dataclass)datetimetimezone)version)Path)AnyAsyncGenerator AwaitableCallablecast)instrument_tags) Starlette) HTTPException) Middleware)CORSMiddleware)Request) JSONResponseStreamingResponse)Route)SchemaGenerator) StaticFiles)ContextWorkflow)EventInternalDispatchEvent StartEvent StepStateStepStateChanged StopEvent)WorkflowHandler) CancelHandlerResponse HandlerDataHandlersListResponseHealthResponseSendEventResponseWorkflowEventsListResponseWorkflowGraphResponseWorkflowSchemaResponseis_status_completed) EventEnvelopeEventEnvelopeWithMetadataEventValidationError)extract_workflow_structure)AbstractWorkflowStore HandlerQueryPersistentHandlerStatus)MemoryWorkflowStore) RunResultT)_nanoidcfeZdZddddgd d dZ d! d"dZd#dZed$dZd%d Z d& d'd Z d(d Z d)d Z d)d Z d)dZ d)dZd)dZd)dZd)dZd*dZd)dZd)dZd+dZd)dZd)dZd)dZd,dZ d-dZ d. d/dZ d0dZd1dZd2dZy)3WorkflowServerNg?) middlewareworkflow_storepersistence_backoffcdi_i_i_i_i_||n t _ttjdz _ t|_ |xsttddgdgdg_t!dj"dgt!d j$d gt!d j&d gt!d j(dgt!d j*dgt!dj,dgt!dj.d gt!dj0dgt!dj2dgt!dj4dgt!dj6d gt!dj8dgt!dj:dgg _t>dfd }tAj<j|_!jBjEdtGjddy)Nstaticz.**T)allow_origin_regex allow_methods allow_headersallow_credentialsz /workflowsGET)methodsz/workflows/{name}/runPOSTz/workflows/{name}/run-nowaitz/workflows/{name}/schemaz/results/{handler_id}z/events/{handler_id}z/healthz /handlersz/handlers/{handler_id}z/handlers/{handler_id}/cancelz /workflows/{name}/representationz/workflows/{name}/eventscKj4d{ddddd{y77#1d{7swYyxYwwN)contextmanager)appselfs i/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/server/server.pylifespanz)WorkflowServer.__init__..lifespansE**,         s:A 3A 7 A 5A A A A A A )routesr:rM/) directoryhtmlui)rJname)rJrreturnzAsyncGenerator[None, None])$ _workflows_additional_events _contexts _handlers_resultsr4_workflow_storer __file__parent _assets_pathlist_persistence_backoffrr _middlewarer_list_workflows _run_workflow_run_workflow_nowait_get_events_schema_get_workflow_result_stream_events _post_event _health_check _get_handlers_get_workflow_handler_cancel_handler_get_workflow_representation_list_workflow_events_routesrrrJmountr)rKr:r;r<rMs` rL__init__zWorkflowServer.__init__Fs^02GI-/68/1 ,8N>Q>S !N11H<$()<$=!%  #'"e"e"&   *  $$  '""  .))  *''  '))  &##  &    ""  ""  (**  /$$  211  *** {B  H    <<''   [4+<+<4Ht  cF||j|<|||j|<yyrH)rUrV)rKrSworkflowadditional_eventss rL add_workflowzWorkflowServer.add_workflows- !)  (,=D # #D ) )rqcK|jjtdgt|jj d{}|D]v}|j|j } |jt|j ||jtj||jd{x|S77 #t$r}tjd|jd|j d | t!j"t$j&}|jj)t+|j|j d |j,t/|d|j0|||j d{7n#t$rYnwxYwYd}~fd}~wwxYww) zPResumes previously running workflows, if they were not complete at last shutdownrunning status_inworkflow_name_inNrSrsrsdata)rs handler_idcontextzFailed to resume handler z for workflow z: failed r~ workflow_namestatusrun_iderrorresult started_at updated_at completed_atctx)rZqueryr1r^rUkeysr_start_workflow_NamedWorkflowr~r from_dictr Exceptionloggerrrnowrutcupdater2rstrr)rKhandlers persistentrsers rLstartzWorkflowServer.starts--33 $+T__=Q=Q=S8T    #! Jz'?'?@H **+'55 *44#--xjnnU +! F Q   / 0E0E/FnU_UmUmTnnpqrpst",,x||4C..55)'1'<'<*4*B*B#+#-#4#4"%a&#''1'<'<'*), *    !/ sA GC"G2ACCC GC G2F;BF&F" F&%F;& F2/F;1F22F;5G;GGcK|jd{ ||jd{y7#7#|jd{7wxYww)z9Use this server as a context manager to start and stop itNrstoprKs rLrIzWorkflowServer.contextmanagersJjjl J))+    $))+  s<A:A>A<AAAAAAcKtjdt|jdt j t |jjDcgc]}|j|c}d{|jj|jjycc}w7>w)Nz*Shutting down Workflow server. Cancelling z handlers.) rinfolenrXasynciogatherr^values_close_handlerclearrYrKhandlers rLrzWorkflowServer.stops 8T^^9L8MZ X nn:>t~~?T?T?V:W Xwd!!'* X     Y sA!C#B<;CC?Cc K|xsi}tj|jf||d|}tj|}tj d|d||j dd|jd{y7w)zRun the server.)hostportz#Starting Workflow server at http://: root_pathrON)uvicornConfigrJServerrrgetserve)rKrruvicorn_configconfigservers rLrzWorkflowServer.serves(-2Qt$Q.Q' 1$q~?Q?QR]_b?c>d e llnsA?B BB c<|j}tddtddddddiddd id dd idd id d dgd ddd idd id ddd gddd ddd dddd dddd dddddiddigdd gdd dddddid idgd did}|j|jS) Nz3.0.0z Workflows APIzllama-index-workflows)titlerschemasobjecttypearraystring)ritems)valuetypesrqualified_namer)r propertiesrequiredT)rnullable)rw completedr cancelled)renumz date-time)rformat)rrrzWorkflow result valuez$refz.#/components/schemas/EventEnvelopeWithMetadatanull) descriptiononeOf r~rrrrrrrr)r~rrrrz#/components/schemas/Handler)r-Handler HandlersList)openapir components)rJrr get_schemarN)rKrJgens rLopenapi_schemazWorkflowServer.openapi_schemasQhh",&'>? $,*0();29VXDV)W)/(:3982D + *1&(9 6%-/5x.@28(1C3;*N,4-&+"8@;.W,4.904/" -5.9041" 3;)M3J-34d)**0(8 .&+"3"+F)K+$Z%- *,3.46T-U-"+ *4  )oA C J L \~~cjj))rqcPKttdjSw)a --- summary: Health check description: Returns the server health status. responses: 200: description: Successful health check content: application/json: schema: type: object properties: status: type: string example: healthy required: [status] healthyr)rr& model_dump)rKrequests rLrhzWorkflowServer._health_checkgs!$N)<GGIJJs$&cjKt|jj}td|iSw)a --- summary: List workflows description: Returns the list of registered workflow names. responses: 200: description: List of workflows content: application/json: schema: type: object properties: workflows: type: array items: type: string required: [workflows] workflows)r^rUrr)rKrworkflow_namess rLrazWorkflowServer._list_workflows{s/&doo2245[.9::s13cKd|jvr tdd|jd}||jvrtdd|d|j|j}|jj |g}|r|j |g}|D]!}|j|j#tt|jSw) a --- summary: List workflow events description: Returns the list of registered workflow event schemas. parameters: - in: path name: name required: true schema: type: string description: Registered workflow name. responses: 200: description: List of workflow event schemas content: application/json: schema: type: object properties: events: type: array description: List of workflow event JSON schemas items: type: object required: [events] rSzname param is required) status_codedetailz Workflow 'z ' not found)events) path_paramsrrUrrVrextendappendmodel_json_schemarr(r)rKrrSrrt event_objsevents rLrmz$WorkflowServer._list_workflow_eventss6 ,, ,C8PQ Q""6* t &C*TF+8VW W&-- 3377bA  MM+ ,  9E   e557 8 96jITTVWWsCCchK|j|}|j||j|jd{\}}}|&|jjj |}nd} |j t|j|j|||d{}|j} |d{d} |j |jd{|j|d{t|jj!| S777m#t$r&} d} tjd| dYd} ~ d} ~ wwxYw7~#t$rYwxYw7v#t$r1} d} tjd| dt#d| | d} ~ wwxYww) a --- summary: Run workflow (wait) description: | Runs the specified workflow synchronously and returns the final result. The request body may include an optional serialized start event, an optional context object, and optional keyword arguments passed to the workflow run. parameters: - in: path name: name required: true schema: type: string description: Registered workflow name. requestBody: required: false content: application/json: schema: type: object properties: start_event: type: object description: 'Plain JSON object representing the start event (e.g., {"message": "..."}).' context: type: object description: Serialized workflow Context. handler_id: type: string description: Workflow handler identifier to continue from a previous completed run. kwargs: type: object description: Additional keyword arguments for the workflow. responses: 200: description: Workflow completed successfully content: application/json: schema: $ref: '#/components/schemas/Handler' 400: description: Invalid start_event payload 404: description: Workflow or handler identifier not found 500: description: Error running workflow or invalid request body Nr{rsr~r start_eventzError running workflow: Texc_inforrr)_extract_workflow_extract_run_paramsrsrSstart_event_classmodel_validaterr run_handlerrrrtaskrrto_response_modelrr) rKrrsrrr~input_evwrapperrrrs rLrbzWorkflowServer._run_workflows`))'2151I1I X&& 2 , (j  "((::II+VHH  00'X]]XEVEVW%$ 1G ))G L ||'!,,&& %%g. . .))+668f A,  L 7s;d KK L '  /  F LL3A37$L G1!56  s,F**F//F2cK|j|} |jjj} |jj j}tt||jS#t$r}t d|dd}~wwxYw#t$r}t d|dd}~wwxYww)ac --- summary: Get JSON schema for start event description: | Gets the JSON schema of the start and stop events from the specified workflow and returns it under "start" (start event) and "stop" (stop event) parameters: - in: path name: name required: true schema: type: string description: Registered workflow name. requestBody: required: false responses: 200: description: JSON schema successfully retrieved for start event content: application/json: schema: type: object properties: start: description: JSON schema for the start event stop: description: JSON schema for the stop event required: [start, stop] 404: description: Workflow not found 500: description: Error while getting the JSON schema for the start or stop event z2Error getting schema of start event for workflow: rrNz1Error getting schema of stop event for workflow: r) rrsrrrrstop_event_classrr*r)rKrrsstart_event_schemarstop_event_schemas rLrdz!WorkflowServer._get_events_schemasB))'2 !)!2!2!D!D!V!V!X    ( 1 1 B B T T V  "(/@ jl   KA3O   J1#N  sEC$B$B#$C B  BB  C# C,B<<CCcK|j|} t|j}t t |jjS#t$r}t d|dd}~wwxYww)a --- summary: Get the representation of the workflow description: | Get the representation of the workflow as a directed graph in JSON format parameters: - in: path name: name required: true schema: type: string description: Registered workflow name. requestBody: required: false responses: 200: description: JSON representation successfully retrieved content: application/json: schema: type: object properties: graph: description: the elements of the JSON representation of the workflow required: [graph] 404: description: Workflow not found 500: description: Error while getting JSON workflow representation z2Error while getting JSON workflow representation: rrN)graph) rr/rsrrrr)rr)rKrrsworkflow_graphrs rLrlz+WorkflowServer._get_workflow_representationPs>))'2 78I8IJN  !(H(H(J K V V X   KA3O  s'A<A1A< A9$A44A99A<cK|j|}|j||j|jd{\}}}|&|jjj |}nd} |j t|j|j|||d{}t|jjS77-#t$r}td|dd}~wwxYww)a --- summary: Run workflow (no-wait) description: | Starts the specified workflow asynchronously and returns a handler identifier which can be used to query results or stream events. parameters: - in: path name: name required: true schema: type: string description: Registered workflow name. requestBody: required: false content: application/json: schema: type: object properties: start_event: type: object description: 'Plain JSON object representing the start event (e.g., {"message": "..."}).' context: type: object description: Serialized workflow Context. handler_id: type: string description: Workflow handler identifier to continue from a previous completed run. kwargs: type: object description: Additional keyword arguments for the workflow. responses: 200: description: Workflow started content: application/json: schema: $ref: '#/components/schemas/Handler' 400: description: Invalid start_event payload 404: description: Workflow or handler identifier not found Nr{rzInitial persistence failed: rr) rrrsrSrrrrrrrrr) rKrrsrrr~rrrs rLrcz#WorkflowServer._run_workflow_nowait{sZ))'2151I1I X&& 2 , (j  "((::II+VHH  00'X]]XEVEVW%$ 1GG557BBDEE+,  5aS9s  sECPPQxH#@@J J""'')gll.F!,,&& ,,. .7,,. .' ,,. . sIAC4 C AC4C-C.C2!C4C C1C40C11C4cK|jd}|s tdd|j|d{}|jdvrdn|jdk(rd nd }|j }|j sd|d <nY|j j }|d k(r%|j jjd n|j j|d <t|| S7w)a/ --- summary: Get workflow result (deprecated) description: | Deprecated. Use GET /handlers/{handler_id} instead. Returns the final result of an asynchronously started workflow, if available. parameters: - in: path name: handler_id required: true schema: type: string description: Workflow run identifier returned from the no-wait run endpoint. deprecated: true responses: 200: description: Result is available content: application/json: schema: type: object 202: description: Result not ready yet content: application/json: schema: type: object 404: description: Handler not found 500: description: Error computing result content: text/plain: schema: type: string r~Handler ID is requiredrrNrwrrrrzworkflows.events.StopEventr) rrr rrrrrrr)rKrr~ handler_datarresponse_modelrs rLrez#WorkflowServer._get_workflow_resultsH((6 '?SQ Q!// ;; ""i/ ""k1 &002""'+N8 $&&55D77##))--h7!((.. 8 $ N??), "types": [], "type": , "qualified_name": , } Event queue is mutable. Elements are added to the queue by the workflow handler, and removed by any consumer of the queue. The queue is protected by a lock that is acquired by the consumer, so only one consumer of the queue at a time is allowed. parameters: - in: path name: handler_id required: true schema: type: string description: Identifier returned from the no-wait run endpoint. - in: query name: sse required: false schema: type: boolean default: true description: If false, as NDJSON instead of Server-Sent Events. - in: query name: include_internal required: false schema: type: boolean default: false description: If true, include internal workflow events (e.g., step state changes). - in: query name: acquire_timeout required: false schema: type: number default: 1 description: Timeout for acquiring the lock to iterate over the events. - in: query name: include_qualified_name required: false schema: type: boolean default: true description: If true, include the qualified name of the event in the response body. responses: 200: description: Streaming started content: text/event-stream: schema: type: object description: Server-Sent Events stream of event data. properties: value: type: object description: The event value. type: type: string description: The class name of the event. types: type: array description: Superclass names from MRO (excluding the event class and base Event). items: type: string qualified_name: type: string description: The qualified name of the event. required: [value, type] 404: description: Handler not found r~acquire_timeout1include_internalfalsetrueinclude_qualified_namessezInvalid acquire_timeout: ''rrNrr>rrrzHandler is completedrrztext/event-streamzapplication/x-ndjsontimeout#No lock available to acquire after s timeoutcK23d{}st|trtj|}|j }r d|dn|dt j dd{u7p76yw)N)rzdata: z  r) isinstancerr- from_eventmodel_dump_jsonrsleep)rrenvelopepayload generatorrrrs rL event_streamz3WorkflowServer._stream_events..event_streams( ' 'e'Ju>S,T4??2H#224"7)400$IR.(mmA&&& '' )s8BA?A;A?A'B5A=6B;A?=B?B) media_type)rrrTzAsyncGenerator[str, None])r query_paramsrlowerfloat ValueErrorrrXrZrr1rqueueemptyrracquire_events_streamNoLockAvailabler)rKrr~rr persistedrr)rr(r'rrrs @@@@rLrfzWorkflowServer._stream_eventsBs*b((6 &&**+C  $M ._parse_list_paramsu'..66zBCF --11*=C 17I5;;=ekkmIFIMJs A=$A=rr>rrwrrNrxr5r)r)r8rrTzlist[str] | None)rr^r3set intersectionrZrr1r$r~rrrr isoformatrrrrr-r"rr%r) rKrr: status_valuesrzallowed_status_valuesrypersistent_handlershrs ` rLrizWorkflowServer._get_handlerssfL *(3 ,_=. ( V S./<<]KL   %)$8$8$>$> 9?O P%  ")  <<ooxxxx78||1<<11378||1<<113;<>>Q^^557tgg881;;AHHE   0%@KKMNN'  s%BF#FF# C*F6&F#F#cBK|jd}|jj|}|"t|jr t dd|a|j |d{}t|jr t ddtjd|dt dd|j}|j}| t d d  |jd{}|jd }|jd } |s t d d tj||j|j} |j'| | t)t+dj-S7%7#t $r} t t#| dd} ~ wt$$r} t d| dd} ~ wwxYw#t$$r} t d| dd} ~ wwxYw#t $rt$$r} t d| d d} ~ wwxYww)a --- summary: Send event to workflow description: Sends an event to a running workflow's context. parameters: - in: path name: handler_id required: true schema: type: string description: Workflow handler identifier. requestBody: required: true content: application/json: schema: type: object properties: event: description: Serialized event. Accepts object or JSON-encoded string for backward compatibility. oneOf: - type: string description: JSON string of the event envelope or value. examples: - '{"type": "ExternalEvent", "value": {"response": "hi"}}' - type: object properties: type: type: string description: The class name of the event. value: type: object description: The event value object (preferred over data). additionalProperties: true step: type: string description: Optional target step name. If not provided, event is sent to all steps. required: [event] responses: 200: description: Event sent successfully content: application/json: schema: type: object properties: status: type: string enum: [sent] required: [status] 400: description: Invalid event data 404: description: Handler not found 409: description: Workflow already completed r~NzWorkflow already completedrrzHandler z is running but not in memory.zHandler expiredzContext not availablerrstepzEvent data is requiredrzFailed to deserialize event: )rCzFailed to send event: sentrzError processing request: )rrXrr+rrr rwarningrrjsonr,parse_event_registryrr.rr send_eventrr'r) rKrr~rr rrbody event_strrCrrs rLrgzWorkflowServer._post_event*s%t((6 ..$$Z0  #6w~~#F'CQTU U ?!%!3!3J!??L"<#6#67#7S *5STU#+<#NN%%kk ;'>CP P$  'D)I88F#D#+CQTUU %++t33G4I4IJ u40  1 @ K K MN Na@((( D#3q6sCC #:1#>C  #3A37S    3A37S  sA%H'F (A3HG5/F05G5&/FG)"G5 HG5 GF00 G<G  GG5 G2G--G22G55HHHHcK|jd}|jjddjdk(}|jj|}||s t dd||j |d{|r@|jjt|g d{}|d k(r t ddtt|rd nd jS7m7?w)a --- summary: Stop and delete handler description: | Stops a running workflow handler by cancelling its tasks. Optionally removes the handler from the persistence store if purge=true. parameters: - in: path name: handler_id required: true schema: type: string description: Workflow handler identifier. - in: query name: purge required: false schema: type: boolean default: false description: If true, also deletes the handler from the store, otherwise updates the status to cancelled. responses: 200: description: Handler cancelled and deleted or cancelled only content: application/json: schema: type: object properties: status: type: string enum: [deleted, cancelled] required: [status] 404: description: Handler not found r~purgerrNrrrrrdeletedrr) rr*rr+rXrrrZdeleter1rr#r)rKrr~rMr n_deleteds rLrkzWorkflowServer._cancel_handlersH((6 $$((':@@BfL..$$Z0 ?5':L L  %%g. . . "2299J<8IA~#+>CPP !$)y{ jl   /s$A?C3C//C31C12>C31C3cd|jvr tdd|jd}||jvr tddt||j|S)NrSz'name' parameter missingrrzWorkflow not foundrr{)rrrUr)rKrrSs rLrz WorkflowServer._extract_workflows^ ,, ,'AsS S""6* t &';M M4$//$2GHHrqcK |jd{}|jd}|jdi}|jd|}|jd} d} |d t j ||j ||j} | 0t| |jstd |jdd} |rtj|| } nr| rp|jjt| g|gd g d{} t| dk(r tddtj|| dj} | xs t!} | | | fS7T#t$r}td|dd}~wwxYw#t$r}td |dd}~wwxYw7#t$rt$r}td|dd}~wwxYww)NzInvalid JSON body: rrrkwargsrr~)explicit_eventz$Validation error for 'start_event': z#Start event must be an instance of r|r)rrzryrrrzError processing request body: r)rFrrrr,rGrHrr!rrrZrr1rrnanoid) rKrrsrrJr context_data run_kwargsstart_event_datar~rrpersisted_handlerss rLrz"WorkflowServer._extract_run_paramss;  W$\\^+ 88I.L(B/J#xx zB ,/JK+ "/"5"5(,,];'/'A'A#K*:!;!;4(!DXE_E_D`a$' G!++XLQ+/+?+?+E+E '1l*7#.-,&")*a/'/BPSTT!++H6H6K6O6OP#/vxJ[*5 5c, W#-@+DRUVV W"!'!EaSI$'$&   8<#  sGE/E,E/A F3(1FA=F3F1AF3+G,E// F 8FF  F3 F.F))F..F33GGGGcKtd|i5|jj||}|j||j|d{}|cdddS7#1swYyxYww)z6Start a workflow and return a wrapper for the handler.r~)rrN)rrsrun_run_workflow_handlerrS)rKrsr~rrrrs rLrzWorkflowServer._start_workflow/s|lJ7 8 ''++',G!66HMM7G      s.A,>A AA  A,A  A)%A,c |Ktj}tjtj }t ||dtj|||djj }|jd{|j<dfd }|j||S7.w)zP Creates a wrapper for the handler and starts streaming events. N) rr.rconsumer_mutexr~rrrrrZr_c~KjjdjjdywrH)rXpoprY)r~rKsrL on_finishz7WorkflowServer._run_workflow_handler..on_finish\s/ NN  z4 0 MM  j$ /s:=rarTNone) rQueuerrrrrLockrZr_ checkpointrXstart_streaming)rKr~rrr.rrras`` rLr\z$WorkflowServer._run_workflow_handlerAs '.mmo\\(,,/ ""<<>!'!! 00!%!:!:    """%,z" 0 )4 #sBB< B: /B<cK|jjs> |jj |jj d{|j |j d{|j j|jd|jj|jdy#t$rYwxYw7#t$rYwxYw7qw)zClose and cleanup a handler.N) rrcancelr cancel_runrrXr`r~rYrs rLrzWorkflowServer._close_handlerds""'') ##**, ))44666 << #,,   7--t4 ',,d3  7   sdC*CCCCC*6C(7AC* CC*CC*C C%"C*$C%%C*c |j|jDcic]}|j|}}|j|jj |dxsgDcic]}|j|c}|Scc}wcc}wrH)rUr__name__rrVr)rKrrrs rLrHzWorkflowServer._event_registryws(, (F(M(MN1QNN 0044]DIOR  A    O s A<B)r:zlist[Middleware] | Noner;zAbstractWorkflowStore | Noner< list[float]rH)rSrrsrrtzlist[type[Event]] | NonerTrd)rTz'WorkflowServer')rTz&AsyncGenerator['WorkflowServer', None]rc) localhostPN)rrrintrzdict[str, Any] | NonerTrd)rTdict)rrrTr)r~rrTr$)rrrTr)rrrTr)rrrsrrrrTz-tuple[Context | None, StartEvent | None, str])NN) rsrr~rrzStartEvent | NonerzContext | NonerTr)r~rrrrr"rTr)rrrTrd)rrrTzdict[str, type[Event]])rm __module__ __qualname__rprurrrIrrrrhrarmrbrdrlrcr rerjrfrirgrkrrrr\rrHrqrLr8r8Es/37;,/8 o ,o 5 o ) o j7; >>>4 >  >*X 04 .   "P*lK(;,+XZYv5 n) VCFJ/(<@|.K`LO\XOtxt< BI>>*2>CF> 6>H*."&  '     $!!.1! > $D &,,;;##****;;**262B2B2F2F  $$,,.B  rqcVK|jj|d{y7wrH)rZrrzs rLpersistz_WorkflowHandler.persists ""))*555s )')c Kt|j} |j} |j|d{y#t$r-}tj d|j d|dd}~wwxYw7>#t$r}|r|jdnd}|Ptj d|j dd |jj#t$rYwxYwtj d|j d|d |tj|d{7Yd}~nd}~wwxYww) zBPersist with retry/backoff; cancel handler when retries exhausted.zFailed to checkpoint handler zV to persistent state. Is there non-serializable state in an event or the state store? TrNrz* after final attempt. Failing the handler.z. Retrying in z seconds: ) r^r_r{rrrr~r}r`rrjrr$)rKbackoffsrrbackoffs rLrgz_WorkflowHandler.checkpointse112 ,,.J -ll:...  LL//@AWXYWZ[     / --5(,,q/4?LL77HHrs!%! ((//1% 3DOO3DNSZR[[efgehimmG,,,! - sEAEBA>BE A;(A66A;;E>B E #>#@'R R R    11 S## !5gYiH  s+A52AA AA5 A#A22A5cK |jjr(|j^|jjsCg}|jjsD|j |jj |jjsD|D]}| t j|jj}|j}t j|||hn|ht jd{\}}||vr|d{n|jnD|jjs|j|jjsC|j2|jjr|jd{|jjy777##|j3|jjr|jd{7|jjwxYww)z Converts the queue to an async generator while the workflow is still running, and there are still events. For better or worse, multiple consumers will compete for events N) return_when)r.r/rrr get_nowaitrrrwaitFIRST_COMPLETEDrjrwrr^release)rKravailable_eventsrqueue_get_task task_waitablerpendings rLrz_WorkflowHandler._iter_eventsPs  *jj&&( %diinn.>#% ****,$++DJJ,A,A,CD****,- EK 6=6I6IJJNN$7!% &-ll$0$]3() ' 7 7 '! g "T) ..."))+-jj&&( %diinn.>.*t/?/?/D/D/Foo'''    ' ' )!/(*t/?/?/D/D/Foo'''    ' ' )slIB!G(&A4G(G"G(+G$,3G(!&G( 9IG&I"G($G(&I(:I"H%# II)rTr2)rr2rTrdrc)rTr$)rr2rTr$)rTr3)rTz str | None)rTzStopEvent | None)razCallable[[], Awaitable[None]]rTrd))rr,rTzAsyncGenerator[Event, None])rmrsrt__doc____annotations__rwr{r}rgr staticmethodrpropertyrrrrhrfr0rrurqrLrrsA   ##  O!!+*%%7;J4;(6-B $  *  55R!$H ! 2 2 $ 2"*rqrceZdZdZy)r1z;Raised when no lock is available to acquire after a timeoutN)rmrsrtrrurqrLr1r1usErqr1c"eZdZUded<ded<y)rrrSrrsN)rmrsrtrrurqrLrr{s Irqr__main__zGenerate OpenAPI schema)rz--outputz openapi.jsonzOutput file path)rdefaulthelpw)indentfpzOpenAPI schema written to )j __future__rrrFlogging contextlibr dataclassesrrrimportlib.metadatarpathlibr typingr r r r rr&llama_index_instrumentation.dispatcherrstarlette.applicationsrstarlette.exceptionsrstarlette.middlewarerstarlette.middleware.corsrstarlette.requestsrstarlette.responsesrrstarlette.routingrstarlette.schemasrstarlette.staticfilesrrrrworkflows.eventsrrrrr r!workflows.handlerr"workflows.protocolr#r$r%r&r'r(r)r*r+&workflows.protocol.serializable_eventsr,r-r.workflows.representation_utilsr/(workflows.server.abstract_workflow_storer0r1r2r3&workflows.server.memory_workflow_storer4workflows.typesr5workflows.utilsr6rU getLoggerrr8rrr1rrmargparseArgumentParserparser add_argumentr parse_argsargsrr dict_schemaopenoutputfdumpprintrurqrLrs# *!'&AAB,.+4&?#--'.    F G&.    zzz) o*o* o*d i     z $X $ $1J KF n;M    D  F'')K dkk3 /1 +aA./ &t{{m 45//s G  G