3535logger .setLevel (logging .INFO )
3636
3737
38- def _AppendAgentEngineTaskEventRequestParameters_to_vertex (
38+ def _AppendRuntimeTaskEventRequestParameters_to_vertex (
3939 from_object : Union [dict [str , Any ], object ],
4040 parent_object : Optional [dict [str , Any ]] = None ,
4141) -> dict [str , Any ]:
@@ -53,7 +53,7 @@ def _AppendAgentEngineTaskEventRequestParameters_to_vertex(
5353 return to_object
5454
5555
56- def _AppendAgentEngineTaskEventResponse_from_vertex (
56+ def _AppendRuntimeTaskEventResponse_from_vertex (
5757 from_object : Union [dict [str , Any ], object ],
5858 parent_object : Optional [dict [str , Any ]] = None ,
5959) -> dict [str , Any ]:
@@ -62,7 +62,7 @@ def _AppendAgentEngineTaskEventResponse_from_vertex(
6262 return to_object
6363
6464
65- def _ListAgentEngineTaskEventsConfig_to_vertex (
65+ def _ListRuntimeTaskEventsConfig_to_vertex (
6666 from_object : Union [dict [str , Any ], object ],
6767 parent_object : Optional [dict [str , Any ]] = None ,
6868) -> dict [str , Any ]:
@@ -83,7 +83,7 @@ def _ListAgentEngineTaskEventsConfig_to_vertex(
8383 return to_object
8484
8585
86- def _ListAgentEngineTaskEventsRequestParameters_to_vertex (
86+ def _ListRuntimeTaskEventsRequestParameters_to_vertex (
8787 from_object : Union [dict [str , Any ], object ],
8888 parent_object : Optional [dict [str , Any ]] = None ,
8989) -> dict [str , Any ]:
@@ -92,9 +92,7 @@ def _ListAgentEngineTaskEventsRequestParameters_to_vertex(
9292 setv (to_object , ["_url" , "name" ], getv (from_object , ["name" ]))
9393
9494 if getv (from_object , ["config" ]) is not None :
95- _ListAgentEngineTaskEventsConfig_to_vertex (
96- getv (from_object , ["config" ]), to_object
97- )
95+ _ListRuntimeTaskEventsConfig_to_vertex (getv (from_object , ["config" ]), to_object )
9896
9997 return to_object
10098
@@ -106,23 +104,23 @@ def append(
106104 * ,
107105 name : str ,
108106 task_events : builtins .list [types .TaskEventOrDict ],
109- config : Optional [types .AppendAgentEngineTaskEventConfigOrDict ] = None ,
110- ) -> types .AppendAgentEngineTaskEventResponse :
107+ config : Optional [types .AppendRuntimeTaskEventConfigOrDict ] = None ,
108+ ) -> types .AppendRuntimeTaskEventResponse :
111109 """
112- Adds events to an Agent Engine task.
110+ Adds events to an Agent Runtime task.
113111
114112 Args:
115- name (str): Required. The name of the Agent Engine task to append the events to. Format:
113+ name (str): Required. The name of the Agent Runtime task to append the events to. Format:
116114 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/a2aTasks/{a2a_task_id}`.
117115 task_events (list[TaskEvent]):
118116 Required. The events to append to the task.
119117
120118 Returns:
121- AppendAgentEngineTaskEventResponse : The response for appending the task events.
119+ AppendRuntimeTaskEventResponse : The response for appending the task events.
122120
123121 """
124122
125- parameter_model = types ._AppendAgentEngineTaskEventRequestParameters (
123+ parameter_model = types ._AppendRuntimeTaskEventRequestParameters (
126124 name = name ,
127125 task_events = task_events ,
128126 config = config ,
@@ -134,7 +132,7 @@ def append(
134132 "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
135133 )
136134 else :
137- request_dict = _AppendAgentEngineTaskEventRequestParameters_to_vertex (
135+ request_dict = _AppendRuntimeTaskEventRequestParameters_to_vertex (
138136 parameter_model
139137 )
140138 request_url_dict = request_dict .get ("_url" )
@@ -164,11 +162,9 @@ def append(
164162 response_dict = {} if not response .body else json .loads (response .body )
165163
166164 if self ._api_client .vertexai :
167- response_dict = _AppendAgentEngineTaskEventResponse_from_vertex (
168- response_dict
169- )
165+ response_dict = _AppendRuntimeTaskEventResponse_from_vertex (response_dict )
170166
171- return_value = types .AppendAgentEngineTaskEventResponse ._from_response (
167+ return_value = types .AppendRuntimeTaskEventResponse ._from_response (
172168 response = response_dict ,
173169 kwargs = (
174170 {
@@ -196,23 +192,23 @@ def _list(
196192 self ,
197193 * ,
198194 name : str ,
199- config : Optional [types .ListAgentEngineTaskEventsConfigOrDict ] = None ,
200- ) -> types .ListAgentEngineTaskEventsResponse :
195+ config : Optional [types .ListRuntimeTaskEventsConfigOrDict ] = None ,
196+ ) -> types .ListRuntimeTaskEventsResponse :
201197 """
202- Lists Agent Engine task events.
198+ Lists Agent Runtime task events.
203199
204200 Args:
205- name (str): Required. The name of the Agent Engine task to list events for. Format:
201+ name (str): Required. The name of the Agent Runtime task to list events for. Format:
206202 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/a2aTasks/{a2a_task_id}`.
207- config (ListAgentEngineTaskEventsConfig ):
208- Optional. Additional configurations for listing the Agent Engine tasks.
203+ config (ListRuntimeTaskEventsConfig ):
204+ Optional. Additional configurations for listing the Agent Runtime tasks.
209205
210206 Returns:
211- ListAgentEngineTaskEventsResponse : The requested Agent Engine tasks.
207+ ListRuntimeTaskEventsResponse : The requested Agent Runtime tasks.
212208
213209 """
214210
215- parameter_model = types ._ListAgentEngineTaskEventsRequestParameters (
211+ parameter_model = types ._ListRuntimeTaskEventsRequestParameters (
216212 name = name ,
217213 config = config ,
218214 )
@@ -223,7 +219,7 @@ def _list(
223219 "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
224220 )
225221 else :
226- request_dict = _ListAgentEngineTaskEventsRequestParameters_to_vertex (
222+ request_dict = _ListRuntimeTaskEventsRequestParameters_to_vertex (
227223 parameter_model
228224 )
229225 request_url_dict = request_dict .get ("_url" )
@@ -252,7 +248,7 @@ def _list(
252248
253249 response_dict = {} if not response .body else json .loads (response .body )
254250
255- return_value = types .ListAgentEngineTaskEventsResponse ._from_response (
251+ return_value = types .ListRuntimeTaskEventsResponse ._from_response (
256252 response = response_dict ,
257253 kwargs = (
258254 {
@@ -280,13 +276,13 @@ def list(
280276 self ,
281277 * ,
282278 name : str ,
283- config : Optional [types .ListAgentEngineTaskEventsConfigOrDict ] = None ,
279+ config : Optional [types .ListRuntimeTaskEventsConfigOrDict ] = None ,
284280 ) -> Iterator [types .TaskEvent ]:
285- """Lists the A2A tasks of an Agent Engine .
281+ """Lists the A2A tasks of a Runtime .
286282
287283 Args:
288284 name (str):
289- Required. The name of the agent engine to list tasks for.
285+ Required. The name of the agent runtime to list tasks for.
290286 config (List):
291287 Optional. The configuration for the tasks to list.
292288
@@ -309,23 +305,23 @@ async def append(
309305 * ,
310306 name : str ,
311307 task_events : builtins .list [types .TaskEventOrDict ],
312- config : Optional [types .AppendAgentEngineTaskEventConfigOrDict ] = None ,
313- ) -> types .AppendAgentEngineTaskEventResponse :
308+ config : Optional [types .AppendRuntimeTaskEventConfigOrDict ] = None ,
309+ ) -> types .AppendRuntimeTaskEventResponse :
314310 """
315- Adds events to an Agent Engine task.
311+ Adds events to an Agent Runtime task.
316312
317313 Args:
318- name (str): Required. The name of the Agent Engine task to append the events to. Format:
314+ name (str): Required. The name of the Agent Runtime task to append the events to. Format:
319315 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/a2aTasks/{a2a_task_id}`.
320316 task_events (list[TaskEvent]):
321317 Required. The events to append to the task.
322318
323319 Returns:
324- AppendAgentEngineTaskEventResponse : The response for appending the task events.
320+ AppendRuntimeTaskEventResponse : The response for appending the task events.
325321
326322 """
327323
328- parameter_model = types ._AppendAgentEngineTaskEventRequestParameters (
324+ parameter_model = types ._AppendRuntimeTaskEventRequestParameters (
329325 name = name ,
330326 task_events = task_events ,
331327 config = config ,
@@ -337,7 +333,7 @@ async def append(
337333 "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
338334 )
339335 else :
340- request_dict = _AppendAgentEngineTaskEventRequestParameters_to_vertex (
336+ request_dict = _AppendRuntimeTaskEventRequestParameters_to_vertex (
341337 parameter_model
342338 )
343339 request_url_dict = request_dict .get ("_url" )
@@ -369,11 +365,9 @@ async def append(
369365 response_dict = {} if not response .body else json .loads (response .body )
370366
371367 if self ._api_client .vertexai :
372- response_dict = _AppendAgentEngineTaskEventResponse_from_vertex (
373- response_dict
374- )
368+ response_dict = _AppendRuntimeTaskEventResponse_from_vertex (response_dict )
375369
376- return_value = types .AppendAgentEngineTaskEventResponse ._from_response (
370+ return_value = types .AppendRuntimeTaskEventResponse ._from_response (
377371 response = response_dict ,
378372 kwargs = (
379373 {
@@ -401,23 +395,23 @@ async def _list(
401395 self ,
402396 * ,
403397 name : str ,
404- config : Optional [types .ListAgentEngineTaskEventsConfigOrDict ] = None ,
405- ) -> types .ListAgentEngineTaskEventsResponse :
398+ config : Optional [types .ListRuntimeTaskEventsConfigOrDict ] = None ,
399+ ) -> types .ListRuntimeTaskEventsResponse :
406400 """
407- Lists Agent Engine task events.
401+ Lists Agent Runtime task events.
408402
409403 Args:
410- name (str): Required. The name of the Agent Engine task to list events for. Format:
404+ name (str): Required. The name of the Agent Runtime task to list events for. Format:
411405 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/a2aTasks/{a2a_task_id}`.
412- config (ListAgentEngineTaskEventsConfig ):
413- Optional. Additional configurations for listing the Agent Engine tasks.
406+ config (ListRuntimeTaskEventsConfig ):
407+ Optional. Additional configurations for listing the Agent Runtime tasks.
414408
415409 Returns:
416- ListAgentEngineTaskEventsResponse : The requested Agent Engine tasks.
410+ ListRuntimeTaskEventsResponse : The requested Agent Runtime tasks.
417411
418412 """
419413
420- parameter_model = types ._ListAgentEngineTaskEventsRequestParameters (
414+ parameter_model = types ._ListRuntimeTaskEventsRequestParameters (
421415 name = name ,
422416 config = config ,
423417 )
@@ -428,7 +422,7 @@ async def _list(
428422 "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
429423 )
430424 else :
431- request_dict = _ListAgentEngineTaskEventsRequestParameters_to_vertex (
425+ request_dict = _ListRuntimeTaskEventsRequestParameters_to_vertex (
432426 parameter_model
433427 )
434428 request_url_dict = request_dict .get ("_url" )
@@ -459,7 +453,7 @@ async def _list(
459453
460454 response_dict = {} if not response .body else json .loads (response .body )
461455
462- return_value = types .ListAgentEngineTaskEventsResponse ._from_response (
456+ return_value = types .ListRuntimeTaskEventsResponse ._from_response (
463457 response = response_dict ,
464458 kwargs = (
465459 {
@@ -487,13 +481,13 @@ async def list(
487481 self ,
488482 * ,
489483 name : str ,
490- config : Optional [types .ListAgentEngineTaskEventsConfigOrDict ] = None ,
484+ config : Optional [types .ListRuntimeTaskEventsConfigOrDict ] = None ,
491485 ) -> AsyncPager [types .TaskEvent ]:
492- """Lists the A2A tasks of an Agent Engine .
486+ """Lists the A2A tasks of an Agent Runtime .
493487
494488 Args:
495489 name (str):
496- Required. The name of the agent engine to list tasks for.
490+ Required. The name of the agent runtime to list tasks for.
497491 config (List):
498492 Optional. The configuration for the tasks to list.
499493
0 commit comments