JiC0ddlmZddlZddlZddlmZmZmZddlm Z ddl m Z erddl m Z ddlmZmZdd lmZmZmZdd lmZmZmZmZmZdd lmZdd lmZdd lm Z ddl!m"Z"m#Z#e e$Z%ejLZ'Gdde(Z)Gdde)Z*y)) annotationsN) TYPE_CHECKINGAnyTuple)get_dispatcher)ValidationErrorContext) StepConfig StepFunction)WorkflowConfigurationErrorWorkflowRuntimeErrorWorkflowValidationError)EventHumanResponseEventInputRequiredEvent StartEvent StopEvent)WorkflowHandler)ResourceManager) RunResultT)get_steps_from_classget_steps_from_instancec eZdZdfd ZxZS) WorkflowMetac6t||||i|_yN)super__init___step_functions)clsnamebasesdct __class__s d/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/workflow.pyr zWorkflowMeta.__init__*s uc*79)r#strr$zTuple[type, ...]r%zdict[str, Any]returnNone)__name__ __module__ __qualname__r __classcell__)r&s@r'rr)s ::r(rc eZdZUdZded< d ddZddZeddZeddZ dd Z dd Z edd Z e dd Zdd Z ddZej$ d ddZddZddZy)Workflowa Event-driven orchestrator to define and run application flows using typed steps. A `Workflow` is composed of `@step`-decorated callables that accept and emit typed [Event][workflows.events.Event]s. Steps can be declared as instance methods or as free functions registered via the decorator. Key features: - Validation of step signatures and event graph before running - Typed start/stop events - Streaming of intermediate events - Optional human-in-the-loop events - Retry policies per step - Resource injection Examples: Basic usage: ```python from workflows import Workflow, step from workflows.events import StartEvent, StopEvent class MyFlow(Workflow): @step async def start(self, ev: StartEvent) -> StopEvent: return StopEvent(result="done") result = await MyFlow(timeout=60).run(topic="Pirates") ``` Custom start/stop events and streaming: ```python handler = MyFlow().run() async for ev in handler.stream_events(): ... result = await handler ``` See Also: - [step][workflows.decorators.step] - [Event][workflows.events.Event] - [Context][workflows.context.context.Context] - [WorkflowHandler][workflows.handler.WorkflowHandler] - [RetryPolicy][workflows.retry_policy.RetryPolicy] dict[str, StepFunction]r!Nc2||_||_||_||_|j |_|j |_|j|_ |rtj|nd|_ |xs t|_t|_y)a. Initialize a workflow instance. Args: timeout (float | None): Max seconds to wait for completion. `None` disables the timeout. disable_validation (bool): Skip pre-run validation of the event graph (not recommended). verbose (bool): If True, print step activity. resource_manager (ResourceManager | None): Custom resource manager for dependency injection. num_concurrent_runs (int | None): Limit on concurrent `run()` calls. N)_timeout_verbose_disable_validation_num_concurrent_runs_ensure_start_event_class_start_event_class_ensure_stop_event_class_stop_event_class_ensure_events_collected_eventsasyncio Semaphore_semr_resource_manager dispatcher _dispatcher)selftimeoutdisable_validationverboseresource_managernum_concurrent_runss r'r zWorkflow.__init__bs,   #5 $7!"&"@"@"B!%!>!>!@446 6IG  1 2t "2!F_5F%r(ct}|jjD]A}|j}|jD]$}t |t s|j|&Ct|}|dk(r'|jj}d|d}t||dkDr-|jj}d|d|d|d}t||jS) z Returns the StartEvent type used in this workflow. It works by inspecting the events received by the step methods. rzOAt least one Event of type StartEvent must be received by any step. (Workflow 'z(' has no @step that accepts StartEvent.)r z;Only one type of StartEvent is allowed per workflow, found :  in workflow ''.) set _get_stepsvalues _step_configaccepted_events issubclassraddlenr&r,rpop)rDstart_events_found step_func step_config event_type num_foundcls_namemsgs r'r8z"Workflow._ensure_start_event_classs 58E*113 7I&/&<&~~..H&Z'OQ -S1 1 ]~~..HMi[XZ[mZno (z- -S1 1%))+ +r(c|jS)zqThe `StartEvent` subclass accepted by this workflow. Determined by inspecting step input types. )r9rDs r'start_event_classzWorkflow.start_event_classs &&&r(c|jS)vReturns all known events emitted by this workflow. Determined by inspecting step input/output types. )r=r_s r'eventszWorkflow.eventss ||r(cvt}|jjD]}|j}|jdk(r|j D]$}t |ts|j|&|jD]$}t |ts|j|&t|S)rb_done) rNrOrPrQr, return_typesrSrrTrRlist)rD events_foundrXrYrZs r'r<z!Workflow._ensure_events_collecteds *- *113 1I&/&<&~~..H&Z'NP -S1 1 ]~~..HLYKWYZkYlm (z- -S1 1$((* *r(c|jS)zwThe `StopEvent` subclass produced by this workflow. Determined by inspecting step return annotations. )r;r_s r'stop_event_classzWorkflow.stop_event_classs %%%r(c t|dd}|sd|jd}t||jit||jvrd|jd}t|||j|j<y)z Adds a free function as step for this workflow instance. It raises an exception if a step with the same name was already added to the workflow. rQNzStep function z" is missing the `@step` decorator.zA step z> is already part of this workflow, please choose another name.)getattrr,rrr!)r"funcrYr]s r'add_stepzWorkflow.add_steps*1~t)L "4==/1STC)#. . ==P3C8PC>3Q3QRRr(c @|;t|ts d}t||rd|}tj ||S |j di|S#t $rC}|j j}d|d|}tj|t|d}~wwxYw)Nz?The 'start_event' argument must be an instance of 'StartEvent'.zKeyword arguments are not supported when 'run()' is invoked with the 'start_event' parameter. These keyword arguments will be ignored: z'Failed creating a start event of type 'z' with the keyword arguments: ) isinstancer ValueErrorloggerwarningr9rr,debugr)rD start_eventkwargsr]eev_names r'_get_start_event_instancez"Workflow._get_start_event_instance s  "k:6W o%AAGJs#  ,*4**4V4 4 ,--66G;G9DbcibjkC LLO&s+ +  ,sA B>BBc ddlm}|j||n||}|jrdn|j|fi|}|j |||j S)aRun the workflow and return a handler for results and streaming. This schedules the workflow execution in the background and returns a [WorkflowHandler][workflows.handler.WorkflowHandler] that can be awaited for the final result or used to stream intermediate events. You may pass either a concrete `start_event` instance or keyword arguments that will be used to construct the inferred [StartEvent][workflows.events.StartEvent] subclass. Args: ctx (Context | None): Optional context to resume or share state across runs. If omitted, a fresh context is created. start_event (StartEvent | None): Optional explicit start event. **kwargs (Any): Keyword args to initialize the start event when `start_event` is not provided. Returns: WorkflowHandler: A future-like object to await the final result and stream events. Raises: WorkflowValidationError: If validation fails and validation is enabled. WorkflowRuntimeError: If the start event cannot be created from kwargs. WorkflowTimeoutError: If execution exceeds the configured timeout. Examples: ```python # Create and run with kwargs handler = MyFlow().run(topic="Pirates") # Stream events async for ev in handler.stream_events(): ... # Await final result result = await handler ``` If you subclassed the start event, you can also directly pass it in: ```python result = await my_workflow.run(start_event=MyStartEvent(topic="Pirates")) ``` rr N)workflowry semaphore)workflows.contextr _validate is_runningr} _workflow_runr@)rDctxryrzr start_event_instances r'runz Workflow.run%ssj . _c'$-~~ /// FvF    ';tyy!  r(c"|jS)z Validate the workflow to ensure it's well-formed. Returns True if the workflow uses human-in-the-loop, False otherwise. )rr_s r'validatezWorkflow.validatejs ~~r(c |jry|js'|jj}d|d}t ||j |_|j|_|j h}t}g}|jjD]\}}|j}|dk7r4|jD]%} t| ts|j|n|jD]} |j!| |j"D]!} | t%dur|j!| #|r7dj'|} t)|dk(rdnd} d | d | d }t+|d} |D]} t| tsd } n| s d }t+|||z }|Dchc]}t|t,t.tfs|!}}|r&dj'd|D}t+d|||z }|Dchc]%}t|t,t.|jfs|'}}|r&dj'd|D}t+d|t,|vxst.|vScc}wcc}w)NFz Workflow 'z' has no configured steps. Did you forget to annotate methods with @step or to register free-function steps via @step(workflow=...)?rez', 'r sStepz 'zi' cannot accept StopEvent. StopEvent signals the end of the workflow. Use a different Event type instead.Tz'No event of type StopEvent is produced.z, c34K|]}|jywrr,.0evs r' z%Workflow._validate..sFbbkkFz6The following events are consumed but never produced: c34K|]}|jywrrrs r'rz%Workflow._validate..sBbbkkBrz6The following events are produced but never consumed: )r6rOr&r,rr8r9r:r;rNitemsrQrRrSrappendrTrftypejoinrUrrr)rDr\r]produced_eventsconsumed_eventssteps_accepting_stop_eventr#rXrYrZ step_namespluralstop_okrunconsumed_eventsxnames unused_eventss r'rzWorkflow._validaters  # # ~~..HXJ'??  -S1 1#'"@"@"B!%!>!>!@&*&=&=%>%(U13"#0668 0OD)&/&<&''",+>&&22"S,,,8;, ,6__#)-B B 'B  B  B B H ^ r(r1) metaclass)+ __future__rr>loggingtypingrrrllama_index_instrumentationrpydanticrcontextr decoratorsr r errorsrrrrcrrrrrhandlerrresourcertypesrutilsrrr,rB getLoggerrvrrr1rsr(r'rs# 7$ 0 %%@ H %    :4: a a r(