'j) bddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z  ddlZn#e$rddlZYnwxYwe dZe eefZGddZGddZd e ed e ed eefd Zd ee ed fd ee ed fd ee ed ffdZdeeefd eeeffdZdeeefded ee ed ffdZdddeee e ed feeffd eee effdZded efdZdS)N)Path) AnyDictIterableListOptionalSequenceTupleTypeVarUnionTc feZdZdZdeeeffdZdeeefdeeeffdZ de edeeefdefd Z deeefdefd Z deeefd e ed e edeefd ZedefdZede efdZede efdZdS) BayesianNodezs Implementation of a single node in a bayesian network allowing sampling from its conditional distribution node_definitionc||_dSNr)selfrs a/home/ice/.hermes/hermes-agent/venv/lib/python3.11/site-packages/browserforge/bayesian_network.py__init__zBayesianNode.__init__s. parent_valuesreturnc|jd}|jD]T}||}||divr|d|}>|di}U|S)zj Extracts unconditional probabilities of node values given the values of the parent nodes conditionalProbabilitiesdeeperskip)r parent_namesget)rr probabilities parent_name parent_values r$get_probabilities_given_known_valuesz1BayesianNode.get_probabilities_given_known_valuess ,-GH , > >K(,,[99L}002>>>> -h 7 E - 1 1&" = = rpossible_valuesr cptj}d}|D]}|||z }||kr|cS|dS)zV Randomly samples from the given values using the given probabilities gr)random)rr$r anchorcumulative_probabilitypossible_values r&sample_random_value_from_possibilitiesz3BayesianNode.sample_random_value_from_possibilities&s]!$- & &N "mN&C C "%..%%%%/q!!rc||}|t||S)zi Randomly samples from the conditional distribution of this node given values of parents )r#r*listkeys)rrr s rsamplezBayesianNode.sample7sHAA-PP :: ##%% & &    rvalue_possibilities banned_valuesc||fd|D}|r||SdS)z Randomly samples from the conditional distribution of this node given restrictions on the possible values and the values of the parents. c$g|] }|v|v | Sr3).0valuer0r s r zABayesianNode.sample_according_to_restrictions..Js8   M))e}.D.D .D.D.DrN)r#r*)rrr/r0 valid_valuesr s ` @r sample_according_to_restrictionsz-BayesianNode.sample_according_to_restrictions@skAA-PP      ,    >>|][[ [4rc|jdS)Nnamerrs rr:zBayesianNode.nameTs#F++rc8|jdgS)N parentNamesrrr;s rrzBayesianNode.parent_namesXs#'' r:::rc8|jdgS)NpossibleValuesr>r;s rr$zBayesianNode.possible_values\s#''(8"===rN)__name__ __module__ __qualname____doc__rstrrrfloatr#rr*r.rrr8propertyr:rr$r3rrrrs/S#X//// !#s(^  c5j     "#Cy"9=c5j9I" """"" DcN s    CH~&c]Cy  # (,c,,,X,;d3i;;;X;>c>>>X>>>rrc eZdZdZdeddfdZd deeee fdeee ffdZ deee efdeeee ffd Z d eee fdeee efd e deeee ffd ZdS)BayesianNetworkza Implementation of a bayesian network capable of randomly sampling from its distribution pathrNc~t|}d|dD|_d|jD|_dS)Nc,g|]}t|Sr3)r)r4node_defs rr6z,BayesianNetwork.__init__..hs-( ( ( '/L " "( ( ( rnodesci|] }|j| Sr3)r:)r4nodes r z,BayesianNetwork.__init__..ksWWW$diWWWr) extract_jsonnodes_in_sampling_order nodes_by_name)rrJnetwork_definitions rrzBayesianNetwork.__init__fsX)$//( ( 3Eg3N( ( ( $XW$:VWWWr input_valuesc|i}|}|jD](}|j|vr||||j<)|S)z] Randomly samples from the distribution represented by the bayesian network. )copyrSr:r.)rrVr.rPs rgenerate_samplezBayesianNetwork.generate_samplems]  L""$$0 8 8Dy&&$(KK$7$7ty! rr/c0|i|dS)a Randomly samples values from the distribution represented by the bayesian network, making sure the sample is consistent with the provided restrictions on value possibilities. Returns None if no such sample can be generated. r)4recursively_generate_consistent_sample_when_possible)rr/s r(generate_consistent_sample_when_possiblez8BayesianNetwork.generate_consistent_sample_when_possibleysHHM`bcdddr sample_so_fardepthcZ|t|jkr|S|j|}g}d} ||||j|j|}|nF|||j<||||dz}||S||||j=~dS)zr Recursively generates a random sample consistent with the given restrictions on possible values. NT)lenrSr8rr:r$r[append)rr]r/r^rPr0 sample_value next_samples rr[zDBayesianNetwork.recursively_generate_consistent_sample_when_possibles C455 5 5 +E2#%   )@@#'' 43GHHL #'3M$) $SS2EAIK&""   . . .di( ) trr)rArBrCrDrrrrrErrYrr\intr[r3rrrIrIas)XTXdXXXX  HT#s(^,D PTUXZ]U]P^    e#'Xc](:#;e $sCx. !eeeeCH~"#x}"45  $sCx. ! rrIabrc>t|fd|DS)zB Performs a set "intersection" on the given (flat) arrays cg|]}|v| Sr3r3)r4xset_bs rr6z&array_intersection..s ' ' '!AJJAJJJr)set)rfrgrks @rarray_intersectionrms* FFE ' ' ' 'q ' ' ''r.c6dt||DS)z Combines two arrays into a single array using the set union Args: a: First array to be combined. b: Second array to be combined. Returns: Zipped (multi-dimensional) array. crg|]4\}}tt||5Sr3)tuplerlunion)r4rjys rr6zarray_zip..s4 9 9 9tq!E#a&&,,q// " " 9 9 9r)zip)rfrgs r array_ziprts  : 9s1ayy 9 9 99robjct|ts|Si}|D]G\}}|dkr |dkr#|t |5t |||<H|S)zU Removes the "deeper/skip" structures from the conditional probability table rr) isinstancedictitemsupdateundeeper)ruresultkeyr5s rr{r{s c4  Fiikk** U &==  (?? MM(5// * * * *"5//F3KK Mrtree valid_keyscgdtttfdtttdft tfdt tddffd ||gS)a# Performs DFS on the Tree and returns values of the nodes on the paths that end with the given keys (stored by levels - first level is the root) ``` 1 / \ 2 3 / \ / \ 4 5 6 7 ``` filter_by_last_level_keys(tree, ['4', '7']) => [[1], [2,3]] tvk.accrNc&|D]z}t||tr||>||vr9tdkr d|Dnt d|Dc|||||gz{dS)Nrcg|]}|fSr3r3r4rjs rr6z>filter_by_last_level_keys..recurse..s+++!!+++rcg|]}|fSr3r3rs rr6z>filter_by_last_level_keys..recurse..s,?,?,?aaT,?,?,?r)r-rwrxrart)rrrr}outrecurses rrz*filter_by_last_level_keys..recurses6688 1 1Cafd++ 1qv~"99s88q==,+s++++&s,?,?3,?,?,?@@ #C3%K0000 1 1r)rrErr r r)r~rrrs @@rfilter_by_last_level_keysrs"$C 14S> 1uU38_d3i-G'H 1tTWy 1]a 1 1 1 1 1 1 1 GD*b!!! Jrnetworkr$c `g}|D]\}}t|ttfs"t |dkrt d|j|}t|jd}t||}| itt|j |||ii}|D][} | D]D}||vr3t| |||} | st d| ||<9| |||<E\|S)z Given a `generative-bayesian-network` instance and a set of user constraints, returns an extended set of constraints **induced** by the original constraints and network structure rzgThe current constraints are too restrictive. No possible values can be found for the given constraints.r)ryrwr,rpra ValueErrorrTr{rrrbrxrsrr-rm) rr$setsr}r5rPr~ zipped_valuesr|set_dictintersected_valuess rget_possible_valuesrsk D%++-- Q Q U%$//   u::??y $S),-GHII1$>>  OtC 1=AABBOCOOPPPP(*F , ,==?? , ,Cf}}%7 vc{%S%S")$B1s &sms  , MrrJcj|jdkrNt|d5}tj|cdddS#1swxYwYt j|d5} td|D}n#t$ricYcdddSwxYw||5}tj|cdddcdddS#1swxYwY ddddS#1swxYwYdS)z Unzips a zip file if the path points to a zip file, otherwise directly loads a JSON file. Parameters: path: The path to the zip file or JSON file. Returns: A dictionary representing the JSON content. z.ziprbNrc3DK|]}|d|VdS)z.jsonN)endswith)r4files r zextract_json..#s3UUTdmmG>T>TUDUUUUUUr) suffixopenjsonloadsreadzipfileZipFilenextnamelist StopIteration)rJrzffilenamefs rrRrRs {f $   +:diikk** + + + + + + + + + + + + + + + + s # #(r UUR[[]]UUUUUHH   II ((((((((  WWX   (!:affhh'' ( ( ( ( ( ( ( ((((((((  ( ( ( ( ( ( ( ( ( ((((((((((((((((((sk&AAA/D(1+BD( B8)D(7B88D(&D6 D(D D(D D((D,/D,)r&rpathlibrtypingrrrrrr r r r orjsonr ImportErrorr r,rpMaprrIrmrtrEr{rrrxrRr3rrrs WWWWWWWWWWWWWWWWWWWWWWKKKKK GCLL D%KO>O>O>O>O>O>O>O>d@@@@@@@@F((1+((1+($q'(((( :eAsFm$ :eAsFm)< :eAsFmAT : : : :$sCx.T#s(^"DcNUSVX[S[_H]B$ $15c5sCxRVWZR[A[;\6\1]$ #x} $$$$N(t(((((((s + 77