cil8 ddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddl mZddl mZddlmZmZd d lmZej.d Zej3ej4 dd eeeefefd eeeefdeeeffdZ dd eeeefefd eeeefdeeeffdZ dd eeeefefd eeeefdeeeffdZ Gdde jBZ"Gdde jBZ#y)N)AnyIteratorOptionalUnion) urlencode) _api_module)_common)get_value_by_path)set_value_by_path) AsyncPagerPager)typeszvertexai_genai.memoryrevisions from_object parent_objectreturnci}t|dgt|ddgt|dgt|dgt|dgt|dg|SNname_urlconfiggetvsetvrr to_objects G/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/memory_revisions.py8_GetAgentEngineMemoryRevisionRequestParameters_to_vertexr&s^!#I K&". Y(${VH*EF K($0 Y Dxj$AB ci}t|dgt|ddgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S)N page_size_querypageSize page_token pageTokenfilterrrs r/_ListAgentEngineMemoryRevisionsConfig_to_vertexr'4s!#I K+'3 ]Xz2D{m4TU K,(4 ]X{3T+ ~5VW K($0 ]Xx0${XJ2OP rc i}t|dgt|ddgt|dgt|dg#t|dgtt|dg||Sr)rrr'rs r:_ListAgentEngineMemoryRevisionsRequestParameters_to_vertexr)Fso!#I K&". Y(${VH*EF K($0  J ;[8*-y   rceZdZdddedeej dejfdZdddedeejdejfdZ dddedeejde ejfdZ y) MemoryRevisionsNrrrrc6tj||}|jjs t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} | j sint#j$| j } tj&j)| |j+ } |jj-| | S a- Gets an agent engine memory revision. Args: name (str): Required. The name of the Agent Engine memory revision to get. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/memories/{memory_id}/revisions/{revision_id}`. config (GetAgentEngineMemoryRevisionConfig): Optional. Additional configurations for getting the Agent Engine memory revision. Returns: AgentEngineMemoryRevision: The requested Agent Engine memory revision. rr6This method is only supported in the Vertex AI client.rz{name}r"?rNgetresponsekwargs)r._GetAgentEngineMemoryRevisionRequestParameters _api_clientvertexai ValueErrorrr2 format_maprpopr http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsMemoryRevision_from_response model_dump_verify_response selfrrparameter_model request_dictrequest_url_dictpath query_paramsr<r4 response_dict return_values rr2zMemoryRevisions.get\sx( NN  ((UV VSL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N ++::"?+E+E+G;  )),7rc6tj||}|jjs t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} | j sint#j$| j } tj&j)| |j+ } |jj-| | S a+ Lists Agent Engine memory revisions. Args: name (str): Required. The name of the Agent Engine memory to list revisions for. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/memories/{memory_id}`. config (ListAgentEngineMemoryRevisionsConfig): Optional. Additional configurations for listing the Agent Engine memory revisions. Returns: ListAgentEngineMemoryRevisionsResponse: The requested Agent Engine memory revisions. r/r0rz{name}/revisionsr"r1rNr2r3)r0_ListAgentEngineMemoryRevisionsRequestParametersr7r8r9r)r2r:rr;rr<r r=r>r?r@rArB&ListAgentEngineMemoryRevisionsResponserDrErFrGs r_listzMemoryRevisions._listsy( PP  ((UV VUL ,//7 )445EF)#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N CCRR"?+E+E+GS  )),7rc~tdtj|j||j|||S)aLists Agent Engine memory revisions. Args: name (str): Required. The name of the Memory to list revisions for. config (ListAgentEngineMemoryRevisionsConfigOrDict): Optional. The configuration for the memories to list revisions. Returns: Iterable[MemoryRevision]: An iterable of memory revisions. memory_revisionsrr/)r functoolspartialrTrHrrs rlistzMemoryRevisions.lists;$    djjt 4 JJDJ 0    r)__name__ __module__ __qualname__strrr(GetAgentEngineMemoryRevisionConfigOrDictrCr2*ListAgentEngineMemoryRevisionsConfigOrDictrSrTrr[rrr+r+Zs LP ??GGH ?    ?JNR ??IIJ ?  5 5 ?JNR   IIJ  %&& '  rr+ceZdZdddedeej dejfdZdddedeejdejfdZ dddedeejde ejfdZ y) AsyncMemoryRevisionsNr,rrrcRKtj||}|jjs t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} | j sint#j$| j } tj&j)| |j+ } |jj-| | S7}wr.)rr6r7r8r9rr2r:rr;rr<r r=r> async_requestr@rArBrCrDrErFrGs rr2zAsyncMemoryRevisions.gets( NN  ((UV VSL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N ++::"?+E+E+G;  )),7 D%F''F%(A>F'cRKtj||}|jjs t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} | j sint#j$| j } tj&j)| |j+ } |jj-| | S7}wrQ)rrRr7r8r9r)r2r:rr;rr<r r=r>rfr@rArBrSrDrErFrGs rrTzAsyncMemoryRevisions._list=s( PP  ((UV VUL ,//7 )445EF)#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N CCRR"?+E+E+GS  )),7 rgcKtdtj|j||j||d{|S7 w)aLists Agent Engine memory revisions. Args: name (str): Required. The name of the Memory to list revisions for. config (ListAgentEngineMemoryRevisionsConfigOrDict): Optional. The configuration for the memories to list revisions. Returns: AsyncPager[MemoryRevision]: An async pager of memory revisions. rVrWr/N)r rXrYrTrZs rr[zAsyncMemoryRevisions.listsG$    djjt 4**$v*6 6     7s=A A A )r\r]r^r_rrr`rCr2rarSrTr r[rbrrrdrds LP AAGGH A    ANNR AAIIJ A  5 5 ANNR   IIJ  E(( )  rrd)N)$rXrAloggingtypingrrrr urllib.parser google.genairr google.genai._commonr rr rgoogle.genai.pagersr r r getLoggerloggersetLevelINFOdictr_objectrr'r) BaseModuler+rdrbrrrxsZ$ 11"$ ::1   ; <  /3 tCH~v-. DcN+  #s(^  /3tCH~v-.DcN+ #s(^(/3tCH~v-.DcN+ #s(^([ k,,[ |_ ;11_ r