o YiI @sUdZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZmZddlmZddlmZdd lmZejr`ddlZddl Z dd l!m"Z"dd l!mZdd l!mZe #d Z$e j%dkrej&Z'ejej&fZ(ddlm)Z)n ej*Z'ejfZ(ddl+m)Z)ejrddl,m-Z.ddl/m0Z1n*eZ.ej2e3d<eZ1ej2e3d<zddl,m-Z.ddl/m0Z1Wn e4ydZ.dZ1YnwddddZ5dde56DZ7ddde"j8de9de9de:d e9f d!d"Z;de"j8d#e9d e9fd$d%Zd*ej?d e@ej?fd+d,ZAd&e"j8d#e9d ee9fd-d.ZBd/ed ejCfd0d1ZDd2ejEd ejFfd3d4ZGd5eejEeejEfd e@ejFfd6d7ZHd8eejIe@ejIfd e@ejCfd9d:ZJd;ejId ejCfdd?ZLd;ejMd ejCfd@dAZNdBeejOd ejPfdCdDZQdEeee@ejOejOe@ejPfd e@ejPfdFdGZRdHeeee9efd ee@ejSfdIdJZTeejUejVejOfZWdKeeWejVdfd ejUfdLdMZXde"j8dNee@ejUe@ejVeWfd ee@e9e@ejUffdOdPZYdNeeejZej[ejUfd e@ejUfdQdRZ\dSej?d dfdTdUZ]dVed dfdWdXZ^dSedee"j8d dfdYdZZ_ dd[d\dSej?dee"j8d]eej?d^e=d df d_d`Z`daedee"j8d ejafdbdcZbdVeejcefd e)ej?fdddeZddee"j8dVeejcefd eejafdfdgZedVeejfefd eejgfdhdiZhdVejid eejgfdjdkZjde"j8dVed eeej0effdldmZkde"j8dVe@ed e@ej0fdndoZlde"j8dpe9d e9fdqdrZmde"j8dsejnd ejofdtduZpde"j8dsejqd ejrfdvdwZsdxee9ejtfd ejufdydzZvdxewe9efd ewe9effd{d|Zxde"j8dpe9d e9fd}d~Zyde9d e9fddZzdZ{dZ|dZ}dZ~d&e"j8dej?d efddZdpeee9ejejejfd e9fddZde9d eeje9ffddZdKejd ejUfddZdNeeejejfd e@ejUfddZ  [ddeeeejejfde=d ejfddZdeejEeejEfd ejfddZde@ejd e@ewe9effddZd#e9d e=fddZdS)z"Transformers for Google GenAI SDK.N)IterableMapping)EnumEnumMeta)Any GenericAliasListOptionalSequenceUnion)mcp_to_gemini_tool)get_value_by_path)is_duck_type_of) _api_client)_common)typeszgoogle_genai._transformers) ) TypeGuard) ClientSession)ToolMcpClientSessionMcpToolZexactMatchSpecZbleuSpecZ rougeSpec) exact_matchbleu rouge_speccCsi|]\}}||qSr).0kvrrT/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/_transformers.py Ir")collection_hierarchy_depthclient resource_namecollection_identifierr%returncCs||d o|d|dd|k}|jrX|dr!|S|dr/d|jd|S||drDd|jd|jd|S|rVd|jd|jd|d|S|S|ra|d|S|S)a|Prepends resource name with project, location, collection_identifier if needed. The collection_identifier will only be prepended if it's not present and the prepending won't violate the collection hierarchy depth. When the prepending condition doesn't meet, returns the input resource_name. Args: client: The API client. resource_name: The user input resource name to be completed. collection_identifier: The collection identifier to be prepended. See collection identifiers in https://google.aip.dev/122. collection_hierarchy_depth: The collection hierarchy depth. Only set this field when the resource has nested collections. For example, `users/vhugo1802/events/birthday-dinner-226`, the collection_identifier is `users` and collection_hierarchy_depth is 4. See nested collections in https://google.aip.dev/122. Example: resource_name = 'cachedContents/123' client.vertexai = True client.project = 'bar' client.location = 'us-west1' _resource_name(client, 'cachedContents/123', collection_identifier='cachedContents') returns: 'projects/bar/locations/us-west1/cachedContents/123' Example: resource_name = 'projects/foo/locations/us-central1/cachedContents/123' # resource_name = 'locations/us-central1/cachedContents/123' client.vertexai = True client.project = 'bar' client.location = 'us-west1' _resource_name(client, resource_name, collection_identifier='cachedContents') returns: 'projects/foo/locations/us-central1/cachedContents/123' Example: resource_name = '123' # resource_name = 'cachedContents/123' client.vertexai = False _resource_name(client, resource_name, collection_identifier='cachedContents') returns 'cachedContents/123' Example: resource_name = 'some/wrong/cachedContents/resource/name/123' resource_prefix = 'cachedContents' client.vertexai = False # client.vertexai = True _resource_name(client, resource_name, collection_identifier='cachedContents') returns: 'some/wrong/cachedContents/resource/name/123' Returns: The completed resource name. /r projects/z locations/ /locations/) startswithcountvertexaiprojectlocation)r&r'r(r%Z$should_prepend_collection_identifierrrr!_resource_nameLs$E   r2modelcCs|stdd|vsd|vsd|vrtd|jrC|ds(|ds(|dr*|Sd |vr>|d d \}}d|d |Sd |S|drJ|S|d rQ|Sd|S)Nzmodel is required.z..?&zinvalid model parameter.r+models/ publishers/r*r z/models/zpublishers/google/models/z tunedModels/) ValueErrorr/r-split)r&r3Z publisherZmodel_idrrr!t_models*    r: api_client base_modelscCs|jr |rdSdS|r dSdS)Nzpublishers/google/modelsmodels tunedModels)r/)r;r<rrr! t_models_urlsr?responsecCs|sgS|d}|dur|S|d}|dur|S|d}|dur%|S|ddur5|ddur5gStdtd|gS)Nr=r>ZpublisherModelsZ httpHeadersZ jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)r@r=Z tuned_modelsZpublisher_modelsrrr!t_extract_modelss&    rEcCsjt||}|s dS|dr|jrd|jd|jd|S|dr3|jr3d|jd|jd|S|S)Nr7r+r,r*r6z/publishers/google/)r:r-r/r0r1)r;r3rrr!t_caches_models rFimagecCszd}t}|jdkrt|ddr|jdvrd}|jddt}|j||fi||}d| }t j ||d S) NZPNGZJPEGfilename)1LRGBZRGBXZCMYKZkeep)Zqualityimage/)data mime_type) dictformatgetattrmodeupdateioBytesIOsavegetvaluelowerrBlob)rGZ image_formatZ save_paramsZimage_io image_bytesrOrrr! pil_to_blobs    r\function_responsecCsD|stdt|trtj|St|tjr|Stdt|)Nzfunction_response is required.zOCould not parse input as FunctionResponse. Unsupported function_response type: ) r8 isinstancerPrFunctionResponsemodel_validater TypeErrortype)r]rrr!t_function_responses   rcfunction_responsescCs.|stdt|trdd|DSt|gS)Nz function_responses are required.cSg|]}t|qSr)rc)rr@rrr! .z(t_function_responses..)r8r^r rcrdrrr!t_function_responses%s   riblobscC"t|tr dd|DSt|gS)NcSrer)t_blob)rblobrrr!rf7rgzt_blobs..)r^listrl)rjrrr!t_blobs3  rormcCs|stdt|tjr|St|trtj|Sd|jj vrDz ddl }|j j }Wn t y6d}Ynw|durDt||rDt |Stdt|)Nzblob is required.rGrz6Could not parse input as Blob. Unsupported blob type: )r8rrrZr^rPr` __class____name__rY PIL.ImageImage ImportErrorr\rarb)rmPIL PIL_Imagerrr!rl<s$      rlcC.t|}|jr|jdr|Std|j)NrMUnsupported mime type: rlrOr-r8rmrrr! t_image_blobVr|cCrx)Nzaudio/ryrzr{rrr! t_audio_blob]r}r~partcCs|durtdt|trtj|dSt|tjr-|jr|js#tdtjj |j|jdSt|t rNztj |WSt j yMtjtj |dYSwt|tjrV|Sd|jjvrz ddl}|jj}Wn tysd}Ynw|durt||rtjt|dStd t|) Nzcontent part is required.)textz$file uri and mime_type are required.)Zfile_urirO) file_datarGr)Z inline_datazUnsupported content part type: )r8r^strrPartrFileurirOZfrom_urirPr`pydanticValidationErrorFileDatarqrrrYrsrtrur\rbrrvrwrrr!t_partds4        rpartscCs@|dus t|tr|stdt|trdd|DSt|gS)Nzcontent parts are required.cSrer)rrrrrr!rfrgzt_parts..)r^rnr8rrrrr!t_partss   r predictionscCsP|sdSg}|D]}|dr%|tjtj|dd|ddddq|S)NrGZgcsUriZ imageBytes)gcs_urir[)rG)rAappendrGeneratedImagert)rZimagesZ predictionrrr!t_image_predictionss   rcontentcCs|durtdt|tjr|St|tr>ztj|WStjy=t |}|j r4tj |gdYStj |gdYSwt|tj rMtj t |gdSt|tjrd|j r]tj |gdStj |gdStj |dS)Nzcontent is required.r)r8rrContentr^rPr`rrr function_call ModelContent UserContentrr)rZ possible_partrrr! t_contents0       rcontentscCst|tr dd|D}nt|g}|jrGg}|D]+}|durDt|tr*tj|}|jdurD|jD]}|j r>| |j q2t dq2q|S|S)NcSrer)rrrrrr!rfrgz(t_contents_for_embed..z/Non-text part found, only returning text parts.) r^rnrr/rPrrr`rrrrBrC)r&rZtransformed_contentsZ text_partsrrrrr!t_contents_for_embeds"       rcs|dus t|tr|stdt|tst|gSg}g}dttjtfdttjfdd}dtj dt fdddttj dt ffd d d ttj d ttj ddffd d d ttj d ttj dtjddffdd }|D]I}t |tj st|tr||t|tr| tj|dqs| |qs||r||||qst|tr| tj |qstdt||||S)Nzcontents are required.rr)c Sst|tst|tjst|tjrdSt|trE|sdSz tj|WdStj yDz tj |WYdStj yCYYdSwwd|j j vrnz ddl}|jj}Wn tybd}Ynw|durnt||rndSdS)NTFrGr)r^rrrrrrPr`rrrrqrrrYrsrtrurrrr!_is_parts>       zt_contents.._is_partcSs|j SN)r)rrrr! _is_user_partsz!t_contents.._is_user_partrcstfdd|DS)Nc3s|]}|VqdSrrrrrr! sz6t_contents.._are_user_parts..)allrrrr!_are_user_partssz#t_contents.._are_user_partsresultaccumulated_partscs>|sdS||rtj|dntj|dg|dd<dS)Nr)rrrr)rr)rrr!$_append_accumulated_parts_as_contents z8t_contents.._append_accumulated_parts_as_content current_partcsBt|}||kr||dS|||g|dd<dSr)rr)rrrrrrrr!_handle_current_part!s  z(t_contents.._handle_current_partrzUnsupported content type: )r^rnr8rr r PartUnionDictrrrboolrrrrrPr`rb)rrrrrrrrr! t_contentssT    %      rschemacCs|dddkrd|d<|d=dSd|vrO|dD]7}d|vrN|ddkrNd|d<|dddit|ddkrN|ddD]\}}|||<qB|d=qdSdS) aProcess null fields in the schema so it is compatible with OpenAPI. The OpenAPI spec does not support 'type: 'null' in the schema. This function handles this case by adding 'nullable: True' to the null field and removing the {'type': 'null'} entry. https://swagger.io/docs/specification/v3_0/data-models/data-types/#null Example of schema properties before and after handling null fields: Before: { "name": { "title": "Name", "type": "string" }, "total_area_sq_mi": { "anyOf": [ { "type": "integer" }, { "type": "null" } ], "default": None, "title": "Total Area Sq Mi" } } After: { "name": { "title": "Name", "type": "string" }, "total_area_sq_mi": { "type": "integer", "nullable": true, "default": None, "title": "Total Area Sq Mi" } } rbNnullTZnullableanyOfr r)rAremovelenitems)ritemkeyvalrrr!handle_null_fieldsHs,   rorigincCstd|)z2Raises an error if the schema type is unsupported.Unsupported schema type: )r8rrrr!"_raise_for_unsupported_schema_typesrcCs2|r|js|ds|drtddSdSdS)NadditionalPropertiesadditional_propertiesz8additionalProperties is not supported in the Gemini API.)r/rAr8)rr&rrr!'_raise_for_unsupported_mldev_propertiess rTorder_propertiesdefsrcsX|ddkr |d=t|dD]\}}||d}dur#|||<qdur?|diD] \}}t|dq2t||dd} durY|| dd d tj d tj ffd d |d} dur~fdd| D|d<dS|d} t | t r| j } t | t r| } |d} | dur| dkr| g|d<|d=ntd| dkr|d} durt| D] \}}|| |<qt| dkrrd|vrt| |d<|d}durt |tr||d<dSdSdS| dkr(|d}dur||d<|d}dur*fdd|D|d<dSdSdS) a{Updates the schema and each sub-schema inplace to be API-compatible. - Inlines the $defs. Example of a schema before and after (with mldev): Before: `schema` { 'items': { '$ref': '#/$defs/CountryInfo' }, 'title': 'Placeholder', 'type': 'array' } `defs` { 'CountryInfo': { 'properties': { 'continent': { 'title': 'Continent', 'type': 'string' }, 'gdp': { 'title': 'Gdp', 'type': 'integer'} }, } 'required':['continent', 'gdp'], 'title': 'CountryInfo', 'type': 'object' } } After: `schema` { 'items': { 'properties': { 'continent': { 'title': 'Continent', 'type': 'string' }, 'gdp': { 'title': 'Gdp', 'type': 'integer' }, } 'required':['continent', 'gdp'], 'title': 'CountryInfo', 'type': 'object' }, 'type': 'array' } titleZPlaceholderLiteralEnum))rr)any_ofr)Z prefix_items prefixItems)property_orderingpropertyOrderingNz$defsr$refdefs/ sub_schemar)cs:|dd}dur|dd}t|d|S)z@Returns the processed `sub_schema`, resolving its '$ref' if any.rNrrr)popr9process_schema)rref)r&rrrr!_recursesz process_schema.._recursercg|]}|qSrr)rrrrr!rfrgz"process_schema..rbconstSTRINGenumzLiteral values must be strings.ZOBJECT propertiesr rrrZARRAYrrcrrr)rprefixrrr!rf*rg)rArrrrrrTr9r StringDictr^rvaluerupperr8rnrkeysrP)rr&rr from_nameZto_namer_rrrZ schema_typerrname additionalrprefixesr)rr&rrr!rshC           rrcsd}|D]}t|jtrd}qt|jts#td|jdt|jq||r9dd|D}t|j|td}|Gfdd d t j }| }t |||d d }t j|S) NFTz Enum member z( value must be a string or integer, got cSsg|]}t|jqSr)rr)rmemberrrr!rf=r#z!_process_enum..)rbceZdZUed<dS)z"_process_enum..Placeholder placeholderNrr __module__ __qualname____annotations__rZenum_to_processrr! PlaceholderA rrr)r^rintrrarrbrrrr BaseModelmodel_json_schemarrSchemar`)rr&Zis_integer_enumrZ str_membersZstr_enumrZ enum_schemarrr! _process_enum-s*      rcCst|to tdd|DS)zEVerifies the schema is of type dict[str, Any] for mypy type checking.css|]}t|tVqdSr)r^r)rrrrr!rNs  z(_is_type_dict_str_any..)r^rPrrrrr!_is_type_dict_str_anyJsrcsFsdSttrtrt|tjSttr"t|St tjrGtttkr6t j dd}t||tj|Stt sftt rfttjrf}t||tj|Stt s|tt s|tts|ttvrGfdddtj}|}t|||dd}tj|Std)NT)Z exclude_unsetcr)zt_schema..PlaceholderrNrrrrr!rxrrrrr)r^rPrrrrr`rrrr model_dumprrb issubclassrrrVersionedUnionTypetyping get_origin _UNION_TYPESr8)r&rrrrrr!t_schemaSsF              rcCsf|sdSt|tjr |St|trtjtjtj|dddSt|tr*tj|St dt |)N)Z voice_name)Zprebuilt_voice_config)Z voice_configzUnsupported speechConfig type: ) rr SpeechConfigr^rZ VoiceConfigZPrebuiltVoiceConfigrPr`r8rbrrrr!t_speech_configs     rcCs<t|tjr|}t|trtj|}|jdurtd|S)Nz.)r^r r<rrrr!t_contents_strictrpr>turns turn_completec CsZ|dur tj|dSz tjt|d|dWSty,}z tdt|d|d}~ww)N)r@r=)r?r@r;z") to `types.LiveClientContent`)rLiveClientContentr> Exceptionr8rb)r?r@errr!t_client_contents rDinputc CsV|s td|z tjt|ddWSty*}z tdt|d|d}~ww)Nz#A tool response is required, got: rhr;z$") to `types.LiveClientToolResponse`)r8rLiveClientToolResponserirBrb)rErCrrr!t_tool_responsesrGmetricsc CsFg}|D]}t|trztj|}Wn tjyYnwt|tjr4|}ddg|d<||qi}ddg|d<t |dg }|dkrMi|d<nN|dkrVi|d<nE| d rh| d d }d |i|d <n3t |dr|jrd|ji}t |dg}|r||d<t |dg} | rd| i|d<||d<ntd|||q|S)a Prepares the metric payload for the evaluation request. Args: request_dict: The dictionary containing the request details. resolved_metrics: A list of resolved metric objects. Returns: The updated request dictionary with the prepared metric payload. ZAVERAGEZSTANDARD_DEVIATIONZaggregation_metricsrrZexact_match_specrZ bleu_specZrougerrI rouge_typerprompt_templateZmetric_prompt_templateZjudge_model_system_instructionsystem_instructionreturn_raw_outputZcustom_output_format_configZpointwise_metric_specz0Unsupported metric type or invalid metric name: )r^rPrZ UnifiedMetricr`rrrrgetvrYr-replacehasattrrJr8) rHZmetrics_payloadZmetricZunified_metric_payloadZmetric_payload_itemZ metric_namerIZpointwise_specrKrLrrr! t_metricss\           rPcCsd|vr|dkp d|vS)Ngeminizgemini-embedding-001Zmaasr)r3rrr!t_is_vertex_embed_content_modelIsrRr)NT)__doc__base64collections.abcrrrrrrrUloggingrsysr,rZ builtin_typesrrrrr r r Z _mcp_utilsr rrrMr TYPE_CHECKINGrsrvrrIr getLoggerrB version_info UnionTyperrr_UnionGenericAliasZtyping_extensionsZmcprrZ mcp.typesrrTyperruZmetric_name_sdk_api_maprZmetric_name_api_sdk_mapZ BaseApiClientrrr2r:rr?rrnrErFrZr\ZFunctionResponseOrDictr_rcriZBlobImageUnionDictrorlr|Z BlobOrDictr~rrrrrrrZ ContentDictZ ContentTyperrZContentListUnionZContentListUnionDictrrrrrrrZSchemaUnionDictrrZSpeechConfigUnionDictrrZSpeechConfigOrDictrrrrZBatchJobSourceUnionDictrrZEmbeddingsBatchJobSourceOrDictrrZBatchJobDestinationOrDictrrrPrrr"r(r0r)r/r2rr3r4r7r9r:Z ContentOrDictr<r>rArDrFrGZMetricSubclassrPrRrrrr!s$               ]             l;        0      2   "  "      F