JiddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z GddeZGd d eZdd Zdd ZGd deZy)) annotationsN)AnyType) BaseModelValidationErrormodel_validator)!import_module_from_qualified_name)EventcfeZdZUdZded<ded<ded<ded <gfd d Ze d dd Zy )EventEnvelopeWithMetadataz Client readable representation of an Event. Includes class metadata in order to support matching event types semantically in an extendable manner (e.g. "StartEvent", "StopEvent", etc.). zdict[str, Any]value str | Nonequalified_namestrtypelist[str] | Nonetypesc|Dcic]}|j|}}t|j|j|jj }tj ||Scc}w)z Attempts to load the event data as a python class based on the envelope metadata. Looks up the event from the registry, if provided. Falls back to the qualified_name, attempting to load from the module path. )r rr) client_dataregistry)__name__ EventEnveloper rr model_dumpparse)selfreregistry_lookupas_event_envelopes x/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/workflows/protocol/serializable_events.py load_eventz$EventEnvelopeWithMetadata.load_eventsl 3;;Q1::q=;;)**499T=P=P *, "")O#  37 "r,r ceZdZUdZded<dZded<dZded<ed ed d Z edd Z e d dd Z y)rz Client write representation of an Event. Includes class metadata in order to support matching event types semantically in an extendable manner (e.g. "StartEvent", "StopEvent", etc.). z Any | Noner Nrrrbeforer#ct|tr tj|}t|t rd|vrd|vr i|d|di}|S#tj$rY:wxYw)Nr data) isinstancerr"loadsJSONDecodeErrordict)r'r8s r_format_compatibilityz#EventEnvelope._format_compatibilityMso dC  zz$' dD !d"v~6$6f6 ''  sA A"!A"c\||jdt|jS)Nr"r#)r r)rrr)r'r(s rr+zEventEnvelope.from_event[s-"""/e%%  r,c |xsi}g} t|trtj|n|}t|t s t dd|vxsd|vxrd|v}|r1|r/|j|vri||j|i}|j|d} tj|}|jru|j|vr?|jd|jddj|jn(||jj|jS|jr`t!|j}t#|t$s |jd |jd n|j|jS|r|nd g}t dj|#tj$r|}YwxYw#t&$r'} |jd t| Yd } ~ hd } ~ wwxYw)a Parse client data into an Event. Raises an EventValidationError if the client data is invalid. Args: client_data: The client data to parse. Can be a dictionary, a string, or an explicit Event class. registry: The registry of event type names to Event classes. explicit_event: An explicit Event class to treat the dict as Returns: The parsed Event. zNFailed to deserialize event. Must be a json object, or stringified json objectrrr )rr zInvalid event type: z. Expected one of z, z$Invalid client data. Qualified name z) does not correspond to an Event subclasszFailed to deserialize event: NzFInvalid client data. Must have a type or a qualified name, got {event} )r9rr"r:r;r<EventValidationErrorrrmodel_validaterappendjoinkeysr rr issubclassr r) r'rrexplicit_eventerrorsas_dictmissing_qualifiersr( module_classrs rrzEventEnvelope.parsebs$>r "+5k3+G ;'[  '4(&`  G + DvW/D%W$  .&&h6PhP(?(?P&// G D!009Ezz::X-MM.uzzl:LTYYW_WdWdWfMgLhi$EJJ/>>u{{KK##@AUAUV !,6MM>u?S?S>TT}~(66u{{CC  Y #388F#344W## "!G "D D MM9#a&B C C Ds1'F3BG&A+G3G  G  G?G::G?)r8rr-r)r(r r-r)NN)rzdict[str, Any] | strrzdict[str, Type[Event]] | NonerGzType[Event] | Noner-r ) rr/r0r1r2rrrr3r=r+rr4r,rrrCs D*!%NJ%(# $   37-1 B5)B50B5+ B5  B5B5r,rcg}|jddD]8}|turn.t|ts|j|j:|sy|S)zo Traverses the MRO (Module Resolution Order) of a class and returns the list of only Event subclasses. N)mror rFrCr)r'namescs rr&r&sUE WWYqr]% :  a  LL $ %  Lr,c8|jd|jS)N.)r/r)r(s rr%r%s q 0 11r,ceZdZdZy)rAz'Raised when the client data is invalid.N)rr/r0r1r4r,rrArAs1r,rA)r' Type[Event]r-r)r(rTr-r) __future__rr"typingrrpydanticrrrworkflows.context.utilsr workflows.eventsr r rr&r% ExceptionrAr4r,rr[sN# @@E"1 1hb5Ib5J  2292r,