j4dZddlmZddlZddlmZddlmZdZdZ ed Gd d Z ed Gd d Z d:dZ d;dZ ddZd?dZd@d Zd:d!ZdAd%ZdBd'Zd@d(ZdCd*ZdDd.ZdEd1ZdFd5ZdGd9ZdS)HzBPrompt-enhancer input/output routing and comma-separated chaining.) annotationsN) dataclass)CallableaThe user input contains a JSON object with `prompt`, `alt_prompt`, and `requested_output`. Treat `prompt` and `alt_prompt` as input data, use both when relevant, and return only the new value for the field named by `requested_output`. Do not return JSON, field names, or an explanation.zThis prompt enhancer chain produces one alt prompt per prompt, but the model does not declare alt_prompt_inherits_prompt_paragraphs=True.T)frozenc.eZdZUded<ded<ded<dS)PromptEnhancerStepstrmode input_field output_fieldN__name__ __module__ __qualname____annotations__GC:\Users\johnn\OneDrive\Documents\AI\shared\prompt_enhancer\chaining.pyrrs3 IIIrrc8eZdZUded<ded<ded<ded<dS)PromptEnhancerResult list[str]prompts alt_promptsboolprompt_changedalt_prompt_changedNr rrrrrsBrrr r returncLt|pdddS)NK)r replacer s rwithout_thinkingr#"s" tzr?? " "3 + ++renabledrc6t|}|r|r|dn|S)Nr )r#)r r$s r with_thinkingr&&s) D ! !D 3T 3d::::t3rrFvaluechoicesrdefaultrequire_choicect|}||vr|ss|S|dd}||kr ||vr|ss|S||vr|ss|Stfd|DdS)NRrc3$K|] }|s|V dSNr).0choicer*s r z#normalize_choice..5s-NNFN~NNNNNNNr)r#r!next)r'r(r)r*r legacy_modes ` rnormalize_choicer4+s E " "D wD ,,sB''Kd{g55;5n5'wn NNNNgNNNPR S SSrcdt|pdvdt|pddD}fd|DS)Nr rcg|]D}t|#t|ESr)r#strip)r/parts r zsplit_mode..:sD u u uVfgkgqgqgsgsVtVt u djjll + + u u ur,c0g|]}t|Sr)r&)r/r8thinkings rr9zsplit_mode..;s# < < .AsFFFFv~~~~~rTLBPrompt enhancer step 'z"' must use only one of T, L, or B. alt_promptboth)LBrTpromptO)r?len ValueErrorr#getappendr)r steps input_lettersr r8s @r parse_moderT>s E4  ggFFFFeFFF }   ! !p6Ft6L6Lpppqq q(v66::};e=;K;Kbegopp  'k3RV;;<<\deeffff Lrc`tdvptfddDS)Nr:c3 K|]}|vV dSr.r)r/rCr s rr1zuses_routing..Ks'AAfnAAAAAArLBO)r#anyr"s`r uses_routingrYIs< D ! !D $; A#AAAA5AAAAAArcTtdt|pdDdS)Nc3BK|]}||VdSr.)isdigit)r/rCs rr1z!profile_suffix..Os1JJF9I9IJJJJJJJrr0)r2r r"s rprofile_suffixr^Ns+ JJc$*"ooJJJC P PPr originals output_countintc6t|dkr||zn|S)NrE)rN)r_r`s roriginals_for_outputsrcRs"'*9~~':':9| # # Irvaluesct|kr#tdtd|dt|t\fdtDS)NzCannot distribute z alt prompts across z prompts.c tg|]4\}}t|tdz krndzD]}|5S)rEr)rangerN)r/indexr'_ remainderrepeatsrds rr9z!repeat_evenly..Zs| B B BleUw_dhklrhshsvwhw_w_wR[R[}~GBABA B BAE B B B Br)rNrOdivmod enumerate)rdr`rjrks` @@r repeat_evenlyrnVs 6{{\!!fc&kkff|fffggg c&kk::GY B B B B B Bi&7&7 B B BBrcNtdt|DS)Nc3>K|]}|jdko|jdvVdS)rG)rLrHN)r r )r/steps rr1z2requires_alt_prompt_inheritance..^s9yy`dt L0[T5EI[5[yyyyyyr)rXrTr"s rrequires_alt_prompt_inheritancerr]s) yyhrswhxhxyyy y yyr prompt_countcB|dkrt|r |stndS)NrEr)rrALT_PROMPT_INHERITANCE_ERROR)r rsr$s rvalidate_alt_prompt_inheritancervas.+7!+;+;@_`d@e@e+;nu+; ' '{}}ralt_prompt_count prompt_labelalt_prompt_labelc 0||krd|d|d|d|d ndS)N"z " contains z paragraphs, but "z" contains only .rr)rsrwrxrys rvalidate_alt_prompt_countr}esXBRUaBaBa } } },< } }P\ } }nz } } } }giir instructions str | Nonecdt|pdvr|St|pddtS)NrJrz )r rstripCOMBINED_INPUT_SYSTEM_SUFFIXr7)r~r s rappend_combined_input_contractrisW #djb//!!,$"%%,,.. R R4P R R X X Z ZZrrrr cbttt}t|krt|td|fvr0tdtdtdfdt |DS)NrEzCannot combine z prompts with z alt prompts.c g|]R}tjtdkrdn|tdkrdn|ddSS)rEr)rLrGrequested_outputF) ensure_ascii)jsondumpsrN)r/rhrr rs rr9z$_combined_inputs..us b b bLQDJ's7||q/@/@!!e"L\gmpq|m}m}BCnCnChihiIN]OeqrrAF G G G b b br)maxrNrnrOrg)rrr counts``` r_combined_inputsros G c+.. / /E ;5  #K77  7||Au:%%f3w<<ffs;GWGWfffggg b b b b b bUZ[`UaUa b b bbr%alt_prompt_inherits_prompt_paragraphsenhance4Callable[[PromptEnhancerStep, list[str]], list[str]]c ~t|}t|}t|pdg}dx}} t|t||} | rt | |D]} | jdkr|} n$| jdkr|} nt ||| j} t|| | } t| t| krEtdt| j dt| dt| d| jdkr| d }}| d } }t|||| S) NrFrLrGrFz ' returned z outputs for z inputs.T) rTlistrvrNrOr rr RuntimeErrorr#r r)r rrrrrRcurrent_promptscurrent_alt_promptsrrinheritance_errorrqinputsoutputss r run_chainrxs t  E7mmO{++3t*//N'7c/>R>RTyzz,*+++ D D  x ' '$FF   - -(FF%o7JDL]^^FwwtV,,-- w<<3v;; & & Q8H8S8S Q Q`cdk`l`l Q Q{~@F|G|G Q Q QRR R   ( (.5t^OO6=t!3   1DnVh i iir)r r rr )r r r$rrr )rF) r'r r(rr)r r*rrr )r r rr)r r rr@)r r rr)r_rr`rarr)rdrr`rarr)r r rsrar$rrr ) rsrarwrarxr ryr rr )r~rr r rr)rrrrr r rr) r r rrrrrrrrrr)__doc__ __future__rr dataclassesrtypingrrrurrr#r&r4r?rTrYr^rcrnrrrvr}rrrrrrrsLHH"""""" !!!!!!k k $  $,,,,4444 T T T T T==== BBBB QQQQJJJJBBBBzzzz~~~~iiii[[[[ bbbbjjjjjjr