o ih@sdZddlmZddlZddlZddlZddlZddlmZmZddl Z ddl Z ddl Z ddl m Z ddl mZddl mZd d Zd d Zd dZe jddddddddddddfddddddddddddddddddfdd did!dddidddddfd"idd#diddd#didfd$dd#dididd#didfd%dddd!d#diidd#didfd&dd#didd!didddfd'd ddd(dd)idd*d+d,d-d.d/ddd*d0dd,d1d.d/fd2d3d4d5id6d7d8gd9id:d;d7d5gd9d<d:d;fd=d d!diid iid d!diifd>d iid d!diid d!diifd?d@dAdBidCdDdEdFdGdHdDdEdFdGdCfg dIedJedKedLefdMdNZe jdOdPd did did didQdRfdSd d!diid d!diid d!diidQdRfdTd did dUid dUid:dVfdWd d!diid d5id d5id:dXfdYd d5id d!diid d!diid:dZfd[d did!d\idd\ddQdRfgd]d^Ze jd_d`d@dadbdDdadcdDdadbfddd@dadcdDdadbdDdadcfdedfdgdAdRiiidhdidjdkiiidfdgdAdkiiifdldmddRdngidoddpdqgidmddpdqgifdrdmddRdngidoddpdsgidmddpdsgifdtdudvdigidwdxdiddydzgidudxdiddydzgifd{duddRd|dvdigidwdxdigidudxdigifd}d~diddddddfdd@dddDddddDdddfdiddddddfdddiiifddd7idddiidddiifddddiiddiddifddddRddgiidddd5ddddgiidddd5ddddgiifgdIededKedefddZGddde jZGddde jZeGddde jZeGddde jZGdddZ GdddeZ!Gddde jZ"ddZ#ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)ddZ*ddÄZ+ddńZ,ddDŽZ-ddɄZ.dd˄Z/dd̈́Z0ddτZ1ddфZ2ddӄZ3ddՄZ4ddׄZ5ddلZ6ddۄZ7dd݄Z8dd߄Z9ddZ:ddZ;ddZddZ?dS)z"Tests tools in the _common module.)EnumN)ListOptional)_common)types)errorscCsrtddd}tjdd}||Wdn1s wYt|dks-J|djtjks7JdS)NWarning!cSdSNr r r Y/home/ubuntu/.local/lib/python3.10/site-packages/google/genai/tests/common/test_common.pyfunc$ztest_warn_once..funcTrecordr)rexperimental_warningwarningscatch_warningslencategoryrExperimentalWarning)rwr r r test_warn_once#s rcCshtddd}tjdd}tjd}|Wdn1s$wY|dj|ks2JdS)Nr cSr r r r r r r r0rz$test_warn_at_call_line..funcTrrr)rrrrinspect currentframef_linenolineno)rcaptured_warnings call_liner r r test_warn_at_call_line/s r!cCs tttttjfsJttjtjttjfsJttttt fr+Jtttt tjfr9JtttrBJttttjfrNJttjtjtt fr]Jttjtjt tjfrmJttjtrwJttjttjfrJdSr ) r_is_struct_typelistdictstrtypingAnyrDictintr r r r test_is_struct_type;s  r*z5test_id, initial_target, update_dict, expected_target simple_updaterab)r/c)r.r/r1 nested_update )xy()r6z)r5r6r9add_new_nested_dictr.r/ empty_targetr5 empty_updateoverwrite_non_dict_with_dictoverwrite_dict_with_non_dictdeeper_nesting)r1d)r/e)r@f)r/g)r.h)r1r@rC)r/rArEdifferent_value_types string_val nested_intd)key1key2{ nested_list)rr,rT)rLrMkey3)rJrOupdate_with_empty_nested_dicttarget_with_empty_nested_dictkey_case_alignment_checkJohn email_addresszjohn@example.com) first_name contact_infoJanezjane@example.com123)rU phone_number) firstNamerWtest_idinitial_target update_dictexpected_targetcCst||||ks JdSr )rrecursive_dict_update)r\r]r^r_r r r test_recursive_dict_updateIs Traz`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_keynewcCst||||ks J|r-t|jdksJ|jdjdks!J||jdjvs+JdS|jD]}|jdkrH||jvrHtd|d|jq0dS)NrrWARNINGzUnexpected warning logged for z: )rr`rrecords levelnamemessagepytestfail)r\r]r^r_expect_warningexpected_log_message_partcaplogrr r r (test_recursive_dict_update_type_warningss /  rtz8test_id, target_dict, update_dict, expected_aligned_dictsimple_snake_to_camelDoe)rV last_name)r[lastNamesimple_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)rV age_yearsEngineer)r[ AGE_YEARSoccupation_title)rVrrempty_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_dictcCs&t||}||ksJd|dS)NzTest failed for: )ralign_key_case)r\rr^r aligned_dictr r r test_align_key_cases \rc@s:eZdZUeed<eed<dZeed<dZe eed<dS) SimpleModelnamerT is_activeN none_field) __name__ __module__ __qualname__r%__annotations__r)rboolrrr r r r r=s  rc@s&eZdZUeed<dZeded<dS)ChainidNchild)rrrr)rrrr r r r rDs rc@s.eZdZUeed<ejedZe ded<dS)Treer)default_factorychildrenN) rrrr)rpydanticFieldr#rrr r r r rLs rc@s,eZdZUeed<ejdddZeed<dS)ReprFalseModelvisiblesecretF)reprhiddenN)rrrr%rrrrr r r r rTs rc@seZdZddZdS) NonPydanticcCsdS)NzNonPydantic( attr='value' )r )selfr r r __repr__[szNonPydantic.__repr__N)rrrrr r r r rYs rc@seZdZdZdZdS)MyEnumrr,N)rrrONETWOr r r r r_src@s eZdZdS) EmptyModelN)rrrr r r r rdsrcCs.tddd}td}t||ksJdS)Nz Test NamerNrrzZ SimpleModel( is_active=True, name='Test Name', value=123 ) )rtextwrapdedentstriprobjexpectedr r r +test_repr_simple_model_defaults_and_no_nonehs rcCst}d}t||ks JdS)Nz EmptyModel())rrrr r r test_repr_empty_modeltsrcCs4tdtddd}td}t||ksJdS)Nrr,r)rrzJ Chain( child=Chain( id=2 ), id=1 ) )rrrrrrr r r test_repr_nested_modelzs rcCsBtdd}tdd}||_||_td}t||ksJdS)Nrrr,zv Chain( child=Chain( child=<... Circular reference ...>, id=2 ), id=1 ) )rrrrrr)obj1obj2rr r r test_repr_circular_models   rcCs6ddg}||td}t||ksJdS)Nrr,zG [ 1, 2, <... Circular reference ...>, ] )appendrrrr _pretty_repr)my_listrr r r test_repr_circular_lists  rcCs4ddi}||d<td}t||ksJdS)Nr.rrzJ { 'a': 1, 'self': <... Circular reference ...> } rrrrr)my_dictrr r r test_repr_circular_dicts rcCsttd}ddtdD}ttd}ttd}dtj|ddvs&Jdtj|ddvs1Jdtj|ddvssz'test_repr_max_items..z<... 5 more items ...>rB) max_itemsz )r#rangesettuplerr)lstdctsttplr r r test_repr_max_itemss    rcCs<d}ttj|dddksJt|tj|ddksJdS)NsdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZ)max_len]r)rrrr)b_datar r r test_repr_max_len_bytessrcCs0ddddddiii}dtj|ddvsJdS)Nr.r/r-z {<... 2 items at Max depth ...>}rdepthrrnestedr r r test_repr_max_depth_dictsrcCs(gdggg}dtj|ddvsJdS)N)r@rArApz [<... 4 items at Max depth ...>]rrrrr r r test_repr_max_depth_listsrcCs@hdddddgdd}td}t||ksJdS) N>rr,r)r0rBrDr,r)r/r.)rF )rrr$r#z { 'dict': { 'a': 1, 'b': 2 }, 'list': [ 7, 8, 9, ], 'set': { 1, 2, 3, }, 'tuple': ( 4, 5, 6, ) } rrr r r test_repr_collectionssr cCs0dddd}td}t||ksJdS)Nr )r)rr,)tuple0tuple1tuple2z{ { 'tuple0': (), 'tuple1': ( 1, ), 'tuple2': ( 1, 2, ) } rrr r r test_tuple_collectionss  rcCsrtgdks JtidksJttdksJttdks&Jtdtitdks7JdS)Nz[]z{}zset()z() empty_setz8 { 'empty_set': set() } )rrrrrrrr r r r test_repr_empty_collections s rcCsRd}d}t||ks Jd}d}t||ksJd}d}t||ks'JdS)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)s1exp1s2exp2s3exp3r r r test_repr_stringssrcCsJtddd}t|}d|vsJd|vsJtd}||ks#JdS)Nshowhide)rrzvisible='show'rz4 ReprFalseModel( visible='show' ) )rrrrrrresultrr r r test_repr_repr_false+s   rcCs@tdddd}t|}d|vsJtd}||ksJdS)Nz Only Namer)rrrrzX SimpleModel( is_active=True, name='Only Name', value=0 ) )rrrrrrr r r test_repr_none_fields8s rcCs:t}tj}||d}td}t||ksJdS)N)npenzh { 'en': , 'np': NonPydantic( attr='value' ) } )rrrrrrrr)rrrrr r r test_repr_other_typesFs r cCs4tddd}td}tj|dd|ksJdS)Nz Indent Testrrz` SimpleModel( is_active=True, name='Indent Test', value=1 ) r0) indent_delta)rrrrrrrr r r test_repr_indent_deltaUs r"c Csztjgtjtjtjddgddtjjddgdtjdd tjtj j d d gd d d d}t d }t||ks;JdS)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>rF)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 MediaModalityTEXTrrrrrr r r test_repr_complex_objectas> !!"r?cCsdddddigiiddddddigiidddddd igiiddgi}d d i}t||dd dddigidd id dddigidd id ddd igidd igi}||ksZJdS)z>Test move_value_by_path function with array wildcard notation.requestsr'r%r#1@)requestoutputDimensionality23z requests[].*zrequests[].request.*rC)r'rDN)rmove_value_by_path)datapathsrr r r test_move_value_by_paths~   )   )rJcsGdddtjGfdddtj}dddii}|jtjdd  ||}Wd n1s2wY|jjdks?Jt|j d ksHJd S) z0Test that no warning is logged when types match.c@eZdZUeed<dS)zMtest_check_field_type_mismatches_no_warning_for_correct_types..ModelArNrrrr)rr r r r ModelA rMceZdZUed<dS)zPtest_check_field_type_mismatches_no_warning_for_correct_types..TestModelmodel_aNrrrrr rMr r TestModelrNrSrPrrNgoogle_genai._commonloggerNr) r BaseModelat_levelloggingrkmodel_validaterPrrrlrsrSrHrr rRr =test_check_field_type_mismatches_no_warning_for_correct_typess  r\csGdddtjGdddtj}Gfdddtj}|dd}d |i}|jtjd d  ||Wd n1s>wYt|jd ksLJd|jdjvsVJd|jdjvs`Jd S)z?Test that warning is logged when Pydantic model types mismatch.c@rK)zPtest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelArNrLr r r r rMrNrMc@rK)zPtest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelBrNrrrr%rr r r r ModelBrNr^crO)zStest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..TestModel model_fieldNrQr rRr r rSrNrStestrr_rTrUNrz&Type mismatch in TestModel.model_fieldrexpected ModelA, got ModelB rrWrXrYrk_check_field_type_mismatchesrrlrnrsr^rSmodel_b_instancerHr rRr @test_check_field_type_mismatches_warns_on_pydantic_type_mismatchs  rgcsGdddtjGfdddtj}ddi}|jtjdd ||}Wdn1s0wY|jdus.ModelArNrLr r r r rM2rNrMceZdZUdZeed<dS)zNtest_check_field_type_mismatches_no_warning_for_none_values..TestModelNr_rrrr_rrr rRr r rS5 rSr_NrTrUr rrWrXrYrkrZr_rrlr[r rRr ;test_check_field_type_mismatches_no_warning_for_none_values/s rlcsGdddtjGfdddtj}i}|jtjdd ||}Wdn1s.wY|jdus:Jt|jdksCJdS) z2Test that no warning is logged for missing fields.c@rK)zNtest_check_field_type_mismatches_no_warning_for_missing_fields..ModelArNrLr r r r rMDrNrMcrh)zQtest_check_field_type_mismatches_no_warning_for_missing_fields..TestModelNr_rir rRr r rSGrjrSrTrUNrrkr[r rRr >test_check_field_type_mismatches_no_warning_for_missing_fieldsAs rmcCsGdddtj}ddd}|jtjdd ||}Wdn1s&wY|jd ks2J|jdks9Jt|j d ksBJdS) z=Test that no warning is logged for primitive type mismatches.c@seZdZUeed<eed<dS)zRtest_check_field_type_mismatches_no_warning_for_primitive_types..TestModel int_field str_fieldN)rrrr)rr%r r r r rSVs  rSrYr`)rnrorTrUNrNr) rrWrXrYrkrZrnrorrlr[r r r ?test_check_field_type_mismatches_no_warning_for_primitive_typesSs  rpcsGdddtjGdddtj}Gfdddtj}|dd}d |i}|jtjd d  ||Wd n1s>wYt|jd ksLJd|jdjvsVJd S)z@Test that Optional types are properly unwrapped before checking.c@rK)zLtest_check_field_type_mismatches_handles_optional_unwrapping..ModelArNrLr r r r rMjrNrMc@rK)zLtest_check_field_type_mismatches_handles_optional_unwrapping..ModelBrNr]r r r r r^mrNr^crh)zOtest_check_field_type_mismatches_handles_optional_unwrapping..TestModelNr_rir rRr r rSprjrSr`rar_rTrUNrrbrrcrer rRr .ModelArNrLr r r r rMrNrMcrO)z\test_check_field_type_mismatches_no_warning_for_correct_pydantic_instance..TestModelr_NrQr rRr r rSrNrS*rar_rTrUNr) rrWrXrYrkrZr_rrrl)rsrSmodel_a_instancerHrr rRr Itest_check_field_type_mismatches_no_warning_for_correct_pydantic_instance~s  rtcsGdddtjGdddtj}Gfdddtj}|dd}|d d d }|jtjd d  ||Wd n1s@wYt|jdksNJd|jdjvsXJd|jdjvsbJd S)z3Test checking multiple fields with mixed scenarios.c@rK)zEtest_check_field_type_mismatches_with_multiple_fields..ModelArNrLr r r r rMrNrMc@rK)zEtest_check_field_type_mismatches_with_multiple_fields..ModelBrNr]r r r r r^rNr^cs.eZdZUed<dZeed<eed<dS)zHtest_check_field_type_mismatches_with_multiple_fields..TestModelfield_aNfield_bfield_c)rrrrrvrr%r rRr r rSs  rSwrongraNr`)rurvrwrTrUrrurrbrcrer rRr 5test_check_field_type_mismatches_with_multiple_fieldss  rycCsnGdddtj}dddgi}|jtjdd ||Wdn1s'wYt|jd ks5JdS) zCTest that validation doesn't crash on generic types like list[str].c@seZdZUeeed<dS)zItest_check_field_type_mismatches_generic_type_no_error..TestModeltagsN)rrrr#r%rr r r r rSs rSrzr.r/rTrUNr)rrWrXrYrkrZrrl)rsrSrHr r r 6test_check_field_type_mismatches_generic_type_no_errors   r{)@__doc__enumrrrYrr&rrrrrorcrrrrr!r*mark parametrizer%r$rartrrWrr model_rebuildrrrrrrrrrrrrrrrr rrrrrr r"r?rJr\rgrlrmrprqrtryr{r r r r s                      Q          .         Y     !  GZ