o Zht @sdZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZejrIddlZddlZddlmZddlmZe d Ze jd krsejZejejfZ dd lm!Z!n ej"ZejfZ dd l#m!Z!d d dej$de%de%de&fddZ'dej$de%fddZ(dej$de)de%fddZ*dej$de+e%e,ej-fdee,ej-fddZ.dej$de%fdd Z/dej0fd!d"Z1d#eej2dej3fd$d%Z4d&eee,ej2ej2e,ej3fde,ej3fd'd(Z5dej$d)eeee%efdee,ej6fd*d+Z7eej8ej9ej2fZ:dej$d,ee:dej8fd-d.Z;dej$d/ee,ej8e,ej9e:fdee,e%e,ej8ffd0d1Zej8fde,ej8fd2d3Z?d4e+e%effd5d6Z@ dfd7d8d4e+e%efdej$d9ee+e%efd:e)fd;d<ZAd=edej$dejBfd>d?ZCd@eejDefde!e+e%effdAdBZEdej$d@eejDefdeejBfdCdDZFdEej$d@eejGefdeejHfdFdGZIdej$deejJfdHdIZKdej$d@e,ede,ejJfdJdKZLdej$dLe%fdMdNZMdej$dOe%fdPdQZNdej$dRe%fdSdTZOdej$dLe%fdUdVZPdWZQdXZRdYZSdZZTdej$d[e+fd\d]ZUdej$dLeee%ejVejWejXffd^d_ZYdej$d`e%deejZe%ffdadbZ[dej$dce\de%fdddeZ]dS)gz"Transformers for Google GenAI SDK.N)IterableMapping)EnumEnumMeta)Any GenericAliasOptionalUnion) _api_client)typeszgoogle_genai._transformers) ) TypeGuard)collection_hierarchy_depthclient resource_namecollection_identifierrcCs||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)rrrrZ$should_prepend_collection_identifierrQ/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/_transformers.py_resource_name2s$E   rmodelcCs|std|jr3|ds|ds|dr|Sd|vr.|dd\}}d|d|Sd|S|dr:|S|d rA|Sd|S) Nzmodel is required.rmodels/ publishers/rr z/models/zpublishers/google/models/z tunedModels/) ValueErrorrrsplit)rr Z publisherZmodel_idrrrt_models&    r% api_client base_modelsreturncCs|jr |rdSdS|r dSdS)Nzpublishers/google/modelsmodels tunedModels)r)r&r'rrr t_models_urlsr+responsecCs|sgS|ddur|dS|ddur|dS|ddur(|dS|ddur8|ddur8gStdtd|gS)Nr)r*ZpublisherModelsZ httpHeadersZ jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)r&r,rrrt_extract_modelss     r1cCsjt||}|s dS|dr|jrd|jd|jd|S|dr3|jr3d|jd|jd|S|S)Nr"rrrr!z/publishers/google/)r%rrrr)r&r rrrt_caches_models r2cCsz ddl}|j}Wn tyd}Ynwt}|dur#t||js(|jdkr2|j|ddd}n |j|ddd}| d| }t j ||dS) NrZRGBAZPNG)formatz image/pngZJPEGz image/jpeg) mime_typedata) ZPIL.PngImagePluginPngImagePlugin ImportErrorioBytesIO isinstanceZ PngImageFilemodesaveseekreadr Blob)ZimgPILr6Zbytesior4r5rrr pil_to_blobs$     rApartcCsz ddl}|jj}Wn tyd}Ynw|durtdt|tr)tj|dS|dur:t||r:tjt|dSt|tj rT|j rF|j sJtdtjj |j |j dSt|t r_tj|St|tjrg|Stdt|)Nrzcontent part is required.)text)Z inline_dataz$file uri and mime_type are required.)Zfile_urir4zUnsupported content part type: ) PIL.ImageImager7r#r:strr PartrAFileurir4Zfrom_uridictmodel_validatetype)rBr@ PIL_Imagerrrt_parts*         rNpartscCs@|dus t|tr|stdt|trdd|DSt|gS)Nzcontent parts are required.cSsg|]}t|qSr)rN.0rBrrr szt_parts..)r:listr#rNrOrrrt_partss   rU predictionscCsP|sdSg}|D]}|dr%|tjtj|dd|ddddq|S)NimageZgcsUriZ imageBytes)gcs_uriZ image_bytes)rW)r-appendr GeneratedImagerE)rrVZimagesZ predictionrrrt_image_predictionss   r[contentcCs|durtdt|tjr|St|tr@ztj|WStjy?tj|}|j r6tj |gdYStj |gdYSwt|tjrW|j rPtj |gdStj |gdStj |dS)Nzcontent is required.rT) r#r:r ContentrJrKpydanticValidationErrorrG function_call ModelContent UserContent)rr\Z possible_partrrr t_content1s,       rccontentscst|trfdd|D}nt|g}jrJg}|D]+}|durGt|tr-tj|}|jdurG|jD]}|j rA| |j q5t dq5q|S|S)Ncsg|]}t|qSr)rc)rQr\rrrrRQsz(t_contents_for_embed..z/Non-text part found, only returning text parts.) r:rSrcrrJr r]rKrOrCrYr.r/)rrdZtransformed_contentsZ text_partsr\rBrrert_contents_for_embedLs&      rfcs|dus t|tr|stdt|tst||gSz ddl}|jjWn ty/dYnwg}g}dttj t fdt tj ffdd }dtj dt fddd ttj dt ffd d d ttjd ttj ffdd d ttjd ttj dtj ffdd }|D]I}t|tjst|tr||t|tr|tj|dq||q||r||||qt|tr|tj|qtdt||||S)Nzcontents are required.rrBr(cspt|tst|tjsdurt|st|tjrdSt|tr6z tj|WdStjy5YdSwdS)NTF) r:rFr rHrGrJrKr^r_rB)rMrr_is_part}s     zt_contents.._is_partcSs|j SN)r`rgrrr _is_user_partsz!t_contents.._is_user_partrOcstfdd|DS)Nc3s|]}|VqdSrirrPrjrr sz6t_contents.._are_user_parts..)allrTrkrr_are_user_partssz#t_contents.._are_user_partsresultaccumulated_partscs>|sdS||rtj|dntj|dg|dd<dS)NrT)rYr rbra)rorp)rnrr$_append_accumulated_parts_as_contents z8t_contents.._append_accumulated_parts_as_content current_partcsBt|}||kr||dS|||g|dd<dSri)rNrY)rorprr)rqrnrjrr_handle_current_parts  z(t_contents.._handle_current_partrTzUnsupported content type: )r:rSr#rcrDrEr7r r PartUnionDictrrrGboolr]rYrbrJrKrL)rrdr@rorprhrsr\r)rMrqrnrjr t_contentshsV    &       rvschemacCs|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" } } rLNnullTZnullableanyOfr r)r-removelenitems)rwitemkeyvalrrrhandle_null_fieldss,   rTorder_propertiesdefsrcs|jsddddurtdddkr!dddddur0dd<|durIdi}|D] \}}t|||q>td d}|durr||d d }t| D]}|||<qfd =dd} | dur| D]!}|d d} | durt|||q~|| d d }| |q~d d | Dd<dSdd} t | t r| j } | } dd} | dur| dkr| gd<d=ntd| dkrAdd} | durdS| D])\}}|d d} | durt|||q|| d d }t||||| |<qt| dkr;|r=tfdddDr?t| }|d<dSdSdSdS| dkr{dd}|durSdS|d d} | durft|||dS|| d d }t||||d<dSdS)aiUpdates the schema and each sub-schema inplace to be API-compatible. - Removes the `title` field from the schema if the client is not vertexai. - 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': { 'type': 'string' }, 'gdp': { 'type': 'integer'} }, } 'required':['continent', 'gdp'], 'type': 'object' }, 'type': 'array' } titleNdefaultzIDefault value is not supported in the response schema for the Gemini API.ZPlaceholderLiteralEnumany_ofryz$defs$refzdefs/cSsg|]}d|vr|qS)rr)rQr}rrrrRusz"process_schema..rLconstSTRINGenumzLiteral values must be strings.ZOBJECT propertiesr c3s|]}|vVqdSrir)rQZ ordering_keyrwrrrls  z!process_schema..)property_orderingZpropertyOrderingrZARRAYr|)rpopr-r#r|process_schemarr$rSkeysrYr:rvalueupperr{rm)rwrrr_Z sub_schemaZ schema_refrefZ schema_keyrZref_keyZ schema_typerrnameZproperty_namesrrrrsA                          rrcsrD]}t|jtstd|jdt|jqGfdddtj}|}t |||dd}t j |S)Nz Enum member z value must be a string, got ceZdZUed<dS)z"_process_enum..Placeholder placeholderN__name__ __module__ __qualname____annotations__rrrr Placeholder rrr) r:rrF TypeErrorrrLr^ BaseModelmodel_json_schemarr SchemarK)rrmemberrZ enum_schemarrr _process_enums     rorigincCst|to tdd|DS)zEVerifies the schema is of type dict[str, Any] for mypy type checking.css|]}t|tVqdSri)r:rF)rQr~rrrrls  z(_is_type_dict_str_any..)r:rJrmrrrr_is_type_dict_str_anysrcsNsdSttrtrt|ddtjSttr$t|SttjrKtttkr8t dj dd}t||ddtj|Stt sjtt rjt tjrj}t||tj|Stt stt sttsttvrGfdddtj}|}t|||dd }tj|St d ) NFrzUnsupported schema type.T)Z exclude_unsetcr)zt_schema..PlaceholderrNrrrrrrrrrrzUnsupported schema type: )r:rJrrr rrKrrr#Z model_dumprrL issubclassr^rrVersionedUnionTypetyping get_origin _UNION_TYPES)rrrwrrrrt_schemasF            rrcCs|sdSt|tjr |St|trtjtjtj|dddSt|trYd|vrY|ddurYd|dvrY|dddurYd|ddvrYtjtjtj|ddddddStdt |)N) voice_name)prebuilt_voice_config) voice_configrrrzUnsupported speechConfig type: ) r:r SpeechConfigrFZ VoiceConfigZPrebuiltVoiceConfigrJr-r#rL)rrrrrt_speech_configs6       rcCs:|sdSt|st|rtjtjj||dgdS|S)N)rcallablefunction_declarations)inspect isfunctionismethodr ToolZFunctionDeclaration from_callable)rrrrrt_toolsrcCst|sgStjgd}g}|D]!}t||}|dur/|jr*|jdur*|j|j7_q||q|jr8|||S)Nr)r rrrrY)rrZ function_toolZtoolsZtoolZtransformed_toolrrrt_tools$s$     rrcCst||ddS)NZcachedContents)r)r)rrrrrt_cached_content_name=srsrccCs@|dr tjd|gdS|drtjd|dStd|)Ngs://jsonlr3rXbq://bigqueryr3Z bigquery_urizUnsupported source: )rr ZBatchJobSourcer#)rrrrrt_batch_job_sourceAs  rdestcCs>|dr tjd|dS|drtjd|dStd|)NrrrrrrzUnsupported destination: )rr ZBatchJobDestinationr#)rrrrrt_batch_job_destinationPs  rcCsD|js|Sd}t||r|ddS|r|Std|d)Nz:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$rrzInvalid batch job name: .)rrematchr$isdigitr#)rrpatternrrrt_batch_job_name_s rg?g4@g @g?structcCs|d}r^d|vr^|}d}t}|ddkrE|tkr&td|d||jd|id }t|||7}t|tt }|ddks|d }rYtd|d |d ||d S|S)Nrz /operations/gdoneTz Operation z timed out. GET)Z http_methodpathZ request_dicterrorz failed with error: z. r,) r-!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)r&rrZ operation total_secondsZ delay_secondsrrrrt_resolve_operationrs.  rcCst|tjr |j}nt|tjr|j}nt|tjr&|jdur$|jj}nd}|dur.tdt|t s=tdt |d| dra| dd}t d|}|durZtd||d }|S| drm| dd}|S) NzFile name is required.z"Could not convert object of type `z` to a file name.zhttps://zfiles/r z [a-z0-9]+z&Could not extract file name from URI: r)r:r rHrVideorIGeneratedVideoZvideor#rFrLrr$rrgroup)r&rsuffixrrrr t_file_names0          rstatuscCsf|dkrtjjS|dkrtjjS|dkrtjjS|dkr tjjStjD] }t|j|kr0|Sq#|S)NZSTATE_UNSPECIFIEDZCREATINGZACTIVEZFAILED)r JobStateZJOB_STATE_UNSPECIFIEDZJOB_STATE_RUNNINGZJOB_STATE_SUCCEEDEDZJOB_STATE_FAILEDrFr)r&rstaterrrt_tuning_job_statuss rr5cCst|ts|St|dS)Nascii)r:bytesbase64 b64encodedecode)r&r5rrrt_bytess rri)^__doc__rcollections.abcrrrrrrr8loggingrsysrr Z builtin_typesrrrrr TYPE_CHECKINGrDr@r^r getLoggerr. version_info UnionTyperrr_UnionGenericAliasZtyping_extensionsZ BaseApiClientrFintrr%rur+rJrSZ ModelDictr1r2r?rArtrGrNrUrZr[r]Z ContentDictZ ContentTypercrfZContentListUnionZContentListUnionDictrvrrrrZSchemaUnionDictrrZSpeechConfigUnionDictrrrrrrrrrrrrrrrHrrrrrrrrrrrs8       ]          e>  # (  .  "   #