iSSKJr SSKrSSKJr SSKJr SSKrSSKrSSK J r SSK J r SSK JrJrJrJr SSKJrJr SS KJr SSKrSS KJr SS KJr SS KJr SS KJr SSKJ r SSK!J"r"J#r# SSK$J%r% SSK&J'r' SSK(J)r) SSK*J+r+J,r, SSK-J.r.J/r/J0r0J1r1J2r2J3r3 SSK4J5r5 SSK6J7r7J8r8J9r9J:r:J;r;Jr>J?r? SSK@JArAJBrBJCrCJDrD SSKEJFrF SSKGJHrH SSKIJJrK SSKLJMrM SSKNJOrOJPrPJQrQ \R"5rS"SS5rT\"S S!55rU"S"S#\V5rW\"S$S%55rX\YS&:XaSSKZrZ\ZR"S'S(9r\\\RS)\^S*S+S,9 \\R5r`\T"5ra\aR5rc\d"\`RS-5rf\R"\cS.\fS/9 SSS5 \h"S0\`R35 gg!,(df  N%=f)1) annotationsN)asynccontextmanager) dataclass)version)Path)AnyAsyncGeneratorCallable Awaitable)datetimetimezone)instrument_tags) Starlette) HTTPException) Middleware)CORSMiddleware)Request) JSONResponseStreamingResponse)Route)SchemaGenerator) StaticFiles)ContextWorkflow)EventInternalDispatchEvent StartEvent StepStateStepStateChanged StopEvent)WorkflowHandler) CancelHandlerResponse HandlerDataHandlersListResponseHealthResponseSendEventResponseWorkflowEventsListResponseWorkflowGraphResponseWorkflowSchemaResponseis_status_completed)AbstractWorkflowStore HandlerQueryPersistentHandlerStatus)MemoryWorkflowStore) RunResultT)_nanoid)_extract_workflow_structure)EventValidationErrorEventEnvelopeWithMetadata EventEnvelopec\rSrSrSSSS/S.S"SjjrS#S$SjjrS%Sjr\S&S j5rS'S jr S(S)S jjr S*S jr S+S jr S+Sjr S+SjrS+SjrS+SjrS+SjrS+SjrS,SjrS+SjrS+SjrS-SjrS+SjrS+SjrS+SjrS.SjrS/SjrS0S1SjjrS2SjrS3SjrS4S jrS!r g)5WorkflowServerGNg?) middlewareworkflow_storepersistence_backoffc^0Tl0Tl0Tl0Tl0TlUbUO [ 5Tl[[5RS- Tl [U5Tl U=(d [[SS/S/SS9/Tl[!STR"S/S9[!S TR$S /S9[!S TR&S /S9[!S TR(S/S9[!S TR*S/S9[!STR,S/S9[!STR.S /S9[!STR0S/S9[!STR2S/S9[!STR4S/S9[!STR6S /S9[!STR8S/S9[!STR:S/S9/ Tl[>SU4Sjj5n[ATR<TRUS9Tl!TRBRES[GTRSS9SS9 g)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}/eventsc># TR5IShvN S7v SSS5IShvN gNN!,IShvN (df  g=f7fN)contextmanager)appselfs Q/home/james-whalen/.local/lib/python3.13/site-packages/workflows/server/server.pylifespan)WorkflowServer.__init__..lifespans,**,,-,,,,,s:A3A7 A5AAAA A A)routesr;rN/) directoryhtmlui)rKname)rKrreturnzAsyncGenerator[None, None])$ _workflows_additional_events _contexts _handlers_resultsr/_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_routesrrrKmountr)rLr;r<r=rNs` rM__init__WorkflowServer.__init__Hs^02GI-/68/1 ,8N>Q>S !N11H<$()<$=!%  #'"e"e"&   *  $$  '""  .))  *''  '))  &##  &    ""  ""  (**  /$$  211  *** {B  H    <<''   [4+<+<4Ht  cDX RU'UbX0RU'ggrI)rWrX)rLrUworkflowadditional_eventss rM add_workflowWorkflowServer.add_workflows) !)  (,= # #D ) )rtcx# URR[S/[URR 55S95IShvN nUHnnURUR nUR[UR US9UR[R"X2RS9S9IShvN Mp U$NzN ![an[RSURSUR S U35 [ R""[$R&5nURR)[+URUR S UR,[/U5SUR0UUURS 9 5IShvN O![a Of=fSnAGM`SnAff=f7f) zPResumes previously running workflows, if they were not complete at last shutdownrunning status_inworkflow_name_inNrUrvrvdata)rv handler_idcontextzFailed to resume handler z for workflow z: failed r workflow_namestatusrun_iderrorresult started_at updated_at completed_atctx)r\queryr,r`rWkeysr_start_workflow_NamedWorkflowrr from_dictr Exceptionloggerrr nowr utcupdater-rstrr)rLhandlers persistentrvers rMstartWorkflowServer.starts--33 $+T__=Q=Q=S8T    #Jz'?'?@H **+'55 *44#--xnnU +#F Q   / 0E0E/FnU_UmUmTnnpqrpst",,x||4C..55)'1'<'<*4*B*B#+#-#4#4"%a&#''1'<'<'*), *    !/ sA F: C "F:/A C ;C<C F:C  F72F2B FFFF2 F(%F2'F((F2+F:2F77F:c# UR5IShvN U7v UR5IShvN gN#N!UR5IShvN f=f7f)z9Use this server as a context manager to start and stop itNrstoprLs rMrJWorkflowServer.contextmanagersHjjl J))+    $))+  s<A:A>A<AAAAAAc# [RS[UR5S35 [R "[ URR55Vs/sHoRU5PM sn6IShvN URR5 URR5 gs snfN>7f)Nz*Shutting down Workflow server. Cancelling z handlers.) rinfolenrZasynciogatherr`values_close_handlerclearr[rLhandlers rMrWorkflowServer.stops 8T^^9L8MZ X nn:>t~~?T?T?V:W X:Ww!!'*:W X     Y sA"C$B?>CC?Cc "# U=(d 0n[R"UR4XS.UD6n[R"U5n[R SUSUUR SS535 UR5IShvN gN7f)zRun the server.)hostportz#Starting Workflow server at http://: root_pathrQN)uvicornConfigrKServerrrgetserve)rLrruvicorn_configconfigservers rMrWorkflowServer.serves|(-2QtQ.Q' 1$q~?Q?QR]_b?c>d e llnsBBB Bc<URn[SS[S5S.SSSS0SSS 0S .SS 0SS 0S .S S/S .SSS 0SS 0S SS.S /SQS.S SS.S SSS.S SSS.S SS.SSS0SS0/S.S. /SQS .SSSSS0S .0S/S .S.0S.5nURUR5$) Nz3.0.0z Workflows APIzllama-index-workflows)titlerschemasobjecttypearraystring)ritems)valuetypesrqualified_namer)r propertiesrequiredT)rnullable)r{ completedr cancelled)renumz date-time)rformat)rrrzWorkflow result valuez$refz.#/components/schemas/EventEnvelopeWithMetadatanull) descriptiononeOf rrrrrrrrr)rrrrrz#/components/schemas/Handler)r5Handler HandlersList)openapir components)rKrr get_schemarP)rLrKgens rMopenapi_schemaWorkflowServer.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))rtcJ# [[SS9R55$7f)aR --- 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)rLrequests rMrjWorkflowServer._health_checkis $N)<GGIJJs!#cj# [URR55n[SU05$7f)as --- 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`rWrr)rLrworkflow_namess rMrcWorkflowServer._list_workflows}s.&doo2245[.9::s13c# SUR;a [SSS9eURSnX R;a[SSUS3S9eURURnURR U/5nU(aUR U5 /nUH"nURUR55 M$ [[US9R55$7f) aL --- 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] rUzname param is required) status_codedetailz Workflow 'z ' not found)events) path_paramsrrWrrXrextendappendmodel_json_schemarr'r)rLrrUrrw event_objsevents rMro$WorkflowServer._list_workflow_eventss6 ,, ,C8PQ Q""6*  &C*TF+8VW W&-- 3377bA  MM+ , E   e557 86jITTVWWsCC cJ# URU5nURXRUR5IShvN up4nUb&URRR U5nOSnUR [URURS9UUUS9IShvN nURnUIShvN Sn URbURIShvN URU5IShvN [UR5R!5U S 9$NNNk![a#n Sn [RSU 3SS9 Sn A NSn A ff=fNy![a Nf=fNr![a+n Sn [RSU 3SS9 [#SU 3U S 9eSn A ff=f7f) ap --- 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 Nrrvrr start_eventzError running workflow: Texc_inforrr)_extract_workflow_extract_run_paramsrvrUstart_event_classmodel_validaterr run_handlerrrrtaskrrto_response_modelrr) rLrrvrrrinput_evwrapperrrrs rMrdWorkflowServer._run_workflows`))'2151I1I && 2 , (j  "((::II+VHH  00'X]]XEVEVW%$ 1G ))G L ||'!,,&& %%g. . .))+668f A,  L 7s;d K L '  /  F LL3A37$L G1!56  s;F#D!2F#11E+"D##E+4D'9D%:D' E+EEE"E+6E)7)E+ F##E+%D'' E1E E+EE+E E&#E+%E&&E++ F 5&FF  F#c|# URU5nURRR5nURR R5n[[X5S9R55$![an[ SU3SS9eSnAff=f![an[ SU3SS9eSnAff=f7f)ak --- 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) rrvrrrrstop_event_classrr)r)rLrrvstart_event_schemarstop_event_schemas rMrf!WorkflowServer._get_events_schemasB))'2 !)!2!2!D!D!V!V!X    ( 1 1 B B T T V  "( jl   KA3O   J1#N  sEB<$A>$B B<> B BBB< B9' B44B99B<c# URU5n[UR5n[ [ UR5S9R55$![an[ SU3SS9eSnAff=f7f)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) rr3rvrrrr(rr)rLrrvworkflow_graphrs rMrn+WorkflowServer._get_workflow_representationRs>))'2 89J9JKN  !(H(H(J K V V X   KA3O  s'A7A.A7 A4" A//A44A7c# URU5nURXRUR5IShvN up4nUb&URRR U5nOSnUR [URURS9UUUS9IShvN n[UR5R55$NN-![an[SU3SS9eSnAff=f7f)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 NrrzInitial persistence failed: rr) rrrvrUrrrrrrrrr) rLrrvrrrrrrs rMre#WorkflowServer._run_workflow_nowait}sZ))'2151I1I && 2 , (j  "((::II+VHH  00'X]]XEVEVW%$ 1GG557BBDEE+,  5aS9s  sE;C1C2C111C"C#C'(C1C C. C))C..C1c# URRU5nUcVURR[ U/S95IShvN nU(d [ SSS9eUSn[ RU5$URR5(a2URb%URIShvN UR5$UR5$NN&![a UR5$f=f7f)N handler_id_inHandler not foundrrr) rZrr\rr,r_WorkflowHandlerhandler_data_from_persistentrdonerrr)rLrrfoundexistings rM _load_handlerWorkflowServer._load_handlers..$$Z0 ?..44J<8E#+>CPPQxH#@@J J""''))gll.F!,,&& ,,. .7,,. .' ,,. . sIAC;CAC;$C3C4C8!C;C C8&C;7C88C;c# URSnU(d [SSS9eURU5IShvN nURS;aSOURS:XaS OS nUR 5nUR (dSUS 'OZUR R nUS :Xa%UR RRS 5OUR RUS '[XTS 9$N7f)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 rHandler ID is requiredrrNr{rrrrzworkflows.events.StopEventr) rrrrrrrrrr)rLrr handler_datarresponse_modelrs rMrg#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 racquire_timeout1include_internalfalsetrueinclude_qualified_namessezInvalid acquire_timeout: ''rrNrr>rrrzHandler is completedrrztext/event-streamzapplication/x-ndjsontimeout#No lock available to acquire after s timeoutc&># TShvN nT(d[U[5(aM'[R"UTS9nUR 5nT(a SUS37v OUS37v [ R "S5IShvN MNN g7f)N)r)zdata: z  r) isinstancerr5 from_eventmodel_dump_jsonrsleep)rrenvelopepayload generatorr&r)r*s rM event_stream3WorkflowServer._stream_events..event_streams( 'e'Ju>S,T,T4??2H#224"7)400$IR.(mmA&&& '' )s8BBB BA6BB B B BB) media_type)rrrVzAsyncGenerator[str, None])r query_paramsrlowerfloat ValueErrorrrZr\rr,rqueueemptyrracquire_events_streamNoLockAvailabler)rLrrr.r persistedrr=rr;r:r&r)r*s @@@@rMrhWorkflowServer._stream_eventsDs.b((6 &&**+C  $M [TRRU55nU(d'TRRU5=(d SnU/nUVs/sH)o3R 5(dMUR 5PM+ nnU(dgU$s snf)N)r`r>getlistrstrip) param_namersinglerrs rM_parse_list_param7WorkflowServer._get_handlers.._parse_list_paramsv'..66zBCF --11*=C 17I;;=mekkmFIMJs B4Brr>rr{rrNr|rJr)r)rMrrVzlist[str] | None)r`set intersectionr\rr,r#rrrrr isoformatrrrrr5r5rr$r) rLrrO status_valuesr~allowed_status_valuesr}persistent_handlershrs ` rMrkWorkflowServer._get_handlerssVL *(3 ,_=. ( *+88G H  %)$8$8$>$> 9 P%  ") ) <<ooxxxx78||1<<11378||1<<113;<>>Q^^557tgg881;;AHHE )  0%@KKMNN'  s%A*F-F.F6C?F5#FFc,# URSnURRU5nUb$[UR5(a [ SSS9eUc`UR U5IShvN n[UR5(a [ SSS9e[RSUS35 [ SSS9eURnURnUc [ S S S9eUR5IShvN nURS 5nURS 5n U(d [ S SS9e[R"XRUR55n UR'XS9 [)[+SS9R-55$GNN![ an [ [#U 5SS9eSn A f[$an [ SU 3SS9eSn A ff=f![$an [ SU 3SS9eSn A ff=f![ a e[$an [ SU 3S S9eSn A ff=f7f)aX --- 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 rNzWorkflow already completedr1rzHandler z is running but not in memory.zHandler expiredzContext not availablerrstepzEvent data is requiredrzFailed to deserialize event: )rZzFailed to send event: sentrzError processing request: )rrZrr*rrrrwarningrrjsonr6parse_event_registryrr4rr send_eventrr&r) rLrrrrrrbody event_strrZrrs rMriWorkflowServer._post_event*s%t((6 ..$$Z0  #6w~~#F#F'CQTU U ?!%!3!3J!??L"<#6#677#7S *5STU#+<#NN%%kk ;'>CP P$  'D)I88F#D#+CQTUU %++33G4I4IJ u0  1 @ K K MN Na@((( D#3q6sCC #:1#>C  #3A37S    3A37S  sA)H+F,A0HG+0F 17G+)/F G (G+H G+ G F** G 7 GG  G+ G( G##G((G++H? H  HHc# URSnURRSS5R5S:HnURRU5nUcU(d [ SSS9eUbUR U5IShvN U(a;URR[U/S 95IShvN nUS :Xa [ SSS9e[[U(aS OS S 9R55$NoN?7f)as --- 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 rpurger'r(Nrrrrrdeletedrr) rr>rr?rZrrr\deleter,rr"r)rLrrrer n_deleteds rMrmWorkflowServer._cancel_handlersH((6 $$((':@@BfL..$$Z0 ?5':L L  %%g. . . "2299J<8IA~#+>CPP !$)y{ jl   /s$BC9C51C97C78>C97C9cSUR;a [SSS9eURSnX R;a [SSS9e[X RUS9$)NrUz'name' parameter missingrrzWorkflow not foundrr)rrrWr)rLrrUs rMr WorkflowServer._extract_workflowsZ ,, ,'AsS S""6*  &';M M4//$2GHHrtc# UR5IShvN nURS5nURS05nURSU5nURS5n Sn Ubd[R "UUR U5URS9n U b1[XR5(d[S UR3SS9eSn U(a[R"X&S 9n OrU (akURR[U /U/S /S 95IShvN n [U 5S:Xa [SSS9e[R"X,SR5n U =(d [!5n XU 4$GN[![an[SU3SS9eSnAff=f![an[S U3SS9eSnAff=fN![a e[an[SU3SS9eSnAff=f7f)NzInvalid JSON body: rrrkwargsrr)explicit_eventz$Validation error for 'start_event': z#Start event must be an instance of rr)rr~r}rrrzError processing request body: r)r]rrrr6r^r_rr4rrr\rr,rrnanoid) rLrrvrrar context_data run_kwargsstart_event_datarrrpersisted_handlerss rMr"WorkflowServer._extract_run_paramss ;  W$\\^+ 88I.L(B/J#xx zB ,/JK+ "/"5"5(,,];'/'A'A#K*:!;!;44(!DXE_E_D`a$' G!++XQ+/+?+?+E+E '1l*7#.-,&")*a/'/BPSTT!++H6K6O6OP#/vxJ*5 5c, W#-@+DRUVV W"!'!EaSI$'$&   8<#  sGE6E3E6A F6)/FBF6F4AF62G3E66 F F  FF6 F1 F,,F11F66G GGGc# [SU05 URRUUS9nURX!RU5IShvN nUsSSS5 $N!,(df  g=f7f)z6Start a workflow and return a wrapper for the handler.r)rrN)rrvrun_run_workflow_handlerrU)rLrvrrrrrs rMrWorkflowServer._start_workflow/solJ7 8''++',G!66MM7G9 8  9 8s.A-:A A A A-A A*&A-cz^^# [R"5n[R"[R 5n[ UUS[R"5TUUUSTRTRS9 nUR5IShvN UTRT'SUU4SjjnURUS9 U$N-7f)z@ Creates a wrapper for the handler and starts streaming events. N) rrBrconsumer_mutexrrrrrr\rac~># TRRTS5 TRRTS5 g7frI)rZpopr[)rrLsrM on_finish7WorkflowServer._run_workflow_handler..on_finish\s. NN  z4 0 MM  j$ /s:=r}rVNone) rQueuer rr rrLockr\ra checkpointrZstart_streaming)rLrrrrBrrr}s`` rMrw$WorkflowServer._run_workflow_handlerAs '.mmo\\(,,/ ""<<>!'!! 00!%!:!:    """%,z" 0 0 )4 #sBB; B9 .B;c# URR5(d>URR5 URR 5IShvN UR bUR IShvN UR RURS5 URRURS5 g![a Nf=fN![a Nf=fNs7f)zClose and cleanup a handler.N) rrcancelr cancel_runrrZr|rr[rs rMrWorkflowServer._close_handlerds""'')) ##**, ))44666 << #,,   7--t4 ',,d3  7   sd C2CC CC C2<C0=AC2 CC2CC2C C-*C2,C--C2c URURVs0sHo"RU_M nnURURR US5=(d /Vs0sHnURU_M sn5 U$s snfs snfrI)rWr__name__rrXr)rLrrrs rMr_WorkflowServer._event_registryws(, (F(M(MN(M1Q(MN 0044]DIORO OA A O   O s B&B ) rXr_rYrZrbrar[rpr\rWrK)r;zlist[Middleware] | Noner<zAbstractWorkflowStore | Noner= list[float]rI)rUrrvrrwzlist[type[Event]] | NonerVr)rVz'WorkflowServer')rVz&AsyncGenerator['WorkflowServer', None]r) localhostPN)rrrintrzdict[str, Any] | NonerVr)rVdict)rrrVr)rrrVr#)rrrVr)rrrVr)rrrvrrrrVz-tuple[Context | None, StartEvent | None, str])NN) rvrrrrzStartEvent | NonerzContext | NonerVr)rrrrrr!rVr)rrrVr)rrrVzdict[str, type[Event]])!r __module__ __qualname____firstlineno__rrrxrrrJrrrrjrcrordrfrnrerrgrlrhrkrirmrrrrwrr___static_attributes__rtrMr8r8Gs/37;,/8 o ,o 5 o ) o j7; >>>4 >  >*X 04 .   "P*lK(;,+XZYv5 n) VCFJ/(<@|.K`LO\VOpxt< BI>>*2>CF> 6>H*."&  '     $!!.1! > $D &,,;;##****;;**262B2B2F2F  $$,,.B  rtcV# URRU5IShvN gN7frI)r\rrs rMpersist_WorkflowHandler.persists""))*555s )')c # [UR5nUR5nURU5IShvN g![a*n[R SUR SU3SS9 eSnAff=fNA<A>E  A9%A44A99E <A>> E?D?C#"D?# C0-D?/C00AD?4D75D?:E ?EE c [URURURRUR UR R5URR5URbURR5OSURURb"[R"UR5S9 $SS9 $)z.Convert runtime handler to API response model.Nr)r#rrrrrrrSrrrrr5r5rs rMr"_WorkflowHandler.to_response_models,,##**;;002002  ,**446**{{&-77 D    rtc 0[URURURURUR bUR R 5O1[R"[R5R 5URbURR 5OSURbURR 5OSURURb"[R "UR5S9 $SS9 $)Nr)r#rrrrrrSr rr rrrrrr5r5)rs rMr-_WorkflowHandler.handler_data_from_persistents!,,$22$$$$$$0",,668hll+557$$0",,668&&2$00::<""  ,-77 8I8IJ  "#  rtcURR5(dgURR5(agURR5nUbgg)z7Get the current status by inspecting the handler state.r{rrr)rrr exceptionrLexcs rMr_WorkflowHandler.statussS$$&&    % % ' '((* ?rtcURR5(dgURR5nUb [ U5$S$![Ra gf=frI)rrrrCancelledErrorrrs rMr_WorkflowHandler.errorsb$$&& "",,.C?s3x44%%  sA A#"A#cURR5(dgURR5$![Ra g[ a gf=frI)rrget_stop_eventrrrrs rMr_WorkflowHandler.resultsW$$&& ##224 4%%   s<A AAcT[R"URUS95Ulg)z;Start streaming events from the handler and managing state.rN)r create_taskrhr)rLr}s rMr _WorkflowHandler.start_streamings"''(;(;i(;(PQ rtc# [SUR05 UR5IShvN XlURR SS9ShvN n[ U[5(aUR[R:XaURR(a$URRR5OSnUc%[RSURS35 MUR5IShvN URR!U5 MNNN% URIShvN Os!["R$a& [R'SURS35 O:[(a.n[R+SURS U3SS 9 SnAOSnAff=fUR5IShvN SSS5 g!,(df  g=f7f) zHInternal method that streams events, updates status, and persists state.rNT)expose_internalz"Context state is None for handler z. This is not expected.z Workflow run z was cancelledz failed! r)rrrrr stream_eventsr4r step_stater NOT_RUNNINGrrrr\rB put_nowaitrrrrr)rLr}rstaters rMrh_WorkflowHandler._stream_eventss lDOO< =//# # #'O#//==d=S -eu&677((I,A,AA;?:J:J:N:N((,,446TX}@@QQhi!//+++ %%e,# $ -, T" &&&&)) M mDOO+ = =sG0GD,#GD2D.D2B0G D0 "G.D20G2G4E EE  G 7F:G F: $F50G5F::GGG G0 G-)G0c# [R"URR5US9IShvN UR US9$N![Ra [ SUS35ef=f7f)zP Acquires the lock to iterate over the events, and returns generator of events. r-Nr/r0)rwait_forrzacquire TimeoutErrorrE _iter_events)rLr.s rMrD&_WorkflowHandler.acquire_events_streamBsx  ""4#6#6#>#>#@'R R R    11 S## !5gYiH  s+A10A AA A1A $A..A1c# URR5(a.URGb|URR5(Gd\/nURR5(dJUR URR 55 URR5(dMJUHnU7v M [ R"URR55nURn[ R"UbXE1OU1[ RS9IShvN upgXF;aUIShvN 7v OUR5 OQURR5(dGM-URb"URR5(dGM\URb7URR5(aUR5IShvN URR5 gNNN#!URb8URR5(aUR5IShvN URR5 f=f7f)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)rBrCrrr get_nowaitrrrwaitFIRST_COMPLETEDrrrrzrelease)rLr.available_eventsrqueue_get_task task_waitablerpendings rMr_WorkflowHandler._iter_eventsPs  *jj&&(( %diinn.>.>#% ****,,$++DJJ,A,A,CD****,,-EK.6=6I6IJJNN$7!% &-ll$0$3() ' 7 7 '!  ") ..."))+-jj&&(( %diinn.>.>.*t/?/?/D/D/F/Foo'''    ' ' )!/(*t/?/?/D/D/F/Foo'''    ' ' )smI6B6H,H-?I6,H-I6 HHI6AI3I I33I6)rrrr)rVr-)rr-rVrr)rVr#)rr-rVr#)rVr.)rVz str | None)rVzStopEvent | None)r}zCallable[[], Awaitable[None]]rVr)r2)r.r@rVzAsyncGenerator[Event, None])rrrr__doc____annotations__rrrrr staticmethodrpropertyrrrrrhrDrrrrtrMrrsA   ##  O!!+*%%7;J4;(6-B $  *  55R!$H ! 2 2 $ 2"*"*rtrc\rSrSrSrSrg)rEiuz;Raised when no lock is available to acquire after a timeoutrN)rrrrrrrrtrMrErEusErtrEc*\rSrSr%S\S'S\S'Srg)ri{rrUrrvrN)rrrrrrrrtrMrr{s Irtr__main__zGenerate OpenAPI schema)rz--outputz openapi.jsonzOutput file path)rdefaulthelpw)indentfpzOpenAPI schema written to )i __future__rr contextlibr dataclassesrr]loggingimportlib.metadatarpathlibrtypingrr r r r r &llama_index_instrumentation.dispatcherrrstarlette.applicationsrstarlette.exceptionsrstarlette.middlewarerstarlette.middleware.corsrstarlette.requestsrstarlette.responsesrrstarlette.routingrstarlette.schemasrstarlette.staticfilesrrrrworkflows.eventsrrrrrr workflows.handlerr!workflows.protocolr"r#r$r%r&r'r(r)r*(workflows.server.abstract_workflow_storer+r,r-r.&workflows.server.memory_workflow_storer/workflows.typesr0workflows.utilsr1rorepresentation_utilsr3&workflows.protocol.serializable_eventsr4r5r6 getLoggerrr8rrrErrargparseArgumentParserparser add_argumentr parse_argsargsrr dict_schemaopenoutputfdumpprintrrtrMrs#*! &;;'B,.+4&?#--'.    G&.=    xxv) o*o* o*d i     z  $ $1J KF n;M    D  F'')K dkk3 1 +aA.  &t{{m 45 s F?? G