+#h<%fddlZddlZddlZddlZddlZddlmZddlmZddl m Z GddeZ y)N)Example) Teleprompter)LabeledFewShotcVeZdZddiddddfdZddddZd Zd Zdd d Zdd ZdZ y)BootstrapFewShotNrc||_||_||_||_||_||_||_d|_tj|_ y)Nr) metricmetric_thresholdteacher_settingsmax_bootstrapped_demosmax_labeled_demos max_rounds max_errors error_count threadingLock error_lock)selfr rrrrrrs V/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/teleprompt/bootstrap.py__init__zBootstrapFewShot.__init__"sO 0 0&<#!2$$#..*)teachervalsetc*||_||_|j|||j|j |j |_d|j _d|j _d|j _ |j S)NTr) trainsetr_prepare_student_and_teacher_prepare_predictor_mappings _bootstrap_trainstudent _compiled_assert_failures_suggest_failures)rr$rrrs rcompilezBootstrapFewShot.compile7ss    ))'7; ((* {{} !% )* %)* &||rc|j|_||jn|j|_t |jdddusJd|j rkt |jdddurQt |j }|j|jj|j|_yyy)Nr%FzStudent must be uncompiled.)k)r) reset_copyr$deepcopyrgetattrrrr(r)rr$r teleprompters rr z-BootstrapFewShot._prepare_student_and_teacherHs))+ -4-@w'')gFXFXFZ t||[%8EA`C``A  ! !gdllK&OSX&X)D,B,BCL'// 0G0G0ITXTaTa/bDL'Y !rc~ii}}|j|j}}t|jt|jk(sJdt |j |j D]\\}}\}}||k(sJd|j j|j s3Jdt|j dt|j t|t|k7sJdd||<||t|<||t|<||_ ||_ y)Nz ^#^ ^>''..$$ BDT*J^J^E_D``deijtj~j~eeAB Bj>R ^3 e5e e3$(N5 !-2N2j> *.3N2j> * 3"-,r)max_bootstrapsc |xs |j}i}|jDcic]}|gc}|_t|jD]a}t t j |jD]4\}}t||k\rA||vs|j||}|s0d||<6ctjjdt|ddzddt |jD cgc] \}} ||vs | c} }|_ tjdj!|j|j"xs |j|_ ycc}wcc} }w)NTz Bootstrapped z full traces after rz examples in round .r)rr8 name2tracesranger enumeratetqdmrr0_bootstrap_one_exampledspyloggerinfo validationrandomRandomshuffler) rr> bootstrappedname round_idx example_idxexamplesuccessidxxs rr"zBootstrapFewShot._bootstrapnsJ'F4+F+F 151D1DED"HEt/ 9I(1$))DMM2J(K 9$ W|$6l2"99'9MG48 [1 9 9 C -..A+PQ/ARRefoeppq r ,5T]]+C_asR^G^1_ a  1++8-F&`s E!9 E&E&c |j}|j}i} tjjddgi|j 5tjj }|dkDr|jdd|zzn|}|dkDr t|ni}tjjdi|5|jD]9\}} | j||<| jD cgc] } | |k7s |  c} | _ ;|di|j} tjj} |jD]\}} ||| _  ddd|jr2|j|  } |jr| |jk\}n| }nd}dddrh D]c}|\} }}d|vrt-dd|j.d||}nt-dddi||} |j0t3| }||j7|e|Scc} w#1swYxYw#1swYxYw#t$r}d}|j 5|xj"d z c_|j"}dddn #1swYnxYw|j$k\r|t&j(j+d |d |jd |d Yd}~!d}~wwxYw#t4$rY1wxYw)Ntracergffffff?gMbP?) temperature)lmTFrz%Failed to run or to evaluate example z with z due to r@ dspy_uuid) augmentedrYrZ)rArdspsettingscontextrrXcopydictr3demosinputsrVr r ExceptionrrrrFrGerrorrrYr9r7KeyErrorappend)rrQrOrArpredictor_cacherX new_settingsrN predictorrT predictionrV metric_valrRecurrent_error_countsteprboutputsdemopredictor_names rrEz'BootstrapFewShot._bootstrap_one_examples&& ,, p%%HBH$2G2GH #\\__ENQR]RWWuy/@)@WAXZ.7!mtr{ \\))9L9 @+2+C+C+EWi09-6?oo*Vg1*V W")!<7>>+;! >OD)"..t45Rt7R7RSOd44s?7KKSQZ^\Ka-K 9k:I}}$$0"+o"= "1I"=  >||r)r) __name__ __module__ __qualname__rr(r r!r"rEr#r[rrrr!sL +*+/"c-8,09@BHrr) rJrrDr\rFdspy.primitivesr telepromptrvanillarrr[rrrs+  #$#,B|Br