o Zh+*@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)selfrclientrequestresponserR/var/www/html/lang_env/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"Ztotal_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)Zincluding_default_value_fieldsN) isinstancer rdicttypeto_dictitemssetattr)rupdateskeyvaluerrrr/js  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]r? 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>r?r&)cached_content) ValueErrorlenrZ to_contentZto_function_libraryZto_protoZto_tool_configZ to_contentsZrole _USER_ROLErto_optional_ttlto_optional_expire_timerr ZCreateCachedContentRequest) r r r;r<r=r>r?r&Z tools_librMrrr_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. rL)rrSZcreate_cached_contentr r1) r0r r r;r<r=r>r?r&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 r1)r0rrrrrTrrrgets    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. )rXN)rrZListCachedContentsRequestZlist_cached_contentsr r1)r0rXrrrMrrrlists  zCachedContent.listNonecCs"t}tj|jd}||dS)z!Deletes `CachedContent` resource.rN)rrZDeleteCachedContentRequestrZdelete_cached_content)rrrrrrdeletes zCachedContent.delete)r?r&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. rH)rr?r&r?r&zQBad update name: Only `ttl` or `expire_time` can be updated for `CachedContent`.)rMZ update_maskN)rrNrrQrRrr rr Z FieldMaskpathsappendZUpdateCachedContentRequestZupdate_cached_contentr/)rr?r&rr8Z field_maskrZ updated_ccrrrupdate s0    zCachedContent.update)rr)rr!)rr#)r,r-rr )r rr r@r;rAr<rBr=rCr>rDr?rEr&rFrrG)r rr r@r;rAr<rBr=rCr>rDr?rEr&rFrr )rrrr )rW)rXrYrrZ)rr\)r?rEr&rFrr\)__name__ __module__ __qualname____doc__rpropertyrr r r"r$r%r&r+__repr__ classmethodr1r/ staticmethodrSrUrVr[r]r`rrrrr s`         62   r ) __future__rdatetimer)typingrrZgoogle.generativeairZgoogle.generativeai.typesrrZgoogle.generativeai.clientrZgoogle.protobufr rPZ _MODEL_ROLEr rrrrs