o %g u @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%efde,e+e%effddZ-dej$de%fdd Z.dej/fd!d"Z0d#eej1dej2fd$d%Z3d&eee,ej1ej1e,ej2fde,ej2fd'd(Z4dej$d)eeee%efdee,ej5fd*d+Z6eej7ej8ej1fZ9dej$d,ee9dej7fd-d.Z:dej$d/ee,ej7e,ej8e9fdee,e%e,ej7ffd0d1Z;dej$d/eeejd4e+e%effd5d6Z? dfd7d8d4e+e%efdej$d9ee+e%efd:e)fd;d<Z@d=edej$dejAfd>d?ZBd@eejCefde!e+e%effdAdBZDdej$d@eejCefdeejAfdCdDZEdEej$d@eejFefdeejGfdFdGZHdej$deejIfdHdIZJdej$d@e,ede,ejIfdJdKZKdej$dLe%fdMdNZLdej$dOe%fdPdQZMdej$dRe%fdSdTZNdej$dLe%fdUdVZOdWZPdXZQdYZRdZZSdej$d[e+fd\d]ZTdej$dLeee%ejUejVejWffd^d_ZXdej$d`e%deejYe%ffdadbZZdej$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)rrrr$should_prepend_collection_identifierr\/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/google/genai/_transformers.py_resource_name2s$E   r modelcCs|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.responsecCs|sgS|d}|dur|S|d}|dur|S|d}|dur%|S|ddur5|ddur5gStdtd|gS)Nr,r-publisherModels httpHeaders jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)r)r/r, tuned_modelspublisher_modelsrrrt_extract_modelss&    r9cCsjt||}|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 r:cCsz 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)imgPILrBbytesior?r@rrr pil_to_blobs$     rPpartcCsz 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) inline_dataz$file uri and mime_type are required.)file_urir?zUnsupported content part type: ) PIL.ImageImagerCr$rFstrr PartrPFileurir?from_uridictmodel_validatetype)rQrN PIL_Imagerrrt_parts*         r`partscCs@|dus t|tr|stdt|trdd|DSt|gS)Nzcontent parts are required.cSsg|]}t|qSr)r`.0rQrrr szt_parts..)rFlistr$r`rarrrt_partss   rg predictionscCsP|sdSg}|D]}|dr%|tjtj|dd|ddddq|S)NimagegcsUri imageBytes)gcs_uri image_bytes)ri)r3appendr GeneratedImagerV)rrhimages predictionrrrt_image_predictions#s   rrcontentcCs|durtdt|tjr|St|tr@ztj|WStjy?tj|}|j r6tj |gdYStj |gdYSwt|tjrW|j rPtj |gdStj |gdStj |dS)Nzcontent is required.rf) r$rFr Contentr\r]pydanticValidationErrorrX function_call ModelContent UserContent)rrs possible_partrrr t_content:s,       r{contentscst|trfdd|D}nt|g}jrJg}|D]+}|durGt|tr-tj|}|jdurG|jD]}|j rA| |j q5t dq5q|S|S)Ncsg|]}t|qSr)r{)rcrsrrrrdZsz(t_contents_for_embed..z/Non-text part found, only returning text parts.) rFrer{rr\r rtr]rarRrnr4r5)rr|transformed_contents text_partsrsrQrr}rt_contents_for_embedUs&      rcs|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.rrQr+cspt|tst|tjsdurt|st|tjrdSt|tr6z tj|WdStjy5YdSwdS)NTF) rFrWr rYrXr\r]rurvrQ)r_rr_is_parts     zt_contents.._is_partcSs|j SN)rwrrrr _is_user_partsz!t_contents.._is_user_partracstfdd|DS)Nc3s|]}|VqdSrrrbrrr sz6t_contents.._are_user_parts..)allrfrrr_are_user_partssz#t_contents.._are_user_partsresultaccumulated_partscs>|sdS||rtj|dntj|dg|dd<dS)Nrf)rnr ryrx)rr)rrr$_append_accumulated_parts_as_contents z8t_contents.._append_accumulated_parts_as_content current_partcsBt|}||kr||dS|||g|dd<dSr)r`rn)rrr)rrrrr_handle_current_parts  z(t_contents.._handle_current_partrfzUnsupported content type: )rFrer$r{rUrVrCr r PartUnionDictrrrXboolrtrnryr\r]r^)rr|rNrrrrrsr)r_rrrr t_contentsqsV    &       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" } } r^NnullTnullableanyOfr r)r3removelenitems)ritemkeyvalrrrhandle_null_fieldss,   rTorder_propertiesdefsrcs|jsddurtdddkrdddddur*dd<|durCdi}|D] \}}t|||q8td d}|durl||d d }t| D]}|||<q`d =dd} | dur| D]!}|d d} | durt|||qx|| d d }| |qxd d | Dd<dSdd} t | t r| j } | } dd} | dur| dkr| gd<d=ntd| dkr:dd} | durdS| D](\}}|d d} | durt|||q|| d d }t||||| |<qt| dkr4|r6tfdddDr8t| }|d<dSdSdSdS| dkrtdd}|durLdS|d d} | dur_t|||dS|| d d }t||||d<dSdS)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' } defaultNzIDefault value is not supported in the response schema for the Gemini API.titlePlaceholderLiteralEnumany_ofrz$defs$refzdefs/cSsg|]}d|vr|qS)rr)rcrrrrrd~sz"process_schema..r^constSTRINGenumzLiteral values must be strings.OBJECT propertiesr c3s|]}|vVqdSrr)rc ordering_keyrrrrs  z!process_schema..)property_orderingpropertyOrderingrARRAYr)rr3r$poprprocess_schemarr%rekeysrnrFrvalueupperrr)rrrr_ sub_schema schema_refref schema_keyrref_key schema_typerrnameproperty_namesrrrrsC                        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) rFrrW TypeErrorrr^ru BaseModelmodel_json_schemarr Schemar])rrmemberr enum_schemarrr _process_enums     rorigincCst|to tdd|DS)zEVerifies the schema is of type dict[str, Any] for mypy type checking.css|]}t|tVqdSr)rFrW)rcrrrrrs  z(_is_type_dict_str_any..)rFr\rrrrr_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) exclude_unsetcr)zt_schema..PlaceholderrNrrrrrrrrrrzUnsupported schema type: )rFr\rrr rr]rrr$ model_dumprr^ issubclassrurrVersionedUnionTypetyping get_origin _UNION_TYPES)rrrrrrrt_schemasF            rrcCs|sdSt|tjr |St|trtjtjtj|dddSt|trYd|vrY|ddurYd|dvrY|dddurYd|ddvrYtjtjtj|ddddddStdt |)N) voice_name)prebuilt_voice_config) voice_configrrrzUnsupported speechConfig type: ) rFr SpeechConfigrW VoiceConfigPrebuiltVoiceConfigr\r3r$r^)rrrrrt_speech_configs6       rcCs:|sdSt|st|rtjtjj||dgdS|S)N)rcallablefunction_declarations)inspect isfunctionismethodr ToolFunctionDeclaration from_callable)rrrrrt_toolsrcCst|sgStjgd}g}|D]!}t||}|dur/|jr*|jdur*|j|j7_q||q|jr8|||S)Nr)r rrrrn)rr function_tooltoolstooltransformed_toolrrrt_tools-s$     rrcCst||ddS)NcachedContents)r)r )rrrrrt_cached_content_nameFsrsrccCs@|dr tjd|gdS|drtjd|dStd|)Ngs://jsonlr=rlbq://bigqueryr= bigquery_urizUnsupported source: )rr BatchJobSourcer$)rrrrrt_batch_job_sourceJs  rdestcCs>|dr tjd|dS|drtjd|dStd|)NrrrrrrzUnsupported destination: )rr BatchJobDestinationr$)rrrrrt_batch_job_destinationYs  rcCsD|js|Sd}t||r|ddS|r|Std|d)Nz:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$rrzInvalid batch job name: .)rrematchr%isdigitr$)rrpatternrrrt_batch_job_namehs 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/) r3!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)r)rr operation total_seconds delay_secondsr rrrt_resolve_operation{s.  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)rFr rYrVideorZGeneratedVideovideor$rWr^rr%rrgroup)r)rsuffixrrrr t_file_names0          rstatuscCsf|dkrtjjS|dkrtjjS|dkrtjjS|dkr tjjStjD] }t|j|kr0|Sq#|S)NSTATE_UNSPECIFIEDCREATINGACTIVEFAILED)r JobStateJOB_STATE_UNSPECIFIEDJOB_STATE_RUNNINGJOB_STATE_SUCCEEDEDJOB_STATE_FAILEDrWr)r)rstaterrrt_tuning_job_statuss r)r@cCst|ts|St|dS)Nascii)rFbytesbase64 b64encodedecode)r)r@rrrt_bytess r/r)]__doc__r,collections.abcrrrrrrrDloggingrsysrr builtin_typesrrrrr TYPE_CHECKINGrUrNrur getLoggerr4 version_info UnionTyperrr_UnionGenericAliastyping_extensions BaseApiClientrWintr r(rr.r\rer9r:rLrPrrXr`rgrorrrt ContentDict ContentTyper{rContentListUnionContentListUnionDictrrrrrSchemaUnionDictrrSpeechConfigUnionDictrrrrrrrrrr rr rrrYrrrr#r)r+r/rrrrs8       ]           e>  # (  .  "   #