o 0Xxi _ @s&dZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddl mZmZmZmZe jrAddlZddlZddlmZddlmZe jd krbe jjZe je jjfZne jZe jfZd d d ejd edede fddZ!d ejdefddZ"dejde#defddZ$dejde%de&ej'fddZ(dejdefddZ)dej*fdd Z+eej,ej-ed!fZ.d ejd"e.dej,fd#d$Z/d ejd%ee&e.fde&ej,fd&d'Z0d ejd(eeeeefde&ej1fd)d*Z2eej3ej4e.fZ5d ejd+e5fd,d-Z6d ejd.ee&ej3e&ej4e5ffd/d0Z7d ejd.ee&ej3e&ej4e5ffd1d2Z8d3e%eeffd4d5Z9  dcd6d7d3e%eefd eejd8ee%eefd9e#fd:d;Z: dddZdBejd?eej?efdeej@fdCdDZAd ejdejBfdEdFZCd ejd?e&ede&ejBfdGdHZDd ejdIefdJdKZEd ejdLefdMdNZFd ejdOefdPdQZGd ejdIefdRdSZHdTZIdUZJdVZKdWZLdejdXe%fdYdZZMdejdIeeejNffd[d\ZOdejd]edejPfd^d_ZQdejd`eRdefdadbZSdS)ez"Transformers for Google GenAI SDK.N)IterableMapping)EnumEnumMeta)Any GenericAliasOptionalUnion) _api_client)types) )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)rrrr$should_prepend_collection_identifierrY/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/_transformers.py_resource_name-s$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 publishermodel_idrrrt_models&    r' api_client base_modelsreturncCs|jr |rdSdS|r dSdS)Nzpublishers/google/modelsmodels tunedModels)r)r(r)rrr t_models_urlsr-responsecCsX|sgS|ddur|dS|ddur|dS|ddur(|dStd)Nr+r,publisherModelsz!Cannot determine the models type.)getr#)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) NrRGBAPNG)formatz image/pngJPEGz image/jpeg) mime_typedata) PIL.PngImagePluginPngImagePlugin ImportErrorioBytesIO isinstance PngImageFilemodesaveseekreadr Blob)imgPILr:bytesior7r8rrr pil_to_blobs$     rHzPIL.Image.ImagepartcCsz ddl}|jj}Wn tyd}Ynw|stdt|tr'tj|dS|dur8t||r8tjt|dSt|tj rR|j rD|j sHtdtjj |j |j dS|S)Nrzcontent part is required.)text) inline_dataz$file uri and mime_type are required.)file_urir7) PIL.ImageImager;r#r>strr PartrHFileurir7from_uri)rrIrF PIL_Imagerrrt_parts"      rUpartscs8|durtdt|trfdd|DSt|gS)Nzcontent parts are required.cg|]}t|qSr)rU).0rIrrr zt_parts..)r#r>listrU)rrVrrYrt_partss   r] predictionscCsP|sdSg}|D]}|dr%|tjtj|dd|ddddq|S)NimagegcsUri imageBytes)gcs_uri image_bytes)r_)r0appendr GeneratedImagerN)rr^images predictionrrrt_image_predictions s   rhcontentcCsF|stdt|tjr|St|trtj|Stjdt||dS)Nzcontent is required.user)rolerV)r#r>r Contentdictmodel_validater])rrirrr t_content s   rocontentscsdjrt|trfdd|DSjrt|jdjgSt|tr,fdd|DSt|gS)Ncsg|] }t|jdjqS)r)rorVrJrXrirYrrrZ3sz(t_contents_for_embed..rcrWrrorqrYrrrZ7r[)rr>r\rorVrJrrprrYrt_contents_for_embed-s  rtcs4|stdt|trfdd|DSt|gS)Nzcontents are required.crWrrrrqrYrrrZCr[zt_contents..)r#r>r\rorsrrYr t_contents<s   ruschemacCs|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" } } typeNnullTnullableanyOfr r)r0removelenitems)rvitemkeyvalrrrhandle_null_fieldsHs,   rTorder_propertiesdefsrcs |r|jsddddurtdddkr#dddddur2dd<|durKdi}|D] \}}t|||q@td drs|d d d }t| D]}|||<qgd =dd}|dur|jstd |D]!}|d d} | durt|||q|| d d }| |qd d|Dd<dSdd} t | t r| j } | } dd} | dur| dkr| gd<d=ntd| dkrIdd} | durdS| D])\} }|d d} | dur t|||q|| d d }t||||| | <qt| dkrC|rEtfdddDrGt| }|d<dSdSdSdS| dkrdd}|dur[dS|d d} | durnt|||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.PlaceholderLiteralEnumany_ofrzz$defs$refzdefs/zAAnyOf is not supported in the response schema for the Gemini API.cSsg|]}d|vr|qS)rr)rXr~rrrrZsz"process_schema..rwconstSTRINGenumzLiteral values must be strings.OBJECT propertiesr c3s|]}|vVqdSNr)rX ordering_keyrvrr s  z!process_schema..)property_orderingpropertyOrderingrARRAYr})rpopr0r#r}process_schemarr$r\keysrdr>rvalueupperr|all)rvrrr_ sub_schemaref schema_keyrref_key schema_typerrnameproperty_namesrrrrs A                          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 Placeholder1 rrr) r>rrO TypeErrorrrwpydantic BaseModelmodel_json_schemarr Schemarn)rrmemberr enum_schemarrr _process_enum's     rorigincsFsdSttrt|ddtjSttr t|SttjrGtttkr4tdj dd}t||ddtj|Stt sftt rft t jrf}t||tj|Stt s|tt s|tts|ttvrGfdddt j}|}t|||dd }tj|Std ) NFrzUnsupported schema type.T) exclude_unsetcr)zt_schema..PlaceholderrNrrrrrr]rrrrzUnsupported schema type: )r>rmrr rrnrrr# model_dumprrw issubclassrrrVersionedUnionTypetyping get_origin _UNION_TYPES)rrrvrrrrt_schema:sF             rrcCs|sdSt|tjr |St|trtjtjtj|dddSt|trCd|vrCd|dvrCtjtjtj|ddddddStdt |)N) voice_name)prebuilt_voice_config) voice_configrrrzUnsupported speechConfig type: ) r>r SpeechConfigrO VoiceConfigPrebuiltVoiceConfigrmr0r#rw)rrrrrt_speech_confighs0      rcCs:|sdSt|st|rtjtjj||dgdS|S)N)rcallablefunction_declarations)inspect isfunctionismethodr ToolFunctionDeclaration from_callable)rrrrrt_toolsrcCsb|sgStjgd}g}|D]}t||}|jr!|j|j7_q||q|jr/|||S)Nr)r rrrrd)rr function_tooltoolstooltransformed_toolrrrt_toolss    rrcCst||ddS)NcachedContents)r)r)rrrrrt_cached_content_namesrsrccCs@|dr tjd|gdS|drtjd|dStd|)Ngs://jsonlr5rbbq://bigqueryr5 bigquery_urizUnsupported source: )rr BatchJobSourcer#)rrrrrt_batch_job_sources  rdestcCs>|dr tjd|dS|drtjd|dStd|)NrrrrrrzUnsupported destination: )rr BatchJobDestinationr#)rrrrrt_batch_job_destinations  rcCsD|js|Sd}t||r|ddS|r|Std|d)Nz:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$rrzInvalid batch job name: .)rrematchr$isdigitr#)rrpatternrrrt_batch_job_names 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) http_methodpath request_dicterrorz failed with error: z. r.) r0!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)r(rr operation total_seconds delay_secondsrrrrt_resolve_operations.  rcCst|tjr |j}|durtd|dr5|dd}td|}|dur.td|| d}|S|drA|dd}|S)NzFile name is required.zhttps://zfiles/r z [a-z0-9]+z&Could not extract file name from URI: r) r>r rQrr#rr$rrgroup)r(rsuffixrrrr t_file_names     rstatuscCs4|dkrdS|dkr dS|dkrdS|dkrdS|S) NSTATE_UNSPECIFIEDJOB_STATE_UNSPECIFIEDCREATINGJOB_STATE_RUNNINGACTIVEJOB_STATE_SUCCEEDEDFAILEDJOB_STATE_FAILEDr)r(rrrrt_tuning_job_statussr r8cCst|ts|St|dS)Nascii)r>bytesbase64 b64encodedecode)r(r8rrrt_bytes%s r)NNr)T__doc__rcollections.abcrrrrrrr<rsysrrrrrr TYPE_CHECKINGrMrFrr r version_info UnionTyperr_UnionGenericAlias ApiClientrOintrr'boolr-rmr\Modelr1r2rDrHrPPartDictPartTyperUr]rerhrl ContentDict ContentTyperortrurrrrSchemaUnionDictrSpeechConfigUnionDictrrrrrrrrrrrrrrrQrJobStater rrrrrrs    ]           =  &   .