o Yih@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 _/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/common/test_common.pyfunc$ztest_warn_once..funcTrecordr)rexperimental_warningwarningscatch_warningslencategoryrZExperimentalWarning)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)rZcaptured_warningsZ call_liner r r test_warn_at_call_line/s rcCs tttttjfsJttjtjttjfsJttttt fr+Jtttt tjfr9JtttrBJttttjfrNJttjtjtt fr]Jttjtjt tjfrmJttjtrwJttjttjfrJdSr ) rZ_is_struct_typelistdictstrtypingAnyrDictintr r r r test_is_struct_type;s  r&z5test_id, initial_target, update_dict, expected_targetZ simple_updaterab)r*c)r)r*r,Z nested_update )xy()r0z)r/r0r3Zadd_new_nested_dictr)r*Z empty_targetr/Z empty_updateZoverwrite_non_dict_with_dictZoverwrite_dict_with_non_dictZdeeper_nesting)r,d)r*e)r4f)r*g)r)h)r,r4r7)r*r5r9Zdifferent_value_typesZ string_val nested_intd)key1key2{ nested_list)rr'rT)r>r?Zkey3)r<rAZupdate_with_empty_nested_dictZtarget_with_empty_nested_dictZkey_case_alignment_checkZJohn email_addresszjohn@example.com) first_name contact_infoZJanezjane@example.com123)rBZ phone_number) firstNamerDtest_idinitial_target update_dictexpected_targetcCst||||ks JdSr )rrecursive_dict_update)rGrHrIrJr r r test_recursive_dict_updateIs TrLz`test_id, initial_target, update_dict, expected_target, expect_warning, expected_log_message_partZtype_match_intFZtype_match_dictZtype_mismatch_int_to_strZhelloz^Type mismatch for key 'a'. Existing type: , new type: . Overwriting.Ztype_mismatch_dict_to_intz_Type mismatch for key 'a'. Existing type: , new type: . Overwriting.Ztype_mismatch_int_to_dictz_Type mismatch for key 'a'. Existing type: , new type: . Overwriting.Z add_new_keynewcCst||||ks J|r-t|jdksJ|jdjdks!J||jdjvs+JdS|jD]}|jdkrH||jvrHtd|d|jq0dS)NrrWARNINGzUnexpected warning logged for z: )rrKrrecords levelnamemessagepytestZfail)rGrHrIrJZexpect_warningZexpected_log_message_partcaplogrr r r (test_recursive_dict_update_type_warningss /  rUz8test_id, target_dict, update_dict, expected_aligned_dictZsimple_snake_to_camelZDoe)rC last_name)rFZlastNameZsimple_camel_to_snakeZnested_dict_alignmentZ user_infoZcontact_detailsZuserInfoZcontactDetailsZ emailAddressztest@example.comZlist_of_dicts_alignmentZ users_list)user_idZ user_nameZ usersListZAlice)ZuserIdZuserNameZ,list_of_dicts_alignment_mixed_case_in_update)rWZUserNameZ-list_of_dicts_different_lengths_update_longerZ items_dataitem_idZ itemsDataZitemIdZExtra)rXitemNameZ-list_of_dicts_different_lengths_target_longer)rX item_nameZ&no_matching_keys_preserves_update_caseZkey_one)ZKEY_TWOZkeyThreeZmixed_match_and_no_match)rC age_yearsZEngineer)rFZ AGE_YEARSoccupation_title)rCr[r]Zempty_target_dict new_valueZ anotherValue)Znew_keyZ anotherKeyZempty_update_dictZ existing_keyvalueZ(target_has_non_dict_value_for_nested_keyconfig_settingsZconfigSettingsZ themeNameZdarkZ(update_has_non_dict_value_for_nested_keyZ theme_nameZlightZdark_theme_stringZdeeply_nested_with_listsZ level_oneZ list_items)rZZ item_valueZlevelOneZ listItemsZTest)rYZ itemValueZTest2 target_dictexpected_aligned_dictcCs&t||}||ksJd|dS)NzTest failed for: )rZalign_key_case)rGrbrIrcZ aligned_dictr r r test_align_key_cases \rdc@s:eZdZUeed<eed<dZeed<dZe eed<dS) SimpleModelnamer_T is_activeN none_field) __name__ __module__ __qualname__r!__annotations__r%rgboolrhrr r r r re=s  rec@s&eZdZUeed<dZeded<dS)ChainidNchild)rirjrkr%rlrprr r r r rnDs rnc@s.eZdZUeed<ejedZe ded<dS)Treero)default_factorychildrenN) rirjrkr%rlpydanticFieldrrsrr r r r rqLs rqc@s,eZdZUeed<ejdddZeed<dS)ReprFalseModelvisiblesecretF)reprhiddenN)rirjrkr!rlrtrurzr r r r rvTs rvc@seZdZddZdS) NonPydanticcCsdS)NzNonPydantic( attr='value' )r )selfr r r __repr__[szNonPydantic.__repr__N)rirjrkr}r r r r r{Ys r{c@seZdZdZdZdS)MyEnumrr'N)rirjrkZONETWOr r r r r~_sr~c@s eZdZdS) EmptyModelN)rirjrkr r r r rdsrcCs.tddd}td}t||ksJdS)Nz Test Namer@rfr_zZ SimpleModel( is_active=True, name='Test Name', value=123 ) )retextwrapdedentstripryobjexpectedr r r +test_repr_simple_model_defaults_and_no_nonehs rcCst}d}t||ks JdS)Nz EmptyModel())rryrr r r test_repr_empty_modeltsrcCs4tdtddd}td}t||ksJdS)Nrr'ro)rorpzJ Chain( child=Chain( id=2 ), id=1 ) )rnrrrryrr r r test_repr_nested_modelzs rcCsBtdd}tdd}||_||_td}t||ksJdS)Nrrr'zv Chain( child=Chain( child=<... Circular reference ...>, id=2 ), id=1 ) )rnrprrrry)Zobj1obj2rr r r test_repr_circular_models   rcCs6ddg}||td}t||ksJdS)Nrr'zG [ 1, 2, <... Circular reference ...>, ] )appendrrrr _pretty_repr)Zmy_listrr r r test_repr_circular_lists  rcCs4ddi}||d<td}t||ksJdS)Nr)rr|zJ { 'a': 1, 'self': <... Circular reference ...> } rrrrr)Zmy_dictrr r r test_repr_circular_dicts rcCsttd}ddtdD}ttd}ttd}dtj|ddvs&Jdtj|ddvs1Jdtj|ddvssz'test_repr_max_items..z<... 5 more items ...>r6)Z max_itemsz )rrangesettuplerr)lstdctstZtplr r r test_repr_max_itemss    rcCs<d}ttj|dddksJt|tj|ddksJdS)NsdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZ)max_len]ra)rrrry)Zb_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)r4r5r5pz [<... 4 items at Max depth ...>]rrrrr r r test_repr_max_depth_listsrcCs@hdddddgdd}td}t||ksJdS) N>rr'r)r+r6r8r'r)r*r))r: )rrr rz { 'dict': { 'a': 1, 'b': 2 }, 'list': [ 7, 8, 9, ], 'set': { 1, 2, 3, }, 'tuple': ( 4, 5, 6, ) } rrr r r test_repr_collectionssrcCs0dddd}td}t||ksJdS)Nr )r)rr')Ztuple0Ztuple1Ztuple2z{ { 'tuple0': (), 'tuple1': ( 1, ), 'tuple2': ( 1, 2, ) } rrr r r test_tuple_collectionss  rcCsrtgdks JtidksJttdksJttdks&Jtdtitdks7JdS)Nz[]z{}zset()z()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)s1Zexp1s2Zexp2Zs3Zexp3r r r test_repr_stringssrcCsJtddd}t|}d|vsJd|vsJtd}||ks#JdS)Nshowhide)rwrzzvisible='show'rzz4 ReprFalseModel( visible='show' ) )rvryrrrrresultrr r r test_repr_repr_false+s   rcCs@tdddd}t|}d|vsJtd}||ksJdS)Nz Only Namer)rfr_rhrhzX SimpleModel( is_active=True, name='Only Name', value=0 ) )reryrrrrr r r test_repr_none_fields8s rcCs:t}tj}||d}td}t||ksJdS)N)npenzh { 'en': , 'np': NonPydantic( attr='value' ) } )r{r~rrrrrr)rrrrr r r test_repr_other_typesFs rcCs4tddd}td}tj|dd|ksJdS)Nz Indent Testrrz` SimpleModel( is_active=True, name='Indent Test', value=1 ) r+)Z indent_delta)rerrrrrrr r r test_repr_indent_deltaUs rc 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)partsZroler)contentZ finish_reasonindexz%models/gemini-2.5-flash-preview-05-20i>r:)ZmodalityZ token_counti\i)Zcandidates_token_countZprompt_token_countZprompt_tokens_detailsZthoughts_token_countZtotal_token_count)Z"automatic_function_calling_history candidatesZ model_versionZusage_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 ) ) )rZGenerateContentResponse CandidateZContentZPartZ FinishReasonSTOPZ$GenerateContentResponseUsageMetadataZModalityTokenCountZ MediaModalityZTEXTrrrryrr r r test_repr_complex_objectas> !!"rcCsdddddigiiddddddigiidddddd 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.requestsrrr1@)requestoutputDimensionality23z requests[].*zrequests[].request.*r)rrN)rZmove_value_by_path)datapathsrr r r test_move_value_by_paths~   )   )rcsGdddtjGfdddtj}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..ModelAr_Nrirjrkr%rlr r r r ModelA rceZdZUed<dS)zPtest_check_field_type_mismatches_no_warning_for_correct_types..TestModelmodel_aNrirjrkrlr rr r TestModelrrrr_r@google_genai._commonloggerNr) r BaseModelat_levelloggingrOmodel_validaterr_rrPrTrrrr rr =test_check_field_type_mismatches_no_warning_for_correct_typess  rcsGdddtjGdddtj}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@r)zPtest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelAr_Nrr r r r rrrc@r)zPtest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..ModelBr_Nrirjrkr!rlr r r r ModelBrrcr)zStest_check_field_type_mismatches_warns_on_pydantic_type_mismatch..TestModel model_fieldNrr rr r rrrtestr_rrrNrz&Type mismatch in TestModel.model_fieldrexpected ModelA, got ModelB rrrrrOZ_check_field_type_mismatchesrrPrRrTrrZmodel_b_instancerr rr @test_check_field_type_mismatches_warns_on_pydantic_type_mismatchs  rcsGdddtjGfdddtj}ddi}|jtjdd ||}Wdn1s0wY|jdus.ModelAr_Nrr r r r r2rrceZdZUdZeed<dS)zNtest_check_field_type_mismatches_no_warning_for_none_values..TestModelNrrirjrkrrrlr rr r r5 rrNrrr rrrrrOrrrrPrr rr ;test_check_field_type_mismatches_no_warning_for_none_values/s rcsGdddtjGfdddtj}i}|jtjdd ||}Wdn1s.wY|jdus:Jt|jdksCJdS) z2Test that no warning is logged for missing fields.c@r)zNtest_check_field_type_mismatches_no_warning_for_missing_fields..ModelAr_Nrr r r r rDrrcr)zQtest_check_field_type_mismatches_no_warning_for_missing_fields..TestModelNrrr rr r rGrrrrNrrrr rr >test_check_field_type_mismatches_no_warning_for_missing_fieldsAs rcCsGdddtj}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)rirjrkr%rlr!r r r r rVs  rrEr)rrrrNr@r) rrrrrOrrrrrPrr r r ?test_check_field_type_mismatches_no_warning_for_primitive_typesSs  rcsGdddtjGdddtj}Gfdddtj}|dd}d |i}|jtjd d  ||Wd n1s>wYt|jd ksLJd|jdjvsVJd S)z@Test that Optional types are properly unwrapped before checking.c@r)zLtest_check_field_type_mismatches_handles_optional_unwrapping..ModelAr_Nrr r r r rjrrc@r)zLtest_check_field_type_mismatches_handles_optional_unwrapping..ModelBr_Nrr r r r rmrrcr)zOtest_check_field_type_mismatches_handles_optional_unwrapping..TestModelNrrr rr r rprrrrrrrNrrrrrr rr .ModelAr_Nrr r r r rrrcr)z\test_check_field_type_mismatches_no_warning_for_correct_pydantic_instance..TestModelrNrr rr r rrr*rrrrNr) rrrrrOrrr_rrP)rTrZmodel_a_instancerrr rr Itest_check_field_type_mismatches_no_warning_for_correct_pydantic_instance~s  rcsGdddtjGdddtj}Gfdddtj}|dd}|d d d }|jtjd d  ||Wd n1s@wYt|jdksNJd|jdjvsXJd|jdjvsbJd S)z3Test checking multiple fields with mixed scenarios.c@r)zEtest_check_field_type_mismatches_with_multiple_fields..ModelAr_Nrr r r r rrrc@r)zEtest_check_field_type_mismatches_with_multiple_fields..ModelBr_Nrr r r r rrrcs.eZdZUed<dZeed<eed<dS)zHtest_check_field_type_mismatches_with_multiple_fields..TestModelfield_aNfield_bfield_c)rirjrkrlrrr!r rr r rs  rZwrongrNr)rrrrrrrrrrrr rr 5test_check_field_type_mismatches_with_multiple_fieldss  rcCsnGdddtj}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)rirjrkrr!rlr r r r rs rrr)r*rrNr)rrrrrOrrrP)rTrrr r r 6test_check_field_type_mismatches_generic_type_no_errors   r)@__doc__enumrrrrr"rrrrtrSrMrrrrrr&markZ parametrizer!r rLrUrdrrernZ model_rebuildrqrvr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r r r s                      Q          .         Y     !  GZ