Skip to content

Commit df9fa1c

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Renaming.
PiperOrigin-RevId: 945668576
1 parent 852c546 commit df9fa1c

127 files changed

Lines changed: 28747 additions & 21064 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

agentplatform/_genai/_agent_engines_utils.py renamed to agentplatform/_genai/_runtimes_utils.py

Lines changed: 88 additions & 94 deletions
Large diffs are not rendered by default.

agentplatform/_genai/a2a_task_events.py

Lines changed: 49 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
logger.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

Comments
 (0)