ir!SSKJr SSKrSSKrSSKrSSKJrJrJrJ r J r J r J r J r Jr \(aSSKJr SSKJr SSKJr SSKJr SS KJr SS KJrJr SS KJr S r"S S\5r SSjr!SSjr"SSjr#SSjr$SSjr%SSjr&SSjr'\RP\RR-r*SSSjjr+g!\a SSKJr Nf=f!\a SSKJr Nf=f) ) annotationsN) TYPE_CHECKING AnnotatedAnyCallableOptionalcastget_args get_originget_type_hints StepFunction)Union) UnionType) BaseModel)WorkflowValidationError)Event EventType)ResourceDefinitiong{Gz?cL\rSrSr%SrS\S'S\S'S\S'S \S 'S\S 'S rg )StepSignatureSpec,zIA Pydantic model representing the signature of a step function or method.zdict[str, list[EventType]]accepted_events list[Any] return_typesz str | Nonecontext_parameterz Any | Nonecontext_state_type resourcesN)__name__ __module__ __qualname____firstlineno____doc____annotations____static_attributes__r I/home/james-whalen/.local/lib/python3.13/site-packages/workflows/utils.pyrr,s#S//!!""r(rc"[U5(d![S[U5R35e[R "U5n[ USS9n0nSnSn/nURR5GHupxUS;aMURXxR5n [U 5bG[U 5n [U 5n [U S5(a U RS:XaUnU (aU SnM|[U 5[La'[U 5upUR[!X}S 95 M[U S5(aU RS:XaUnM[#X5n[%S U55(dMXU'GM ['U[)U5UUUS 9$) a Given a function, ensure the signature is compatible with a workflow step. Args: fn (Callable): The function to inspect. Returns: StepSignatureSpec: A specification object containing: - accepted_events: Dictionary mapping parameter names to their event types - return_types: List of return type annotations - context_parameter: Name of the context parameter if present Raises: TypeError: If fn is not a callable object z Expected a callable object, got T)include_extrasN)selfclsr!Contextr)nameresourcec3# UHBnU[:H=(d- [R"U5=(a [U[5v MD g7fN)rinspectisclass issubclass).0param_ts r) $inspect_signature..tsB ' u  I(GZ-G I&sA A )rrrrr)callable TypeErrortyper!r3 signaturer parametersitemsget annotationr r hasattrrappendr_get_param_typesallr_get_return_types)fnsig type_hintsrrrrr/trAoriginargs_r0 param_typess r)inspect_signaturerO6s" B<<:48;L;L:MNOO   B C48J24OI>>'') ? " ^^D,,7  j ! - +FJ'Dvz**v)/K$(!)-a& j !Y .":.KA   /TM N  :z * *z/B/Bi/O $  'q5  '   %0D ! O*R '&r*+-  r(c[UR5nUS:Xa Sn[U5eUS:aSUS3n[U5eUR(d Sn[U5eg)z Validate that a step signature specification meets workflow requirements. Args: spec (StepSignatureSpec): The signature specification to validate. Raises: WorkflowValidationError: If the signature is invalid for a workflow step. rzGStep signature must have at least one parameter annotated as type EventrzJStep signature must contain exactly one parameter of type Event but found .zKReturn types of workflows step functions must be annotated with their type.N)lenrrr)spec num_of_eventsmsgs r)validate_step_signaturerVsn,,-MW%c**  Z[hZiijk%c**   [%c** r(cSSKJn 0n[R"U[RS9nUH%upE[ US5(dM[ X5X$'M' U$)z Given a class, return the list of its methods that were defined as steps. Args: _class (object): The class to inspect for step methods. Returns: dict[str, Callable]: A dictionary mapping step names to their corresponding methods. rr  predicate _step_config)workflows.decoratorsrr3 getmembers isfunctionrBr )_classr step_methods all_methodsr/methods r)get_steps_from_classrbsR2,.L$$Vw7I7IJK#  6> * *!%l!;L $ r(cSSKJn 0n[R"U[RS9nUH%upE[ US5(dM[ X5X$'M' U$)z Given a workflow instance, return the list of its methods that were defined as steps. Args: workflow (object): The workflow instance to inspect. Returns: dict[str, Callable]: A dictionary mapping step names to their corresponding methods. rr rXrZ)r[rr3r\ismethodrBr )workflowrr_r`r/ras r)get_steps_from_instancerfsR2,.L$$X9I9IJK#  6> * *!%l!;L $ r(c8URURUR5nU[RR La[ /$[U5[[[4;a+[U5Vs/sHo3[S5LdMUPM sn$U/$s snf)a Extract and process the types of a parameter. This helper function handles Union and Optional types, returning a list of the actual types. For Union[A, None] (Optional[A]), it returns [A]. Args: param (inspect.Parameter): The parameter to analyze. type_hints (dict): The resolved type hints for the function. Returns: list[Any]: A list of extracted types, excluding None from Unions/Optionals. N) r@r/rAr3 Parameteremptyrr rrrr r<)paramrItyprJs r)rDrDs{ ..U%5%5 6C g%%%u #5(I66#C=@=aT$Z,?=@@ 5LAs 6B Bc[U5nURS5nUc/$[U5nU[[4;a+[ U5Vs/sHoD[ S5LdMUPM sn$U/$s snf)zZ Extract the return type hints from a function. Handles Union, Optional, and List types. returnN)r r@r rrr r<)funcrI return_hintrKrJs r)rFrFsq  %J..*K  $F %###K0H0aT$Z4G0HH}Is A+A+cU(d Sn[U5eURS5n[U5S:XagSU;agUSS:H$)a Determines whether a certain qualified name points to a free function. A free function is either a module-level function or a nested function. This implementation follows PEP-3155 for handling nested function detection. Args: qualname (str): The qualified name to analyze. Returns: bool: True if the name represents a free function, False otherwise. Raises: ValueError: If the qualified name is empty. z"The qualified name cannot be emptyrQrTzF) ValueErrorsplitrR)qualnamerUtokss r)is_free_functionrvsL" 2o >># D 4yA~ 4 Bx:%%r(cDSRS[U555$)z9Returns a unique identifier with the format 'kY2xP9hTnQ'.c3V# UHn[R"[5v M! g7fr2)secretschoice _alphabet)r6rMs r)r8_nanoid..sBk7>>),,ks'))joinrange)sizes r)_nanoidrs 77BeDkB BBr()rGrrmr)rSrrmNone)r^objectrmdict[str, StepFunction])rerrmr)rjzinspect.ParameterrIdictrmr)rnrrmr)rtstrrmbool) )rintrmr), __future__rrzstringr3typingrrrrrr r r r r[rr ImportErrortyping_extensionstypesrpydanticrerrorsreventsrrr0rBUSY_WAIT_DELAYrrOrVrbrfrDrFrv ascii_lettersdigitsr|rr r(r)rs#    1( *+$( L^+0...&&@  6== 0 C}('( *)*s"B/C/ B>=B> CC