עihdZddlmZddlZddlZddlZddlZddlmZmZddl Z ddl Z ddl Z ddl m Z ddl mZddl mZd Zd Zd Ze j&j)d d ddddddddddfddddddddddddddddddfdddiddddidddddfdidd diddd didfd!dd dididd didfd"ddddd diidd didfd#dd didddidddfd$dddd%dd&idd'd(d)d*d+d,ddd'd-dd)d.d+d,fd/d0d1d2id3d4d5gd6id7d8d4d2gd6d9d7d8fd:dddiidiidddiifd;diidddiidddiifdd?id@dAdBdCdDdEdAdBdCdDd@fg dFedGedHedIefdJZe j&j)dKdLddiddiddidMdNfdOdddiidddiidddiidMdNfdPddiddQiddQid7dRfdSdddiidd2idd2id7dTfdUdd2idddiidddiid7dVfdWddiddXiddXddMdNfgdYZe j&j)dZd[d=d\d]dAd\d^dAd\d]fd_d=d\d^dAd\d]dAd\d^fd`dadbd>dNiiidcdddedfiiidadbd>dfiiifdgdhddNdigidjddkdlgidhddkdlgifdmdhddNdigidjddkdngidhddkdngifdodpdqdigidrdsdiddtdugidpdsdiddtdugifdvdpddNdwdqdigidrdsdigidpdsdigifdxdydidddzdddzfd{d=dd|dAd}d~ddAd}d~dfdiddddddfdddiiifddd4idddiidddiifddddiiddiddifddddNddgiidddd2ddddgiidddd2ddddgiifgdFededHedefdZGdde j4ZGdde j4Zej;Gdde j4Zej;Gdde j4ZGddZ GddeZ!Gdde j4Z"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8d„Z9dÄZ:dĄZ;dńZdȄZ?y)z"Tests tools in the _common module.)EnumN)ListOptional)_common)types)errorsctjdd}tjd5}||dddt dk(sJ|dj t jk(sJy#1swY.func$rTrecordr)rexperimental_warningwarningscatch_warningslencategoryr ExperimentalWarning)rws rtest_warn_oncer#s~  + , d+ qFF  Q1 1&44 44 4   s A<<Bctjdd}tjd5}t j j dz}|ddddjk(sJy#1swYxYw)Nr cyr rrrrrz$test_warn_at_call_line..func0rrTrrr)rrrrinspect currentframef_linenolineno)rcaptured_warnings call_lines rtest_warn_at_call_liner&/s  + , d+ /@$$&//!3IF  1  $ $ 11 1   s )A88Bcxtjtttt j fsJtjt jt jtt j fsJtjttttfrJtjtttt j frJtjttrJtjttt j frJtjt jt jttfrJtjt jt jtt j frJtjt jtrJtjt jtt j frJyr ) r_is_struct_typelistdictstrtypingAnyrDictintrrrtest_is_struct_typer0;ss  d3 ?&;!< == =  V[[fjj-I!J KK K  $ $T$sCx.%9 :: :  $ $T$sFJJ*?%@ AA A  $ $T#Y // /  $ $T#vzz/%: ;; ;  $ $V[[S#X1F%G HH H  $ $V[[S&**_1M%N OO O  $ $V[[%5 66 6  $ $V[[fjj%A BB B Brz5test_id, initial_target, update_dict, expected_target simple_updaterab)r5c)r4r5r7 nested_update )xy()r<z)r;r<r?add_new_nested_dictr4r5 empty_targetr; empty_updateoverwrite_non_dict_with_dictoverwrite_dict_with_non_dictdeeper_nesting)r7d)r5e)rFf)r5g)r4h)r7rFrI)r5rGrKdifferent_value_types string_val nested_intd)key1key2{ nested_list)rr2rT)rRrSkey3)rPrUupdate_with_empty_nested_dicttarget_with_empty_nested_dictkey_case_alignment_checkJohn email_addresszjohn@example.com) first_name contact_infoJanezjane@example.com123)r[ phone_number) firstNamer]test_idinitial_target update_dictexpected_targetc>tj||||k(sJyr )rrecursive_dict_update)rbrcrdres rtest_recursive_dict_updaterhIs#h  < ? ** *rz`test_id, initial_target, update_dict, expected_target, expect_warning, expected_log_message_parttype_match_intFtype_match_dicttype_mismatch_int_to_strhelloz^Type mismatch for key 'a'. Existing type: , new type: . Overwriting.type_mismatch_dict_to_intz_Type mismatch for key 'a'. Existing type: , new type: . Overwriting.type_mismatch_int_to_dictz_Type mismatch for key 'a'. Existing type: , new type: . Overwriting. add_new_keynewctj||||k(sJ|rVt|jdk(sJ|jdjdk(sJ||jdj vsJy|jD]F}|jdk(s||j vs"t jd|d|j Hy)NrrWARNINGzUnexpected warning logged for z: )rrgrrecords levelnamemessagepytestfail)rbrcrdreexpect_warningexpected_log_message_partcaplogrs r(test_recursive_dict_update_type_warningsr|s^ !!.+> _ ,, ,6>>"a'''~~a **i777(FNN1,=,E,EEEEnn [F9,1Jfnn1\=gYbHXYZ [rz8test_id, target_dict, update_dict, expected_aligned_dictsimple_snake_to_camelDoe)r\ last_name)ralastNamesimple_camel_to_snakenested_dict_alignment user_infocontact_detailsuserInfocontactDetails emailAddressztest@example.comlist_of_dicts_alignment users_list)user_id user_name usersListAlice)userIduserName,list_of_dicts_alignment_mixed_case_in_update)rUserName-list_of_dicts_different_lengths_update_longer items_dataitem_id itemsDataitemIdExtra)ritemName-list_of_dicts_different_lengths_target_longer)r item_name&no_matching_keys_preserves_update_casekey_one)KEY_TWOkeyThreemixed_match_and_no_match)r\ age_yearsEngineer)ra AGE_YEARSoccupation_title)r\rrempty_target_dict new_value anotherValue)new_key anotherKeyempty_update_dict existing_keyvalue(target_has_non_dict_value_for_nested_keyconfig_settingsconfigSettings themeNamedark(update_has_non_dict_value_for_nested_key theme_namelightdark_theme_stringdeeply_nested_with_lists level_one list_items)r item_valuelevelOne listItemsTest)r itemValueTest2 target_dictexpected_aligned_dictcNtj||}||k(s Jd|y)NzTest failed for: )ralign_key_case)rbrrdr aligned_dicts rtest_align_key_casers3x'' [A, . .M2CG90MM .rcDeZdZUeed<eed<dZeed<dZe eed<y) SimpleModelnamerT is_activeN none_field) __name__ __module__ __qualname__r+__annotations__r/rboolrrrrrrr=s$ ) *)T"*hsm"rrc,eZdZUeed<dZeded<y)ChainidNchild)rrrr/rrrrrrrrDs '!%' !rrcNeZdZUeed<ej eZe ded<y)Treer)default_factorychildrenN) rrrr/rpydanticFieldr)rrrrrrrLs  ')8>>$?(DL?rrcJeZdZUeed<ej ddZeed<y)ReprFalseModelvisiblesecretF)reprhiddenN)rrrr+rrrrrrrrrTs ,xe4&#4rrceZdZdZy) NonPydanticcy)NzNonPydantic( attr='value' )r)selfs r__repr__zNonPydantic.__repr__[s ,rN)rrrrrrrrrYs-rrceZdZdZdZy)MyEnumrr2N)rrrONETWOrrrrr_s  # #rrc eZdZy) EmptyModelN)rrrrrrrrdsrrctdd}tjdj}t ||k(sJy)Nz Test NamerTrrzZ SimpleModel( is_active=True, name='Test Name', value=123 ) )rtextwrapdedentstriprobjexpecteds r+test_repr_simple_model_defaults_and_no_nonerhsAC0# __    ch  rc<t}d}t||k(sJy)Nz EmptyModel())rrrs rtest_repr_empty_modelrts  # ( ch  rctdtd}tjdj}t ||k(sJy)Nrr2r)rrzJ Chain( child=Chain( id=2 ), id=1 ) )rrrrrrs rtest_repr_nested_modelrzsE %1+&# __    ch  rctd}td}||_||_tjdj }t ||k(sJy)Nrrr2zv Chain( child=Chain( child=<... Circular reference ...>, id=2 ), id=1 ) )rrrrrr)obj1obj2rs rtest_repr_circular_modelrsU !$ !$$*$* __    dx  rcddg}|j|tjdj}t j ||k(sJy)Nrr2zG [ 1, 2, <... Circular reference ...>, ] )appendrrrr _pretty_repr)my_listrs rtest_repr_circular_listrsQ F' .. __      g &( 22 2rcddi}||d<tjdj}tj||k(sJy)Nr4rrzJ { 'a': 1, 'self': <... Circular reference ...> } rrrrr)my_dictrs rtest_repr_circular_dictrsN !H''&/ __     g &( 22 2rcttd}tdDcic]}||}}ttd}ttd}dt j |dvsJdt j |dvsJdt j |dvsJdt j |dvsJycc}w)Nr9z<... 5 more items ...>rH) max_itemsz )r)rangesettuplerr)lstidctsttpls rtest_repr_max_itemsrs U2Y#Ry!!A!#! 59~" eBi# 3!,----cQ??A?'"6"6rQ"GGG'"6"6sa"HHH "s Ccd}ttj|ddk(sJt|tj|dk(sJy)NsdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZ)max_len]r)rrrr)b_datas rtest_repr_max_len_bytesrsF & W ! !&" 5 6& @@ @ f--fcB BB BrcPddddddiii}dtj|dvsJy)Nr4r5r3z {<... 2 items at Max depth ...>}rdepthrrnesteds rtest_repr_max_depth_dictrs7 #3S123 4& +w/C/CFRS/T TT TrcHgdggg}dtj|dvsJy)N)rFrGrGpz [<... 4 items at Max depth ...>]rrrrs rtest_repr_max_depth_listr s," # $ %& +w/C/CFRS/T TT Trchdddddgdd}tjdj}tj||k(sJy) N>rr2r)r6rHrJr2r)r5r4)rL )r r r*r)z { 'dict': { 'a': 1, 'b': 2 }, 'list': [ 7, 8, 9, ], 'set': { 1, 2, 3, }, 'tuple': ( 4, 5, 6, ) } rrs rtest_repr_collectionsr$sV A  # __ , - .   c "h .. .rcdddd}tjdj}tj||k(sJy)Nr)r)rr2)tuple0tuple1tuple2z{ { 'tuple0': (), 'tuple1': ( 1, ), 'tuple2': ( 1, 2, ) } rrs rtest_tuple_collectionsr)sN # __        c "h .. .rc~tjgdk(sJtjidk(sJtjtdk(sJtjtdk(sJtjdtit j dj k(sJy)Nz[]z{}zset()z() empty_setz8 { 'empty_set': set() } )rrr r rrrrrrtest_repr_empty_collectionsr, s   b !T )) )   b !T )) ) 35!W,,   eg &$ .. . K/0oo   57 rcd}d}tj||k(sJd}d}tj||k(sJd}d}tj||k(sJy)Nzline onez 'line one'z!line one line two with """ insidez*"""line one line two with \"\"\" inside"""zA string with """ insidez'A string with """ inside'r)s1exp1s2exp2s3exp3s rtest_repr_stringsr4sk" $   b !T )) )+" 9$   b !T )) )!" '$   b !T )) )rctdd}t|}d|vsJd|vsJtjdj }||k(sJy)Nshowhide)rrzvisible='show'rz4 ReprFalseModel( visible='show' ) )rrrrrrresultrs rtest_repr_repr_falser:+sbvf5# 9& V ## #    __    8  rctddd}t|}d|vsJtjdj }||k(sJy)Nz Only Namer)rrrrzX SimpleModel( is_active=True, name='Only Name', value=0 ) )rrrrrr8s rtest_repr_none_fieldsr<8sVA$?# 9& V ## # __    8  rct}tj}||d}tjdj }t j||k(sJy)N)npenzh { 'en': , 'np': NonPydantic( attr='value' ) } )rrrrrrrr)r>r?rrs rtest_repr_other_typesr@FsV}" zz"# __      c "h .. .rctdd}tjdj}t j |d|k(sJy)Nz Indent Testrrz` SimpleModel( is_active=True, name='Indent Test', value=1 ) r6) indent_delta)rrrrrrrs rtest_repr_indent_deltarCUsJa0# __      c 2h >> >rc tjgtjtjtjdgdtj j dgdtjdd tjtjjd gd d }tjdj}t||k(sJy)NaAThere isn't a single "best" LLM, as the ideal choice highly depends on your specific needs, use case, budget, and priorities. The field is evolving incredibly fast, with new models and improvements being released constantly. However, we can talk about the **leading contenders** and what they are generally known for:...)textmodel)partsroler)content finish_reasonindexz%models/gemini-2.5-flash-preview-05-20i>rL)modality token_counti\i)candidates_token_countprompt_token_countprompt_tokens_detailsthoughts_token_counttotal_token_count)"automatic_function_calling_history candidates model_versionusage_metadataa GenerateContentResponse( automatic_function_calling_history=[], candidates=[ Candidate( content=Content( parts=[ Part( text="""There isn't a single "best" LLM, as the ideal choice highly depends on your specific needs, use case, budget, and priorities. The field is evolving incredibly fast, with new models and improvements being released constantly. However, we can talk about the **leading contenders** and what they are generally known for:...""" ), ], role='model' ), finish_reason=, index=0 ), ], model_version='models/gemini-2.5-flash-preview-05-20', usage_metadata=GenerateContentResponseUsageMetadata( candidates_token_count=1086, prompt_token_count=7, prompt_tokens_details=[ ModalityTokenCount( modality=, token_count=7 ), ], thoughts_token_count=860, total_token_count=1953 ) ) )rGenerateContentResponse CandidateContentPart FinishReasonSTOP$GenerateContentResponseUsageMetadataModalityTokenCount MediaModalityTEXTrrrrrs rtest_repr_complex_objectraas %%)+ //mmjj c "..33  <??!%&& ..33! #  ' #B__!!B uwC D ch  rcdddddigiiddddddigiidddddd igiiddgi}d d i}tj||dd dddigidd id dddigidd id ddd igidd igi}||k(sJy)z>Test move_value_by_path function with array wildcard notation.requestsrIrGrE1@)requestoutputDimensionality23z requests[].*zrequests[].request.*rf)rIrgN)rmove_value_by_path)datapathsrs rtest_move_value_by_pathrmsS$c ') $c ') $c ') 3%' $R 1 2% T5)$c +-  $c +-  $c +-  3%'(R   rcrGddtjGfddtj}dddii}|jtjd 5|j |}d d d j jdk(sJt|jd k(sJy #1swY?xYw) z0Test that no warning is logged when types match.ceZdZUeed<y)Mtest_check_field_type_mismatches_no_warning_for_correct_types..ModelArNrrrr/rrrrModelArp JrrrceZdZUWed<y)Ptest_check_field_type_mismatches_no_warning_for_correct_types..TestModelmodel_aNrrrrrrsr TestModelrus OrryrvrrTgoogle_genai._commonloggerNr) r BaseModelat_levelloggingrsmodel_validatervrrrtr{ryrkr9rrs @r=test_check_field_type_mismatches_no_warning_for_correct_typesrsw  '## gs^ $$ w/EF,  % %d +F,    $$ $ V^^  !! ! ,,s B--B6cGddtjGddtj}Gfddtj}|d}d |i}|jtjd 5|j |d d d t |jd k(sJd|jdjvsJd|jdjvsJy #1swY^xYw)z?Test that warning is logged when Pydantic model types mismatch.ceZdZUeed<y)Ptest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelArNrqrrrrrrrsrrrceZdZUeed<y)Ptest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelBrNrrrr+rrrrModelBrrsrrceZdZUWed<y)Stest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..TestModel model_fieldNrwrxsrryr rrytestrrrzr{Nrz&Type mismatch in TestModel.model_fieldrexpected ModelA, got ModelB rr}r~rrs_check_field_type_mismatchesrrtrvr{rrymodel_b_instancerkrrs @r@test_check_field_type_mismatches_warns_on_pydantic_type_mismatchrsw  w  '##&) ) *$ w/EF1 **401 V^^  !! ! 1V^^A5F5N5N NN N &&..*;*C*C CC C 11s =C,,C5cTGddtjGfddtj}ddi}|jtjd5|j |}dddj Jt|jd k(sJy#1swY2xYw) z/Test that no warning is logged for None values.ceZdZUeed<y)Ktest_check_field_type_mismatches_no_warning_for_none_values..ModelArNrqrrrrrr2rsrrrc&eZdZUdZeWed<y)Ntest_check_field_type_mismatches_no_warning_for_none_values..TestModelNrrrrrrrrxsrryr5$(K&!(rryrNrzr{r rr}r~rrsrrrrtrs @r;test_check_field_type_mismatches_no_warning_for_none_valuesr/sw  )'##)  $ w/EF,  % %d +F,    ## # V^^  !! ! ,,s BB'cPGddtjGfddtj}i}|jtjd5|j |}dddj Jt|jdk(sJy#1swY2xYw) z2Test that no warning is logged for missing fields.ceZdZUeed<y)Ntest_check_field_type_mismatches_no_warning_for_missing_fields..ModelArNrqrrrrrrDrsrrrc&eZdZUdZeWed<y)Qtest_check_field_type_mismatches_no_warning_for_missing_fields..TestModelNrrrxsrryrGrrryrzr{Nrrrs @r>test_check_field_type_mismatches_no_warning_for_missing_fieldsrAsw  )'##) $ w/EF,  % %d +F,    ## # V^^  !! ! ,,s BB%cFGddtj}ddd}|jtjd5|j |}dddj d k(sJ|jdk(sJt|jd k(sJy#1swYFxYw) z=Test that no warning is logged for primitive type mismatches.c"eZdZUeed<eed<y)Rtest_check_field_type_mismatches_no_warning_for_primitive_types..TestModel int_field str_fieldN)rrrr/rr+rrrryrVs NNrryr_r)rrrzr{NrTr) rr}r~rrsrrrrrt)r{ryrkr9s r?test_check_field_type_mismatches_no_warning_for_primitive_typesrSs'## 6 2$ w/EF,  % %d +F,   S    V ## # V^^  !! ! ,,s BB cGddtjGddtj}Gfddtj}|d}d |i}|jtjd 5|j |d d d t |jd k(sJd|jdjvsJy #1swYAxYw)z@Test that Optional types are properly unwrapped before checking.ceZdZUeed<y)Ltest_check_field_type_mismatches_handles_optional_unwrapping..ModelArNrqrrrrrrjrsrrrceZdZUeed<y)Ltest_check_field_type_mismatches_handles_optional_unwrapping..ModelBrNrrrrrrmrsrrc&eZdZUdZeWed<y)Otest_check_field_type_mismatches_handles_optional_unwrapping..TestModelNrrrxsrryrprrryrrrrzr{Nrrrrrs @r.ModelArNrqrrrrrrrsrrrceZdZUWed<y)\test_check_field_type_mismatches_no_warning_for_correct_pydantic_instance..TestModelrNrwrxsrryrrrry*rrrzr{Nr) rr}r~rrsrrrrrt)r{rymodel_a_instancerkr9rrs @rItest_check_field_type_mismatches_no_warning_for_correct_pydantic_instancer~sw  '##"% ) *$ w/EF,  % %d +F,    ! !R '' ' V^^  !! ! ,,s $B44B=cGddtjGddtj}Gfddtj}|d}|d d d }|jtjd 5|j |d d d t |jdk(sJd|jdjvsJd|jdjvsJy #1swY^xYw)z3Test checking multiple fields with mixed scenarios.ceZdZUeed<y)Etest_check_field_type_mismatches_with_multiple_fields..ModelArNrqrrrrrrrsrrrceZdZUeed<y)Etest_check_field_type_mismatches_with_multiple_fields..ModelBrNrrrrrrrsrrc<eZdZUWed<dZeWed<eed<y)Htest_check_field_type_mismatches_with_multiple_fields..TestModelfield_aNfield_bfield_c)rrrrrrr+rxsrryrs O $GXf $ LrrywrongrNr)rrrrzr{rrrrrrs @r5test_check_field_type_mismatches_with_multiple_fieldsrsw  w  '## '*! $ w/EF1 **401 V^^  !! ! fnnQ'// // / &&..*;*C*C CC C 11s ?C..C7cGddtj}dddgi}|jtjd5|j |dddt |jd k(sJy#1swY$xYw) zCTest that validation doesn't crash on generic types like list[str].ceZdZUeeed<y)Itest_check_field_type_mismatches_generic_type_no_error..TestModeltagsN)rrrr)r+rrrrryrs s)Orryrr4r5rzr{Nr)rr}r~rrsrrrt)r{ryrks r6test_check_field_type_mismatches_generic_type_no_errorrsu'## 3* $ w/EF# T"# V^^  !! !##s A66A?)@__doc__enumrr rrr,rrrrrwrjrrr rr&r0mark parametrizer+r*rhr|rr}rr model_rebuildrrrrrrrrrrrrrrr r$r)r,r4r:r<r@rCrarmrrrrrrrrrrrrrs") !  5 2 C; ! ! !! $   , -$1 -"5A >   " !H $ %, -    3) $3) $   3) $ 3) $   +!  3) 3) $   +3) $ !H!     aa(q1 2a(q1 :aa0qqA J   $!L#+> ?=)" $! 6 e 4! 6   $ e 4! 6 e 4   $ ,.CD E *^=O,PQ R ,@R.ST U   & ;< = aW=> ? qg>? @   ; ;< = qg>? @ w?@ A   < YN+ , Ha=aW*MN O XqMqg+NO P   < ;i^L M Hb>* + XrN+ ,   5 Nq )q )   '! 3 rz R! S   #> B#> B   W %     7  $  V4 5 f 5 6   7 w 7 8 2 3  3 4   ' <!*L)MN O +V#(N]dsvPw'xy z