hjhdZddlmZddlZddlZddlZddlZddlmZmZddl Z ddl Z ddl Z ddl m Z ddl mZddl mZd Zd Zd Ze jd 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 jdKdLddiddiddidMdNfdOdddiidddiidddiidMdNfdPddiddQiddQid7dRfdSdddiidd2idd2id7dTfdUdd2idddiidddiid7dVfdWddiddXiddXddMdNfgdYZe jdZd[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 jZGdde jZeGdde jZeGdde jZGddZ GddeZ!Gdde jZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8d„Z9dÄZ:dĄZ;dńZdȄZ?dS)z"Tests tools in the _common module.)EnumN)ListOptional)_common)types)errorsc&tjdd}tjd5}||dddn #1swxYwYt |dksJ|djt jksJdS)NWarning!cdSNri/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/google/genai/tests/common/test_common.pyfuncztest_warn_once..func$DrTrecordr)rexperimental_warningwarningscatch_warningslencategoryr ExperimentalWarning)rws rtest_warn_oncer#s  ++  ,+ d+++ qDFFFDFFF                Q1 1&4 4 4 4 4 4 4sAAAc tjdd}tjd5}t jjdz}|dddn #1swxYwY|dj|ksJdS)Nr cdSr rrrrrz$test_warn_at_call_line..func0rrTrrr)rrrrinspect currentframef_linenolineno)rcaptured_warnings call_lines rtest_warn_at_call_liner&/s  ++  ,+ d+++ /@$&&/!3IDFFF                1  $ 1 1 1 1 1 1s&A%%A),A)c~tjtttt jfsJtjt jt jtt jfsJtjttttfrJtjtttt jfrJtjttrJtjttt jfrJtjt jt jttfrJtjt jt jtt jfrJtjt jtrJtjt jtt jfrJdSr ) r_is_struct_typelistdictstrtypingAnyrDictintrrrtest_is_struct_typer0;sx  d3 ?&;!< = ====  V[fj-I!J K KKKK  $T$sCx.%9 : ::::  $T$sFJ*?%@ A AAAA  $T#Y / ////  $T#vz/%: ; ;;;;  $V[S#X1F%G H HHHH  $V[S&*_1M%N O OOOO  $V[%5 6 6666  $V[fj%A B BBBBBBrz5test_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||||ksJdSr )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_keynewcRtj||||ksJ|rJt|jdksJ|jdjdksJ||jdjvsJdS|jD]5}|jdkr(||jvrt jd|d|j6dS)NrrWARNINGzUnexpected warning logged for z: )rrgrrecords levelnamemessagepytestfail)rbrcrdreexpect_warningexpected_log_message_partcaplogrs r(test_recursive_dict_update_type_warningsr|s^ !.+>>> _ , , , ,[6>""a''''~a *i7777(FN1,=,EEEEEEEn [ [F9,,1Jfn1\1\YgYYYYZZZ [ [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_dictcVtj||}||ks Jd|dS)NzTest failed for: )ralign_key_case)rbrrdr aligned_dicts rtest_align_key_casers@x' [AA, . . . .0MG0M0M . . . . .rcLeZdZUeed<eed<dZeed<dZe eed<dS) SimpleModelnamerT is_activeN none_field) __name__ __module__ __qualname__r+__annotations__r/rboolrrrrrrr=sH ))) ***)T"*hsm"""""rrc4eZdZUeed<dZeded<dS)ChainidNchild)rrrr/rrrrrrrrDs0 '''!%' !!!!!rrcReZdZUeed<ejeZe ded<dS)Treer)default_factorychildrenN) rrrr/rpydanticFieldr)rrrrrrrLs> ''')8>$???(DL?????rrcHeZdZUeed<ejddZeed<dS)ReprFalseModelvisiblesecretF)reprhiddenN)rrrr+rrrrrrrrrTs< ,,,xe444&#44444rrceZdZdZdS) NonPydanticcdS)NzNonPydantic( attr='value' )r)selfs r__repr__zNonPydantic.__repr__[s , ,rN)rrrrrrrrrYs#-----rrceZdZdZdZdS)MyEnumrr2N)rrrONETWOrrrrr_s # ###rrceZdZdS) EmptyModelN)rrrrrrrrds$rrctdd}tjd}t ||ksJdS)Nz Test NamerTrrzZ SimpleModel( is_active=True, name='Test Name', value=123 ) )rtextwrapdedentstriprobjexpecteds r+test_repr_simple_model_defaults_and_no_nonerhsWC000# _     ch      rcPt}d}t||ksJdS)Nz EmptyModel())rrrs rtest_repr_empty_modelrts- # ( ch      rctdtd}tjd}t ||ksJdS)Nrr2r)rrzJ Chain( child=Chain( id=2 ), id=1 ) )rrrrrrs rtest_repr_nested_modelrzs_ %1+++&&&# _     ch      rctd}td}||_||_tjd}t ||ksJdS)Nrrr2zv Chain( child=Chain( child=<... Circular reference ...>, id=2 ), id=1 ) )rrrrrr)obj1obj2rs rtest_repr_circular_modelrsm !$ !$$*$* _     dx      rcddg}||tjd}t j||ksJdS)Nrr2zG [ 1, 2, <... Circular reference ...>, ] )appendrrrr _pretty_repr)my_listrs rtest_repr_circular_listrse F' .. _      g & &( 2 2 2 2 2 2rcddi}||d<tjd}tj||ksJdS)Nr4rrzJ { 'a': 1, 'self': <... Circular reference ...> } rrrrr)my_dictrs rtest_repr_circular_dictrs\ !H''&/ _     g & &( 2 2 2 2 2 2rcttd}dtdD}ttd}ttd}dt j|dvsJdt j|dvsJdt j|dvsJdt j|dvsJdS)Nr9ci|]}||Srr).0is r z'test_repr_max_items..s!!!!A!!!rz<... 5 more items ...>rH) max_itemsz )r)rangesettuplerr)lstdctsttpls rtest_repr_max_itemsrs U2YY#!!uRyy!!!# 599~~" eBii# 3!,,,-----cQ???????'"6rQ"G"G"GGGGG'"6sa"H"H"HHHHHHHrcd}ttj|ddksJt|tj|dksJdS)NsdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZ)max_len]r)rrrr)b_datas rtest_repr_max_len_bytesrs] & W !&" 5 5 5 6 6& @ @ @ @ f-fcBBB B B B B B BrcPddddddiii}dtj|dvsJdS)Nr4r5r3z {<... 2 items at Max depth ...>}rdepthrrnesteds rtest_repr_max_depth_dictr sF #3S1123 4& +w/CFRS/T/T/T T T T T T TrcHgdggg}dtj|dvsJdS)N)rFrGrGpz [<... 4 items at Max depth ...>]rrrrs rtest_repr_max_depth_listr#s>""" # $ %& +w/CFRS/T/T/T T T T T T Trchdddddgdd}tjd}tj||ksJdS) N>rr2r)r6rHrJr2r)r5r4)rL )r rr*r)z { 'dict': { 'a': 1, 'b': 2 }, 'list': [ 7, 8, 9, ], 'set': { 1, 2, 3, }, 'tuple': ( 4, 5, 6, ) } rrs rtest_repr_collectionsr'st YYAii   # _  , - .  c " "h . . . . . .rcdddd}tjd}tj||ksJdS)Nr)r)rr2)tuple0tuple1tuple2z{ { 'tuple0': (), 'tuple1': ( 1, ), 'tuple2': ( 1, 2, ) } rrs rtest_tuple_collectionsr,sa  # _        c " "h . . . . . .rctjgdksJtjidksJtjtdksJtjtdksJtjdtit jdksJdS)Nz[]z{}zset()z() empty_setz8 { 'empty_set': set() } )rrr rrrrrrrtest_repr_empty_collectionsr/ s  b ! !T ) ) ) )  b ! !T ) ) ) ) 355!!W,,,,  egg & &$ . . . . K/00o    577 rcd}d}tj||ksJd}d}tj||ksJd}d}tj||ksJdS)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_stringsr7sz" $  b ! !T ) ) ) )+" 9$  b ! !T ) ) ) )!" '$  b ! !T ) ) ) ) ) )rctdd}t|}d|vsJd|vsJtjd}||ksJdS)Nshowhide)rrzvisible='show'rz4 ReprFalseModel( visible='show' ) )rrrrrrresultrs rtest_repr_repr_falser=+s~vf555# 99& V # # # #      _     8      rctddd}t|}d|vsJtjd}||ksJdS)Nz Only Namer)rrrrzX SimpleModel( is_active=True, name='Only Name', value=0 ) )rrrrrr;s rtest_repr_none_fieldsr?8soA$???# 99& V # # # # _     8      rct}tj}||d}tjd}t j||ksJdS)N)npenzh { 'en': , 'np': NonPydantic( attr='value' ) } )rrrrrrrr)rArBrrs rtest_repr_other_typesrCFsg}}" z"# _      c " "h . . . . . .rctdd}tjd}t j|d|ksJdS)Nz Indent Testrrz` SimpleModel( is_active=True, name='Indent Test', value=1 ) r6) indent_delta)rrrrrrrs rtest_repr_indent_deltarFUsaa000# _      c 2 2 2h > > > > > >rc tjgtjtjtjdgdtjjdgdtjdd tjtj j d gd d }tj d }t||ksJdS)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_objectrdas %)+ /mj c   ".3  <?!%& .3! #    '   #B_!!!B uwwC D ch      rcdddddigiiddddddigiidddddd igiiddgi}d d i}tj||dd dddigidd id dddigidd id ddd igidd igi}||ksJdS)z>Test move_value_by_path function with array wildcard notation.requestsrLrJrH1@)requestoutputDimensionality23z requests[].*zrequests[].request.*ri)rLrjN)rmove_value_by_path)datapathsrs rtest_move_value_by_pathrps}$c ') $c ') $c ') 3%' $R 1 2% T5)))$c +-   $c +-   $c +-   3%'(R       rclGddtjGfddtj}dddii}|tjd 5||}d d d n #1swxYwY|jjdksJt|j d ksJd S) z0Test that no warning is logged when types match.ceZdZUeed<dS)Mtest_check_field_type_mismatches_no_warning_for_correct_types..ModelArNrrrr/rrrrModelArs JJJJJrruceZdZUed<dS)Ptest_check_field_type_mismatches_no_warning_for_correct_types..TestModelmodel_aNrrrrrusr TestModelrxs OOOOOrr|ryrrTgoogle_genai._commonloggerNr) r BaseModelat_levelloggingrsmodel_validateryrrrtr{r|rnr<rus @r=test_check_field_type_mismatches_no_warning_for_correct_typesrs1w '# gs^ $$ w/EFF,,  % %d + +F,,,,,,,,,,,,,,,   $ $ $ $ V^   ! ! ! ! ! !sA==BBcGddtjGddtj}Gfddtj}|d}d |i}|tjd 5||d d d n #1swxYwYt |jd ksJd|jdjvsJd|jdjvsJd S)z?Test that warning is logged when Pydantic model types mismatch.ceZdZUeed<dS)Ptest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelArNrtrrrrurrvrruceZdZUeed<dS)Ptest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelBrNrrrr+rrrrModelBrrvrrceZdZUed<dS)Stest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..TestModel model_fieldNrzr{srr|rrr|testrrr}r~Nrz&Type mismatch in TestModel.model_fieldrexpected ModelA, got ModelB rrrrrs_check_field_type_mismatchesrrtrvr{rr|model_b_instancernrus @r@test_check_field_type_mismatches_warns_on_pydantic_type_mismatchrsw w '#V&))) ) *$ w/EFF11 **4000111111111111111 V^   ! ! ! ! 1V^A5F5N N N N N &&.*;*C C C C C C C=BB#&B#cVGddtjGfddtj}ddi}|tjd5||}dddn #1swxYwY|jJt|jd ksJdS) z/Test that no warning is logged for None values.ceZdZUeed<dS)Ktest_check_field_type_mismatches_no_warning_for_none_values..ModelArNrtrrrrur2rvrruc,eZdZUdZeed<dS)Ntest_check_field_type_mismatches_no_warning_for_none_values..TestModelNrrrrrrrr{srr|r5($(K&!(((((rr|rNr}r~r rrrrrsrrrrtrs @r;test_check_field_type_mismatches_no_warning_for_none_valuesr/s&w )))))))'#)))  $ w/EFF,,  % %d + +F,,,,,,,,,,,,,,,   # # # V^   ! ! ! ! ! !sA;;A?A?cRGddtjGfddtj}i}|tjd5||}dddn #1swxYwY|jJt|jdksJdS) z2Test that no warning is logged for missing fields.ceZdZUeed<dS)Ntest_check_field_type_mismatches_no_warning_for_missing_fields..ModelArNrtrrrrurDrvrruc,eZdZUdZeed<dS)Qtest_check_field_type_mismatches_no_warning_for_missing_fields..TestModelNrrr{srr|rGrrr|r}r~Nrrrs @r>test_check_field_type_mismatches_no_warning_for_missing_fieldsrAs!w )))))))'#))) $ w/EFF,,  % %d + +F,,,,,,,,,,,,,,,   # # # V^   ! ! ! ! ! !sA99A=A=cDGddtj}ddd}|tjd5||}dddn #1swxYwY|jd ksJ|jdksJt|j d ksJdS) z=Test that no warning is logged for primitive type mismatches.c$eZdZUeed<eed<dS)Rtest_check_field_type_mismatches_no_warning_for_primitive_types..TestModel int_field str_fieldN)rrrr/rr+rrrr|rVs"NNNNNNNNrr|r_r)rrr}r~NrTr) rrrrrsrrrrrt)r{r|rnr<s r?test_check_field_type_mismatches_no_warning_for_primitive_typesrSs '# 6 2 2$ w/EFF,,  % %d + +F,,,,,,,,,,,,,,,  S  V # # # # V^   ! ! ! ! ! !sA!!A%(A%cGddtjGddtj}Gfddtj}|d}d |i}|tjd 5||d d d n #1swxYwYt |jd ksJd|jdjvsJd S)z@Test that Optional types are properly unwrapped before checking.ceZdZUeed<dS)Ltest_check_field_type_mismatches_handles_optional_unwrapping..ModelArNrtrrrrurjrvrruceZdZUeed<dS)Ltest_check_field_type_mismatches_handles_optional_unwrapping..ModelBrNrrrrrrmrvrrc,eZdZUdZeed<dS)Otest_check_field_type_mismatches_handles_optional_unwrapping..TestModelNrrr{srr|rprrr|rrrr}r~Nrrrrrs @r.ModelArNrtrrrrurrvrruceZdZUed<dS)\test_check_field_type_mismatches_no_warning_for_correct_pydantic_instance..TestModelrNrzr{srr|rrrr|*rrr}r~Nr) rrrrrsrrrrrt)r{r|model_a_instancernr<rus @rItest_check_field_type_mismatches_no_warning_for_correct_pydantic_instancer~sBw '#V"%%% ) *$ w/EFF,,  % %d + +F,,,,,,,,,,,,,,,   !R ' ' ' ' V^   ! ! ! ! ! !s%BB B cGddtjGddtj}Gfddtj}|d}|d d d }|tjd 5||d d d n #1swxYwYt |jdksJd|jdjvsJd|jdjvsJd S)z3Test checking multiple fields with mixed scenarios.ceZdZUeed<dS)Etest_check_field_type_mismatches_with_multiple_fields..ModelArNrtrrrrurrvrruceZdZUeed<dS)Etest_check_field_type_mismatches_with_multiple_fields..ModelBrNrrrrrrrvrrc@eZdZUed<dZeed<eed<dS)Htest_check_field_type_mismatches_with_multiple_fields..TestModelfield_aNfield_bfield_c)rrrrrrr+r{srr|rs8 OOO $GXf $$$ LLLLLrr|wrongrNr)rrrr}r~rrrrrrs @r5test_check_field_type_mismatches_with_multiple_fieldsrsw w '# V'***!  $ w/EFF11 **4000111111111111111 V^   ! ! ! ! fnQ'/ / / / / &&.*;*C C C C C C Cs?B!!B%(B%cGddtj}dddgi}|tjd5||dddn #1swxYwYt |jd ksJdS) zCTest that validation doesn't crash on generic types like list[str].c&eZdZUeeed<dS)Itest_check_field_type_mismatches_generic_type_no_error..TestModeltagsN)rrrr)r+rrrrr|rs s)OOOOOrr|rr4r5r}r~Nr)rrrrrsrrrt)r{r|rns r6test_check_field_type_mismatches_generic_type_no_errorrs'# 3* $ w/EFF## T"""############### V^   ! ! ! ! ! !sA""A&)A&)@__doc__enumrr rrr,rrrrrwrjrrr rr&r0mark parametrizer+r*rhr|rrrr model_rebuildrrrrrrrrrrrrrr r#r'r,r/r7r=r?rCrFrdrprrrrrrrrrrrrrs ")( !!!!!!!!  5 5 5 2 2 2 C C C; !  !  !! $ $   ,, - -$$1 - -"55A > >   " !H $$ %,, - -    3) $ $3) $ $   3) $ $ 3) $ $   +!   3) 3) $ $   +3) $ $ !H!      aa((q11 2a((q11 : :aa00qqAA J J   $!L#+> ? ?=)))" $! 6 6 e 4 4! 6 6   $ e 4 4! 6 6 e 4 4   $ ,.CD E *^=O,PQ R ,@R.ST U   & ;;< = aW==> ? qg>>? @   ; ;;< = qg>>? @ w??@ A   < YN+ , Ha=aW*M*MN O XqMqg+N+NO P   < ;;i^L M Hb>* + XrN+ ,   5 Nq ) )q ) )   '! 3 3 rz R R! S S   #> B B#> B B   W %     7  $  V4 5 f 5 6   7 w 7 8 2 3  3 4   ' <!*L*L)MN O +V#(N(N]dsvPwPw'xy z