iSSKJr SSKJrJrJrJrJrJrJ r J r J r SSK r \ RS:aSSKJr OSSKJr SSKJrJrJr SSKJr SS KJr SS KJrJrJr \(aSS KJr SS KJr "S S\5r \"S5r!\ "S5r"\ "SSS9r#"SS\\\!\#45r$\ SSj5r%\ SSSS.SSjj5r%SSSSS.SSjjjr%S S!Sjjr&g)") annotations) TYPE_CHECKINGAnyCallableGenericProtocolTypeTypeVarcastoverloadN) ) ParamSpec) BaseModel ConfigDictField)WorkflowValidationError)ResourceDefinition)inspect_signatureis_free_functionvalidate_step_signature)Workflow) RetryPolicyc~\rSrSr%\"SS9rS\S'S\S'S\S'S \S 'S \S 'S \S'S\S'\"SS9rS\S'Sr g) StepConfig(T)arbitrary_types_allowedz list[Any]accepted_eventsstr event_name return_typesz str | Nonecontext_parameterint num_workersRetryPolicy | None retry_policyzlist[ResourceDefinition] resourcesN)defaultzType[BaseModel] | Nonecontext_state_type) __name__ __module__ __qualname____firstlineno__r model_config__annotations__rr*__static_attributes__r+N/home/james-whalen/.local/lib/python3.13/site-packages/workflows/decorators.pyrr(sFd;LO!!$$''16t1D.Dr3rPRR_coT) covariantcB\rSrSr%SrS\S'S\S'S\S'S SjrS rg ) StepFunction:zRA decorated function, that has some _step_config metadata from the @step decoratorr _step_configr r,r.cgNr+)selfargskwargss r4__call__StepFunction.__call__Bs3r3r+N)r@zP.argsrAzP.kwargsreturnr7)r,r-r.r/__doc__r1rBr2r+r3r4r:r::s\MFr3r:cgr>r+)funcs r4steprHEs69r3workflowr%r'cgr>r+rJs r4rHrHIs 69r3c6^^^SUUU4SjjnUbU"U5$U$)a Decorate a callable to declare it as a workflow step. The decorator inspects the function signature to infer the accepted event type, return event types, optional `Context` parameter (optionally with a typed state model), and any resource injections via `typing.Annotated`. When applied to free functions, provide the workflow class via `workflow=MyWorkflow`. For instance methods, the association is automatic. Args: workflow (type[Workflow] | None): Workflow class to attach the free function step to. Not required for methods. num_workers (int): Number of workers for this step. Defaults to 4. retry_policy (RetryPolicy | None): Optional retry policy for failures. Returns: Callable: The original function, annotated with internal step metadata. Raises: WorkflowValidationError: If signature validation fails or when decorating a free function without specifying `workflow`. Examples: Method step: ```python class MyFlow(Workflow): @step async def start(self, ev: StartEvent) -> StopEvent: return StopEvent(result="done") ``` Free function step: ```python class MyWorkflow(Workflow): pass @step(workflow=MyWorkflow) async def generate(ev: StartEvent) -> NextEvent: ... ``` c>[T[5(aTS::a [S5e[UTT5n[ UR 5(a/TcSUR S3n[U5eTRU5 U$)Nrz-num_workers must be an integer greater than 0z To decorate z5 please pass a workflow class to the @step decorator.) isinstancer$rmake_step_functionrr.r,add_step)rGmsgr%r'rKs r4 decoratorstep..decorators+s++{a/?)? "$ \B D-- . .$T]]O3hi-c22   d # r3rGCallable[P, R]rDStepFunction[P, R]r+)rGrKr%r'rSs ``` r4rHrHRs&f"  r3c B[U5n[U5 [[URR 555upE[ [[[4U5n[UUURURURUUURS9UlU$)N)rr!r"r#r*r%r'r()rrnextiterritemsr r:r5r6rr"r#r*r(r<)rGr%r'specr!rcasteds r4rPrPs T "DD!"&tD,@,@,F,F,H'I"JJ ,q!t$d +F$'&&0022!.. F Mr3rU)rKType['Workflow'] | Noner%r$r'r&rDz.Callable[[Callable[P, R]], StepFunction[P, R]]r>) rGzCallable[P, R] | NonerKr^r%r$r'r&rDzCCallable[[Callable[P, R]], StepFunction[P, R]] | StepFunction[P, R])rIN)rGrVr%r$r'r&rDrW)' __future__rtypingrrrrrr r r r sys version_infortyping_extensionspydanticrrrerrorsrresourcerutilsrrrrKrr'rrr5r6r7r:rHrPr+r3r4rhsn#    w +11+( "% E EcN CLv&G8WQW-G 9 9 )-'+ 9%99% 9 4 9 9#'G)-'+ G G&G G % G I GVTX '*>Pr3