Jir!ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZerddlmZ ddlmZ ddlmZddlmZdd lmZdd lmZmZdd lmZd ZGd deZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'ejPejRzZ*dddZ+y#e$r ddlmZYvwxYw#e$r ddlmZYwxYw) ) annotationsN) TYPE_CHECKING AnnotatedAnyCallableOptionalcastget_args get_originget_type_hints StepFunction)Union) UnionType) BaseModel)WorkflowValidationError)Event EventType)ResourceDefinitiong{Gz?cDeZdZUdZded<ded<ded<ded <ded <y ) StepSignatureSpeczIA 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____doc____annotations__a/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/utils.pyrr,s#S//!!""r%rct|s!tdt|jt j |}t |d}i}d}d}g}|jjD]\}}|dvr |j||j} t| ;t| } t| } t| dr| jdk(r |}| r| d}mt| tur+t| \} } |jt!|| t| dr| jdk(r|}t#||}t%d |Ds|||<t'|t)|||| S) 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)selfclsrContextr)nameresourcec3K|]6}|tk(xs'tj|xrt|t8ywN)rinspectisclass issubclass).0param_ts r& z$inspect_signature..tsC  u  I(GZ-G I s<>)rrrrr)callable TypeErrortyperr0 signaturer parametersitemsget annotationr r hasattrrappendr_get_param_typesallr_get_return_types)fnsig type_hintsrrrrr,tr=originargs_r- param_typess r&inspect_signaturerK6s" B<:48;L;L:MNOO   B C48J24OI>>'')'a ? " ^^D!,,7  j ! - +FJ'Dvz*v)/K$(!)-a& j !Y .":.KAx   /THM N  :z *z/B/Bi/O $  'q*5  '  %0OD ! O'R '&r*+-  r%ct|j}|dk(r d}t||dkDrd|d}t||js d}t|y)a  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_signaturerRsn,,-MW%c**  Z[hZiijk%c**   [%c** r%cddlm}i}tj|tj}|D]!\}}t |dst ||||<#|S)a  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.decoratorsrr0 getmembers isfunctionr>r )_classr step_methods all_methodsr,methods r&get_steps_from_classr^sZ2,.L$$Vw7I7IJK#< f 6> *!%lF!;L < r%cddlm}i}tj|tj}|D]!\}}t |dst ||||<#|S)a 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 rTrV)rWrr0rXismethodr>r )workflowrr[r\r,r]s r&get_steps_from_instancerbsZ2,.L$$X9I9IJK#< f 6> *!%lF!;L < r%c*|j|j|j}|tjj urt gSt|tttfvr't|Dcgc]}|tdus|c}S|gScc}w)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,r=r0 Parameteremptyrr rrrr r8)paramrEtyprFs r&r@r@sy ..U%5%5 6C g%%%u #5(I66#C=@aAT$Z,?@@ 5LAs 3BBct|}|jd}|gSt|}|ttfvr't |Dcgc]}|t dus|c}S|gScc}w)zf Extract the return type hints from a function. Handles Union, Optional, and List types. returnN)r r<r rrr r8)funcrE return_hintrGrFs r&rBrBsm  %J..*K  $F %###K0HaAT$Z4GHH}Is A$A$cz|s d}t||jd}t|dk(ryd|vry|ddk(S)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 emptyrMrTzF) ValueErrorsplitrN)qualnamerQtokss r&is_free_functionrrsL" 2o >># D 4yA~ 4 Bx:%%r%cDdjdt|DS)z9Returns a unique identifier with the format 'kY2xP9hTnQ'.c3NK|]}tjtywr/)secretschoice _alphabet)r3rIs r&r5z_nanoid..sB7>>),Bs#%)joinrange)sizes r&_nanoidr|s 77BeDkB BBr%)rCrrir)rOrriNone)rZobjectridict[str, StepFunction])rar~rir)rfzinspect.ParameterrEdictrir)rjrrir)rpstrribool) )r{intrir), __future__rr0rvstringtypingrrrrrr r r r rWrr ImportErrortyping_extensionstypesrpydanticrerrorsreventsrrr-rBUSY_WAIT_DELAYrrKrRr^rbr@rBrr ascii_lettersdigitsrxr|r$r%r&rs#    1( *+$( L^+0...&&@  6== 0 C}('( *)*s"B!B2! B/.B/2 C?C