ciئ ddlZddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl m Z ddlmZddlmZddlmZmZd d lmZd d lmZej2rd d lmZeZej:d Zej?ej@ ddee!e"efe#fdee!e"efde!e"effdZ$ ddee!e"efe#fdee!e"efde!e"effdZ% ddee!e"efe#fdee!e"efde!e"effdZ& ddee!e"efe#fdee!e"efde!e"effdZ' ddee!e"efe#fdee!e"efde!e"effdZ( ddee!e"efe#fdee!e"efde!e"effdZ) ddee!e"efe#fdee!e"efde!e"effdZ* ddee!e"efe#fdee!e"efde!e"effdZ+ ddee!e"efe#fdee!e"efde!e"effdZ,Gdde jZZ.Gdde jZZ/y)N)AnyIteratorOptionalUnion) urlencode) _api_module)_common)get_value_by_path)set_value_by_path) AsyncPagerPager)_agent_engines_utils)types)session_eventszvertexai_genai.sessions from_object parent_objectreturnci}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg|S) N display_name displayName session_state sessionStatettl expire_time expireTimelabelsgetvsetvrr to_objects ?/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/sessions.py)_CreateAgentEngineSessionConfig_to_vertexr$.s!#I K.)*6 ]]OT+?O-PQ K/*+7 ]^,d;@Q.RS K%!- ]UGT+w%?@ K-)5 ]\ND}o,NO K($0 ]XJ[8*(EF c i}t|dgt|ddgt|dgt|dgt|dgt|dgt|dg#t|dgtt|dg||S)Nname_urluser_iduserIdconfig)rr r$r!s r#4_CreateAgentEngineSessionRequestParameters_to_vertexr,Fs!#I K&". Y(${VH*EF K)%1 Y Dyk$BC K($0  J 5[8*-y   r%ci}t|dgt|ddgt|dgt|dgt|dgt|dg|SNr'r(r+rr!s r#4_DeleteAgentEngineSessionRequestParameters_to_vertexr/]^!#I K&". Y(${VH*EF K($0 Y Dxj$AB r%ci}t|dgt|ddgt|dgt|dgt|dgt|dg|S)Noperation_namer( operationNamer+rr!s r#3_GetAgentEngineSessionOperationParameters_to_vertexr4ksg!#I K*+,8 0${EUDV2W  K($0 Y Dxj$AB r%ci}t|dgt|ddgt|dgt|dgt|dgt|dg|Sr.rr!s r#1_GetAgentEngineSessionRequestParameters_to_vertexr6{r0r%ci}t|dgt|ddgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S)N page_size_querypageSize page_token pageTokenfilterrr!s r#(_ListAgentEngineSessionsConfig_to_vertexr>s!#I K+'3 ]Xz2D{m4TU K,(4 ]X{3T+ ~5VW K($0 ]Xx0${XJ2OP r%c i}t|dgt|ddgt|dgt|dg#t|dgtt|dg||Sr.)rr r>r!s r#3_ListAgentEngineSessionsRequestParameters_to_vertexr@so!#I K&". Y(${VH*EF K($0  J 4[8*-y   r%c i}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|d gt|d d gt|d gt|d gt|d gt|d g|S)Nrrrrrrrr update_maskr9 updateMaskr)r*rr!s r#)_UpdateAgentEngineSessionConfig_to_vertexrDs&!#I K.)*6 ]]OT+?O-PQ K/*+7 ]^,d;@Q.RS K%!- ]UGT+w%?@ K-)5 ]\ND}o,NO K($0 ]XJ[8*(EF K-)5 Hl3T+ 5W  K)%1 ]XJ[9+(FG r%c i}t|dgt|ddgt|dgt|dg#t|dgtt|dg||Sr.)rr rDr!s r#4_UpdateAgentEngineSessionRequestParameters_to_vertexrFso!#I K&". Y(${VH*EF K($0  J 5[8*-y   r%c eZdZdddededeej dejfdZdddedeejdejfdZ dddedeejdejfd Zdddedeejdej fd Zddd edeej$dejfd Zdddedeej(dejfd ZdZeej2dddZdddededeej dejfdZdddedeejdeejfdZy)SessionsNr+r'r)r+rc8tj|||}|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 aK Creates a new session in the Agent Engine. Args: name (str): Required. The name of the Agent Engine to create the session under. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`. user_id (str): Required. The user ID of the session. config (CreateAgentEngineSessionConfig): Optional. Additional configurations for creating the Agent Engine session. Returns: AgentEngineSessionOperation: The operation for creating the Agent Engine session. r'r)r+6This method is only supported in the Vertex AI client.r({name}/sessionsr9?r+Npostresponsekwargs)r*_CreateAgentEngineSessionRequestParameters _api_clientvertexai ValueErrorr,get format_maprpopr+ http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsAgentEngineSessionOperation_from_response model_dump_verify_response selfr'r)r+parameter_model request_dictrequest_url_dictpath query_paramsr[rR response_dict return_values r#_createzSessions._creates|, JJ ((UV VOL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++FD, U"*--TZZ 5N 88GG"?+E+E+GH  )),7r%c6tj||}|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 session. Args: name (str): Required. The name of the Agent Engine session to be deleted. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/sessions/{session_id}`. config (DeleteAgentEngineSessionConfig): Optional. Additional configurations for deleting the Agent Engine session. Returns: DeleteAgentEngineSessionOperation: The operation for deleting the Agent Engine session. r'r+rMr({name}r9rOr+NdeleterQ)r*_DeleteAgentEngineSessionRequestParametersrUrVrWr/rXrYrrZr+r[r r\r]r^r_r`ra!DeleteAgentEngineSessionOperationrcrdre rgr'r+rhrirjrkrlr[rRrmrns r#rtzSessions.delete)sy( JJ  ((UV VOL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++HdL,W"*--TZZ 5N >>MM"?+E+E+GN  )),7r%c6tj||}|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 session. Args: name (str): Required. The name of the Agent Engine session to get. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/sessions/{session_id}`. config (GetAgentEngineSessionConfig): Optional. Additional configurations for getting the Agent Engine session. Returns: AgentEngineSession: The requested Agent Engine session. rrrMr(rsr9rOr+NrXrQ)r'_GetAgentEngineSessionRequestParametersrUrVrWr6rXrYrrZr+r[r r\r]r^r_r`raSessionrcrdrerws r#rXz Sessions.getjsv( GG  ((UV VLL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N }}33"?+E+E+G4  )),7r%c6tj||}|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 sessions. Args: name (str): Required. The name of the Agent Engine to list sessions for. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`. config (ListAgentEngineSessionsConfig): Optional. Additional configurations for listing the Agent Engine sessions. Returns: ListReasoningEnginesSessionsResponse: The requested Agent Engine sessions. rrrMr(rNr9rOr+NrXrQ)r)_ListAgentEngineSessionsRequestParametersrUrVrWr@rXrYrrZr+r[r r\r]r^r_r`ra$ListReasoningEnginesSessionsResponsercrdrerws r#_listzSessions._listsy( II  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N AAPP"?+E+E+GQ  )),7r%r2c6tj||}|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)r2r+rMr(z{operationName}r9rOr+rXrQ)r)_GetAgentEngineSessionOperationParametersrUrVrWr4rXrYrrZr+r[r r\r]r^r_r`rarbrcrdre rgr2r+rhrirjrkrlr[rRrmrns r#_get_session_operationzSessions._get_session_operationsy  II)  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++E4|T"*--TZZ 5N 88GG"?+E+E+GH  )),7r%c6tj||}|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 Updates an Agent Engine session. Args: name (str): Required. The name of the Agent Engine session to be updated. Format: `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/sessions/{session_id}`. config (UpdateAgentEngineSessionConfig): Optional. Additional configurations for updating the Agent Engine session. Returns: AgentEngineSessionOperation: The operation for updating the Agent Engine session. rrrMr(rsr9rOr+NpatchrQ)r*_UpdateAgentEngineSessionRequestParametersrUrVrWrFrXrYrrZr+r[r r\r]r^r_r`rarbrcrdrerws r#_updatezSessions._updatesy( JJ  ((UV VOL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ##++GT<V"*--TZZ 5N 88GG"?+E+E+GH  )),7r%mThe Vertex SDK GenAI agent_engines.sessions.events module is experimental, and may change in future versions.c|j tjdt|_|jj |j S#t$r}t d|d}~wwxYwNz.session_eventszThe 'agent_engines.sessions.events' module requiresadditional packages. Please install them using pip install google-cloud-aiplatform[agent_engines])_events importlib import_module __package__ ImportError SessionEventsrUrges r#eventszSessions.eventsbsp <<   )667H+V ||))$*:*:;;  !=  A A, A''A,c,|tj}n/t|trtjj |}|j |||}|j r|jstj|j|jd}|jr-|j|jj|_ |S|jrtd|jtd|j|S)Creates a new session in the Agent Engine. Args: name (str): Required. The name of the agent engine to create the session for. user_id (str): Required. The user ID of the session. config (CreateAgentEngineSessionConfig): Optional. The configuration for the session to create. Returns: AgentEngineSessionOperation: The operation for creating the session. rL?r2get_operation_fnpoll_interval_secondsr'Failed to create session: FError retrieving session from the operation response. Operation name: )rCreateAgentEngineSessionConfig isinstancedictmodel_validaterowait_for_completiondoner_await_operationr'rrRrXerror RuntimeErrorrgr'r)r+ operations r#createzSessions.createus( >99;F  %99HHPFLL!  % %inn,==(~~!%!>L..|< ::<H ))77 D,   #+--TZZ 5N 88GG"?+E+E+GH  )),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}wrq)rrurUrVrWr/rXrYrrZr+r[r r\r]rr_r`rarvrcrdrerws r#rtzAsyncSessions.deletes( JJ  ((UV VOL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 dL,  #+--TZZ 5N >>MM"?+E+E+GN  )),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}wry)rrzrUrVrWr6rXrYrrZr+r[r r\r]rr_r`rar{rcrdrerws r#rXzAsyncSessions.getHs( GG  ((UV VLL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N }}33"?+E+E+G4  )),7 rcRKtj||}|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})rr~rUrVrWr@rXrYrrZr+r[r r\r]rr_r`rarrcrdrerws r#rzAsyncSessions._lists( II  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N AAPP"?+E+E+GQ  )),7 rr2cRKtj||}|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)rrrUrVrWr4rXrYrrZr+r[r r\r]rr_r`rarbrcrdrers r#rz$AsyncSessions._get_session_operations  II)  ((UV VNL ,//7 (334DE(#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 4|  #+--TZZ 5N 88GG"?+E+E+GH  )),7 rcRKtj||}|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)rrrUrVrWrFrXrYrrZr+r[r r\r]rr_r`rarbrcrdrerws r#rzAsyncSessions._updates( JJ  ((UV VOL ,//7 **+;<#''1 V1Y|456D4(48  " " .&&33?*11>>L..|< ::<H ))77 T<  #+--TZZ 5N 88GG"?+E+E+GH  )),7 rrc|j tjdt|_|jj |j S#t$r}t d|d}~wwxYwr)rrrrrAsyncSessionEventsrUrs r#rzAsyncSessions.eventsHsp <<   )667H+V ||..t/?/?@@  !=  rcpK|tj}n/t|trtjj |}|j |||d{}|j r|jstj|j|jdd{}|jr5|j|jjd{|_ |S|jrtd|jtd|j|S777Mw)rNrLrrrrr)rrrrrrorrr_await_async_operationr'rrRrXrrrs r#rzAsyncSessions.create[s&( >99;F  %99HHPF,,'   % %inn2II(~~!%!r@rDrF BaseModulerHrrr%r#rs7$  11"$ ::1" 9A   4 5  /3tCH~v-.DcN+ #s(^4/3tCH~v-.DcN+ #s(^2/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(^D/3tCH~v-.DcN+ #s(^(W {%%W tc K**c r%