ci7Z ddlZddlZddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddl mZddlmZddlmZddlmZmZd d lmZd d lmZej4rd d lmZeZej<d ZejAejB d%de e"e#efe$fdee"e#efde"e#effdZ% d%de e"e#efe$fdee"e#efde"e#effdZ& d%de e"e#efe$fdee"e#efde"e#effdZ' d%de e"e#efe$fdee"e#efde"e#effdZ( d%de e"e#efe$fdee"e#efde"e#effdZ) d%de e"e#efe$fdee"e#efde"e#effdZ* d%de e"e#efe$fdee"e#efde"e#effdZ+ d%de e"e#efe$fdee"e#efde"e#effdZ, d%de e"e#efe$fdee"e#efde"e#effdZ- d%de e"e#efe$fdee"e#efde"e#effdZ. d%de e"e#efe$fdee"e#efde"e#effdZ/ d%de e"e#efe$fdee"e#efde"e#effdZ0 d%de e"e#efe$fdee"e#efde"e#effdZ1 d%de e"e#efe$fdee"e#efde"e#effdZ2 d%de e"e#efe$fdee"e#efde"e#effdZ3 d%de e"e#efe$fdee"e#efde"e#effd Z4Gd!d"e jjZ6Gd#d$e jjZ7y)&N)AnyIteratorListOptionalUnion) urlencode) _api_module)_common)get_value_by_path)set_value_by_path) AsyncPagerPager)_agent_engines_utils)types)memory_revisionszvertexai_genai.memories from_object parent_objectreturnci}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|d gt|d gt|d gt|d gt|d gt|d gt|d g't|d gt|d gDcgc]}|c}t|dgt|dgt|dg|Scc}w)N display_name displayName descriptionttl expire_time expireTimerevision_expire_timerevisionExpireTime revision_ttl revisionTtldisable_memory_revisionsdisableMemoryRevisionstopicsmetadatagetvsetvrr to_objectitems ?/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/memories.py"_AgentEngineMemoryConfig_to_vertexr,.s!#I K.)*6 ]]OT+?O-PQ K-)5 ]]OT+ -OP K%!- ]UGT+w%?@ K-)5 ]\ND}o,NO K012>  ! " 56 7  K.)*6 ]]OT+?O-PQ K456B  % & 9: ;  K($0 H:kH:9V'W'W  K*&2 ]ZL${ZL*IJ  (Xs1 E) c Pi}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg#t|dgtt|dg||SNname_urlfactscopeconfig)r&r'r,rrr)s r+3_CreateAgentEngineMemoryRequestParameters_to_vertexr5\s!#I K&". Y(${VH*EF K&". Y${VH"=> K'#/ Y 4 gY#?@ K($0  J .[8*-y   ci}t|dgt|ddgt|dgt|dgt|dgt|dg|SNr/r0r3r%r4s r+3_DeleteAgentEngineMemoryRequestParameters_to_vertexr9v^!#I K&". Y(${VH*EF K($0 Y Dxj$AB r6ci}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|d gt|d gt|d gt|d gt|d gt|d gt|d gt|d gt|d g|S)Ndisable_consolidationdisableConsolidationrevision_labelsrevisionLabelsrrrr r!r"r$metadata_merge_strategymetadataMergeStrategyr%r4s r+,_GenerateAgentEngineMemoriesConfig_to_vertexrBsQ!#I K123?  # $ 67 8  K+,-9 ]-.[CTBU0VW K012>  ! " 56 7  K.)*6 ]]OT+?O-PQ K456B  % & 9: ;  K*&2 ]ZL${ZL*IJ K345A  $ % 89 : r6c i}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|d gt|d gt|d gt|d g#t|d gtt|d g||S) Nr/r0vertex_session_sourcevertexSessionSourcedirect_contents_sourcedirectContentsSourcedirect_memories_sourcedirectMemoriesSourcer2r3)r&r'rBr4s r+7_GenerateAgentEngineMemoriesRequestParameters_to_vertexrJs+!#I K&". Y(${VH*EF K123?  " # 67 8  K234@  # $ 78 9  K234@  # $ 78 9  K'#/ Y 4 gY#?@ K($0  J 8[8*-y   r6ci}t|dgt|ddgt|dgt|dgt|dgt|dg|SNoperation_namer0 operationNamer3r%r4s r+<_GetAgentEngineGenerateMemoriesOperationParameters_to_vertexrOg!#I K*+,8 0${EUDV2W  K($0 Y Dxj$AB r6ci}t|dgt|ddgt|dgt|dgt|dgt|dg|SrLr%r4s r+2_GetAgentEngineMemoryOperationParameters_to_vertexrRrPr6ci}t|dgt|ddgt|dgt|dgt|dgt|dg|Sr8r%r4s r+0_GetAgentEngineMemoryRequestParameters_to_vertexrTr:r6cBi}t|dgt|ddgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S) N page_size_querypageSize page_token pageTokenfilterorder_byorderByr%r4s r+&_ListAgentEngineMemoryConfig_to_vertexr^ s!#I K+'3 ]Xz2D{m4TU K,(4 ]X{3T+ ~5VW K($0 ]Xx0${XJ2OP K*&2 ]Xy14 j\3RS r6c i}t|dgt|ddgt|dgt|dg#t|dgtt|dg||Sr8)r&r'r^r4s r+1_ListAgentEngineMemoryRequestParameters_to_vertexr`!so!#I K&". Y(${VH*EF K($0  J 2[8*-y   r6ci}t|dgt|ddgt|dgt|dgt|dgt|dgt|dg't|dgt|dgDcgc]}|c}t|dgt|dgt|dgt|dgt|dgt|dg|Scc}w)Nr/r0r[ filter_groups filterGroupsforcer3r%r(s r+4_PurgeAgentEngineMemoriesRequestParameters_to_vertexre5s!#I K&". Y(${VH*EF K($0 Y Dxj$AB K/*+7   ";0AB CdT C  K'#/ Y 4 gY#?@ K($0 Y Dxj$AB  Ds4 C ci}t|dgt|dgt|dgt|dg't|dgt|dgDcgc]}|c}|Scc}w)Nr[rbrcr%r(s r+,_RetrieveAgentEngineMemoriesConfig_to_vertexrgPsv!#I K($0 ]XJ[8*(EF K/*+7   ";0AB CdT C  Ds A c i}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg#t|dgtt|dg||S) Nr/r0r2similarity_search_paramssimilaritySearchParamssimple_retrieval_paramssimpleRetrievalParamsr3)r&r'rgr4s r+7_RetrieveAgentEngineMemoriesRequestParameters_to_vertexrmcs!#I K&". Y(${VH*EF K'#/ Y 4 gY#?@ K456B  % & 9: ;  K345A  $ % 89 :  K($0  J 8[8*-y   r6ci}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgt|dgt|dg|S)Nr/r0target_revision_idtargetRevisionIdr3r%r4s r+5_RollbackAgentEngineMemoryRequestParameters_to_vertexrqs!#I K&". Y(${VH*EF K./0< Y+,d;AU@V.WX K($0 Y Dxj$AB r6c*i}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|d gt|d gt|d gt|d gt|d gt|d gt|d g't|d gt|d gDcgc]}|c}t|dgt|dgt|dgt|dgt|ddgt|dg|Scc}w)Nrrrrrrrrrr r!r"r#r$ update_maskrW updateMaskr%r(s r+(_UpdateAgentEngineMemoryConfig_to_vertexrus!#I K.)*6 ]]OT+?O-PQ K-)5 ]]OT+ -OP K%!- ]UGT+w%?@ K-)5 ]\ND}o,NO K012>  ! " 56 7  K.)*6 ]]OT+?O-PQ K456B  % & 9: ;  K($0 H:kH:9V'W'W  K*&2 ]ZL${ZL*IJ K-)5 Hl3T+ 5W  (Xs1 F c Pi}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg#t|dgtt|dg||Sr.)r&r'rur4s r+3_UpdateAgentEngineMemoryRequestParameters_to_vertexrws!#I K&". Y(${VH*EF K&". Y${VH"=> K'#/ Y 4 gY#?@ K($0  J 4[8*-y   r6cBeZdZdddededeeefdeejdejf dZ dddedeejdejfd Z dddddd ded eejd eejd eejdeeeefdeej dej"fdZdddedeej&dej(fdZdddedeej,dej.fdZdddedeej2dejfdZdddedeej2dej"fdZdddddedeeefdeej8deej:deej<dej>f dZ dddededeejBdejDfdZ#dddddedeedeeeefdeejHdejf dZ%ddddddedeedee&ejNdee(deejRdejTf d Z+dZ,e-d)d!Z.dddededeeefdeejdejf d"Z/dddddd ded eejd eejd eejdeeeefdeej dej"fd#Z0dddedeej,de1ej(fd$Z&dddddedeeefdeej8deej:deej<de1ejdf d%Z3dddededeejBdejDfd&Z4ddd'dddedeedee5ejlde(deejRdejTf d(Z7y)*MemoriesNr3r/r1r2r3rc:tj||||}|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 z; Creates a new memory in the Agent Engine. r/r1r2r36This method is only supported in the Vertex AI client.r0{name}/memoriesrW?r3Npostresponsekwargs)r)_CreateAgentEngineMemoryRequestParameters _api_clientvertexai ValueErrorr5get format_maprpopr3 http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsAgentEngineMemoryOperation_from_response model_dump_verify_responseselfr/r1r2r3parameter_model request_dictrequest_url_dictpath query_paramsrr response_dict return_values r+_createzMemories._creates II  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N 77FF"?+E+E+GG  )),7r6c6tj||}|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 Delete an Agent Engine memory. Args: name (str): Required. The name of the Agent Engine memory to be deleted. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/memories/{memory}`. config (DeleteAgentEngineMemoryConfig): Optional. Additional configurations for deleting the Agent Engine. r/r3r~r0{name}rWrr3Ndeleter)r)_DeleteAgentEngineMemoryRequestParametersrrrr9rrrrr3rr rrrrrr DeleteAgentEngineMemoryOperationrrr rr/r3rrrrrrrrrs r+rzMemories.delete#sy$ II  ((UV VNL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++HdL,W"*--TZZ 5N ==LL"?+E+E+GM  )),7r6rDrFrHr2r3rDrFrHc>tj||||||}|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 z9 Generates memories for an Agent Engine. r/rDrFrHr2r3r~r0z{name}/memories:generaterWrr3Nrr)r-_GenerateAgentEngineMemoriesRequestParametersrrrrJrrrrr3rr rrrrrr$AgentEngineGenerateMemoriesOperationrrrrr/rDrFrHr2r3rrrrrrrrrs r+ _generatezMemories._generatebs( MM"7#9#9  ((UV VRL ,//7 1<<=MN1#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N AAPP"?+E+E+GQ  )),7r6c6tj||}|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 a2 Gets an agent engine memory. Args: name (str): Required. A fully-qualified resource name or ID such as "projects/123/locations/us-central1/reasoningEngines/456/memories/789" or a shortened name such as "reasoningEngines/456/memories/789". rr~r0rrWrr3Nrr)r&_GetAgentEngineMemoryRequestParametersrrrrTrrrrr3rr rrrrrrMemoryrrrrs r+rz Memories.getsv  FF  ((UV VKL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N ||22"?+E+E+G3  )),7r6c6tj||}|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 z. Lists Agent Engine memories. rr~r0rrWrr3Nrr)r'_ListAgentEngineMemoryRequestParametersrrrr`rrrrr3rr rrrrrr$ListReasoningEnginesMemoriesResponserrrrs r+_listzMemories._listsy GG  ((UV VLL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N AAPP"?+E+E+GQ  )),7r6rMc6tj||}|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 N)rMr3r~r0z{operationName}rWrr3rr)r(_GetAgentEngineMemoryOperationParametersrrrrRrrrrr3rr rrrrrrrrrr rrMr3rrrrrrrrrs r+_get_memory_operationzMemories._get_memory_operationsy  HH)  ((UV VML ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N 77FF"?+E+E+GG  )),7r6c6tj||}|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-| | Sr)r2_GetAgentEngineGenerateMemoriesOperationParametersrrrrOrrrrr3rr rrrrrrrrrrrs r+ _get_generate_memories_operationz)Memories._get_generate_memories_operationNsy  RR)  ((UV VWL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N AAPP"?+E+E+GQ  )),7r6rirkr3rirkc<tj|||||}|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 z9 Retrieves memories for an Agent Engine. )r/r2rirkr3r~r0z{name}/memories:retrieverWrr3Nrr)r-_RetrieveAgentEngineMemoriesRequestParametersrrrrmrrrrr3rr rrrrrrRetrieveMemoriesResponserrrrr/r2rirkr3rrrrrrrrrs r+ _retrievezMemories._retrieves" MM%=$;  ((UV VRL ,//7 1<<=MN1#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N 55DD"?+E+E+GE  )),7r6roc8tj|||}|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 z; Rollback a memory to a previous revision. r/ror3r~r0z{name}:rollbackrWrr3Nrr)r+_RollbackAgentEngineMemoryRequestParametersrrrrqrrrrr3rr rrrrrr"AgentEngineRollbackMemoryOperationrrr rr/ror3rrrrrrrrrs r+ _rollbackzMemories._rollbacks| KK1 ((UV VPL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N ??NN"?+E+E+GO  )),7r6r1r2r3c:tj||||}|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 z1 Updates an Agent Engine memory. r}r~r0rrWrr3Npatchr)r)_UpdateAgentEngineMemoryRequestParametersrrrrwrrrrr3rr rrrrrrrrrrrs r+_updatezMemories._updates II  ((UV VNL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++GT<V"*--TZZ 5N 77FF"?+E+E+GG  )),7r6r[rbrdr3r[rbrdc<tj|||||}|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 z7 Purges memories from an Agent Engine. r/r[rbrdr3r~r0z{name}/memories:purgerWrr3Nrr)r*_PurgeAgentEngineMemoriesRequestParametersrrrrerrrrr3rr rrrrrr!AgentEnginePurgeMemoriesOperationrrrrr/r[rbrdr3rrrrrrrrrs r+_purgezMemories._purge6s JJ'  ((UV VOL ,//7 .99:JK.#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N >>MM"?+E+E+GN  )),7r6c|j tjdt|_|jj |j S#t$r}t d|d}~wwxYwNz.memory_revisionszThe 'agent_engines.memories.revisions' module requires additional packages. Please install them using pip install google-cloud-aiplatform[agent_engines]) _revisions importlib import_module __package__ ImportErrorMemoryRevisionsrres r+ revisionszMemories.revisionsusp ?? " #,"9"9'#..t/?/?@@  !=  A A, A''A,c|tj}n/t|trtjj |}|j ||||}|j r|js,tj|j|jd}|jr-|j|jj|_ |S|jrtd|jtd|S)5Creates a new memory in the Agent Engine. Args: name (str): Required. The name of the memory to create. fact (str): Required. The fact to be stored in the memory. scope (dict[str, str]): Required. The scope of the memory. For example, {"user_id": "123"}. config (AgentEngineMemoryConfigOrDict): Optional. The configuration for the memory. Returns: AgentEngineMemoryOperation: The operation for creating the memory. r}?rMget_operation_fnpoll_interval_secondsr/Failed to create memory: Error creating memory.)rAgentEngineMemoryConfig isinstancedictmodel_validaterwait_for_completiondoner_await_operationr/rrrerror RuntimeErrorrr/r1r2r3 operations r+createzMemories.creates. >224F  %22AA&IFLL !  % %>>0AA#,>>%)%?%?*- !!%)XX93E3E3J3JX%K "  "%>y>O#PQQ"#;<<r6c|tj}n/t|trtjj |}|j ||||||}|j r\|jsPtj|j|jd}|jrtd|j|S)Generates memories for the agent engine. Args: name (str): Required. The name of the agent engine to generate memories for. vertex_session_source (GenerateMemoriesRequestVertexSessionSource): Optional. The vertex session source to use for generating memories. Only one of vertex_session_source, direct_contents_source, or direct_memories_source can be specified. direct_contents_source(GenerateMemoriesRequestDirectContentsSource): Optional. The direct contents source to use for generating memories. Only one of vertex_session_source, direct_contents_source, or direct_memories_source can be specified. direct_memories_source (GenerateMemoriesRequestDirectMemoriesSource): Optional. The direct memories source to use for generating memories. Only one of vertex_session_source, direct_contents_source, or direct_memories_source can be specified. scope (dict[str, str]): Optional. The scope of the memories to generate. This is optional if vertex_session_source is used, otherwise it must be specified. config (GenerateMemoriesConfig): Optional. The configuration for the memories to generate. Returns: AgentEngineGenerateMemoriesOperation: The operation for generating the memories. rrrFailed to generate memory: )r!GenerateAgentEngineMemoriesConfigrrrrrrrrr/rrrrr/rDrFrHr2r3rs r+generatezMemories.generatesX ><<>F  %<<KKFSFNN"7#9#9 #   % %inn,==(~~!%!F!F&)I "%@@Q#RSSr6c~tdtj|j||j|||S)aZLists Agent Engine memories. Args: name (str): Required. The name of the agent engine to list memories for. config (ListAgentEngineMemoryConfig): Optional. The configuration for the memories to list. Returns: Iterable[Memory]: An iterable of memories. memoriesrr)r functoolspartialrrr/r3s r+listz Memories.lists;$    djjt 4 JJDJ 0    r6c ^tdfdj||S)aRetrieves memories for the agent. Args: name (str): Required. The name of the agent engine to retrieve memories for. scope (dict[str, str]): Required. The scope of the memories to retrieve. For example, {"user_id": "123"}. similarity_search_params (RetrieveMemoriesRequestSimilaritySearchParams): Optional. The similarity search parameters to use for retrieving memories. simple_retrieval_params (RetrieveMemoriesRequestSimpleRetrievalParams): Optional. The simple retrieval parameters to use for retrieving memories. config (RetrieveAgentEngineMemoriesConfig): Optional. The configuration for the memories to retrieve. Returns: Iterator[RetrieveMemoriesResponseRetrievedMemory]: An iterable of retrieved memories. retrieved_memoriesc0j|SNr/rirkr2r3rr3r/r2rrirks r+z#Memories.retrieve..7$4>>)A(? *r6r)rrrr/r2rirkr3s````` r+retrievezMemories.retrievesBD   NN)A(?   !  r6c|tj}n/t|trtjj |}|j |||}|j r\|jsPtj|j|jd}|jrtd|j|S)Rolls back a memory to a previous revision. Args: name (str): Required. The name of the memory to rollback. target_revision_id (str): Required. The revision ID to roll back to config (RollbackAgentEngineMemoryConfig): Optional. The configuration for the rollback. Returns: AgentEngineRollbackMemoryOperation: The operation for rolling back the memory. rrrFailed to rollback memory: )rRollbackAgentEngineMemoryConfigrrrrrrrrr/rrrrr/ror3rs r+rollbackzMemories.rollbackHs* >::99;F  %99HHPFKK'     % %inn,==(~~!%!;!;&)I "%? ?P#QRRr6)rz'memory_revisions_module.MemoryRevisions)8__name__ __module__ __qualname__strrrrAgentEngineMemoryConfigOrDictrr#DeleteAgentEngineMemoryConfigOrDictrr0GenerateMemoriesRequestVertexSessionSourceOrDict1GenerateMemoriesRequestDirectContentsSourceOrDict1GenerateMemoriesRequestDirectMemoriesSourceOrDict'GenerateAgentEngineMemoriesConfigOrDictrr GetAgentEngineMemoryConfigOrDictrr!ListAgentEngineMemoryConfigOrDictrr#GetAgentEngineOperationConfigOrDictrr3RetrieveMemoriesRequestSimilaritySearchParamsOrDict2RetrieveMemoriesRequestSimpleRetrievalParamsOrDict'RetrieveAgentEngineMemoriesConfigOrDictrr%RollbackAgentEngineMemoryConfigOrDictrr#UpdateAgentEngineMemoryConfigOrDictrr MemoryConjunctionFilterOrDictbool$PurgeAgentEngineMemoriesConfigOrDictrrrpropertyrrrr'RetrieveMemoriesResponseRetrievedMemoryrrrMemoryConjunctionFilterr!r6r+ryrysAE 99 9 CH~ 9 <<= 9  ) )9~GK ==BBC =  / / =J   *.JNCC (  B B C!)  C C! C!)  C C! CS#X'CFFGC  3 3CRDH ;;??@ ;  ;BEI 55@@A 5  3 3 5vGK 11BBC 1  ) ) 1nGK 11BBC 1  3 3 1t  JN??CH~ ? #+  E E# ?"*  D D" ?FFG?  ' '?LIM 77 7 DDE 7  1 1 7z#*.FJ 99sm 9 S#X' 9 BBC 9  ) )9~!%MQ $GK;; ;  U%H%H IJ ; ~ ;CCD;  0 0;zJ AA,AE 00 0 CH~ 0 <<= 0  ) )0p   *.JN@@ (  B B @!)  C C! @!)  C C! @S#X'@FFG@  3 3@LEI   @@A  %,,   @  JN3 3 CH~ 3 #+  E E# 3 "*  D D" 3 FFG3  %?? @3 tIM && & DDE &  1 1 &X!%GKGK11 1  U%B%B CD 1  1CCD1  0 01r6rycBeZdZdddededeeefdeejdejf dZ dddedeejdejfd Z dddddd ded eejd eejd eejdeeeefdeej dej"fdZdddedeej&dej(fdZdddedeej,dej.fdZdddedeej2dejfdZdddedeej2dej"fdZdddddedeeefdeej8deej:deej<dej>f dZ dddededeejBdejDfdZ#dddddedeedeeeefdeejHdejf dZ%ddddddedeedee&ejNdee(deejRdejTf d Z+dZ,e-d)d!Z.dddededeeefdeejdejf d"Z/dddddd ded eejd eejd eejdeeeefdeej dej"fd#Z0dddedeej,de1ej(fd$Z&dddddedeeefdeej8deej:deej<de1ejdf d%Z3dddededeejBdejDfd&Z4ddd'dddedeedee5ejlde(deejRdejTf d(Z7y)* AsyncMemoriesNrzr/r1r2r3rcVKtj||||}|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|)rrrrrr5rrrrr3rr rr async_requestrrrrrrrrs r+rzAsyncMemories._creates II  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 D,   #+--TZZ 5N 77FF"?+E+E+GG  )),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}wr)rrrrrr9rrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories.deletes$ II  ((UV VNL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 dL,  #+--TZZ 5N ==LL"?+E+E+GM  )),7 D%F''F%(A>F'rrDrFrHcZKtj||||||}|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)rrrrrrJrrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._generate$s( MM"7#9#9  ((UV VRL ,//7 1<<=MN1#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 D,   #+--TZZ 5N AAPP"?+E+E+GQ  )),7 sD)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}wr)rrrrrrTrrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories.getks  FF  ((UV VKL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N ||22"?+E+E+G3  )),7 rAcRKtj||}|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)rrrrrr`rrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._lists GG  ((UV VLL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N AAPP"?+E+E+GQ  )),7 rArMcRKtj||}|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)rrrrrrRrrrrr3rr rrr>rrrrrrrrs r+rz#AsyncMemories._get_memory_operations  HH)  ((UV VML ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N 77FF"?+E+E+GG  )),7 rAcRKtj||}|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)rrrrrrOrrrrr3rr rrr>rrrrrrrrs r+rz.AsyncMemories._get_generate_memories_operations  RR)  ((UV VWL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N AAPP"?+E+E+GQ  )),7 rArrirkcXKtj|||||}|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)rrrrrrmrrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._retrieveMs" MM%=$;  ((UV VRL ,//7 1<<=MN1#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 D,   #+--TZZ 5N 55DD"?+E+E+GE  )),7 D(F**F(+A>F*rocTKtj|||}|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)rrrrrrqrrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._rollbacks KK1 ((UV VPL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 D,   #+--TZZ 5N ??NN"?+E+E+GO  )),7 sD&F((F&)A>F(rcVKtj||||}|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)rrrrrrwrrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._updates II  ((UV VNL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 T<  #+--TZZ 5N 77FF"?+E+E+GG  )),7 r?rr[rbrdcXKtj|||||}|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)rrrrrrerrrrr3rr rrr>rrrrrrrrs r+rzAsyncMemories._purges JJ'  ((UV VOL ,//7 .99:JK.#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 D,   #+--TZZ 5N >>MM"?+E+E+GN  )),7 rHc|j tjdt|_|jj |j S#t$r}t d|d}~wwxYwr)rrrrrAsyncMemoryRevisionsrrs r+rzAsyncMemories.revisionsIsp ?? " #,"9"9'#33D4D4DEE  !=  rcXK|tj}n/t|trtjj |}|j ||||d{}|j r|js4tj|j|jdd{}|jr5|j|jjd{|_ |S|jrtd|jtd|S77x7@w)rNr}rrrrr)rrrrrrrrr_await_async_operationr/rrrrrrs r+rzAsyncMemories.createZs. >224F  %22AA&IF,, '   % %>>"6"M"M#,>>%)%?%?*-# !!+/889K9K9P9P8+Q%Q "  "%>y>O#PQQ"#;<<+ &Rs7AD*!D$"A D*-D&.9D*'D((=D*&D*(D*cK|tj}n/t|trtjj |}|j ||||||d{}|j rd|jsXtj|j|jdd{}|jrtd|j|S7v7,w)rNrrrr)rrrrrrrrrrOr/rrrrs r+rzAsyncMemories.generatesX ><<>F  %<<KKFSF.."7#9#9 )    % %inn2II(~~!%!F!F&)I "%@@Q#RSS! s%A!C#C$A C/C0+CCcKtdtj|j||j||d{|S7 w)a_Lists Agent Engine memories. Args: name (str): Required. The name of the agent engine to list memories for. config (ListAgentEngineMemoryConfig): Optional. The configuration for the memories to list. Returns: AsyncPager[Memory]: An async pager of memories. rrrN)r rrrr s r+r zAsyncMemories.listsG$    djjt 4**$v*6 6     7s=A A A c zKtdfdj|d{|S7 w)aRetrieves memories for the agent. Args: name (str): Required. The name of the agent engine to retrieve memories for. scope (dict[str, str]): Required. The scope of the memories to retrieve. For example, {"user_id": "123"}. similarity_search_params (RetrieveMemoriesRequestSimilaritySearchParams): Optional. The similarity search parameters to use for retrieving memories. simple_retrieval_params (RetrieveMemoriesRequestSimpleRetrievalParams): Optional. The simple retrieval parameters to use for retrieving memories. config (RetrieveAgentEngineMemoriesConfig): Optional. The configuration for the memories to retrieve. Returns: AsyncPager[RetrieveMemoriesResponseRetrievedMemory]: An async pager of retrieved memories. r c0j|Srrrs r+rz(AsyncMemories.retrieve.. rr6rN)r rrs````` r+rzAsyncMemories.retrievesPD  ..)A(? !  !   s (;9 ;cK|tj}n/t|trtjj |}|j |||d{}|j rd|jsXtj|j|jdd{}|jrtd|j|S7v7,w)rNrrrr)rrrrrrrrrrOr/rrrrs r+rzAsyncMemories.rollback s* >::99;F  %99HHPF++' &    % %inn2II(~~!%!;!;&)I "%? ?P#QRR s%A C"C#A C.C/+CC)rz,memory_revisions_module.AsyncMemoryRevisions)8r"r#r$r%rrrr&rrr'rrr(r)r*r+rrr,rrr-rrr.rrr/r0r1rrr2rrr3rr r4r5r6rrrr7rrrr r8rrrr9r!r:r6r+r<r<sAE ;; ; CH~ ; <<= ;  ) );BGK ??BBC ?  / / ?N   *.JNEE (  B B E!)  C C! E!)  C C! ES#X'EFFGE  3 3EVDH ==??@ =  =FEI 77@@A 7  3 3 7zGK 33BBC 3  ) ) 3rGK 33BBC 3  3 3 3x  JNAACH~ A #+  E E# A"*  D D" AFFGA  ' 'APIM 99 9 DDE 9  1 1 9~#*.FJ ;;sm ; S#X' ; BBC ;  ) );B!%MQ $GK== =  U%H%H IJ = ~ =CCD=  0 0=~J FF,AE 00 0 CH~ 0 <<= 0  ) )0p   *.JN@@ (  B B @!)  C C! @!)  C C! @S#X'@FFG@  3 3@LEI   @@A  ELL !  @  JN3 3 CH~ 3 #+  E E# 3 "*  D D" 3 FFG3  EAA B3 tIM && & DDE &  1 1 &X!%GKGK11 1  U%B%B CD 1  1CCD1  0 01r6r<)N)8rrrloggingtypingrrrrr urllib.parser google.genair r google.genai._commonr r&r r'google.genai.pagersr rrr TYPE_CHECKINGrmemory_revisions_module_ getLoggerloggersetLevelINFOrr%objectr,r5r9rBrJrOrRrTr^r`rergrmrqrurw BaseModuleryr<r:r6r+rfsH$  77"$ ::1" =A   4 5  /3+tCH~v-.+DcN++ #s(^+`/3tCH~v-.DcN+ #s(^8/3 tCH~v-. DcN+  #s(^  /3+tCH~v-.+DcN++ #s(^+`/3)tCH~v-.)DcN+) #s(^)\/3 tCH~v-. DcN+  #s(^ $/3 tCH~v-. DcN+  #s(^ $/3 tCH~v-. DcN+  #s(^  /3tCH~v-.DcN+ #s(^./3tCH~v-.DcN+ #s(^,/3tCH~v-.DcN+ #s(^:/3tCH~v-.DcN+ #s(^*/3"tCH~v-."DcN+" #s(^"N/3tCH~v-.DcN+ #s(^&/30tCH~v-.0DcN+0 #s(^0j/3tCH~v-.DcN+ #s(^4{{%%{|QK**Qr6