o j6h+*@sddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddl m Z ddlmZd Zd ZGd d d ZdS) ) annotationsN)IterableOptional)protos) caching_types) content_types)get_default_cache_client)field_mask_pb2usermodelc @seZdZdZddZedCddZedCdd ZedCd d ZedDd dZ edEddZ edEddZ edEddZ ddZ e ZedFddZddZedddddddddGd0d1ZedddddddddHd2d3ZedId5d6ZedJdKd;d<ZdLd>d?Zddd@dMdAdBZdS)N CachedContentzCached content resource.cCs6t}d|vr d|}tj|d}||}||_dS)zFetches a `CachedContent` resource. Identical to `CachedContent.get`. Args: name: The resource name referring to the cached content. cachedContents/nameN)rrGetCachedContentRequestget_cached_content_proto)selfrclientrequestresponserj/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/caching.py__init__#s    zCachedContent.__init__returnstrcC|jjSN)rrrrrrr4zCachedContent.namecCrr)rr rrrrr 8rzCachedContent.modelcCrr)r display_namerrrrr <rzCachedContent.display_name"protos.CachedContent.UsageMetadatacCrr)rusage_metadatarrrrr"@rzCachedContent.usage_metadatadatetime.datetimecCrr)r create_timerrrrr$DrzCachedContent.create_timecCrr)r update_timerrrrr%HrzCachedContent.update_timecCrr)r expire_timerrrrr&LrzCachedContent.expire_timecCsRtd|jd|jd|jddd|jjddd |jd |jd |j d S) Nz1 CachedContent( name='z', model='z!', display_name='z"', usage_metadata={z* 'total_token_count': z, }z, create_time=z, update_time=z, expire_time=z )) textwrapdedentrr r r"total_token_countr$r%r&rrrr__str__Ps*  zCachedContent.__str__obj+CachedContent | protos.CachedContent | dictcCs"||}t|_|||S)zKCreates an instance of CachedContent form an object, without calling `get`.)__new__rr r_update)clsr-rrrr _from_objbs   zCachedContent._from_objcCsPt|tr|j}t|tst|j|dd}|D] \}}t|j||qdS)zFUpdates this instance inplace, does not call the API's `update` methodF)including_default_value_fieldsN) isinstancer rdicttypeto_dictitemssetattr)rupdateskeyvaluerrrr0js  zCachedContent._updateN)r system_instructioncontentstools tool_configttlr&r r str | Noner=#Optional[content_types.ContentType]r>$Optional[content_types.ContentsType]r?+Optional[content_types.FunctionLibraryType]r@&Optional[content_types.ToolConfigType]rA Optional[caching_types.TTLTypes]r&'Optional[caching_types.ExpireTimeTypes]!protos.CreateCachedContentRequestc Cs|r|rtdd|vrd|}|rt|dkrtd|r#t|}t|}|r.|}|r5t|}|rFt|}|djsFt |d_t |}t |}t j||||||||d} t j| dS) z&Prepares a CreateCachedContentRequest.LExclusive arguments: Please provide either `ttl` or `expire_time`, not both./zmodels/z;`display_name` must be no more than 128 unicode characters.r r r=r>r?r@rAr&)cached_content) ValueErrorlenr to_contentto_function_libraryto_prototo_tool_config to_contentsrole _USER_ROLErto_optional_ttlto_optional_expire_timerr CreateCachedContentRequest) r r r=r>r?r@rAr& tools_librOrrr_prepare_create_requestus@          z%CachedContent._prepare_create_requestc Cs8t} |j||||||||d} | | } t| } | S)aCreates `CachedContent` resource. Args: model: The name of the `model` to use for cached content creation. Any `CachedContent` resource can be only used with the `model` it was created for. display_name: The user-generated meaningful display name of the cached content. `display_name` must be no more than 128 unicode characters. system_instruction: Developer set system instruction. contents: Contents to cache. tools: A list of `Tools` the model may use to generate response. tool_config: Config to apply to all tools. ttl: TTL for cached resource (in seconds). Defaults to 1 hour. `ttl` and `expire_time` are exclusive arguments. expire_time: Expiration time for cached resource. `ttl` and `expire_time` are exclusive arguments. Returns: `CachedContent` resource with specified name. rN)rr]create_cached_contentr r2) r1r r r=r>r?r@rAr&rrrresultrrrcreates" zCachedContent.creatercCs:t}d|vr d|}tj|d}||}t|}|S)zFetches required `CachedContent` resource. Args: name: The resource name referring to the cached content. Returns: `CachedContent` resource with specified `name`. r r)rrrrr r2)r1rrrrr_rrrgets    zCachedContent.get page_size Optional[int]Iterable[CachedContent]ccs8t}tj|d}||D] }t|}|VqdS)a.Lists `CachedContent` objects associated with the project. Args: page_size: The maximum number of permissions to return (per page). The service may return fewer `CachedContent` objects. Returns: A paginated list of `CachedContent` objects. )rcN)rrListCachedContentsRequestlist_cached_contentsr r2)r1rcrrrOrrrlists  zCachedContent.listNonecCs"t}tj|jd}||dS)z!Deletes `CachedContent` resource.rN)rrDeleteCachedContentRequestrdelete_cached_content)rrrrrrdeletes zCachedContent.delete)rAr&cCst}|r |r tdt|}t|}tj|j||d}t }|r+|j dn |r4|j dntdtj ||d}| |}||dS)aXUpdates requested `CachedContent` resource. Args: ttl: TTL for cached resource (in seconds). Defaults to 1 hour. `ttl` and `expire_time` are exclusive arguments. expire_time: Expiration time for cached resource. `ttl` and `expire_time` are exclusive arguments. rJ)rrAr&rAr&zQBad update name: Only `ttl` or `expire_time` can be updated for `CachedContent`.)rO update_maskN)rrPrrYrZrr rr FieldMaskpathsappendUpdateCachedContentRequestupdate_cached_contentr0)rrAr&rr: field_maskr updated_ccrrrupdate s0    zCachedContent.update)rr)rr!)rr#)r-r.rr )r rr rBr=rCr>rDr?rEr@rFrArGr&rHrrI)r rr rBr=rCr>rDr?rEr@rFrArGr&rHrr )rrrr )rb)rcrdrre)rri)rArGr&rHrri)__name__ __module__ __qualname____doc__rpropertyrr r r"r$r%r&r,__repr__ classmethodr2r0 staticmethodr]r`rarhrlrurrrrr s`         62   r ) __future__rdatetimer)typingrrgoogle.generativeairgoogle.generativeai.typesrrgoogle.generativeai.clientrgoogle.protobufr rX _MODEL_ROLEr rrrrs