Skip to content

Commit e1b6541

Browse files
matthew29tangcopybara-github
matthew29tang
authored and
copybara-github
committed
chore: Exclude config from request body
PiperOrigin-RevId: 883246061
1 parent e11ff3e commit e1b6541

8 files changed

Lines changed: 32 additions & 181 deletions

File tree

‎vertexai/_genai/a2a_task_events.py‎

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,6 @@ def _AppendAgentEngineTaskEventRequestParameters_to_vertex(
5050
[item for item in getv(from_object, ["task_events"])],
5151
)
5252

53-
if getv(from_object, ["config"]) is not None:
54-
setv(to_object, ["config"], getv(from_object, ["config"]))
55-
5653
return to_object
5754

5855

@@ -95,12 +92,8 @@ def _ListAgentEngineTaskEventsRequestParameters_to_vertex(
9592
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
9693

9794
if getv(from_object, ["config"]) is not None:
98-
setv(
99-
to_object,
100-
["config"],
101-
_ListAgentEngineTaskEventsConfig_to_vertex(
102-
getv(from_object, ["config"]), to_object
103-
),
95+
_ListAgentEngineTaskEventsConfig_to_vertex(
96+
getv(from_object, ["config"]), to_object
10497
)
10598

10699
return to_object

‎vertexai/_genai/a2a_tasks.py‎

Lines changed: 2 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,7 @@ def _CreateAgentEngineTaskRequestParameters_to_vertex(
7575
setv(to_object, ["_query", "a2a_task_id"], getv(from_object, ["a2a_task_id"]))
7676

7777
if getv(from_object, ["config"]) is not None:
78-
setv(
79-
to_object,
80-
["config"],
81-
_CreateAgentEngineTaskConfig_to_vertex(
82-
getv(from_object, ["config"]), to_object
83-
),
84-
)
78+
_CreateAgentEngineTaskConfig_to_vertex(getv(from_object, ["config"]), to_object)
8579

8680
return to_object
8781

@@ -94,9 +88,6 @@ def _GetAgentEngineTaskRequestParameters_to_vertex(
9488
if getv(from_object, ["name"]) is not None:
9589
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
9690

97-
if getv(from_object, ["config"]) is not None:
98-
setv(to_object, ["config"], getv(from_object, ["config"]))
99-
10091
return to_object
10192

10293

@@ -130,13 +121,7 @@ def _ListAgentEngineTasksRequestParameters_to_vertex(
130121
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
131122

132123
if getv(from_object, ["config"]) is not None:
133-
setv(
134-
to_object,
135-
["config"],
136-
_ListAgentEngineTasksConfig_to_vertex(
137-
getv(from_object, ["config"]), to_object
138-
),
139-
)
124+
_ListAgentEngineTasksConfig_to_vertex(getv(from_object, ["config"]), to_object)
140125

141126
return to_object
142127

‎vertexai/_genai/agent_engines.py‎

Lines changed: 4 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -119,13 +119,7 @@ def _CreateAgentEngineRequestParameters_to_vertex(
119119
) -> dict[str, Any]:
120120
to_object: dict[str, Any] = {}
121121
if getv(from_object, ["config"]) is not None:
122-
setv(
123-
to_object,
124-
["config"],
125-
_CreateAgentEngineConfig_to_vertex(
126-
getv(from_object, ["config"]), to_object
127-
),
128-
)
122+
_CreateAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object)
129123

130124
return to_object
131125

@@ -141,9 +135,6 @@ def _DeleteAgentEngineRequestParameters_to_vertex(
141135
if getv(from_object, ["force"]) is not None:
142136
setv(to_object, ["force"], getv(from_object, ["force"]))
143137

144-
if getv(from_object, ["config"]) is not None:
145-
setv(to_object, ["config"], getv(from_object, ["config"]))
146-
147138
return to_object
148139

149140

@@ -157,9 +148,6 @@ def _GetAgentEngineOperationParameters_to_vertex(
157148
to_object, ["_url", "operationName"], getv(from_object, ["operation_name"])
158149
)
159150

160-
if getv(from_object, ["config"]) is not None:
161-
setv(to_object, ["config"], getv(from_object, ["config"]))
162-
163151
return to_object
164152

165153

@@ -171,9 +159,6 @@ def _GetAgentEngineRequestParameters_to_vertex(
171159
if getv(from_object, ["name"]) is not None:
172160
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
173161

174-
if getv(from_object, ["config"]) is not None:
175-
setv(to_object, ["config"], getv(from_object, ["config"]))
176-
177162
return to_object
178163

179164

@@ -201,11 +186,7 @@ def _ListAgentEngineRequestParameters_to_vertex(
201186
) -> dict[str, Any]:
202187
to_object: dict[str, Any] = {}
203188
if getv(from_object, ["config"]) is not None:
204-
setv(
205-
to_object,
206-
["config"],
207-
_ListAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object),
208-
)
189+
_ListAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object)
209190

210191
return to_object
211192

@@ -237,11 +218,7 @@ def _QueryAgentEngineRequestParameters_to_vertex(
237218
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
238219

239220
if getv(from_object, ["config"]) is not None:
240-
setv(
241-
to_object,
242-
["config"],
243-
_QueryAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object),
244-
)
221+
_QueryAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object)
245222

246223
return to_object
247224

@@ -324,13 +301,7 @@ def _UpdateAgentEngineRequestParameters_to_vertex(
324301
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
325302

326303
if getv(from_object, ["config"]) is not None:
327-
setv(
328-
to_object,
329-
["config"],
330-
_UpdateAgentEngineConfig_to_vertex(
331-
getv(from_object, ["config"]), to_object
332-
),
333-
)
304+
_UpdateAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object)
334305

335306
return to_object
336307

‎vertexai/_genai/memories.py‎

Lines changed: 8 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -108,13 +108,7 @@ def _CreateAgentEngineMemoryRequestParameters_to_vertex(
108108
setv(to_object, ["scope"], getv(from_object, ["scope"]))
109109

110110
if getv(from_object, ["config"]) is not None:
111-
setv(
112-
to_object,
113-
["config"],
114-
_AgentEngineMemoryConfig_to_vertex(
115-
getv(from_object, ["config"]), to_object
116-
),
117-
)
111+
_AgentEngineMemoryConfig_to_vertex(getv(from_object, ["config"]), to_object)
118112

119113
return to_object
120114

@@ -127,9 +121,6 @@ def _DeleteAgentEngineMemoryRequestParameters_to_vertex(
127121
if getv(from_object, ["name"]) is not None:
128122
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
129123

130-
if getv(from_object, ["config"]) is not None:
131-
setv(to_object, ["config"], getv(from_object, ["config"]))
132-
133124
return to_object
134125

135126

@@ -216,12 +207,8 @@ def _GenerateAgentEngineMemoriesRequestParameters_to_vertex(
216207
setv(to_object, ["scope"], getv(from_object, ["scope"]))
217208

218209
if getv(from_object, ["config"]) is not None:
219-
setv(
220-
to_object,
221-
["config"],
222-
_GenerateAgentEngineMemoriesConfig_to_vertex(
223-
getv(from_object, ["config"]), to_object
224-
),
210+
_GenerateAgentEngineMemoriesConfig_to_vertex(
211+
getv(from_object, ["config"]), to_object
225212
)
226213

227214
return to_object
@@ -237,9 +224,6 @@ def _GetAgentEngineGenerateMemoriesOperationParameters_to_vertex(
237224
to_object, ["_url", "operationName"], getv(from_object, ["operation_name"])
238225
)
239226

240-
if getv(from_object, ["config"]) is not None:
241-
setv(to_object, ["config"], getv(from_object, ["config"]))
242-
243227
return to_object
244228

245229

@@ -253,9 +237,6 @@ def _GetAgentEngineMemoryOperationParameters_to_vertex(
253237
to_object, ["_url", "operationName"], getv(from_object, ["operation_name"])
254238
)
255239

256-
if getv(from_object, ["config"]) is not None:
257-
setv(to_object, ["config"], getv(from_object, ["config"]))
258-
259240
return to_object
260241

261242

@@ -267,9 +248,6 @@ def _GetAgentEngineMemoryRequestParameters_to_vertex(
267248
if getv(from_object, ["name"]) is not None:
268249
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
269250

270-
if getv(from_object, ["config"]) is not None:
271-
setv(to_object, ["config"], getv(from_object, ["config"]))
272-
273251
return to_object
274252

275253

@@ -303,13 +281,7 @@ def _ListAgentEngineMemoryRequestParameters_to_vertex(
303281
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
304282

305283
if getv(from_object, ["config"]) is not None:
306-
setv(
307-
to_object,
308-
["config"],
309-
_ListAgentEngineMemoryConfig_to_vertex(
310-
getv(from_object, ["config"]), to_object
311-
),
312-
)
284+
_ListAgentEngineMemoryConfig_to_vertex(getv(from_object, ["config"]), to_object)
313285

314286
return to_object
315287

@@ -335,9 +307,6 @@ def _PurgeAgentEngineMemoriesRequestParameters_to_vertex(
335307
if getv(from_object, ["force"]) is not None:
336308
setv(to_object, ["force"], getv(from_object, ["force"]))
337309

338-
if getv(from_object, ["config"]) is not None:
339-
setv(to_object, ["config"], getv(from_object, ["config"]))
340-
341310
return to_object
342311

343312

@@ -386,12 +355,8 @@ def _RetrieveAgentEngineMemoriesRequestParameters_to_vertex(
386355
)
387356

388357
if getv(from_object, ["config"]) is not None:
389-
setv(
390-
to_object,
391-
["config"],
392-
_RetrieveAgentEngineMemoriesConfig_to_vertex(
393-
getv(from_object, ["config"]), to_object
394-
),
358+
_RetrieveAgentEngineMemoriesConfig_to_vertex(
359+
getv(from_object, ["config"]), to_object
395360
)
396361

397362
return to_object
@@ -408,9 +373,6 @@ def _RollbackAgentEngineMemoryRequestParameters_to_vertex(
408373
if getv(from_object, ["target_revision_id"]) is not None:
409374
setv(to_object, ["targetRevisionId"], getv(from_object, ["target_revision_id"]))
410375

411-
if getv(from_object, ["config"]) is not None:
412-
setv(to_object, ["config"], getv(from_object, ["config"]))
413-
414376
return to_object
415377

416378

@@ -484,12 +446,8 @@ def _UpdateAgentEngineMemoryRequestParameters_to_vertex(
484446
setv(to_object, ["scope"], getv(from_object, ["scope"]))
485447

486448
if getv(from_object, ["config"]) is not None:
487-
setv(
488-
to_object,
489-
["config"],
490-
_UpdateAgentEngineMemoryConfig_to_vertex(
491-
getv(from_object, ["config"]), to_object
492-
),
449+
_UpdateAgentEngineMemoryConfig_to_vertex(
450+
getv(from_object, ["config"]), to_object
493451
)
494452

495453
return to_object

‎vertexai/_genai/memory_revisions.py‎

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,6 @@ def _GetAgentEngineMemoryRevisionRequestParameters_to_vertex(
4343
if getv(from_object, ["name"]) is not None:
4444
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
4545

46-
if getv(from_object, ["config"]) is not None:
47-
setv(to_object, ["config"], getv(from_object, ["config"]))
48-
4946
return to_object
5047

5148

@@ -76,12 +73,8 @@ def _ListAgentEngineMemoryRevisionsRequestParameters_to_vertex(
7673
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
7774

7875
if getv(from_object, ["config"]) is not None:
79-
setv(
80-
to_object,
81-
["config"],
82-
_ListAgentEngineMemoryRevisionsConfig_to_vertex(
83-
getv(from_object, ["config"]), to_object
84-
),
76+
_ListAgentEngineMemoryRevisionsConfig_to_vertex(
77+
getv(from_object, ["config"]), to_object
8578
)
8679

8780
return to_object

‎vertexai/_genai/sandboxes.py‎

Lines changed: 4 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -72,12 +72,8 @@ def _CreateAgentEngineSandboxRequestParameters_to_vertex(
7272
setv(to_object, ["spec"], getv(from_object, ["spec"]))
7373

7474
if getv(from_object, ["config"]) is not None:
75-
setv(
76-
to_object,
77-
["config"],
78-
_CreateAgentEngineSandboxConfig_to_vertex(
79-
getv(from_object, ["config"]), to_object
80-
),
75+
_CreateAgentEngineSandboxConfig_to_vertex(
76+
getv(from_object, ["config"]), to_object
8177
)
8278

8379
return to_object
@@ -91,9 +87,6 @@ def _DeleteAgentEngineSandboxRequestParameters_to_vertex(
9187
if getv(from_object, ["name"]) is not None:
9288
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
9389

94-
if getv(from_object, ["config"]) is not None:
95-
setv(to_object, ["config"], getv(from_object, ["config"]))
96-
9790
return to_object
9891

9992

@@ -108,9 +101,6 @@ def _ExecuteCodeAgentEngineSandboxRequestParameters_to_vertex(
108101
if getv(from_object, ["inputs"]) is not None:
109102
setv(to_object, ["inputs"], [item for item in getv(from_object, ["inputs"])])
110103

111-
if getv(from_object, ["config"]) is not None:
112-
setv(to_object, ["config"], getv(from_object, ["config"]))
113-
114104
return to_object
115105

116106

@@ -124,9 +114,6 @@ def _GetAgentEngineSandboxOperationParameters_to_vertex(
124114
to_object, ["_url", "operationName"], getv(from_object, ["operation_name"])
125115
)
126116

127-
if getv(from_object, ["config"]) is not None:
128-
setv(to_object, ["config"], getv(from_object, ["config"]))
129-
130117
return to_object
131118

132119

@@ -138,9 +125,6 @@ def _GetAgentEngineSandboxRequestParameters_to_vertex(
138125
if getv(from_object, ["name"]) is not None:
139126
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
140127

141-
if getv(from_object, ["config"]) is not None:
142-
setv(to_object, ["config"], getv(from_object, ["config"]))
143-
144128
return to_object
145129

146130

@@ -171,12 +155,8 @@ def _ListAgentEngineSandboxesRequestParameters_to_vertex(
171155
setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
172156

173157
if getv(from_object, ["config"]) is not None:
174-
setv(
175-
to_object,
176-
["config"],
177-
_ListAgentEngineSandboxesConfig_to_vertex(
178-
getv(from_object, ["config"]), to_object
179-
),
158+
_ListAgentEngineSandboxesConfig_to_vertex(
159+
getv(from_object, ["config"]), to_object
180160
)
181161

182162
return to_object

0 commit comments

Comments
 (0)