+#h8 ddlZddlZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z m Z mZddlmZddlZddlmZmZmZdej,fdZdefd ZGd d ee ZGd d e eZdde ee efdefdZ dde eeeeeefffdedede efdZdedee e ffdZddedeeeffdZ ddefdZ!dedefdZ"y)N)deepcopy)AnyDictTupleTypeUnion) BaseModelField create_model) FieldInfo) InputField OutputFieldnew_to_old_fieldreturnc tj|jfi|jj Dcic]\}}|t |c}}Scc}}w)z"Convert from new to legacy format.)dspTemplate instructionsfieldsitemsr) signaturenamefields V/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/signatures/signature.pysignature_to_templatersP << d|d |jd<R|S) N__annotations______doc__prefix:descz${})getr isinstancer startswithstrr-__new__r6r(getattrr&_validate_fields model_fieldsjson_schema_extra infer_prefix) mcssignature_namebases namespacer0raw_annotationsrrr#basedocr1s rr@zSignatureMeta.__new__#sb#--(92>$??, ,KD%eY/??4(T-H(+%  , (7 #$goc>5)NvN ;;  *dM2!$ 26Cby&)  * ;; /4CK ++113 AKD%u6664@4F4L''1U44447vR.''/  A  r2c |jjD]R\}}|jxsi}|jd}|dvs,t d|d|j d|jy)N__dspy_field_type)inputoutputzField 'z' in 'zK' must be declared with InputField or OutputField. field.json_schema_extra=)rCrrDr< TypeError__name__)r#rrextra field_types rrBzSignatureMeta._validate_fieldsJs++113 KD%++1rE#67J!44dV6#,,8DkplClCkEF  r2rcdj|jj}dj|jj}|d|S)Nrz -> )r r!keysr")r#in_argsout_argss rrzSignatureMeta.signatureSsM))C,,113499S..3356$xj))r2ct|ddS)Nr6r7)rAr#s rrzSignatureMeta.instructionsYssIr**r2rr+c.t|j|Sr*)r+r)r#rs rwith_instructionszSignatureMeta.with_instructions]s\22r2c6i|j|jSr*)r!r"rZs rrzSignatureMeta.fields`s 9#""8c&7&788r2c t|j}i||j|||_| |||_t ||j S)zUpdate the field, name, in a new Signature type. Returns a new Signature type with the field, name, updated with fields[name].json_schema_extra[key] = value. )rrrD annotationr+r)r#rtype_r0 fields_copys rwith_updated_fieldsz!SignatureMeta.with_updated_fieldsesd szz* / $11/ / D+  +0K  (c&6&677r2c$|jdS)NrO_get_fields_with_typerZs rr!zSignatureMeta.input_fieldsts((11r2c$|jdS)NrPrdrZs rr"zSignatureMeta.output_fieldsxs((22r2c|jjDcic]\}}|jd|k(s||c}}Scc}}w)NrN)rCrrD)r#rTkvs rrez#SignatureMeta._get_fields_with_type|sA!$!1!1!7!7!9tAQ=P=PQd=eis=s1ttts AAc*|jd|||S)Nrinsertr#rrr`s rprependzSignatureMeta.prependszz!T5%00r2c*|jd|||S)Nrkrms rappendzSignatureMeta.appendszz"dE511r2indexrr`c| |j}|t}t|jj }t|j j }|j ddk(r|n|}|dkr|t|dzz }|dks|t|kDrtd||j||||fft||z}t||jS)NrNrOrzInvalid index: ) r_r?listr!rr"rDlen ValueErrorrldictr+r) r#rrrrr`r!r"lst new_fieldss rrlzSignatureMeta.inserts =$$E =EC,,2245 S..4467 $556IJgUl[h 19 SX\ !E 19C(ug67 7 54%01,67 S%5%566r2c*t|trt|tsy|j|jk7ry|j j |j j zD] }||j vs||j vs yy)z/Compare the JSON schema of two Pydantic models.FT)r=type issubclassr rrrV)r#otherrs requalszSignatureMeta.equalssz%&j .J   u11 1JJOO% (9(9(;; D5<<'4szz+A r2cg}|jjD]\}}|j|d|ddj|}|jd|j dt |jd|dS)a#Output a representation of the signature. Uses the form: Signature(question, context -> answer question: str = InputField(desc="..."), context: List[str] = InputField(desc="..."), answer: int = OutputField(desc="..."), ). z = Field()z (z instructions=z ))rrrqr rRrreprr)r# field_reprsrr field_reprs r__repr__zSignatureMeta.__repr__s ::++- ;KD%   $yq9 : ;]];/ ,,q/B4HXHXCYBZZ`ak`lloppr2r*)rR __module__ __qualname__r.r@rBpropertyr?rrrr\rxr rrbr!r"rernrqintrlboolrr __classcell__)r1s@rr(r(sO1 %N*#** +S++3S3T+5F39tCN+99 8[@Q 82T#y.1223d3 >233u$sI~2Fu1k1B12[0A2737c77kIZ7. d qr2r(ceZdZdZy)r+r7N)rRrrr6r2rr+r+s  r2r+) metaclassrc^|yt|tr t||S| td|S)Nz=Don't specify instructions when initializing with a Signature)r=r?r+rw)rrs rensure_signaturers:)S!L11XYY r2rrGct|tr t|n|}i}|jD]\}}t|tst dt |t|t r|j}|}n,t|tst dt ||\}}|t}t|t tjtjfst dt |t|t st dt |||f||<t|jt|jk(sJ|t|d} t!| }t#|ft|d|S)aCreate a new Signature type with the given fields and instructions. Note: Even though we're calling a type, we're not making an instance of the type. In general, instances of Signature types are not allowed to be made. The call syntax is provided for convenience. Args: signature: The signature format, specified as "input1, input2 -> output1, output2". instructions: An optional prompt for the signature. signature_name: An optional name for the new signature type. z!Field names must be strings, not z!Field values must be tuples, not zField types must be types, not z*Field values must be Field instances, not r7)__base__r6)r=r?_parse_signaturerrwr|r r_tupletyping _GenericAliastypes GenericAliasrurVr+r&r ) rrrGr fixed_fieldsr type_fieldr`rsigs rr,r,s}"-7y#,F i (IFL"LLN,j$$@d MN N j) ,))EEj%0 #DT*EUDV!WXX%LE5 =E%$(<(P>P!QR>tE{mLM M%+I$u+WX X#U^ T',,  !!# $V[[](; ;; ; 2&,S1      r2c|jddk7rtd|d|jd\}}i}t|D]\}}|t f||<t|D]\}}|t f||<|S)Nz->rtzInvalid signature format: 'z!', must contain exactly one '->'.)countrwsplit_parse_arg_stringr r)r inputs_str outputs_strrrr`s rrrst!6ykAbcdd'ood3J F(4- ez|,t -(5. e{}-t . Mr2stringc@tjd|zdzjdjj}|Dcgc]}|j}}|Dcgc])}|j t nt|j +}}t||Scc}wcc}w)Nzdef f(z): passr) astparsebodyr/argr_r?_parse_type_nodezip)rnamesr/rrs rrr%s 99X&2 3 8 8 ; @ @ E ED $ %SWW %E %\` aUXCNN*S0@0P P aE a ue  & as B.Bc tjt|tjr7|j }t |dk7rtd|t|dSt|tjr|j}t|St|tjr`|j}|vr|Stttt t"t$t&fD]}|j(|k(s|cStd|t|tj*r1t|j}t|j,}||St|tj.r |j0}t%fd|DSt|tj2r|j4jdk(rn|j6D cgc]} | j8} } |j6D cgc]} | jj} } t;dit't=| | Std|cc} wcc} w)zRecursively parse an AST node representing a type annotation. without using structural pattern matching introduced in Python 3.10. rtz!Code is not syntactically valid: rzUnknown name: c36K|]}t|ywr*)r).0eltrs r z#_parse_type_node..NsBc%c51Bsr r)r__dict__r=rModulerrvrwrExprvalueNameidrr?floatrrurrxrR SubscriptslicereltsCallfunckeywordsrr r) noderrrid_r` base_typearg_typerkwrVvaluess ` rrr,s  }$ #yy t9>@GH HQ//$! u--$!gg %<: 3tT5$? E~~$  >#/00$ &$TZZ7 #DJJ6""$ "yyBTBBB$! 99<<7 "%)]]3rBFF3D3/3}}=bhhnn=F=34D& 123 3 8? @@ 4=s I:I attribute_namectjdd|}tjdd|}tjdd|}tjdd|}|jd}g}|D]C}|jr|j |%|j |j Edj |S)z&Infer a prefix from an attribute name.z(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z])z([a-zA-Z])(\d)z(\d)([a-zA-Z])_ )resubrisupperrq capitalizer )rs1intermediate_namewith_underscores_around_numberswordstitle_cased_wordswords rrErEYs "Hn =B2HbA')ff'# ')ff''# , 1 1# 6E8 <<>  $ $T *  $ $T__%6 7 8 88% &&r2r*)NStringSignature)#rrrrcopyrrrrrrpydanticr r r pydantic.fieldsr rdspy.signatures.fieldr rrrrr?r&r|r(r+rr,rrrrErr2rrs; 0033% KK I#I YqDOYqX ] c4 ?&: ;S\+8S$sE$ /$::;;<888 )_ 8v  dEk(: c$sCx.*A#*AZ'''r2