google-genai 1.61.0__py3-none-any.whl → 1.62.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
google/genai/models.py CHANGED
@@ -57,6 +57,7 @@ def _VideoGenerationReferenceType_to_mldev_enum_validate(
57
57
  def _Blob_to_mldev(
58
58
  from_object: Union[dict[str, Any], object],
59
59
  parent_object: Optional[dict[str, Any]] = None,
60
+ root_object: Optional[Union[dict[str, Any], object]] = None,
60
61
  ) -> dict[str, Any]:
61
62
  to_object: dict[str, Any] = {}
62
63
  if getv(from_object, ['data']) is not None:
@@ -74,6 +75,7 @@ def _Blob_to_mldev(
74
75
  def _Candidate_from_mldev(
75
76
  from_object: Union[dict[str, Any], object],
76
77
  parent_object: Optional[dict[str, Any]] = None,
78
+ root_object: Optional[Union[dict[str, Any], object]] = None,
77
79
  ) -> dict[str, Any]:
78
80
  to_object: dict[str, Any] = {}
79
81
  if getv(from_object, ['content']) is not None:
@@ -84,7 +86,7 @@ def _Candidate_from_mldev(
84
86
  to_object,
85
87
  ['citation_metadata'],
86
88
  _CitationMetadata_from_mldev(
87
- getv(from_object, ['citationMetadata']), to_object
89
+ getv(from_object, ['citationMetadata']), to_object, root_object
88
90
  ),
89
91
  )
90
92
 
@@ -130,6 +132,7 @@ def _Candidate_from_mldev(
130
132
  def _CitationMetadata_from_mldev(
131
133
  from_object: Union[dict[str, Any], object],
132
134
  parent_object: Optional[dict[str, Any]] = None,
135
+ root_object: Optional[Union[dict[str, Any], object]] = None,
133
136
  ) -> dict[str, Any]:
134
137
  to_object: dict[str, Any] = {}
135
138
  if getv(from_object, ['citationSources']) is not None:
@@ -146,6 +149,7 @@ def _ComputeTokensParameters_to_vertex(
146
149
  api_client: BaseApiClient,
147
150
  from_object: Union[dict[str, Any], object],
148
151
  parent_object: Optional[dict[str, Any]] = None,
152
+ root_object: Optional[Union[dict[str, Any], object]] = None,
149
153
  ) -> dict[str, Any]:
150
154
  to_object: dict[str, Any] = {}
151
155
  if getv(from_object, ['model']) is not None:
@@ -168,6 +172,7 @@ def _ComputeTokensParameters_to_vertex(
168
172
  def _ComputeTokensResponse_from_vertex(
169
173
  from_object: Union[dict[str, Any], object],
170
174
  parent_object: Optional[dict[str, Any]] = None,
175
+ root_object: Optional[Union[dict[str, Any], object]] = None,
171
176
  ) -> dict[str, Any]:
172
177
  to_object: dict[str, Any] = {}
173
178
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -188,6 +193,7 @@ def _ComputeTokensResponse_from_vertex(
188
193
  def _ContentEmbeddingStatistics_from_vertex(
189
194
  from_object: Union[dict[str, Any], object],
190
195
  parent_object: Optional[dict[str, Any]] = None,
196
+ root_object: Optional[Union[dict[str, Any], object]] = None,
191
197
  ) -> dict[str, Any]:
192
198
  to_object: dict[str, Any] = {}
193
199
  if getv(from_object, ['truncated']) is not None:
@@ -202,6 +208,7 @@ def _ContentEmbeddingStatistics_from_vertex(
202
208
  def _ContentEmbedding_from_vertex(
203
209
  from_object: Union[dict[str, Any], object],
204
210
  parent_object: Optional[dict[str, Any]] = None,
211
+ root_object: Optional[Union[dict[str, Any], object]] = None,
205
212
  ) -> dict[str, Any]:
206
213
  to_object: dict[str, Any] = {}
207
214
  if getv(from_object, ['values']) is not None:
@@ -212,7 +219,7 @@ def _ContentEmbedding_from_vertex(
212
219
  to_object,
213
220
  ['statistics'],
214
221
  _ContentEmbeddingStatistics_from_vertex(
215
- getv(from_object, ['statistics']), to_object
222
+ getv(from_object, ['statistics']), to_object, root_object
216
223
  ),
217
224
  )
218
225
 
@@ -222,6 +229,7 @@ def _ContentEmbedding_from_vertex(
222
229
  def _Content_to_mldev(
223
230
  from_object: Union[dict[str, Any], object],
224
231
  parent_object: Optional[dict[str, Any]] = None,
232
+ root_object: Optional[Union[dict[str, Any], object]] = None,
225
233
  ) -> dict[str, Any]:
226
234
  to_object: dict[str, Any] = {}
227
235
  if getv(from_object, ['parts']) is not None:
@@ -229,7 +237,7 @@ def _Content_to_mldev(
229
237
  to_object,
230
238
  ['parts'],
231
239
  [
232
- _Part_to_mldev(item, to_object)
240
+ _Part_to_mldev(item, to_object, root_object)
233
241
  for item in getv(from_object, ['parts'])
234
242
  ],
235
243
  )
@@ -243,6 +251,7 @@ def _Content_to_mldev(
243
251
  def _ControlReferenceConfig_to_vertex(
244
252
  from_object: Union[dict[str, Any], object],
245
253
  parent_object: Optional[dict[str, Any]] = None,
254
+ root_object: Optional[Union[dict[str, Any], object]] = None,
246
255
  ) -> dict[str, Any]:
247
256
  to_object: dict[str, Any] = {}
248
257
  if getv(from_object, ['control_type']) is not None:
@@ -261,6 +270,7 @@ def _ControlReferenceConfig_to_vertex(
261
270
  def _CountTokensConfig_to_mldev(
262
271
  from_object: Union[dict[str, Any], object],
263
272
  parent_object: Optional[dict[str, Any]] = None,
273
+ root_object: Optional[Union[dict[str, Any], object]] = None,
264
274
  ) -> dict[str, Any]:
265
275
  to_object: dict[str, Any] = {}
266
276
 
@@ -283,6 +293,7 @@ def _CountTokensConfig_to_mldev(
283
293
  def _CountTokensConfig_to_vertex(
284
294
  from_object: Union[dict[str, Any], object],
285
295
  parent_object: Optional[dict[str, Any]] = None,
296
+ root_object: Optional[Union[dict[str, Any], object]] = None,
286
297
  ) -> dict[str, Any]:
287
298
  to_object: dict[str, Any] = {}
288
299
 
@@ -298,7 +309,7 @@ def _CountTokensConfig_to_vertex(
298
309
  parent_object,
299
310
  ['tools'],
300
311
  [
301
- _Tool_to_vertex(item, to_object)
312
+ _Tool_to_vertex(item, to_object, root_object)
302
313
  for item in getv(from_object, ['tools'])
303
314
  ],
304
315
  )
@@ -308,7 +319,7 @@ def _CountTokensConfig_to_vertex(
308
319
  parent_object,
309
320
  ['generationConfig'],
310
321
  _GenerationConfig_to_vertex(
311
- getv(from_object, ['generation_config']), to_object
322
+ getv(from_object, ['generation_config']), to_object, root_object
312
323
  ),
313
324
  )
314
325
 
@@ -319,6 +330,7 @@ def _CountTokensParameters_to_mldev(
319
330
  api_client: BaseApiClient,
320
331
  from_object: Union[dict[str, Any], object],
321
332
  parent_object: Optional[dict[str, Any]] = None,
333
+ root_object: Optional[Union[dict[str, Any], object]] = None,
322
334
  ) -> dict[str, Any]:
323
335
  to_object: dict[str, Any] = {}
324
336
  if getv(from_object, ['model']) is not None:
@@ -333,13 +345,15 @@ def _CountTokensParameters_to_mldev(
333
345
  to_object,
334
346
  ['contents'],
335
347
  [
336
- _Content_to_mldev(item, to_object)
348
+ _Content_to_mldev(item, to_object, root_object)
337
349
  for item in t.t_contents(getv(from_object, ['contents']))
338
350
  ],
339
351
  )
340
352
 
341
353
  if getv(from_object, ['config']) is not None:
342
- _CountTokensConfig_to_mldev(getv(from_object, ['config']), to_object)
354
+ _CountTokensConfig_to_mldev(
355
+ getv(from_object, ['config']), to_object, root_object
356
+ )
343
357
 
344
358
  return to_object
345
359
 
@@ -348,6 +362,7 @@ def _CountTokensParameters_to_vertex(
348
362
  api_client: BaseApiClient,
349
363
  from_object: Union[dict[str, Any], object],
350
364
  parent_object: Optional[dict[str, Any]] = None,
365
+ root_object: Optional[Union[dict[str, Any], object]] = None,
351
366
  ) -> dict[str, Any]:
352
367
  to_object: dict[str, Any] = {}
353
368
  if getv(from_object, ['model']) is not None:
@@ -365,7 +380,9 @@ def _CountTokensParameters_to_vertex(
365
380
  )
366
381
 
367
382
  if getv(from_object, ['config']) is not None:
368
- _CountTokensConfig_to_vertex(getv(from_object, ['config']), to_object)
383
+ _CountTokensConfig_to_vertex(
384
+ getv(from_object, ['config']), to_object, root_object
385
+ )
369
386
 
370
387
  return to_object
371
388
 
@@ -373,6 +390,7 @@ def _CountTokensParameters_to_vertex(
373
390
  def _CountTokensResponse_from_mldev(
374
391
  from_object: Union[dict[str, Any], object],
375
392
  parent_object: Optional[dict[str, Any]] = None,
393
+ root_object: Optional[Union[dict[str, Any], object]] = None,
376
394
  ) -> dict[str, Any]:
377
395
  to_object: dict[str, Any] = {}
378
396
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -396,6 +414,7 @@ def _CountTokensResponse_from_mldev(
396
414
  def _CountTokensResponse_from_vertex(
397
415
  from_object: Union[dict[str, Any], object],
398
416
  parent_object: Optional[dict[str, Any]] = None,
417
+ root_object: Optional[Union[dict[str, Any], object]] = None,
399
418
  ) -> dict[str, Any]:
400
419
  to_object: dict[str, Any] = {}
401
420
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -413,6 +432,7 @@ def _DeleteModelParameters_to_mldev(
413
432
  api_client: BaseApiClient,
414
433
  from_object: Union[dict[str, Any], object],
415
434
  parent_object: Optional[dict[str, Any]] = None,
435
+ root_object: Optional[Union[dict[str, Any], object]] = None,
416
436
  ) -> dict[str, Any]:
417
437
  to_object: dict[str, Any] = {}
418
438
  if getv(from_object, ['model']) is not None:
@@ -429,6 +449,7 @@ def _DeleteModelParameters_to_vertex(
429
449
  api_client: BaseApiClient,
430
450
  from_object: Union[dict[str, Any], object],
431
451
  parent_object: Optional[dict[str, Any]] = None,
452
+ root_object: Optional[Union[dict[str, Any], object]] = None,
432
453
  ) -> dict[str, Any]:
433
454
  to_object: dict[str, Any] = {}
434
455
  if getv(from_object, ['model']) is not None:
@@ -444,6 +465,7 @@ def _DeleteModelParameters_to_vertex(
444
465
  def _DeleteModelResponse_from_mldev(
445
466
  from_object: Union[dict[str, Any], object],
446
467
  parent_object: Optional[dict[str, Any]] = None,
468
+ root_object: Optional[Union[dict[str, Any], object]] = None,
447
469
  ) -> dict[str, Any]:
448
470
  to_object: dict[str, Any] = {}
449
471
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -457,6 +479,7 @@ def _DeleteModelResponse_from_mldev(
457
479
  def _DeleteModelResponse_from_vertex(
458
480
  from_object: Union[dict[str, Any], object],
459
481
  parent_object: Optional[dict[str, Any]] = None,
482
+ root_object: Optional[Union[dict[str, Any], object]] = None,
460
483
  ) -> dict[str, Any]:
461
484
  to_object: dict[str, Any] = {}
462
485
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -470,6 +493,7 @@ def _DeleteModelResponse_from_vertex(
470
493
  def _EditImageConfig_to_vertex(
471
494
  from_object: Union[dict[str, Any], object],
472
495
  parent_object: Optional[dict[str, Any]] = None,
496
+ root_object: Optional[Union[dict[str, Any], object]] = None,
473
497
  ) -> dict[str, Any]:
474
498
  to_object: dict[str, Any] = {}
475
499
 
@@ -591,6 +615,7 @@ def _EditImageParameters_to_vertex(
591
615
  api_client: BaseApiClient,
592
616
  from_object: Union[dict[str, Any], object],
593
617
  parent_object: Optional[dict[str, Any]] = None,
618
+ root_object: Optional[Union[dict[str, Any], object]] = None,
594
619
  ) -> dict[str, Any]:
595
620
  to_object: dict[str, Any] = {}
596
621
  if getv(from_object, ['model']) is not None:
@@ -608,13 +633,15 @@ def _EditImageParameters_to_vertex(
608
633
  to_object,
609
634
  ['instances[0]', 'referenceImages'],
610
635
  [
611
- _ReferenceImageAPI_to_vertex(item, to_object)
636
+ _ReferenceImageAPI_to_vertex(item, to_object, root_object)
612
637
  for item in getv(from_object, ['reference_images'])
613
638
  ],
614
639
  )
615
640
 
616
641
  if getv(from_object, ['config']) is not None:
617
- _EditImageConfig_to_vertex(getv(from_object, ['config']), to_object)
642
+ _EditImageConfig_to_vertex(
643
+ getv(from_object, ['config']), to_object, root_object
644
+ )
618
645
 
619
646
  return to_object
620
647
 
@@ -622,6 +649,7 @@ def _EditImageParameters_to_vertex(
622
649
  def _EditImageResponse_from_vertex(
623
650
  from_object: Union[dict[str, Any], object],
624
651
  parent_object: Optional[dict[str, Any]] = None,
652
+ root_object: Optional[Union[dict[str, Any], object]] = None,
625
653
  ) -> dict[str, Any]:
626
654
  to_object: dict[str, Any] = {}
627
655
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -634,7 +662,7 @@ def _EditImageResponse_from_vertex(
634
662
  to_object,
635
663
  ['generated_images'],
636
664
  [
637
- _GeneratedImage_from_vertex(item, to_object)
665
+ _GeneratedImage_from_vertex(item, to_object, root_object)
638
666
  for item in getv(from_object, ['predictions'])
639
667
  ],
640
668
  )
@@ -645,6 +673,7 @@ def _EditImageResponse_from_vertex(
645
673
  def _EmbedContentConfig_to_mldev(
646
674
  from_object: Union[dict[str, Any], object],
647
675
  parent_object: Optional[dict[str, Any]] = None,
676
+ root_object: Optional[Union[dict[str, Any], object]] = None,
648
677
  ) -> dict[str, Any]:
649
678
  to_object: dict[str, Any] = {}
650
679
 
@@ -677,6 +706,7 @@ def _EmbedContentConfig_to_mldev(
677
706
  def _EmbedContentConfig_to_vertex(
678
707
  from_object: Union[dict[str, Any], object],
679
708
  parent_object: Optional[dict[str, Any]] = None,
709
+ root_object: Optional[Union[dict[str, Any], object]] = None,
680
710
  ) -> dict[str, Any]:
681
711
  to_object: dict[str, Any] = {}
682
712
 
@@ -718,6 +748,7 @@ def _EmbedContentParameters_to_mldev(
718
748
  api_client: BaseApiClient,
719
749
  from_object: Union[dict[str, Any], object],
720
750
  parent_object: Optional[dict[str, Any]] = None,
751
+ root_object: Optional[Union[dict[str, Any], object]] = None,
721
752
  ) -> dict[str, Any]:
722
753
  to_object: dict[str, Any] = {}
723
754
  if getv(from_object, ['model']) is not None:
@@ -740,7 +771,9 @@ def _EmbedContentParameters_to_mldev(
740
771
  )
741
772
 
742
773
  if getv(from_object, ['config']) is not None:
743
- _EmbedContentConfig_to_mldev(getv(from_object, ['config']), to_object)
774
+ _EmbedContentConfig_to_mldev(
775
+ getv(from_object, ['config']), to_object, root_object
776
+ )
744
777
 
745
778
  setv(
746
779
  to_object,
@@ -754,6 +787,7 @@ def _EmbedContentParameters_to_vertex(
754
787
  api_client: BaseApiClient,
755
788
  from_object: Union[dict[str, Any], object],
756
789
  parent_object: Optional[dict[str, Any]] = None,
790
+ root_object: Optional[Union[dict[str, Any], object]] = None,
757
791
  ) -> dict[str, Any]:
758
792
  to_object: dict[str, Any] = {}
759
793
  if getv(from_object, ['model']) is not None:
@@ -776,7 +810,9 @@ def _EmbedContentParameters_to_vertex(
776
810
  )
777
811
 
778
812
  if getv(from_object, ['config']) is not None:
779
- _EmbedContentConfig_to_vertex(getv(from_object, ['config']), to_object)
813
+ _EmbedContentConfig_to_vertex(
814
+ getv(from_object, ['config']), to_object, root_object
815
+ )
780
816
 
781
817
  return to_object
782
818
 
@@ -784,6 +820,7 @@ def _EmbedContentParameters_to_vertex(
784
820
  def _EmbedContentResponse_from_mldev(
785
821
  from_object: Union[dict[str, Any], object],
786
822
  parent_object: Optional[dict[str, Any]] = None,
823
+ root_object: Optional[Union[dict[str, Any], object]] = None,
787
824
  ) -> dict[str, Any]:
788
825
  to_object: dict[str, Any] = {}
789
826
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -807,6 +844,7 @@ def _EmbedContentResponse_from_mldev(
807
844
  def _EmbedContentResponse_from_vertex(
808
845
  from_object: Union[dict[str, Any], object],
809
846
  parent_object: Optional[dict[str, Any]] = None,
847
+ root_object: Optional[Union[dict[str, Any], object]] = None,
810
848
  ) -> dict[str, Any]:
811
849
  to_object: dict[str, Any] = {}
812
850
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -819,7 +857,7 @@ def _EmbedContentResponse_from_vertex(
819
857
  to_object,
820
858
  ['embeddings'],
821
859
  [
822
- _ContentEmbedding_from_vertex(item, to_object)
860
+ _ContentEmbedding_from_vertex(item, to_object, root_object)
823
861
  for item in getv(from_object, ['predictions[]', 'embeddings'])
824
862
  ],
825
863
  )
@@ -833,6 +871,7 @@ def _EmbedContentResponse_from_vertex(
833
871
  def _Endpoint_from_vertex(
834
872
  from_object: Union[dict[str, Any], object],
835
873
  parent_object: Optional[dict[str, Any]] = None,
874
+ root_object: Optional[Union[dict[str, Any], object]] = None,
836
875
  ) -> dict[str, Any]:
837
876
  to_object: dict[str, Any] = {}
838
877
  if getv(from_object, ['endpoint']) is not None:
@@ -849,6 +888,7 @@ def _Endpoint_from_vertex(
849
888
  def _FileData_to_mldev(
850
889
  from_object: Union[dict[str, Any], object],
851
890
  parent_object: Optional[dict[str, Any]] = None,
891
+ root_object: Optional[Union[dict[str, Any], object]] = None,
852
892
  ) -> dict[str, Any]:
853
893
  to_object: dict[str, Any] = {}
854
894
  if getv(from_object, ['display_name']) is not None:
@@ -866,6 +906,7 @@ def _FileData_to_mldev(
866
906
  def _FunctionCall_to_mldev(
867
907
  from_object: Union[dict[str, Any], object],
868
908
  parent_object: Optional[dict[str, Any]] = None,
909
+ root_object: Optional[Union[dict[str, Any], object]] = None,
869
910
  ) -> dict[str, Any]:
870
911
  to_object: dict[str, Any] = {}
871
912
  if getv(from_object, ['id']) is not None:
@@ -889,6 +930,7 @@ def _FunctionCall_to_mldev(
889
930
  def _FunctionCallingConfig_to_mldev(
890
931
  from_object: Union[dict[str, Any], object],
891
932
  parent_object: Optional[dict[str, Any]] = None,
933
+ root_object: Optional[Union[dict[str, Any], object]] = None,
892
934
  ) -> dict[str, Any]:
893
935
  to_object: dict[str, Any] = {}
894
936
  if getv(from_object, ['allowed_function_names']) is not None:
@@ -913,6 +955,7 @@ def _FunctionCallingConfig_to_mldev(
913
955
  def _FunctionDeclaration_to_vertex(
914
956
  from_object: Union[dict[str, Any], object],
915
957
  parent_object: Optional[dict[str, Any]] = None,
958
+ root_object: Optional[Union[dict[str, Any], object]] = None,
916
959
  ) -> dict[str, Any]:
917
960
  to_object: dict[str, Any] = {}
918
961
  if getv(from_object, ['description']) is not None:
@@ -951,6 +994,7 @@ def _GenerateContentConfig_to_mldev(
951
994
  api_client: BaseApiClient,
952
995
  from_object: Union[dict[str, Any], object],
953
996
  parent_object: Optional[dict[str, Any]] = None,
997
+ root_object: Optional[Union[dict[str, Any], object]] = None,
954
998
  ) -> dict[str, Any]:
955
999
  to_object: dict[str, Any] = {}
956
1000
 
@@ -959,7 +1003,9 @@ def _GenerateContentConfig_to_mldev(
959
1003
  parent_object,
960
1004
  ['systemInstruction'],
961
1005
  _Content_to_mldev(
962
- t.t_content(getv(from_object, ['system_instruction'])), to_object
1006
+ t.t_content(getv(from_object, ['system_instruction'])),
1007
+ to_object,
1008
+ root_object,
963
1009
  ),
964
1010
  )
965
1011
 
@@ -1042,7 +1088,7 @@ def _GenerateContentConfig_to_mldev(
1042
1088
  parent_object,
1043
1089
  ['safetySettings'],
1044
1090
  [
1045
- _SafetySetting_to_mldev(item, to_object)
1091
+ _SafetySetting_to_mldev(item, to_object, root_object)
1046
1092
  for item in getv(from_object, ['safety_settings'])
1047
1093
  ],
1048
1094
  )
@@ -1052,7 +1098,7 @@ def _GenerateContentConfig_to_mldev(
1052
1098
  parent_object,
1053
1099
  ['tools'],
1054
1100
  [
1055
- _Tool_to_mldev(t.t_tool(api_client, item), to_object)
1101
+ _Tool_to_mldev(t.t_tool(api_client, item), to_object, root_object)
1056
1102
  for item in t.t_tools(api_client, getv(from_object, ['tools']))
1057
1103
  ],
1058
1104
  )
@@ -1061,7 +1107,9 @@ def _GenerateContentConfig_to_mldev(
1061
1107
  setv(
1062
1108
  parent_object,
1063
1109
  ['toolConfig'],
1064
- _ToolConfig_to_mldev(getv(from_object, ['tool_config']), to_object),
1110
+ _ToolConfig_to_mldev(
1111
+ getv(from_object, ['tool_config']), to_object, root_object
1112
+ ),
1065
1113
  )
1066
1114
 
1067
1115
  if getv(from_object, ['labels']) is not None:
@@ -1107,7 +1155,9 @@ def _GenerateContentConfig_to_mldev(
1107
1155
  setv(
1108
1156
  to_object,
1109
1157
  ['imageConfig'],
1110
- _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
1158
+ _ImageConfig_to_mldev(
1159
+ getv(from_object, ['image_config']), to_object, root_object
1160
+ ),
1111
1161
  )
1112
1162
 
1113
1163
  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
@@ -1129,6 +1179,7 @@ def _GenerateContentConfig_to_vertex(
1129
1179
  api_client: BaseApiClient,
1130
1180
  from_object: Union[dict[str, Any], object],
1131
1181
  parent_object: Optional[dict[str, Any]] = None,
1182
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1132
1183
  ) -> dict[str, Any]:
1133
1184
  to_object: dict[str, Any] = {}
1134
1185
 
@@ -1227,7 +1278,7 @@ def _GenerateContentConfig_to_vertex(
1227
1278
  parent_object,
1228
1279
  ['tools'],
1229
1280
  [
1230
- _Tool_to_vertex(t.t_tool(api_client, item), to_object)
1281
+ _Tool_to_vertex(t.t_tool(api_client, item), to_object, root_object)
1231
1282
  for item in t.t_tools(api_client, getv(from_object, ['tools']))
1232
1283
  ],
1233
1284
  )
@@ -1276,7 +1327,9 @@ def _GenerateContentConfig_to_vertex(
1276
1327
  setv(
1277
1328
  to_object,
1278
1329
  ['imageConfig'],
1279
- _ImageConfig_to_vertex(getv(from_object, ['image_config']), to_object),
1330
+ _ImageConfig_to_vertex(
1331
+ getv(from_object, ['image_config']), to_object, root_object
1332
+ ),
1280
1333
  )
1281
1334
 
1282
1335
  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
@@ -1298,6 +1351,7 @@ def _GenerateContentParameters_to_mldev(
1298
1351
  api_client: BaseApiClient,
1299
1352
  from_object: Union[dict[str, Any], object],
1300
1353
  parent_object: Optional[dict[str, Any]] = None,
1354
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1301
1355
  ) -> dict[str, Any]:
1302
1356
  to_object: dict[str, Any] = {}
1303
1357
  if getv(from_object, ['model']) is not None:
@@ -1312,7 +1366,7 @@ def _GenerateContentParameters_to_mldev(
1312
1366
  to_object,
1313
1367
  ['contents'],
1314
1368
  [
1315
- _Content_to_mldev(item, to_object)
1369
+ _Content_to_mldev(item, to_object, root_object)
1316
1370
  for item in t.t_contents(getv(from_object, ['contents']))
1317
1371
  ],
1318
1372
  )
@@ -1322,7 +1376,7 @@ def _GenerateContentParameters_to_mldev(
1322
1376
  to_object,
1323
1377
  ['generationConfig'],
1324
1378
  _GenerateContentConfig_to_mldev(
1325
- api_client, getv(from_object, ['config']), to_object
1379
+ api_client, getv(from_object, ['config']), to_object, root_object
1326
1380
  ),
1327
1381
  )
1328
1382
 
@@ -1333,6 +1387,7 @@ def _GenerateContentParameters_to_vertex(
1333
1387
  api_client: BaseApiClient,
1334
1388
  from_object: Union[dict[str, Any], object],
1335
1389
  parent_object: Optional[dict[str, Any]] = None,
1390
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1336
1391
  ) -> dict[str, Any]:
1337
1392
  to_object: dict[str, Any] = {}
1338
1393
  if getv(from_object, ['model']) is not None:
@@ -1354,7 +1409,7 @@ def _GenerateContentParameters_to_vertex(
1354
1409
  to_object,
1355
1410
  ['generationConfig'],
1356
1411
  _GenerateContentConfig_to_vertex(
1357
- api_client, getv(from_object, ['config']), to_object
1412
+ api_client, getv(from_object, ['config']), to_object, root_object
1358
1413
  ),
1359
1414
  )
1360
1415
 
@@ -1364,6 +1419,7 @@ def _GenerateContentParameters_to_vertex(
1364
1419
  def _GenerateContentResponse_from_mldev(
1365
1420
  from_object: Union[dict[str, Any], object],
1366
1421
  parent_object: Optional[dict[str, Any]] = None,
1422
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1367
1423
  ) -> dict[str, Any]:
1368
1424
  to_object: dict[str, Any] = {}
1369
1425
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -1376,7 +1432,7 @@ def _GenerateContentResponse_from_mldev(
1376
1432
  to_object,
1377
1433
  ['candidates'],
1378
1434
  [
1379
- _Candidate_from_mldev(item, to_object)
1435
+ _Candidate_from_mldev(item, to_object, root_object)
1380
1436
  for item in getv(from_object, ['candidates'])
1381
1437
  ],
1382
1438
  )
@@ -1399,6 +1455,7 @@ def _GenerateContentResponse_from_mldev(
1399
1455
  def _GenerateContentResponse_from_vertex(
1400
1456
  from_object: Union[dict[str, Any], object],
1401
1457
  parent_object: Optional[dict[str, Any]] = None,
1458
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1402
1459
  ) -> dict[str, Any]:
1403
1460
  to_object: dict[str, Any] = {}
1404
1461
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -1434,6 +1491,7 @@ def _GenerateContentResponse_from_vertex(
1434
1491
  def _GenerateImagesConfig_to_mldev(
1435
1492
  from_object: Union[dict[str, Any], object],
1436
1493
  parent_object: Optional[dict[str, Any]] = None,
1494
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1437
1495
  ) -> dict[str, Any]:
1438
1496
  to_object: dict[str, Any] = {}
1439
1497
 
@@ -1546,6 +1604,7 @@ def _GenerateImagesConfig_to_mldev(
1546
1604
  def _GenerateImagesConfig_to_vertex(
1547
1605
  from_object: Union[dict[str, Any], object],
1548
1606
  parent_object: Optional[dict[str, Any]] = None,
1607
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1549
1608
  ) -> dict[str, Any]:
1550
1609
  to_object: dict[str, Any] = {}
1551
1610
 
@@ -1667,6 +1726,7 @@ def _GenerateImagesParameters_to_mldev(
1667
1726
  api_client: BaseApiClient,
1668
1727
  from_object: Union[dict[str, Any], object],
1669
1728
  parent_object: Optional[dict[str, Any]] = None,
1729
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1670
1730
  ) -> dict[str, Any]:
1671
1731
  to_object: dict[str, Any] = {}
1672
1732
  if getv(from_object, ['model']) is not None:
@@ -1680,7 +1740,9 @@ def _GenerateImagesParameters_to_mldev(
1680
1740
  setv(to_object, ['instances[0]', 'prompt'], getv(from_object, ['prompt']))
1681
1741
 
1682
1742
  if getv(from_object, ['config']) is not None:
1683
- _GenerateImagesConfig_to_mldev(getv(from_object, ['config']), to_object)
1743
+ _GenerateImagesConfig_to_mldev(
1744
+ getv(from_object, ['config']), to_object, root_object
1745
+ )
1684
1746
 
1685
1747
  return to_object
1686
1748
 
@@ -1689,6 +1751,7 @@ def _GenerateImagesParameters_to_vertex(
1689
1751
  api_client: BaseApiClient,
1690
1752
  from_object: Union[dict[str, Any], object],
1691
1753
  parent_object: Optional[dict[str, Any]] = None,
1754
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1692
1755
  ) -> dict[str, Any]:
1693
1756
  to_object: dict[str, Any] = {}
1694
1757
  if getv(from_object, ['model']) is not None:
@@ -1702,7 +1765,9 @@ def _GenerateImagesParameters_to_vertex(
1702
1765
  setv(to_object, ['instances[0]', 'prompt'], getv(from_object, ['prompt']))
1703
1766
 
1704
1767
  if getv(from_object, ['config']) is not None:
1705
- _GenerateImagesConfig_to_vertex(getv(from_object, ['config']), to_object)
1768
+ _GenerateImagesConfig_to_vertex(
1769
+ getv(from_object, ['config']), to_object, root_object
1770
+ )
1706
1771
 
1707
1772
  return to_object
1708
1773
 
@@ -1710,6 +1775,7 @@ def _GenerateImagesParameters_to_vertex(
1710
1775
  def _GenerateImagesResponse_from_mldev(
1711
1776
  from_object: Union[dict[str, Any], object],
1712
1777
  parent_object: Optional[dict[str, Any]] = None,
1778
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1713
1779
  ) -> dict[str, Any]:
1714
1780
  to_object: dict[str, Any] = {}
1715
1781
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -1722,7 +1788,7 @@ def _GenerateImagesResponse_from_mldev(
1722
1788
  to_object,
1723
1789
  ['generated_images'],
1724
1790
  [
1725
- _GeneratedImage_from_mldev(item, to_object)
1791
+ _GeneratedImage_from_mldev(item, to_object, root_object)
1726
1792
  for item in getv(from_object, ['predictions'])
1727
1793
  ],
1728
1794
  )
@@ -1732,7 +1798,9 @@ def _GenerateImagesResponse_from_mldev(
1732
1798
  to_object,
1733
1799
  ['positive_prompt_safety_attributes'],
1734
1800
  _SafetyAttributes_from_mldev(
1735
- getv(from_object, ['positivePromptSafetyAttributes']), to_object
1801
+ getv(from_object, ['positivePromptSafetyAttributes']),
1802
+ to_object,
1803
+ root_object,
1736
1804
  ),
1737
1805
  )
1738
1806
 
@@ -1742,6 +1810,7 @@ def _GenerateImagesResponse_from_mldev(
1742
1810
  def _GenerateImagesResponse_from_vertex(
1743
1811
  from_object: Union[dict[str, Any], object],
1744
1812
  parent_object: Optional[dict[str, Any]] = None,
1813
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1745
1814
  ) -> dict[str, Any]:
1746
1815
  to_object: dict[str, Any] = {}
1747
1816
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -1754,7 +1823,7 @@ def _GenerateImagesResponse_from_vertex(
1754
1823
  to_object,
1755
1824
  ['generated_images'],
1756
1825
  [
1757
- _GeneratedImage_from_vertex(item, to_object)
1826
+ _GeneratedImage_from_vertex(item, to_object, root_object)
1758
1827
  for item in getv(from_object, ['predictions'])
1759
1828
  ],
1760
1829
  )
@@ -1764,7 +1833,9 @@ def _GenerateImagesResponse_from_vertex(
1764
1833
  to_object,
1765
1834
  ['positive_prompt_safety_attributes'],
1766
1835
  _SafetyAttributes_from_vertex(
1767
- getv(from_object, ['positivePromptSafetyAttributes']), to_object
1836
+ getv(from_object, ['positivePromptSafetyAttributes']),
1837
+ to_object,
1838
+ root_object,
1768
1839
  ),
1769
1840
  )
1770
1841
 
@@ -1774,6 +1845,7 @@ def _GenerateImagesResponse_from_vertex(
1774
1845
  def _GenerateVideosConfig_to_mldev(
1775
1846
  from_object: Union[dict[str, Any], object],
1776
1847
  parent_object: Optional[dict[str, Any]] = None,
1848
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1777
1849
  ) -> dict[str, Any]:
1778
1850
  to_object: dict[str, Any] = {}
1779
1851
 
@@ -1845,7 +1917,9 @@ def _GenerateVideosConfig_to_mldev(
1845
1917
  setv(
1846
1918
  parent_object,
1847
1919
  ['instances[0]', 'lastFrame'],
1848
- _Image_to_mldev(getv(from_object, ['last_frame']), to_object),
1920
+ _Image_to_mldev(
1921
+ getv(from_object, ['last_frame']), to_object, root_object
1922
+ ),
1849
1923
  )
1850
1924
 
1851
1925
  if getv(from_object, ['reference_images']) is not None:
@@ -1853,7 +1927,9 @@ def _GenerateVideosConfig_to_mldev(
1853
1927
  parent_object,
1854
1928
  ['instances[0]', 'referenceImages'],
1855
1929
  [
1856
- _VideoGenerationReferenceImage_to_mldev(item, to_object)
1930
+ _VideoGenerationReferenceImage_to_mldev(
1931
+ item, to_object, root_object
1932
+ )
1857
1933
  for item in getv(from_object, ['reference_images'])
1858
1934
  ],
1859
1935
  )
@@ -1872,6 +1948,7 @@ def _GenerateVideosConfig_to_mldev(
1872
1948
  def _GenerateVideosConfig_to_vertex(
1873
1949
  from_object: Union[dict[str, Any], object],
1874
1950
  parent_object: Optional[dict[str, Any]] = None,
1951
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1875
1952
  ) -> dict[str, Any]:
1876
1953
  to_object: dict[str, Any] = {}
1877
1954
 
@@ -1955,7 +2032,9 @@ def _GenerateVideosConfig_to_vertex(
1955
2032
  setv(
1956
2033
  parent_object,
1957
2034
  ['instances[0]', 'lastFrame'],
1958
- _Image_to_vertex(getv(from_object, ['last_frame']), to_object),
2035
+ _Image_to_vertex(
2036
+ getv(from_object, ['last_frame']), to_object, root_object
2037
+ ),
1959
2038
  )
1960
2039
 
1961
2040
  if getv(from_object, ['reference_images']) is not None:
@@ -1963,7 +2042,9 @@ def _GenerateVideosConfig_to_vertex(
1963
2042
  parent_object,
1964
2043
  ['instances[0]', 'referenceImages'],
1965
2044
  [
1966
- _VideoGenerationReferenceImage_to_vertex(item, to_object)
2045
+ _VideoGenerationReferenceImage_to_vertex(
2046
+ item, to_object, root_object
2047
+ )
1967
2048
  for item in getv(from_object, ['reference_images'])
1968
2049
  ],
1969
2050
  )
@@ -1972,7 +2053,9 @@ def _GenerateVideosConfig_to_vertex(
1972
2053
  setv(
1973
2054
  parent_object,
1974
2055
  ['instances[0]', 'mask'],
1975
- _VideoGenerationMask_to_vertex(getv(from_object, ['mask']), to_object),
2056
+ _VideoGenerationMask_to_vertex(
2057
+ getv(from_object, ['mask']), to_object, root_object
2058
+ ),
1976
2059
  )
1977
2060
 
1978
2061
  if getv(from_object, ['compression_quality']) is not None:
@@ -1988,6 +2071,7 @@ def _GenerateVideosConfig_to_vertex(
1988
2071
  def _GenerateVideosOperation_from_mldev(
1989
2072
  from_object: Union[dict[str, Any], object],
1990
2073
  parent_object: Optional[dict[str, Any]] = None,
2074
+ root_object: Optional[Union[dict[str, Any], object]] = None,
1991
2075
  ) -> dict[str, Any]:
1992
2076
  to_object: dict[str, Any] = {}
1993
2077
  if getv(from_object, ['name']) is not None:
@@ -2007,7 +2091,9 @@ def _GenerateVideosOperation_from_mldev(
2007
2091
  to_object,
2008
2092
  ['response'],
2009
2093
  _GenerateVideosResponse_from_mldev(
2010
- getv(from_object, ['response', 'generateVideoResponse']), to_object
2094
+ getv(from_object, ['response', 'generateVideoResponse']),
2095
+ to_object,
2096
+ root_object,
2011
2097
  ),
2012
2098
  )
2013
2099
 
@@ -2016,7 +2102,9 @@ def _GenerateVideosOperation_from_mldev(
2016
2102
  to_object,
2017
2103
  ['result'],
2018
2104
  _GenerateVideosResponse_from_mldev(
2019
- getv(from_object, ['response', 'generateVideoResponse']), to_object
2105
+ getv(from_object, ['response', 'generateVideoResponse']),
2106
+ to_object,
2107
+ root_object,
2020
2108
  ),
2021
2109
  )
2022
2110
 
@@ -2026,6 +2114,7 @@ def _GenerateVideosOperation_from_mldev(
2026
2114
  def _GenerateVideosOperation_from_vertex(
2027
2115
  from_object: Union[dict[str, Any], object],
2028
2116
  parent_object: Optional[dict[str, Any]] = None,
2117
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2029
2118
  ) -> dict[str, Any]:
2030
2119
  to_object: dict[str, Any] = {}
2031
2120
  if getv(from_object, ['name']) is not None:
@@ -2045,7 +2134,7 @@ def _GenerateVideosOperation_from_vertex(
2045
2134
  to_object,
2046
2135
  ['response'],
2047
2136
  _GenerateVideosResponse_from_vertex(
2048
- getv(from_object, ['response']), to_object
2137
+ getv(from_object, ['response']), to_object, root_object
2049
2138
  ),
2050
2139
  )
2051
2140
 
@@ -2054,7 +2143,7 @@ def _GenerateVideosOperation_from_vertex(
2054
2143
  to_object,
2055
2144
  ['result'],
2056
2145
  _GenerateVideosResponse_from_vertex(
2057
- getv(from_object, ['response']), to_object
2146
+ getv(from_object, ['response']), to_object, root_object
2058
2147
  ),
2059
2148
  )
2060
2149
 
@@ -2065,6 +2154,7 @@ def _GenerateVideosParameters_to_mldev(
2065
2154
  api_client: BaseApiClient,
2066
2155
  from_object: Union[dict[str, Any], object],
2067
2156
  parent_object: Optional[dict[str, Any]] = None,
2157
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2068
2158
  ) -> dict[str, Any]:
2069
2159
  to_object: dict[str, Any] = {}
2070
2160
  if getv(from_object, ['model']) is not None:
@@ -2081,21 +2171,25 @@ def _GenerateVideosParameters_to_mldev(
2081
2171
  setv(
2082
2172
  to_object,
2083
2173
  ['instances[0]', 'image'],
2084
- _Image_to_mldev(getv(from_object, ['image']), to_object),
2174
+ _Image_to_mldev(getv(from_object, ['image']), to_object, root_object),
2085
2175
  )
2086
2176
 
2087
2177
  if getv(from_object, ['video']) is not None:
2088
2178
  setv(
2089
2179
  to_object,
2090
2180
  ['instances[0]', 'video'],
2091
- _Video_to_mldev(getv(from_object, ['video']), to_object),
2181
+ _Video_to_mldev(getv(from_object, ['video']), to_object, root_object),
2092
2182
  )
2093
2183
 
2094
2184
  if getv(from_object, ['source']) is not None:
2095
- _GenerateVideosSource_to_mldev(getv(from_object, ['source']), to_object)
2185
+ _GenerateVideosSource_to_mldev(
2186
+ getv(from_object, ['source']), to_object, root_object
2187
+ )
2096
2188
 
2097
2189
  if getv(from_object, ['config']) is not None:
2098
- _GenerateVideosConfig_to_mldev(getv(from_object, ['config']), to_object)
2190
+ _GenerateVideosConfig_to_mldev(
2191
+ getv(from_object, ['config']), to_object, root_object
2192
+ )
2099
2193
 
2100
2194
  return to_object
2101
2195
 
@@ -2104,6 +2198,7 @@ def _GenerateVideosParameters_to_vertex(
2104
2198
  api_client: BaseApiClient,
2105
2199
  from_object: Union[dict[str, Any], object],
2106
2200
  parent_object: Optional[dict[str, Any]] = None,
2201
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2107
2202
  ) -> dict[str, Any]:
2108
2203
  to_object: dict[str, Any] = {}
2109
2204
  if getv(from_object, ['model']) is not None:
@@ -2120,21 +2215,25 @@ def _GenerateVideosParameters_to_vertex(
2120
2215
  setv(
2121
2216
  to_object,
2122
2217
  ['instances[0]', 'image'],
2123
- _Image_to_vertex(getv(from_object, ['image']), to_object),
2218
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
2124
2219
  )
2125
2220
 
2126
2221
  if getv(from_object, ['video']) is not None:
2127
2222
  setv(
2128
2223
  to_object,
2129
2224
  ['instances[0]', 'video'],
2130
- _Video_to_vertex(getv(from_object, ['video']), to_object),
2225
+ _Video_to_vertex(getv(from_object, ['video']), to_object, root_object),
2131
2226
  )
2132
2227
 
2133
2228
  if getv(from_object, ['source']) is not None:
2134
- _GenerateVideosSource_to_vertex(getv(from_object, ['source']), to_object)
2229
+ _GenerateVideosSource_to_vertex(
2230
+ getv(from_object, ['source']), to_object, root_object
2231
+ )
2135
2232
 
2136
2233
  if getv(from_object, ['config']) is not None:
2137
- _GenerateVideosConfig_to_vertex(getv(from_object, ['config']), to_object)
2234
+ _GenerateVideosConfig_to_vertex(
2235
+ getv(from_object, ['config']), to_object, root_object
2236
+ )
2138
2237
 
2139
2238
  return to_object
2140
2239
 
@@ -2142,6 +2241,7 @@ def _GenerateVideosParameters_to_vertex(
2142
2241
  def _GenerateVideosResponse_from_mldev(
2143
2242
  from_object: Union[dict[str, Any], object],
2144
2243
  parent_object: Optional[dict[str, Any]] = None,
2244
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2145
2245
  ) -> dict[str, Any]:
2146
2246
  to_object: dict[str, Any] = {}
2147
2247
  if getv(from_object, ['generatedSamples']) is not None:
@@ -2149,7 +2249,7 @@ def _GenerateVideosResponse_from_mldev(
2149
2249
  to_object,
2150
2250
  ['generated_videos'],
2151
2251
  [
2152
- _GeneratedVideo_from_mldev(item, to_object)
2252
+ _GeneratedVideo_from_mldev(item, to_object, root_object)
2153
2253
  for item in getv(from_object, ['generatedSamples'])
2154
2254
  ],
2155
2255
  )
@@ -2174,6 +2274,7 @@ def _GenerateVideosResponse_from_mldev(
2174
2274
  def _GenerateVideosResponse_from_vertex(
2175
2275
  from_object: Union[dict[str, Any], object],
2176
2276
  parent_object: Optional[dict[str, Any]] = None,
2277
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2177
2278
  ) -> dict[str, Any]:
2178
2279
  to_object: dict[str, Any] = {}
2179
2280
  if getv(from_object, ['videos']) is not None:
@@ -2181,7 +2282,7 @@ def _GenerateVideosResponse_from_vertex(
2181
2282
  to_object,
2182
2283
  ['generated_videos'],
2183
2284
  [
2184
- _GeneratedVideo_from_vertex(item, to_object)
2285
+ _GeneratedVideo_from_vertex(item, to_object, root_object)
2185
2286
  for item in getv(from_object, ['videos'])
2186
2287
  ],
2187
2288
  )
@@ -2206,6 +2307,7 @@ def _GenerateVideosResponse_from_vertex(
2206
2307
  def _GenerateVideosSource_to_mldev(
2207
2308
  from_object: Union[dict[str, Any], object],
2208
2309
  parent_object: Optional[dict[str, Any]] = None,
2310
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2209
2311
  ) -> dict[str, Any]:
2210
2312
  to_object: dict[str, Any] = {}
2211
2313
  if getv(from_object, ['prompt']) is not None:
@@ -2217,14 +2319,14 @@ def _GenerateVideosSource_to_mldev(
2217
2319
  setv(
2218
2320
  parent_object,
2219
2321
  ['instances[0]', 'image'],
2220
- _Image_to_mldev(getv(from_object, ['image']), to_object),
2322
+ _Image_to_mldev(getv(from_object, ['image']), to_object, root_object),
2221
2323
  )
2222
2324
 
2223
2325
  if getv(from_object, ['video']) is not None:
2224
2326
  setv(
2225
2327
  parent_object,
2226
2328
  ['instances[0]', 'video'],
2227
- _Video_to_mldev(getv(from_object, ['video']), to_object),
2329
+ _Video_to_mldev(getv(from_object, ['video']), to_object, root_object),
2228
2330
  )
2229
2331
 
2230
2332
  return to_object
@@ -2233,6 +2335,7 @@ def _GenerateVideosSource_to_mldev(
2233
2335
  def _GenerateVideosSource_to_vertex(
2234
2336
  from_object: Union[dict[str, Any], object],
2235
2337
  parent_object: Optional[dict[str, Any]] = None,
2338
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2236
2339
  ) -> dict[str, Any]:
2237
2340
  to_object: dict[str, Any] = {}
2238
2341
  if getv(from_object, ['prompt']) is not None:
@@ -2244,14 +2347,14 @@ def _GenerateVideosSource_to_vertex(
2244
2347
  setv(
2245
2348
  parent_object,
2246
2349
  ['instances[0]', 'image'],
2247
- _Image_to_vertex(getv(from_object, ['image']), to_object),
2350
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
2248
2351
  )
2249
2352
 
2250
2353
  if getv(from_object, ['video']) is not None:
2251
2354
  setv(
2252
2355
  parent_object,
2253
2356
  ['instances[0]', 'video'],
2254
- _Video_to_vertex(getv(from_object, ['video']), to_object),
2357
+ _Video_to_vertex(getv(from_object, ['video']), to_object, root_object),
2255
2358
  )
2256
2359
 
2257
2360
  return to_object
@@ -2260,13 +2363,16 @@ def _GenerateVideosSource_to_vertex(
2260
2363
  def _GeneratedImageMask_from_vertex(
2261
2364
  from_object: Union[dict[str, Any], object],
2262
2365
  parent_object: Optional[dict[str, Any]] = None,
2366
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2263
2367
  ) -> dict[str, Any]:
2264
2368
  to_object: dict[str, Any] = {}
2265
2369
  if getv(from_object, ['_self']) is not None:
2266
2370
  setv(
2267
2371
  to_object,
2268
2372
  ['mask'],
2269
- _Image_from_vertex(getv(from_object, ['_self']), to_object),
2373
+ _Image_from_vertex(
2374
+ getv(from_object, ['_self']), to_object, root_object
2375
+ ),
2270
2376
  )
2271
2377
 
2272
2378
  if getv(from_object, ['labels']) is not None:
@@ -2280,13 +2386,14 @@ def _GeneratedImageMask_from_vertex(
2280
2386
  def _GeneratedImage_from_mldev(
2281
2387
  from_object: Union[dict[str, Any], object],
2282
2388
  parent_object: Optional[dict[str, Any]] = None,
2389
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2283
2390
  ) -> dict[str, Any]:
2284
2391
  to_object: dict[str, Any] = {}
2285
2392
  if getv(from_object, ['_self']) is not None:
2286
2393
  setv(
2287
2394
  to_object,
2288
2395
  ['image'],
2289
- _Image_from_mldev(getv(from_object, ['_self']), to_object),
2396
+ _Image_from_mldev(getv(from_object, ['_self']), to_object, root_object),
2290
2397
  )
2291
2398
 
2292
2399
  if getv(from_object, ['raiFilteredReason']) is not None:
@@ -2300,7 +2407,9 @@ def _GeneratedImage_from_mldev(
2300
2407
  setv(
2301
2408
  to_object,
2302
2409
  ['safety_attributes'],
2303
- _SafetyAttributes_from_mldev(getv(from_object, ['_self']), to_object),
2410
+ _SafetyAttributes_from_mldev(
2411
+ getv(from_object, ['_self']), to_object, root_object
2412
+ ),
2304
2413
  )
2305
2414
 
2306
2415
  return to_object
@@ -2309,13 +2418,16 @@ def _GeneratedImage_from_mldev(
2309
2418
  def _GeneratedImage_from_vertex(
2310
2419
  from_object: Union[dict[str, Any], object],
2311
2420
  parent_object: Optional[dict[str, Any]] = None,
2421
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2312
2422
  ) -> dict[str, Any]:
2313
2423
  to_object: dict[str, Any] = {}
2314
2424
  if getv(from_object, ['_self']) is not None:
2315
2425
  setv(
2316
2426
  to_object,
2317
2427
  ['image'],
2318
- _Image_from_vertex(getv(from_object, ['_self']), to_object),
2428
+ _Image_from_vertex(
2429
+ getv(from_object, ['_self']), to_object, root_object
2430
+ ),
2319
2431
  )
2320
2432
 
2321
2433
  if getv(from_object, ['raiFilteredReason']) is not None:
@@ -2329,7 +2441,9 @@ def _GeneratedImage_from_vertex(
2329
2441
  setv(
2330
2442
  to_object,
2331
2443
  ['safety_attributes'],
2332
- _SafetyAttributes_from_vertex(getv(from_object, ['_self']), to_object),
2444
+ _SafetyAttributes_from_vertex(
2445
+ getv(from_object, ['_self']), to_object, root_object
2446
+ ),
2333
2447
  )
2334
2448
 
2335
2449
  if getv(from_object, ['prompt']) is not None:
@@ -2341,13 +2455,14 @@ def _GeneratedImage_from_vertex(
2341
2455
  def _GeneratedVideo_from_mldev(
2342
2456
  from_object: Union[dict[str, Any], object],
2343
2457
  parent_object: Optional[dict[str, Any]] = None,
2458
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2344
2459
  ) -> dict[str, Any]:
2345
2460
  to_object: dict[str, Any] = {}
2346
2461
  if getv(from_object, ['video']) is not None:
2347
2462
  setv(
2348
2463
  to_object,
2349
2464
  ['video'],
2350
- _Video_from_mldev(getv(from_object, ['video']), to_object),
2465
+ _Video_from_mldev(getv(from_object, ['video']), to_object, root_object),
2351
2466
  )
2352
2467
 
2353
2468
  return to_object
@@ -2356,13 +2471,16 @@ def _GeneratedVideo_from_mldev(
2356
2471
  def _GeneratedVideo_from_vertex(
2357
2472
  from_object: Union[dict[str, Any], object],
2358
2473
  parent_object: Optional[dict[str, Any]] = None,
2474
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2359
2475
  ) -> dict[str, Any]:
2360
2476
  to_object: dict[str, Any] = {}
2361
2477
  if getv(from_object, ['_self']) is not None:
2362
2478
  setv(
2363
2479
  to_object,
2364
2480
  ['video'],
2365
- _Video_from_vertex(getv(from_object, ['_self']), to_object),
2481
+ _Video_from_vertex(
2482
+ getv(from_object, ['_self']), to_object, root_object
2483
+ ),
2366
2484
  )
2367
2485
 
2368
2486
  return to_object
@@ -2371,6 +2489,7 @@ def _GeneratedVideo_from_vertex(
2371
2489
  def _GenerationConfig_to_vertex(
2372
2490
  from_object: Union[dict[str, Any], object],
2373
2491
  parent_object: Optional[dict[str, Any]] = None,
2492
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2374
2493
  ) -> dict[str, Any]:
2375
2494
  to_object: dict[str, Any] = {}
2376
2495
  if getv(from_object, ['model_selection_config']) is not None:
@@ -2485,6 +2604,7 @@ def _GetModelParameters_to_mldev(
2485
2604
  api_client: BaseApiClient,
2486
2605
  from_object: Union[dict[str, Any], object],
2487
2606
  parent_object: Optional[dict[str, Any]] = None,
2607
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2488
2608
  ) -> dict[str, Any]:
2489
2609
  to_object: dict[str, Any] = {}
2490
2610
  if getv(from_object, ['model']) is not None:
@@ -2501,6 +2621,7 @@ def _GetModelParameters_to_vertex(
2501
2621
  api_client: BaseApiClient,
2502
2622
  from_object: Union[dict[str, Any], object],
2503
2623
  parent_object: Optional[dict[str, Any]] = None,
2624
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2504
2625
  ) -> dict[str, Any]:
2505
2626
  to_object: dict[str, Any] = {}
2506
2627
  if getv(from_object, ['model']) is not None:
@@ -2516,6 +2637,7 @@ def _GetModelParameters_to_vertex(
2516
2637
  def _GoogleMaps_to_mldev(
2517
2638
  from_object: Union[dict[str, Any], object],
2518
2639
  parent_object: Optional[dict[str, Any]] = None,
2640
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2519
2641
  ) -> dict[str, Any]:
2520
2642
  to_object: dict[str, Any] = {}
2521
2643
  if getv(from_object, ['auth_config']) is not None:
@@ -2530,6 +2652,7 @@ def _GoogleMaps_to_mldev(
2530
2652
  def _GoogleSearch_to_mldev(
2531
2653
  from_object: Union[dict[str, Any], object],
2532
2654
  parent_object: Optional[dict[str, Any]] = None,
2655
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2533
2656
  ) -> dict[str, Any]:
2534
2657
  to_object: dict[str, Any] = {}
2535
2658
  if getv(from_object, ['exclude_domains']) is not None:
@@ -2553,6 +2676,7 @@ def _GoogleSearch_to_mldev(
2553
2676
  def _ImageConfig_to_mldev(
2554
2677
  from_object: Union[dict[str, Any], object],
2555
2678
  parent_object: Optional[dict[str, Any]] = None,
2679
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2556
2680
  ) -> dict[str, Any]:
2557
2681
  to_object: dict[str, Any] = {}
2558
2682
  if getv(from_object, ['aspect_ratio']) is not None:
@@ -2582,6 +2706,7 @@ def _ImageConfig_to_mldev(
2582
2706
  def _ImageConfig_to_vertex(
2583
2707
  from_object: Union[dict[str, Any], object],
2584
2708
  parent_object: Optional[dict[str, Any]] = None,
2709
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2585
2710
  ) -> dict[str, Any]:
2586
2711
  to_object: dict[str, Any] = {}
2587
2712
  if getv(from_object, ['aspect_ratio']) is not None:
@@ -2617,6 +2742,7 @@ def _ImageConfig_to_vertex(
2617
2742
  def _Image_from_mldev(
2618
2743
  from_object: Union[dict[str, Any], object],
2619
2744
  parent_object: Optional[dict[str, Any]] = None,
2745
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2620
2746
  ) -> dict[str, Any]:
2621
2747
  to_object: dict[str, Any] = {}
2622
2748
 
@@ -2636,6 +2762,7 @@ def _Image_from_mldev(
2636
2762
  def _Image_from_vertex(
2637
2763
  from_object: Union[dict[str, Any], object],
2638
2764
  parent_object: Optional[dict[str, Any]] = None,
2765
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2639
2766
  ) -> dict[str, Any]:
2640
2767
  to_object: dict[str, Any] = {}
2641
2768
  if getv(from_object, ['gcsUri']) is not None:
@@ -2657,6 +2784,7 @@ def _Image_from_vertex(
2657
2784
  def _Image_to_mldev(
2658
2785
  from_object: Union[dict[str, Any], object],
2659
2786
  parent_object: Optional[dict[str, Any]] = None,
2787
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2660
2788
  ) -> dict[str, Any]:
2661
2789
  to_object: dict[str, Any] = {}
2662
2790
  if getv(from_object, ['gcs_uri']) is not None:
@@ -2678,6 +2806,7 @@ def _Image_to_mldev(
2678
2806
  def _Image_to_vertex(
2679
2807
  from_object: Union[dict[str, Any], object],
2680
2808
  parent_object: Optional[dict[str, Any]] = None,
2809
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2681
2810
  ) -> dict[str, Any]:
2682
2811
  to_object: dict[str, Any] = {}
2683
2812
  if getv(from_object, ['gcs_uri']) is not None:
@@ -2700,6 +2829,7 @@ def _ListModelsConfig_to_mldev(
2700
2829
  api_client: BaseApiClient,
2701
2830
  from_object: Union[dict[str, Any], object],
2702
2831
  parent_object: Optional[dict[str, Any]] = None,
2832
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2703
2833
  ) -> dict[str, Any]:
2704
2834
  to_object: dict[str, Any] = {}
2705
2835
 
@@ -2732,6 +2862,7 @@ def _ListModelsConfig_to_vertex(
2732
2862
  api_client: BaseApiClient,
2733
2863
  from_object: Union[dict[str, Any], object],
2734
2864
  parent_object: Optional[dict[str, Any]] = None,
2865
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2735
2866
  ) -> dict[str, Any]:
2736
2867
  to_object: dict[str, Any] = {}
2737
2868
 
@@ -2764,11 +2895,12 @@ def _ListModelsParameters_to_mldev(
2764
2895
  api_client: BaseApiClient,
2765
2896
  from_object: Union[dict[str, Any], object],
2766
2897
  parent_object: Optional[dict[str, Any]] = None,
2898
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2767
2899
  ) -> dict[str, Any]:
2768
2900
  to_object: dict[str, Any] = {}
2769
2901
  if getv(from_object, ['config']) is not None:
2770
2902
  _ListModelsConfig_to_mldev(
2771
- api_client, getv(from_object, ['config']), to_object
2903
+ api_client, getv(from_object, ['config']), to_object, root_object
2772
2904
  )
2773
2905
 
2774
2906
  return to_object
@@ -2778,11 +2910,12 @@ def _ListModelsParameters_to_vertex(
2778
2910
  api_client: BaseApiClient,
2779
2911
  from_object: Union[dict[str, Any], object],
2780
2912
  parent_object: Optional[dict[str, Any]] = None,
2913
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2781
2914
  ) -> dict[str, Any]:
2782
2915
  to_object: dict[str, Any] = {}
2783
2916
  if getv(from_object, ['config']) is not None:
2784
2917
  _ListModelsConfig_to_vertex(
2785
- api_client, getv(from_object, ['config']), to_object
2918
+ api_client, getv(from_object, ['config']), to_object, root_object
2786
2919
  )
2787
2920
 
2788
2921
  return to_object
@@ -2791,6 +2924,7 @@ def _ListModelsParameters_to_vertex(
2791
2924
  def _ListModelsResponse_from_mldev(
2792
2925
  from_object: Union[dict[str, Any], object],
2793
2926
  parent_object: Optional[dict[str, Any]] = None,
2927
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2794
2928
  ) -> dict[str, Any]:
2795
2929
  to_object: dict[str, Any] = {}
2796
2930
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -2806,7 +2940,7 @@ def _ListModelsResponse_from_mldev(
2806
2940
  to_object,
2807
2941
  ['models'],
2808
2942
  [
2809
- _Model_from_mldev(item, to_object)
2943
+ _Model_from_mldev(item, to_object, root_object)
2810
2944
  for item in t.t_extract_models(getv(from_object, ['_self']))
2811
2945
  ],
2812
2946
  )
@@ -2817,6 +2951,7 @@ def _ListModelsResponse_from_mldev(
2817
2951
  def _ListModelsResponse_from_vertex(
2818
2952
  from_object: Union[dict[str, Any], object],
2819
2953
  parent_object: Optional[dict[str, Any]] = None,
2954
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2820
2955
  ) -> dict[str, Any]:
2821
2956
  to_object: dict[str, Any] = {}
2822
2957
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -2832,7 +2967,7 @@ def _ListModelsResponse_from_vertex(
2832
2967
  to_object,
2833
2968
  ['models'],
2834
2969
  [
2835
- _Model_from_vertex(item, to_object)
2970
+ _Model_from_vertex(item, to_object, root_object)
2836
2971
  for item in t.t_extract_models(getv(from_object, ['_self']))
2837
2972
  ],
2838
2973
  )
@@ -2843,6 +2978,7 @@ def _ListModelsResponse_from_vertex(
2843
2978
  def _MaskReferenceConfig_to_vertex(
2844
2979
  from_object: Union[dict[str, Any], object],
2845
2980
  parent_object: Optional[dict[str, Any]] = None,
2981
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2846
2982
  ) -> dict[str, Any]:
2847
2983
  to_object: dict[str, Any] = {}
2848
2984
  if getv(from_object, ['mask_mode']) is not None:
@@ -2862,6 +2998,7 @@ def _MaskReferenceConfig_to_vertex(
2862
2998
  def _Model_from_mldev(
2863
2999
  from_object: Union[dict[str, Any], object],
2864
3000
  parent_object: Optional[dict[str, Any]] = None,
3001
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2865
3002
  ) -> dict[str, Any]:
2866
3003
  to_object: dict[str, Any] = {}
2867
3004
  if getv(from_object, ['name']) is not None:
@@ -2919,6 +3056,7 @@ def _Model_from_mldev(
2919
3056
  def _Model_from_vertex(
2920
3057
  from_object: Union[dict[str, Any], object],
2921
3058
  parent_object: Optional[dict[str, Any]] = None,
3059
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2922
3060
  ) -> dict[str, Any]:
2923
3061
  to_object: dict[str, Any] = {}
2924
3062
  if getv(from_object, ['name']) is not None:
@@ -2938,7 +3076,7 @@ def _Model_from_vertex(
2938
3076
  to_object,
2939
3077
  ['endpoints'],
2940
3078
  [
2941
- _Endpoint_from_vertex(item, to_object)
3079
+ _Endpoint_from_vertex(item, to_object, root_object)
2942
3080
  for item in getv(from_object, ['deployedModels'])
2943
3081
  ],
2944
3082
  )
@@ -2950,7 +3088,9 @@ def _Model_from_vertex(
2950
3088
  setv(
2951
3089
  to_object,
2952
3090
  ['tuned_model_info'],
2953
- _TunedModelInfo_from_vertex(getv(from_object, ['_self']), to_object),
3091
+ _TunedModelInfo_from_vertex(
3092
+ getv(from_object, ['_self']), to_object, root_object
3093
+ ),
2954
3094
  )
2955
3095
 
2956
3096
  if getv(from_object, ['defaultCheckpointId']) is not None:
@@ -2973,6 +3113,7 @@ def _Model_from_vertex(
2973
3113
  def _Part_to_mldev(
2974
3114
  from_object: Union[dict[str, Any], object],
2975
3115
  parent_object: Optional[dict[str, Any]] = None,
3116
+ root_object: Optional[Union[dict[str, Any], object]] = None,
2976
3117
  ) -> dict[str, Any]:
2977
3118
  to_object: dict[str, Any] = {}
2978
3119
  if getv(from_object, ['media_resolution']) is not None:
@@ -2994,14 +3135,18 @@ def _Part_to_mldev(
2994
3135
  setv(
2995
3136
  to_object,
2996
3137
  ['fileData'],
2997
- _FileData_to_mldev(getv(from_object, ['file_data']), to_object),
3138
+ _FileData_to_mldev(
3139
+ getv(from_object, ['file_data']), to_object, root_object
3140
+ ),
2998
3141
  )
2999
3142
 
3000
3143
  if getv(from_object, ['function_call']) is not None:
3001
3144
  setv(
3002
3145
  to_object,
3003
3146
  ['functionCall'],
3004
- _FunctionCall_to_mldev(getv(from_object, ['function_call']), to_object),
3147
+ _FunctionCall_to_mldev(
3148
+ getv(from_object, ['function_call']), to_object, root_object
3149
+ ),
3005
3150
  )
3006
3151
 
3007
3152
  if getv(from_object, ['function_response']) is not None:
@@ -3015,7 +3160,9 @@ def _Part_to_mldev(
3015
3160
  setv(
3016
3161
  to_object,
3017
3162
  ['inlineData'],
3018
- _Blob_to_mldev(getv(from_object, ['inline_data']), to_object),
3163
+ _Blob_to_mldev(
3164
+ getv(from_object, ['inline_data']), to_object, root_object
3165
+ ),
3019
3166
  )
3020
3167
 
3021
3168
  if getv(from_object, ['text']) is not None:
@@ -3040,13 +3187,16 @@ def _Part_to_mldev(
3040
3187
  def _ProductImage_to_vertex(
3041
3188
  from_object: Union[dict[str, Any], object],
3042
3189
  parent_object: Optional[dict[str, Any]] = None,
3190
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3043
3191
  ) -> dict[str, Any]:
3044
3192
  to_object: dict[str, Any] = {}
3045
3193
  if getv(from_object, ['product_image']) is not None:
3046
3194
  setv(
3047
3195
  to_object,
3048
3196
  ['image'],
3049
- _Image_to_vertex(getv(from_object, ['product_image']), to_object),
3197
+ _Image_to_vertex(
3198
+ getv(from_object, ['product_image']), to_object, root_object
3199
+ ),
3050
3200
  )
3051
3201
 
3052
3202
  return to_object
@@ -3055,6 +3205,7 @@ def _ProductImage_to_vertex(
3055
3205
  def _RecontextImageConfig_to_vertex(
3056
3206
  from_object: Union[dict[str, Any], object],
3057
3207
  parent_object: Optional[dict[str, Any]] = None,
3208
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3058
3209
  ) -> dict[str, Any]:
3059
3210
  to_object: dict[str, Any] = {}
3060
3211
 
@@ -3134,6 +3285,7 @@ def _RecontextImageParameters_to_vertex(
3134
3285
  api_client: BaseApiClient,
3135
3286
  from_object: Union[dict[str, Any], object],
3136
3287
  parent_object: Optional[dict[str, Any]] = None,
3288
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3137
3289
  ) -> dict[str, Any]:
3138
3290
  to_object: dict[str, Any] = {}
3139
3291
  if getv(from_object, ['model']) is not None:
@@ -3144,10 +3296,14 @@ def _RecontextImageParameters_to_vertex(
3144
3296
  )
3145
3297
 
3146
3298
  if getv(from_object, ['source']) is not None:
3147
- _RecontextImageSource_to_vertex(getv(from_object, ['source']), to_object)
3299
+ _RecontextImageSource_to_vertex(
3300
+ getv(from_object, ['source']), to_object, root_object
3301
+ )
3148
3302
 
3149
3303
  if getv(from_object, ['config']) is not None:
3150
- _RecontextImageConfig_to_vertex(getv(from_object, ['config']), to_object)
3304
+ _RecontextImageConfig_to_vertex(
3305
+ getv(from_object, ['config']), to_object, root_object
3306
+ )
3151
3307
 
3152
3308
  return to_object
3153
3309
 
@@ -3155,6 +3311,7 @@ def _RecontextImageParameters_to_vertex(
3155
3311
  def _RecontextImageResponse_from_vertex(
3156
3312
  from_object: Union[dict[str, Any], object],
3157
3313
  parent_object: Optional[dict[str, Any]] = None,
3314
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3158
3315
  ) -> dict[str, Any]:
3159
3316
  to_object: dict[str, Any] = {}
3160
3317
  if getv(from_object, ['predictions']) is not None:
@@ -3162,7 +3319,7 @@ def _RecontextImageResponse_from_vertex(
3162
3319
  to_object,
3163
3320
  ['generated_images'],
3164
3321
  [
3165
- _GeneratedImage_from_vertex(item, to_object)
3322
+ _GeneratedImage_from_vertex(item, to_object, root_object)
3166
3323
  for item in getv(from_object, ['predictions'])
3167
3324
  ],
3168
3325
  )
@@ -3173,6 +3330,7 @@ def _RecontextImageResponse_from_vertex(
3173
3330
  def _RecontextImageSource_to_vertex(
3174
3331
  from_object: Union[dict[str, Any], object],
3175
3332
  parent_object: Optional[dict[str, Any]] = None,
3333
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3176
3334
  ) -> dict[str, Any]:
3177
3335
  to_object: dict[str, Any] = {}
3178
3336
  if getv(from_object, ['prompt']) is not None:
@@ -3184,7 +3342,9 @@ def _RecontextImageSource_to_vertex(
3184
3342
  setv(
3185
3343
  parent_object,
3186
3344
  ['instances[0]', 'personImage', 'image'],
3187
- _Image_to_vertex(getv(from_object, ['person_image']), to_object),
3345
+ _Image_to_vertex(
3346
+ getv(from_object, ['person_image']), to_object, root_object
3347
+ ),
3188
3348
  )
3189
3349
 
3190
3350
  if getv(from_object, ['product_images']) is not None:
@@ -3192,7 +3352,7 @@ def _RecontextImageSource_to_vertex(
3192
3352
  parent_object,
3193
3353
  ['instances[0]', 'productImages'],
3194
3354
  [
3195
- _ProductImage_to_vertex(item, to_object)
3355
+ _ProductImage_to_vertex(item, to_object, root_object)
3196
3356
  for item in getv(from_object, ['product_images'])
3197
3357
  ],
3198
3358
  )
@@ -3203,13 +3363,16 @@ def _RecontextImageSource_to_vertex(
3203
3363
  def _ReferenceImageAPI_to_vertex(
3204
3364
  from_object: Union[dict[str, Any], object],
3205
3365
  parent_object: Optional[dict[str, Any]] = None,
3366
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3206
3367
  ) -> dict[str, Any]:
3207
3368
  to_object: dict[str, Any] = {}
3208
3369
  if getv(from_object, ['reference_image']) is not None:
3209
3370
  setv(
3210
3371
  to_object,
3211
3372
  ['referenceImage'],
3212
- _Image_to_vertex(getv(from_object, ['reference_image']), to_object),
3373
+ _Image_to_vertex(
3374
+ getv(from_object, ['reference_image']), to_object, root_object
3375
+ ),
3213
3376
  )
3214
3377
 
3215
3378
  if getv(from_object, ['reference_id']) is not None:
@@ -3223,7 +3386,7 @@ def _ReferenceImageAPI_to_vertex(
3223
3386
  to_object,
3224
3387
  ['maskImageConfig'],
3225
3388
  _MaskReferenceConfig_to_vertex(
3226
- getv(from_object, ['mask_image_config']), to_object
3389
+ getv(from_object, ['mask_image_config']), to_object, root_object
3227
3390
  ),
3228
3391
  )
3229
3392
 
@@ -3232,7 +3395,7 @@ def _ReferenceImageAPI_to_vertex(
3232
3395
  to_object,
3233
3396
  ['controlImageConfig'],
3234
3397
  _ControlReferenceConfig_to_vertex(
3235
- getv(from_object, ['control_image_config']), to_object
3398
+ getv(from_object, ['control_image_config']), to_object, root_object
3236
3399
  ),
3237
3400
  )
3238
3401
 
@@ -3256,6 +3419,7 @@ def _ReferenceImageAPI_to_vertex(
3256
3419
  def _SafetyAttributes_from_mldev(
3257
3420
  from_object: Union[dict[str, Any], object],
3258
3421
  parent_object: Optional[dict[str, Any]] = None,
3422
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3259
3423
  ) -> dict[str, Any]:
3260
3424
  to_object: dict[str, Any] = {}
3261
3425
  if getv(from_object, ['safetyAttributes', 'categories']) is not None:
@@ -3279,6 +3443,7 @@ def _SafetyAttributes_from_mldev(
3279
3443
  def _SafetyAttributes_from_vertex(
3280
3444
  from_object: Union[dict[str, Any], object],
3281
3445
  parent_object: Optional[dict[str, Any]] = None,
3446
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3282
3447
  ) -> dict[str, Any]:
3283
3448
  to_object: dict[str, Any] = {}
3284
3449
  if getv(from_object, ['safetyAttributes', 'categories']) is not None:
@@ -3302,6 +3467,7 @@ def _SafetyAttributes_from_vertex(
3302
3467
  def _SafetySetting_to_mldev(
3303
3468
  from_object: Union[dict[str, Any], object],
3304
3469
  parent_object: Optional[dict[str, Any]] = None,
3470
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3305
3471
  ) -> dict[str, Any]:
3306
3472
  to_object: dict[str, Any] = {}
3307
3473
  if getv(from_object, ['category']) is not None:
@@ -3319,13 +3485,14 @@ def _SafetySetting_to_mldev(
3319
3485
  def _ScribbleImage_to_vertex(
3320
3486
  from_object: Union[dict[str, Any], object],
3321
3487
  parent_object: Optional[dict[str, Any]] = None,
3488
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3322
3489
  ) -> dict[str, Any]:
3323
3490
  to_object: dict[str, Any] = {}
3324
3491
  if getv(from_object, ['image']) is not None:
3325
3492
  setv(
3326
3493
  to_object,
3327
3494
  ['image'],
3328
- _Image_to_vertex(getv(from_object, ['image']), to_object),
3495
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
3329
3496
  )
3330
3497
 
3331
3498
  return to_object
@@ -3334,6 +3501,7 @@ def _ScribbleImage_to_vertex(
3334
3501
  def _SegmentImageConfig_to_vertex(
3335
3502
  from_object: Union[dict[str, Any], object],
3336
3503
  parent_object: Optional[dict[str, Any]] = None,
3504
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3337
3505
  ) -> dict[str, Any]:
3338
3506
  to_object: dict[str, Any] = {}
3339
3507
 
@@ -3378,6 +3546,7 @@ def _SegmentImageParameters_to_vertex(
3378
3546
  api_client: BaseApiClient,
3379
3547
  from_object: Union[dict[str, Any], object],
3380
3548
  parent_object: Optional[dict[str, Any]] = None,
3549
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3381
3550
  ) -> dict[str, Any]:
3382
3551
  to_object: dict[str, Any] = {}
3383
3552
  if getv(from_object, ['model']) is not None:
@@ -3388,10 +3557,14 @@ def _SegmentImageParameters_to_vertex(
3388
3557
  )
3389
3558
 
3390
3559
  if getv(from_object, ['source']) is not None:
3391
- _SegmentImageSource_to_vertex(getv(from_object, ['source']), to_object)
3560
+ _SegmentImageSource_to_vertex(
3561
+ getv(from_object, ['source']), to_object, root_object
3562
+ )
3392
3563
 
3393
3564
  if getv(from_object, ['config']) is not None:
3394
- _SegmentImageConfig_to_vertex(getv(from_object, ['config']), to_object)
3565
+ _SegmentImageConfig_to_vertex(
3566
+ getv(from_object, ['config']), to_object, root_object
3567
+ )
3395
3568
 
3396
3569
  return to_object
3397
3570
 
@@ -3399,6 +3572,7 @@ def _SegmentImageParameters_to_vertex(
3399
3572
  def _SegmentImageResponse_from_vertex(
3400
3573
  from_object: Union[dict[str, Any], object],
3401
3574
  parent_object: Optional[dict[str, Any]] = None,
3575
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3402
3576
  ) -> dict[str, Any]:
3403
3577
  to_object: dict[str, Any] = {}
3404
3578
  if getv(from_object, ['predictions']) is not None:
@@ -3406,7 +3580,7 @@ def _SegmentImageResponse_from_vertex(
3406
3580
  to_object,
3407
3581
  ['generated_masks'],
3408
3582
  [
3409
- _GeneratedImageMask_from_vertex(item, to_object)
3583
+ _GeneratedImageMask_from_vertex(item, to_object, root_object)
3410
3584
  for item in getv(from_object, ['predictions'])
3411
3585
  ],
3412
3586
  )
@@ -3417,6 +3591,7 @@ def _SegmentImageResponse_from_vertex(
3417
3591
  def _SegmentImageSource_to_vertex(
3418
3592
  from_object: Union[dict[str, Any], object],
3419
3593
  parent_object: Optional[dict[str, Any]] = None,
3594
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3420
3595
  ) -> dict[str, Any]:
3421
3596
  to_object: dict[str, Any] = {}
3422
3597
  if getv(from_object, ['prompt']) is not None:
@@ -3428,7 +3603,7 @@ def _SegmentImageSource_to_vertex(
3428
3603
  setv(
3429
3604
  parent_object,
3430
3605
  ['instances[0]', 'image'],
3431
- _Image_to_vertex(getv(from_object, ['image']), to_object),
3606
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
3432
3607
  )
3433
3608
 
3434
3609
  if getv(from_object, ['scribble_image']) is not None:
@@ -3436,7 +3611,7 @@ def _SegmentImageSource_to_vertex(
3436
3611
  parent_object,
3437
3612
  ['instances[0]', 'scribble'],
3438
3613
  _ScribbleImage_to_vertex(
3439
- getv(from_object, ['scribble_image']), to_object
3614
+ getv(from_object, ['scribble_image']), to_object, root_object
3440
3615
  ),
3441
3616
  )
3442
3617
 
@@ -3446,6 +3621,7 @@ def _SegmentImageSource_to_vertex(
3446
3621
  def _ToolConfig_to_mldev(
3447
3622
  from_object: Union[dict[str, Any], object],
3448
3623
  parent_object: Optional[dict[str, Any]] = None,
3624
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3449
3625
  ) -> dict[str, Any]:
3450
3626
  to_object: dict[str, Any] = {}
3451
3627
  if getv(from_object, ['retrieval_config']) is not None:
@@ -3458,7 +3634,9 @@ def _ToolConfig_to_mldev(
3458
3634
  to_object,
3459
3635
  ['functionCallingConfig'],
3460
3636
  _FunctionCallingConfig_to_mldev(
3461
- getv(from_object, ['function_calling_config']), to_object
3637
+ getv(from_object, ['function_calling_config']),
3638
+ to_object,
3639
+ root_object,
3462
3640
  ),
3463
3641
  )
3464
3642
 
@@ -3468,6 +3646,7 @@ def _ToolConfig_to_mldev(
3468
3646
  def _Tool_to_mldev(
3469
3647
  from_object: Union[dict[str, Any], object],
3470
3648
  parent_object: Optional[dict[str, Any]] = None,
3649
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3471
3650
  ) -> dict[str, Any]:
3472
3651
  to_object: dict[str, Any] = {}
3473
3652
  if getv(from_object, ['retrieval']) is not None:
@@ -3498,14 +3677,18 @@ def _Tool_to_mldev(
3498
3677
  setv(
3499
3678
  to_object,
3500
3679
  ['googleMaps'],
3501
- _GoogleMaps_to_mldev(getv(from_object, ['google_maps']), to_object),
3680
+ _GoogleMaps_to_mldev(
3681
+ getv(from_object, ['google_maps']), to_object, root_object
3682
+ ),
3502
3683
  )
3503
3684
 
3504
3685
  if getv(from_object, ['google_search']) is not None:
3505
3686
  setv(
3506
3687
  to_object,
3507
3688
  ['googleSearch'],
3508
- _GoogleSearch_to_mldev(getv(from_object, ['google_search']), to_object),
3689
+ _GoogleSearch_to_mldev(
3690
+ getv(from_object, ['google_search']), to_object, root_object
3691
+ ),
3509
3692
  )
3510
3693
 
3511
3694
  if getv(from_object, ['google_search_retrieval']) is not None:
@@ -3524,6 +3707,7 @@ def _Tool_to_mldev(
3524
3707
  def _Tool_to_vertex(
3525
3708
  from_object: Union[dict[str, Any], object],
3526
3709
  parent_object: Optional[dict[str, Any]] = None,
3710
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3527
3711
  ) -> dict[str, Any]:
3528
3712
  to_object: dict[str, Any] = {}
3529
3713
  if getv(from_object, ['retrieval']) is not None:
@@ -3550,7 +3734,7 @@ def _Tool_to_vertex(
3550
3734
  to_object,
3551
3735
  ['functionDeclarations'],
3552
3736
  [
3553
- _FunctionDeclaration_to_vertex(item, to_object)
3737
+ _FunctionDeclaration_to_vertex(item, to_object, root_object)
3554
3738
  for item in getv(from_object, ['function_declarations'])
3555
3739
  ],
3556
3740
  )
@@ -3577,6 +3761,7 @@ def _Tool_to_vertex(
3577
3761
  def _TunedModelInfo_from_vertex(
3578
3762
  from_object: Union[dict[str, Any], object],
3579
3763
  parent_object: Optional[dict[str, Any]] = None,
3764
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3580
3765
  ) -> dict[str, Any]:
3581
3766
  to_object: dict[str, Any] = {}
3582
3767
  if (
@@ -3601,6 +3786,7 @@ def _TunedModelInfo_from_vertex(
3601
3786
  def _UpdateModelConfig_to_mldev(
3602
3787
  from_object: Union[dict[str, Any], object],
3603
3788
  parent_object: Optional[dict[str, Any]] = None,
3789
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3604
3790
  ) -> dict[str, Any]:
3605
3791
  to_object: dict[str, Any] = {}
3606
3792
 
@@ -3623,6 +3809,7 @@ def _UpdateModelConfig_to_mldev(
3623
3809
  def _UpdateModelConfig_to_vertex(
3624
3810
  from_object: Union[dict[str, Any], object],
3625
3811
  parent_object: Optional[dict[str, Any]] = None,
3812
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3626
3813
  ) -> dict[str, Any]:
3627
3814
  to_object: dict[str, Any] = {}
3628
3815
 
@@ -3646,6 +3833,7 @@ def _UpdateModelParameters_to_mldev(
3646
3833
  api_client: BaseApiClient,
3647
3834
  from_object: Union[dict[str, Any], object],
3648
3835
  parent_object: Optional[dict[str, Any]] = None,
3836
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3649
3837
  ) -> dict[str, Any]:
3650
3838
  to_object: dict[str, Any] = {}
3651
3839
  if getv(from_object, ['model']) is not None:
@@ -3656,7 +3844,9 @@ def _UpdateModelParameters_to_mldev(
3656
3844
  )
3657
3845
 
3658
3846
  if getv(from_object, ['config']) is not None:
3659
- _UpdateModelConfig_to_mldev(getv(from_object, ['config']), to_object)
3847
+ _UpdateModelConfig_to_mldev(
3848
+ getv(from_object, ['config']), to_object, root_object
3849
+ )
3660
3850
 
3661
3851
  return to_object
3662
3852
 
@@ -3665,6 +3855,7 @@ def _UpdateModelParameters_to_vertex(
3665
3855
  api_client: BaseApiClient,
3666
3856
  from_object: Union[dict[str, Any], object],
3667
3857
  parent_object: Optional[dict[str, Any]] = None,
3858
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3668
3859
  ) -> dict[str, Any]:
3669
3860
  to_object: dict[str, Any] = {}
3670
3861
  if getv(from_object, ['model']) is not None:
@@ -3675,7 +3866,9 @@ def _UpdateModelParameters_to_vertex(
3675
3866
  )
3676
3867
 
3677
3868
  if getv(from_object, ['config']) is not None:
3678
- _UpdateModelConfig_to_vertex(getv(from_object, ['config']), to_object)
3869
+ _UpdateModelConfig_to_vertex(
3870
+ getv(from_object, ['config']), to_object, root_object
3871
+ )
3679
3872
 
3680
3873
  return to_object
3681
3874
 
@@ -3683,6 +3876,7 @@ def _UpdateModelParameters_to_vertex(
3683
3876
  def _UpscaleImageAPIConfig_to_vertex(
3684
3877
  from_object: Union[dict[str, Any], object],
3685
3878
  parent_object: Optional[dict[str, Any]] = None,
3879
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3686
3880
  ) -> dict[str, Any]:
3687
3881
  to_object: dict[str, Any] = {}
3688
3882
 
@@ -3762,6 +3956,7 @@ def _UpscaleImageAPIParameters_to_vertex(
3762
3956
  api_client: BaseApiClient,
3763
3957
  from_object: Union[dict[str, Any], object],
3764
3958
  parent_object: Optional[dict[str, Any]] = None,
3959
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3765
3960
  ) -> dict[str, Any]:
3766
3961
  to_object: dict[str, Any] = {}
3767
3962
  if getv(from_object, ['model']) is not None:
@@ -3775,7 +3970,7 @@ def _UpscaleImageAPIParameters_to_vertex(
3775
3970
  setv(
3776
3971
  to_object,
3777
3972
  ['instances[0]', 'image'],
3778
- _Image_to_vertex(getv(from_object, ['image']), to_object),
3973
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
3779
3974
  )
3780
3975
 
3781
3976
  if getv(from_object, ['upscale_factor']) is not None:
@@ -3786,7 +3981,9 @@ def _UpscaleImageAPIParameters_to_vertex(
3786
3981
  )
3787
3982
 
3788
3983
  if getv(from_object, ['config']) is not None:
3789
- _UpscaleImageAPIConfig_to_vertex(getv(from_object, ['config']), to_object)
3984
+ _UpscaleImageAPIConfig_to_vertex(
3985
+ getv(from_object, ['config']), to_object, root_object
3986
+ )
3790
3987
 
3791
3988
  return to_object
3792
3989
 
@@ -3794,6 +3991,7 @@ def _UpscaleImageAPIParameters_to_vertex(
3794
3991
  def _UpscaleImageResponse_from_vertex(
3795
3992
  from_object: Union[dict[str, Any], object],
3796
3993
  parent_object: Optional[dict[str, Any]] = None,
3994
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3797
3995
  ) -> dict[str, Any]:
3798
3996
  to_object: dict[str, Any] = {}
3799
3997
  if getv(from_object, ['sdkHttpResponse']) is not None:
@@ -3806,7 +4004,7 @@ def _UpscaleImageResponse_from_vertex(
3806
4004
  to_object,
3807
4005
  ['generated_images'],
3808
4006
  [
3809
- _GeneratedImage_from_vertex(item, to_object)
4007
+ _GeneratedImage_from_vertex(item, to_object, root_object)
3810
4008
  for item in getv(from_object, ['predictions'])
3811
4009
  ],
3812
4010
  )
@@ -3817,13 +4015,14 @@ def _UpscaleImageResponse_from_vertex(
3817
4015
  def _VideoGenerationMask_to_vertex(
3818
4016
  from_object: Union[dict[str, Any], object],
3819
4017
  parent_object: Optional[dict[str, Any]] = None,
4018
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3820
4019
  ) -> dict[str, Any]:
3821
4020
  to_object: dict[str, Any] = {}
3822
4021
  if getv(from_object, ['image']) is not None:
3823
4022
  setv(
3824
4023
  to_object,
3825
4024
  ['_self'],
3826
- _Image_to_vertex(getv(from_object, ['image']), to_object),
4025
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
3827
4026
  )
3828
4027
 
3829
4028
  if getv(from_object, ['mask_mode']) is not None:
@@ -3835,13 +4034,14 @@ def _VideoGenerationMask_to_vertex(
3835
4034
  def _VideoGenerationReferenceImage_to_mldev(
3836
4035
  from_object: Union[dict[str, Any], object],
3837
4036
  parent_object: Optional[dict[str, Any]] = None,
4037
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3838
4038
  ) -> dict[str, Any]:
3839
4039
  to_object: dict[str, Any] = {}
3840
4040
  if getv(from_object, ['image']) is not None:
3841
4041
  setv(
3842
4042
  to_object,
3843
4043
  ['image'],
3844
- _Image_to_mldev(getv(from_object, ['image']), to_object),
4044
+ _Image_to_mldev(getv(from_object, ['image']), to_object, root_object),
3845
4045
  )
3846
4046
 
3847
4047
  if getv(from_object, ['reference_type']) is not None:
@@ -3856,13 +4056,14 @@ def _VideoGenerationReferenceImage_to_mldev(
3856
4056
  def _VideoGenerationReferenceImage_to_vertex(
3857
4057
  from_object: Union[dict[str, Any], object],
3858
4058
  parent_object: Optional[dict[str, Any]] = None,
4059
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3859
4060
  ) -> dict[str, Any]:
3860
4061
  to_object: dict[str, Any] = {}
3861
4062
  if getv(from_object, ['image']) is not None:
3862
4063
  setv(
3863
4064
  to_object,
3864
4065
  ['image'],
3865
- _Image_to_vertex(getv(from_object, ['image']), to_object),
4066
+ _Image_to_vertex(getv(from_object, ['image']), to_object, root_object),
3866
4067
  )
3867
4068
 
3868
4069
  if getv(from_object, ['reference_type']) is not None:
@@ -3874,6 +4075,7 @@ def _VideoGenerationReferenceImage_to_vertex(
3874
4075
  def _Video_from_mldev(
3875
4076
  from_object: Union[dict[str, Any], object],
3876
4077
  parent_object: Optional[dict[str, Any]] = None,
4078
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3877
4079
  ) -> dict[str, Any]:
3878
4080
  to_object: dict[str, Any] = {}
3879
4081
  if getv(from_object, ['uri']) is not None:
@@ -3895,6 +4097,7 @@ def _Video_from_mldev(
3895
4097
  def _Video_from_vertex(
3896
4098
  from_object: Union[dict[str, Any], object],
3897
4099
  parent_object: Optional[dict[str, Any]] = None,
4100
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3898
4101
  ) -> dict[str, Any]:
3899
4102
  to_object: dict[str, Any] = {}
3900
4103
  if getv(from_object, ['gcsUri']) is not None:
@@ -3916,6 +4119,7 @@ def _Video_from_vertex(
3916
4119
  def _Video_to_mldev(
3917
4120
  from_object: Union[dict[str, Any], object],
3918
4121
  parent_object: Optional[dict[str, Any]] = None,
4122
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3919
4123
  ) -> dict[str, Any]:
3920
4124
  to_object: dict[str, Any] = {}
3921
4125
  if getv(from_object, ['uri']) is not None:
@@ -3937,6 +4141,7 @@ def _Video_to_mldev(
3937
4141
  def _Video_to_vertex(
3938
4142
  from_object: Union[dict[str, Any], object],
3939
4143
  parent_object: Optional[dict[str, Any]] = None,
4144
+ root_object: Optional[Union[dict[str, Any], object]] = None,
3940
4145
  ) -> dict[str, Any]:
3941
4146
  to_object: dict[str, Any] = {}
3942
4147
  if getv(from_object, ['uri']) is not None:
@@ -3974,7 +4179,7 @@ class Models(_api_module.BaseModule):
3974
4179
 
3975
4180
  if self._api_client.vertexai:
3976
4181
  request_dict = _GenerateContentParameters_to_vertex(
3977
- self._api_client, parameter_model
4182
+ self._api_client, parameter_model, None, parameter_model
3978
4183
  )
3979
4184
  request_url_dict = request_dict.get('_url')
3980
4185
  if request_url_dict:
@@ -3983,7 +4188,7 @@ class Models(_api_module.BaseModule):
3983
4188
  path = '{model}:generateContent'
3984
4189
  else:
3985
4190
  request_dict = _GenerateContentParameters_to_mldev(
3986
- self._api_client, parameter_model
4191
+ self._api_client, parameter_model, None, parameter_model
3987
4192
  )
3988
4193
  request_url_dict = request_dict.get('_url')
3989
4194
  if request_url_dict:
@@ -4020,10 +4225,14 @@ class Models(_api_module.BaseModule):
4020
4225
  response_dict = {} if not response.body else json.loads(response.body)
4021
4226
 
4022
4227
  if self._api_client.vertexai:
4023
- response_dict = _GenerateContentResponse_from_vertex(response_dict)
4228
+ response_dict = _GenerateContentResponse_from_vertex(
4229
+ response_dict, None, parameter_model
4230
+ )
4024
4231
 
4025
4232
  if not self._api_client.vertexai:
4026
- response_dict = _GenerateContentResponse_from_mldev(response_dict)
4233
+ response_dict = _GenerateContentResponse_from_mldev(
4234
+ response_dict, None, parameter_model
4235
+ )
4027
4236
 
4028
4237
  return_value = types.GenerateContentResponse._from_response(
4029
4238
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4051,7 +4260,7 @@ class Models(_api_module.BaseModule):
4051
4260
 
4052
4261
  if self._api_client.vertexai:
4053
4262
  request_dict = _GenerateContentParameters_to_vertex(
4054
- self._api_client, parameter_model
4263
+ self._api_client, parameter_model, None, parameter_model
4055
4264
  )
4056
4265
  request_url_dict = request_dict.get('_url')
4057
4266
  if request_url_dict:
@@ -4062,7 +4271,7 @@ class Models(_api_module.BaseModule):
4062
4271
  path = '{model}:streamGenerateContent?alt=sse'
4063
4272
  else:
4064
4273
  request_dict = _GenerateContentParameters_to_mldev(
4065
- self._api_client, parameter_model
4274
+ self._api_client, parameter_model, None, parameter_model
4066
4275
  )
4067
4276
  request_url_dict = request_dict.get('_url')
4068
4277
  if request_url_dict:
@@ -4102,10 +4311,14 @@ class Models(_api_module.BaseModule):
4102
4311
  response_dict = {} if not response.body else json.loads(response.body)
4103
4312
 
4104
4313
  if self._api_client.vertexai:
4105
- response_dict = _GenerateContentResponse_from_vertex(response_dict)
4314
+ response_dict = _GenerateContentResponse_from_vertex(
4315
+ response_dict, None, parameter_model
4316
+ )
4106
4317
 
4107
4318
  if not self._api_client.vertexai:
4108
- response_dict = _GenerateContentResponse_from_mldev(response_dict)
4319
+ response_dict = _GenerateContentResponse_from_mldev(
4320
+ response_dict, None, parameter_model
4321
+ )
4109
4322
 
4110
4323
  return_value = types.GenerateContentResponse._from_response(
4111
4324
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4156,7 +4369,7 @@ class Models(_api_module.BaseModule):
4156
4369
 
4157
4370
  if self._api_client.vertexai:
4158
4371
  request_dict = _EmbedContentParameters_to_vertex(
4159
- self._api_client, parameter_model
4372
+ self._api_client, parameter_model, None, parameter_model
4160
4373
  )
4161
4374
  request_url_dict = request_dict.get('_url')
4162
4375
  if request_url_dict:
@@ -4165,7 +4378,7 @@ class Models(_api_module.BaseModule):
4165
4378
  path = '{model}:predict'
4166
4379
  else:
4167
4380
  request_dict = _EmbedContentParameters_to_mldev(
4168
- self._api_client, parameter_model
4381
+ self._api_client, parameter_model, None, parameter_model
4169
4382
  )
4170
4383
  request_url_dict = request_dict.get('_url')
4171
4384
  if request_url_dict:
@@ -4195,10 +4408,14 @@ class Models(_api_module.BaseModule):
4195
4408
  response_dict = {} if not response.body else json.loads(response.body)
4196
4409
 
4197
4410
  if self._api_client.vertexai:
4198
- response_dict = _EmbedContentResponse_from_vertex(response_dict)
4411
+ response_dict = _EmbedContentResponse_from_vertex(
4412
+ response_dict, None, parameter_model
4413
+ )
4199
4414
 
4200
4415
  if not self._api_client.vertexai:
4201
- response_dict = _EmbedContentResponse_from_mldev(response_dict)
4416
+ response_dict = _EmbedContentResponse_from_mldev(
4417
+ response_dict, None, parameter_model
4418
+ )
4202
4419
 
4203
4420
  return_value = types.EmbedContentResponse._from_response(
4204
4421
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4228,7 +4445,7 @@ class Models(_api_module.BaseModule):
4228
4445
 
4229
4446
  if self._api_client.vertexai:
4230
4447
  request_dict = _GenerateImagesParameters_to_vertex(
4231
- self._api_client, parameter_model
4448
+ self._api_client, parameter_model, None, parameter_model
4232
4449
  )
4233
4450
  request_url_dict = request_dict.get('_url')
4234
4451
  if request_url_dict:
@@ -4237,7 +4454,7 @@ class Models(_api_module.BaseModule):
4237
4454
  path = '{model}:predict'
4238
4455
  else:
4239
4456
  request_dict = _GenerateImagesParameters_to_mldev(
4240
- self._api_client, parameter_model
4457
+ self._api_client, parameter_model, None, parameter_model
4241
4458
  )
4242
4459
  request_url_dict = request_dict.get('_url')
4243
4460
  if request_url_dict:
@@ -4267,10 +4484,14 @@ class Models(_api_module.BaseModule):
4267
4484
  response_dict = {} if not response.body else json.loads(response.body)
4268
4485
 
4269
4486
  if self._api_client.vertexai:
4270
- response_dict = _GenerateImagesResponse_from_vertex(response_dict)
4487
+ response_dict = _GenerateImagesResponse_from_vertex(
4488
+ response_dict, None, parameter_model
4489
+ )
4271
4490
 
4272
4491
  if not self._api_client.vertexai:
4273
- response_dict = _GenerateImagesResponse_from_mldev(response_dict)
4492
+ response_dict = _GenerateImagesResponse_from_mldev(
4493
+ response_dict, None, parameter_model
4494
+ )
4274
4495
 
4275
4496
  return_value = types.GenerateImagesResponse._from_response(
4276
4497
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4303,7 +4524,7 @@ class Models(_api_module.BaseModule):
4303
4524
  raise ValueError('This method is only supported in the Vertex AI client.')
4304
4525
  else:
4305
4526
  request_dict = _EditImageParameters_to_vertex(
4306
- self._api_client, parameter_model
4527
+ self._api_client, parameter_model, None, parameter_model
4307
4528
  )
4308
4529
  request_url_dict = request_dict.get('_url')
4309
4530
  if request_url_dict:
@@ -4334,7 +4555,9 @@ class Models(_api_module.BaseModule):
4334
4555
  response_dict = {} if not response.body else json.loads(response.body)
4335
4556
 
4336
4557
  if self._api_client.vertexai:
4337
- response_dict = _EditImageResponse_from_vertex(response_dict)
4558
+ response_dict = _EditImageResponse_from_vertex(
4559
+ response_dict, None, parameter_model
4560
+ )
4338
4561
 
4339
4562
  return_value = types.EditImageResponse._from_response(
4340
4563
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4367,7 +4590,7 @@ class Models(_api_module.BaseModule):
4367
4590
  raise ValueError('This method is only supported in the Vertex AI client.')
4368
4591
  else:
4369
4592
  request_dict = _UpscaleImageAPIParameters_to_vertex(
4370
- self._api_client, parameter_model
4593
+ self._api_client, parameter_model, None, parameter_model
4371
4594
  )
4372
4595
  request_url_dict = request_dict.get('_url')
4373
4596
  if request_url_dict:
@@ -4398,7 +4621,9 @@ class Models(_api_module.BaseModule):
4398
4621
  response_dict = {} if not response.body else json.loads(response.body)
4399
4622
 
4400
4623
  if self._api_client.vertexai:
4401
- response_dict = _UpscaleImageResponse_from_vertex(response_dict)
4624
+ response_dict = _UpscaleImageResponse_from_vertex(
4625
+ response_dict, None, parameter_model
4626
+ )
4402
4627
 
4403
4628
  return_value = types.UpscaleImageResponse._from_response(
4404
4629
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4475,7 +4700,7 @@ class Models(_api_module.BaseModule):
4475
4700
  raise ValueError('This method is only supported in the Vertex AI client.')
4476
4701
  else:
4477
4702
  request_dict = _RecontextImageParameters_to_vertex(
4478
- self._api_client, parameter_model
4703
+ self._api_client, parameter_model, None, parameter_model
4479
4704
  )
4480
4705
  request_url_dict = request_dict.get('_url')
4481
4706
  if request_url_dict:
@@ -4506,7 +4731,9 @@ class Models(_api_module.BaseModule):
4506
4731
  response_dict = {} if not response.body else json.loads(response.body)
4507
4732
 
4508
4733
  if self._api_client.vertexai:
4509
- response_dict = _RecontextImageResponse_from_vertex(response_dict)
4734
+ response_dict = _RecontextImageResponse_from_vertex(
4735
+ response_dict, None, parameter_model
4736
+ )
4510
4737
 
4511
4738
  return_value = types.RecontextImageResponse._from_response(
4512
4739
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4558,7 +4785,7 @@ class Models(_api_module.BaseModule):
4558
4785
  raise ValueError('This method is only supported in the Vertex AI client.')
4559
4786
  else:
4560
4787
  request_dict = _SegmentImageParameters_to_vertex(
4561
- self._api_client, parameter_model
4788
+ self._api_client, parameter_model, None, parameter_model
4562
4789
  )
4563
4790
  request_url_dict = request_dict.get('_url')
4564
4791
  if request_url_dict:
@@ -4589,7 +4816,9 @@ class Models(_api_module.BaseModule):
4589
4816
  response_dict = {} if not response.body else json.loads(response.body)
4590
4817
 
4591
4818
  if self._api_client.vertexai:
4592
- response_dict = _SegmentImageResponse_from_vertex(response_dict)
4819
+ response_dict = _SegmentImageResponse_from_vertex(
4820
+ response_dict, None, parameter_model
4821
+ )
4593
4822
 
4594
4823
  return_value = types.SegmentImageResponse._from_response(
4595
4824
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4610,7 +4839,7 @@ class Models(_api_module.BaseModule):
4610
4839
 
4611
4840
  if self._api_client.vertexai:
4612
4841
  request_dict = _GetModelParameters_to_vertex(
4613
- self._api_client, parameter_model
4842
+ self._api_client, parameter_model, None, parameter_model
4614
4843
  )
4615
4844
  request_url_dict = request_dict.get('_url')
4616
4845
  if request_url_dict:
@@ -4619,7 +4848,7 @@ class Models(_api_module.BaseModule):
4619
4848
  path = '{name}'
4620
4849
  else:
4621
4850
  request_dict = _GetModelParameters_to_mldev(
4622
- self._api_client, parameter_model
4851
+ self._api_client, parameter_model, None, parameter_model
4623
4852
  )
4624
4853
  request_url_dict = request_dict.get('_url')
4625
4854
  if request_url_dict:
@@ -4647,10 +4876,10 @@ class Models(_api_module.BaseModule):
4647
4876
  response_dict = {} if not response.body else json.loads(response.body)
4648
4877
 
4649
4878
  if self._api_client.vertexai:
4650
- response_dict = _Model_from_vertex(response_dict)
4879
+ response_dict = _Model_from_vertex(response_dict, None, parameter_model)
4651
4880
 
4652
4881
  if not self._api_client.vertexai:
4653
- response_dict = _Model_from_mldev(response_dict)
4882
+ response_dict = _Model_from_mldev(response_dict, None, parameter_model)
4654
4883
 
4655
4884
  return_value = types.Model._from_response(
4656
4885
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4670,7 +4899,7 @@ class Models(_api_module.BaseModule):
4670
4899
 
4671
4900
  if self._api_client.vertexai:
4672
4901
  request_dict = _ListModelsParameters_to_vertex(
4673
- self._api_client, parameter_model
4902
+ self._api_client, parameter_model, None, parameter_model
4674
4903
  )
4675
4904
  request_url_dict = request_dict.get('_url')
4676
4905
  if request_url_dict:
@@ -4679,7 +4908,7 @@ class Models(_api_module.BaseModule):
4679
4908
  path = '{models_url}'
4680
4909
  else:
4681
4910
  request_dict = _ListModelsParameters_to_mldev(
4682
- self._api_client, parameter_model
4911
+ self._api_client, parameter_model, None, parameter_model
4683
4912
  )
4684
4913
  request_url_dict = request_dict.get('_url')
4685
4914
  if request_url_dict:
@@ -4712,10 +4941,14 @@ class Models(_api_module.BaseModule):
4712
4941
  response_dict = {} if not response.body else json.loads(response.body)
4713
4942
 
4714
4943
  if self._api_client.vertexai:
4715
- response_dict = _ListModelsResponse_from_vertex(response_dict)
4944
+ response_dict = _ListModelsResponse_from_vertex(
4945
+ response_dict, None, parameter_model
4946
+ )
4716
4947
 
4717
4948
  if not self._api_client.vertexai:
4718
- response_dict = _ListModelsResponse_from_mldev(response_dict)
4949
+ response_dict = _ListModelsResponse_from_mldev(
4950
+ response_dict, None, parameter_model
4951
+ )
4719
4952
 
4720
4953
  return_value = types.ListModelsResponse._from_response(
4721
4954
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4741,7 +4974,7 @@ class Models(_api_module.BaseModule):
4741
4974
 
4742
4975
  if self._api_client.vertexai:
4743
4976
  request_dict = _UpdateModelParameters_to_vertex(
4744
- self._api_client, parameter_model
4977
+ self._api_client, parameter_model, None, parameter_model
4745
4978
  )
4746
4979
  request_url_dict = request_dict.get('_url')
4747
4980
  if request_url_dict:
@@ -4750,7 +4983,7 @@ class Models(_api_module.BaseModule):
4750
4983
  path = '{model}'
4751
4984
  else:
4752
4985
  request_dict = _UpdateModelParameters_to_mldev(
4753
- self._api_client, parameter_model
4986
+ self._api_client, parameter_model, None, parameter_model
4754
4987
  )
4755
4988
  request_url_dict = request_dict.get('_url')
4756
4989
  if request_url_dict:
@@ -4780,10 +5013,10 @@ class Models(_api_module.BaseModule):
4780
5013
  response_dict = {} if not response.body else json.loads(response.body)
4781
5014
 
4782
5015
  if self._api_client.vertexai:
4783
- response_dict = _Model_from_vertex(response_dict)
5016
+ response_dict = _Model_from_vertex(response_dict, None, parameter_model)
4784
5017
 
4785
5018
  if not self._api_client.vertexai:
4786
- response_dict = _Model_from_mldev(response_dict)
5019
+ response_dict = _Model_from_mldev(response_dict, None, parameter_model)
4787
5020
 
4788
5021
  return_value = types.Model._from_response(
4789
5022
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4807,7 +5040,7 @@ class Models(_api_module.BaseModule):
4807
5040
 
4808
5041
  if self._api_client.vertexai:
4809
5042
  request_dict = _DeleteModelParameters_to_vertex(
4810
- self._api_client, parameter_model
5043
+ self._api_client, parameter_model, None, parameter_model
4811
5044
  )
4812
5045
  request_url_dict = request_dict.get('_url')
4813
5046
  if request_url_dict:
@@ -4816,7 +5049,7 @@ class Models(_api_module.BaseModule):
4816
5049
  path = '{name}'
4817
5050
  else:
4818
5051
  request_dict = _DeleteModelParameters_to_mldev(
4819
- self._api_client, parameter_model
5052
+ self._api_client, parameter_model, None, parameter_model
4820
5053
  )
4821
5054
  request_url_dict = request_dict.get('_url')
4822
5055
  if request_url_dict:
@@ -4846,10 +5079,14 @@ class Models(_api_module.BaseModule):
4846
5079
  response_dict = {} if not response.body else json.loads(response.body)
4847
5080
 
4848
5081
  if self._api_client.vertexai:
4849
- response_dict = _DeleteModelResponse_from_vertex(response_dict)
5082
+ response_dict = _DeleteModelResponse_from_vertex(
5083
+ response_dict, None, parameter_model
5084
+ )
4850
5085
 
4851
5086
  if not self._api_client.vertexai:
4852
- response_dict = _DeleteModelResponse_from_mldev(response_dict)
5087
+ response_dict = _DeleteModelResponse_from_mldev(
5088
+ response_dict, None, parameter_model
5089
+ )
4853
5090
 
4854
5091
  return_value = types.DeleteModelResponse._from_response(
4855
5092
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4898,7 +5135,7 @@ class Models(_api_module.BaseModule):
4898
5135
 
4899
5136
  if self._api_client.vertexai:
4900
5137
  request_dict = _CountTokensParameters_to_vertex(
4901
- self._api_client, parameter_model
5138
+ self._api_client, parameter_model, None, parameter_model
4902
5139
  )
4903
5140
  request_url_dict = request_dict.get('_url')
4904
5141
  if request_url_dict:
@@ -4907,7 +5144,7 @@ class Models(_api_module.BaseModule):
4907
5144
  path = '{model}:countTokens'
4908
5145
  else:
4909
5146
  request_dict = _CountTokensParameters_to_mldev(
4910
- self._api_client, parameter_model
5147
+ self._api_client, parameter_model, None, parameter_model
4911
5148
  )
4912
5149
  request_url_dict = request_dict.get('_url')
4913
5150
  if request_url_dict:
@@ -4937,10 +5174,14 @@ class Models(_api_module.BaseModule):
4937
5174
  response_dict = {} if not response.body else json.loads(response.body)
4938
5175
 
4939
5176
  if self._api_client.vertexai:
4940
- response_dict = _CountTokensResponse_from_vertex(response_dict)
5177
+ response_dict = _CountTokensResponse_from_vertex(
5178
+ response_dict, None, parameter_model
5179
+ )
4941
5180
 
4942
5181
  if not self._api_client.vertexai:
4943
- response_dict = _CountTokensResponse_from_mldev(response_dict)
5182
+ response_dict = _CountTokensResponse_from_mldev(
5183
+ response_dict, None, parameter_model
5184
+ )
4944
5185
 
4945
5186
  return_value = types.CountTokensResponse._from_response(
4946
5187
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -4992,7 +5233,7 @@ class Models(_api_module.BaseModule):
4992
5233
  raise ValueError('This method is only supported in the Vertex AI client.')
4993
5234
  else:
4994
5235
  request_dict = _ComputeTokensParameters_to_vertex(
4995
- self._api_client, parameter_model
5236
+ self._api_client, parameter_model, None, parameter_model
4996
5237
  )
4997
5238
  request_url_dict = request_dict.get('_url')
4998
5239
  if request_url_dict:
@@ -5023,7 +5264,9 @@ class Models(_api_module.BaseModule):
5023
5264
  response_dict = {} if not response.body else json.loads(response.body)
5024
5265
 
5025
5266
  if self._api_client.vertexai:
5026
- response_dict = _ComputeTokensResponse_from_vertex(response_dict)
5267
+ response_dict = _ComputeTokensResponse_from_vertex(
5268
+ response_dict, None, parameter_model
5269
+ )
5027
5270
 
5028
5271
  return_value = types.ComputeTokensResponse._from_response(
5029
5272
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -5059,7 +5302,7 @@ class Models(_api_module.BaseModule):
5059
5302
 
5060
5303
  if self._api_client.vertexai:
5061
5304
  request_dict = _GenerateVideosParameters_to_vertex(
5062
- self._api_client, parameter_model
5305
+ self._api_client, parameter_model, None, parameter_model
5063
5306
  )
5064
5307
  request_url_dict = request_dict.get('_url')
5065
5308
  if request_url_dict:
@@ -5068,7 +5311,7 @@ class Models(_api_module.BaseModule):
5068
5311
  path = '{model}:predictLongRunning'
5069
5312
  else:
5070
5313
  request_dict = _GenerateVideosParameters_to_mldev(
5071
- self._api_client, parameter_model
5314
+ self._api_client, parameter_model, None, parameter_model
5072
5315
  )
5073
5316
  request_url_dict = request_dict.get('_url')
5074
5317
  if request_url_dict:
@@ -5098,10 +5341,14 @@ class Models(_api_module.BaseModule):
5098
5341
  response_dict = {} if not response.body else json.loads(response.body)
5099
5342
 
5100
5343
  if self._api_client.vertexai:
5101
- response_dict = _GenerateVideosOperation_from_vertex(response_dict)
5344
+ response_dict = _GenerateVideosOperation_from_vertex(
5345
+ response_dict, None, parameter_model
5346
+ )
5102
5347
 
5103
5348
  if not self._api_client.vertexai:
5104
- response_dict = _GenerateVideosOperation_from_mldev(response_dict)
5349
+ response_dict = _GenerateVideosOperation_from_mldev(
5350
+ response_dict, None, parameter_model
5351
+ )
5105
5352
 
5106
5353
  return_value = types.GenerateVideosOperation._from_response(
5107
5354
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -5813,7 +6060,7 @@ class AsyncModels(_api_module.BaseModule):
5813
6060
 
5814
6061
  if self._api_client.vertexai:
5815
6062
  request_dict = _GenerateContentParameters_to_vertex(
5816
- self._api_client, parameter_model
6063
+ self._api_client, parameter_model, None, parameter_model
5817
6064
  )
5818
6065
  request_url_dict = request_dict.get('_url')
5819
6066
  if request_url_dict:
@@ -5822,7 +6069,7 @@ class AsyncModels(_api_module.BaseModule):
5822
6069
  path = '{model}:generateContent'
5823
6070
  else:
5824
6071
  request_dict = _GenerateContentParameters_to_mldev(
5825
- self._api_client, parameter_model
6072
+ self._api_client, parameter_model, None, parameter_model
5826
6073
  )
5827
6074
  request_url_dict = request_dict.get('_url')
5828
6075
  if request_url_dict:
@@ -5859,10 +6106,14 @@ class AsyncModels(_api_module.BaseModule):
5859
6106
  response_dict = {} if not response.body else json.loads(response.body)
5860
6107
 
5861
6108
  if self._api_client.vertexai:
5862
- response_dict = _GenerateContentResponse_from_vertex(response_dict)
6109
+ response_dict = _GenerateContentResponse_from_vertex(
6110
+ response_dict, None, parameter_model
6111
+ )
5863
6112
 
5864
6113
  if not self._api_client.vertexai:
5865
- response_dict = _GenerateContentResponse_from_mldev(response_dict)
6114
+ response_dict = _GenerateContentResponse_from_mldev(
6115
+ response_dict, None, parameter_model
6116
+ )
5866
6117
 
5867
6118
  return_value = types.GenerateContentResponse._from_response(
5868
6119
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -5890,7 +6141,7 @@ class AsyncModels(_api_module.BaseModule):
5890
6141
 
5891
6142
  if self._api_client.vertexai:
5892
6143
  request_dict = _GenerateContentParameters_to_vertex(
5893
- self._api_client, parameter_model
6144
+ self._api_client, parameter_model, None, parameter_model
5894
6145
  )
5895
6146
  request_url_dict = request_dict.get('_url')
5896
6147
  if request_url_dict:
@@ -5901,7 +6152,7 @@ class AsyncModels(_api_module.BaseModule):
5901
6152
  path = '{model}:streamGenerateContent?alt=sse'
5902
6153
  else:
5903
6154
  request_dict = _GenerateContentParameters_to_mldev(
5904
- self._api_client, parameter_model
6155
+ self._api_client, parameter_model, None, parameter_model
5905
6156
  )
5906
6157
  request_url_dict = request_dict.get('_url')
5907
6158
  if request_url_dict:
@@ -5944,10 +6195,14 @@ class AsyncModels(_api_module.BaseModule):
5944
6195
  response_dict = {} if not response.body else json.loads(response.body)
5945
6196
 
5946
6197
  if self._api_client.vertexai:
5947
- response_dict = _GenerateContentResponse_from_vertex(response_dict)
6198
+ response_dict = _GenerateContentResponse_from_vertex(
6199
+ response_dict, None, parameter_model
6200
+ )
5948
6201
 
5949
6202
  if not self._api_client.vertexai:
5950
- response_dict = _GenerateContentResponse_from_mldev(response_dict)
6203
+ response_dict = _GenerateContentResponse_from_mldev(
6204
+ response_dict, None, parameter_model
6205
+ )
5951
6206
 
5952
6207
  return_value = types.GenerateContentResponse._from_response(
5953
6208
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6000,7 +6255,7 @@ class AsyncModels(_api_module.BaseModule):
6000
6255
 
6001
6256
  if self._api_client.vertexai:
6002
6257
  request_dict = _EmbedContentParameters_to_vertex(
6003
- self._api_client, parameter_model
6258
+ self._api_client, parameter_model, None, parameter_model
6004
6259
  )
6005
6260
  request_url_dict = request_dict.get('_url')
6006
6261
  if request_url_dict:
@@ -6009,7 +6264,7 @@ class AsyncModels(_api_module.BaseModule):
6009
6264
  path = '{model}:predict'
6010
6265
  else:
6011
6266
  request_dict = _EmbedContentParameters_to_mldev(
6012
- self._api_client, parameter_model
6267
+ self._api_client, parameter_model, None, parameter_model
6013
6268
  )
6014
6269
  request_url_dict = request_dict.get('_url')
6015
6270
  if request_url_dict:
@@ -6039,10 +6294,14 @@ class AsyncModels(_api_module.BaseModule):
6039
6294
  response_dict = {} if not response.body else json.loads(response.body)
6040
6295
 
6041
6296
  if self._api_client.vertexai:
6042
- response_dict = _EmbedContentResponse_from_vertex(response_dict)
6297
+ response_dict = _EmbedContentResponse_from_vertex(
6298
+ response_dict, None, parameter_model
6299
+ )
6043
6300
 
6044
6301
  if not self._api_client.vertexai:
6045
- response_dict = _EmbedContentResponse_from_mldev(response_dict)
6302
+ response_dict = _EmbedContentResponse_from_mldev(
6303
+ response_dict, None, parameter_model
6304
+ )
6046
6305
 
6047
6306
  return_value = types.EmbedContentResponse._from_response(
6048
6307
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6072,7 +6331,7 @@ class AsyncModels(_api_module.BaseModule):
6072
6331
 
6073
6332
  if self._api_client.vertexai:
6074
6333
  request_dict = _GenerateImagesParameters_to_vertex(
6075
- self._api_client, parameter_model
6334
+ self._api_client, parameter_model, None, parameter_model
6076
6335
  )
6077
6336
  request_url_dict = request_dict.get('_url')
6078
6337
  if request_url_dict:
@@ -6081,7 +6340,7 @@ class AsyncModels(_api_module.BaseModule):
6081
6340
  path = '{model}:predict'
6082
6341
  else:
6083
6342
  request_dict = _GenerateImagesParameters_to_mldev(
6084
- self._api_client, parameter_model
6343
+ self._api_client, parameter_model, None, parameter_model
6085
6344
  )
6086
6345
  request_url_dict = request_dict.get('_url')
6087
6346
  if request_url_dict:
@@ -6111,10 +6370,14 @@ class AsyncModels(_api_module.BaseModule):
6111
6370
  response_dict = {} if not response.body else json.loads(response.body)
6112
6371
 
6113
6372
  if self._api_client.vertexai:
6114
- response_dict = _GenerateImagesResponse_from_vertex(response_dict)
6373
+ response_dict = _GenerateImagesResponse_from_vertex(
6374
+ response_dict, None, parameter_model
6375
+ )
6115
6376
 
6116
6377
  if not self._api_client.vertexai:
6117
- response_dict = _GenerateImagesResponse_from_mldev(response_dict)
6378
+ response_dict = _GenerateImagesResponse_from_mldev(
6379
+ response_dict, None, parameter_model
6380
+ )
6118
6381
 
6119
6382
  return_value = types.GenerateImagesResponse._from_response(
6120
6383
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6147,7 +6410,7 @@ class AsyncModels(_api_module.BaseModule):
6147
6410
  raise ValueError('This method is only supported in the Vertex AI client.')
6148
6411
  else:
6149
6412
  request_dict = _EditImageParameters_to_vertex(
6150
- self._api_client, parameter_model
6413
+ self._api_client, parameter_model, None, parameter_model
6151
6414
  )
6152
6415
  request_url_dict = request_dict.get('_url')
6153
6416
  if request_url_dict:
@@ -6178,7 +6441,9 @@ class AsyncModels(_api_module.BaseModule):
6178
6441
  response_dict = {} if not response.body else json.loads(response.body)
6179
6442
 
6180
6443
  if self._api_client.vertexai:
6181
- response_dict = _EditImageResponse_from_vertex(response_dict)
6444
+ response_dict = _EditImageResponse_from_vertex(
6445
+ response_dict, None, parameter_model
6446
+ )
6182
6447
 
6183
6448
  return_value = types.EditImageResponse._from_response(
6184
6449
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6211,7 +6476,7 @@ class AsyncModels(_api_module.BaseModule):
6211
6476
  raise ValueError('This method is only supported in the Vertex AI client.')
6212
6477
  else:
6213
6478
  request_dict = _UpscaleImageAPIParameters_to_vertex(
6214
- self._api_client, parameter_model
6479
+ self._api_client, parameter_model, None, parameter_model
6215
6480
  )
6216
6481
  request_url_dict = request_dict.get('_url')
6217
6482
  if request_url_dict:
@@ -6242,7 +6507,9 @@ class AsyncModels(_api_module.BaseModule):
6242
6507
  response_dict = {} if not response.body else json.loads(response.body)
6243
6508
 
6244
6509
  if self._api_client.vertexai:
6245
- response_dict = _UpscaleImageResponse_from_vertex(response_dict)
6510
+ response_dict = _UpscaleImageResponse_from_vertex(
6511
+ response_dict, None, parameter_model
6512
+ )
6246
6513
 
6247
6514
  return_value = types.UpscaleImageResponse._from_response(
6248
6515
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6319,7 +6586,7 @@ class AsyncModels(_api_module.BaseModule):
6319
6586
  raise ValueError('This method is only supported in the Vertex AI client.')
6320
6587
  else:
6321
6588
  request_dict = _RecontextImageParameters_to_vertex(
6322
- self._api_client, parameter_model
6589
+ self._api_client, parameter_model, None, parameter_model
6323
6590
  )
6324
6591
  request_url_dict = request_dict.get('_url')
6325
6592
  if request_url_dict:
@@ -6350,7 +6617,9 @@ class AsyncModels(_api_module.BaseModule):
6350
6617
  response_dict = {} if not response.body else json.loads(response.body)
6351
6618
 
6352
6619
  if self._api_client.vertexai:
6353
- response_dict = _RecontextImageResponse_from_vertex(response_dict)
6620
+ response_dict = _RecontextImageResponse_from_vertex(
6621
+ response_dict, None, parameter_model
6622
+ )
6354
6623
 
6355
6624
  return_value = types.RecontextImageResponse._from_response(
6356
6625
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6405,7 +6674,7 @@ class AsyncModels(_api_module.BaseModule):
6405
6674
  raise ValueError('This method is only supported in the Vertex AI client.')
6406
6675
  else:
6407
6676
  request_dict = _SegmentImageParameters_to_vertex(
6408
- self._api_client, parameter_model
6677
+ self._api_client, parameter_model, None, parameter_model
6409
6678
  )
6410
6679
  request_url_dict = request_dict.get('_url')
6411
6680
  if request_url_dict:
@@ -6436,7 +6705,9 @@ class AsyncModels(_api_module.BaseModule):
6436
6705
  response_dict = {} if not response.body else json.loads(response.body)
6437
6706
 
6438
6707
  if self._api_client.vertexai:
6439
- response_dict = _SegmentImageResponse_from_vertex(response_dict)
6708
+ response_dict = _SegmentImageResponse_from_vertex(
6709
+ response_dict, None, parameter_model
6710
+ )
6440
6711
 
6441
6712
  return_value = types.SegmentImageResponse._from_response(
6442
6713
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6457,7 +6728,7 @@ class AsyncModels(_api_module.BaseModule):
6457
6728
 
6458
6729
  if self._api_client.vertexai:
6459
6730
  request_dict = _GetModelParameters_to_vertex(
6460
- self._api_client, parameter_model
6731
+ self._api_client, parameter_model, None, parameter_model
6461
6732
  )
6462
6733
  request_url_dict = request_dict.get('_url')
6463
6734
  if request_url_dict:
@@ -6466,7 +6737,7 @@ class AsyncModels(_api_module.BaseModule):
6466
6737
  path = '{name}'
6467
6738
  else:
6468
6739
  request_dict = _GetModelParameters_to_mldev(
6469
- self._api_client, parameter_model
6740
+ self._api_client, parameter_model, None, parameter_model
6470
6741
  )
6471
6742
  request_url_dict = request_dict.get('_url')
6472
6743
  if request_url_dict:
@@ -6496,10 +6767,10 @@ class AsyncModels(_api_module.BaseModule):
6496
6767
  response_dict = {} if not response.body else json.loads(response.body)
6497
6768
 
6498
6769
  if self._api_client.vertexai:
6499
- response_dict = _Model_from_vertex(response_dict)
6770
+ response_dict = _Model_from_vertex(response_dict, None, parameter_model)
6500
6771
 
6501
6772
  if not self._api_client.vertexai:
6502
- response_dict = _Model_from_mldev(response_dict)
6773
+ response_dict = _Model_from_mldev(response_dict, None, parameter_model)
6503
6774
 
6504
6775
  return_value = types.Model._from_response(
6505
6776
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6519,7 +6790,7 @@ class AsyncModels(_api_module.BaseModule):
6519
6790
 
6520
6791
  if self._api_client.vertexai:
6521
6792
  request_dict = _ListModelsParameters_to_vertex(
6522
- self._api_client, parameter_model
6793
+ self._api_client, parameter_model, None, parameter_model
6523
6794
  )
6524
6795
  request_url_dict = request_dict.get('_url')
6525
6796
  if request_url_dict:
@@ -6528,7 +6799,7 @@ class AsyncModels(_api_module.BaseModule):
6528
6799
  path = '{models_url}'
6529
6800
  else:
6530
6801
  request_dict = _ListModelsParameters_to_mldev(
6531
- self._api_client, parameter_model
6802
+ self._api_client, parameter_model, None, parameter_model
6532
6803
  )
6533
6804
  request_url_dict = request_dict.get('_url')
6534
6805
  if request_url_dict:
@@ -6563,10 +6834,14 @@ class AsyncModels(_api_module.BaseModule):
6563
6834
  response_dict = {} if not response.body else json.loads(response.body)
6564
6835
 
6565
6836
  if self._api_client.vertexai:
6566
- response_dict = _ListModelsResponse_from_vertex(response_dict)
6837
+ response_dict = _ListModelsResponse_from_vertex(
6838
+ response_dict, None, parameter_model
6839
+ )
6567
6840
 
6568
6841
  if not self._api_client.vertexai:
6569
- response_dict = _ListModelsResponse_from_mldev(response_dict)
6842
+ response_dict = _ListModelsResponse_from_mldev(
6843
+ response_dict, None, parameter_model
6844
+ )
6570
6845
 
6571
6846
  return_value = types.ListModelsResponse._from_response(
6572
6847
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6592,7 +6867,7 @@ class AsyncModels(_api_module.BaseModule):
6592
6867
 
6593
6868
  if self._api_client.vertexai:
6594
6869
  request_dict = _UpdateModelParameters_to_vertex(
6595
- self._api_client, parameter_model
6870
+ self._api_client, parameter_model, None, parameter_model
6596
6871
  )
6597
6872
  request_url_dict = request_dict.get('_url')
6598
6873
  if request_url_dict:
@@ -6601,7 +6876,7 @@ class AsyncModels(_api_module.BaseModule):
6601
6876
  path = '{model}'
6602
6877
  else:
6603
6878
  request_dict = _UpdateModelParameters_to_mldev(
6604
- self._api_client, parameter_model
6879
+ self._api_client, parameter_model, None, parameter_model
6605
6880
  )
6606
6881
  request_url_dict = request_dict.get('_url')
6607
6882
  if request_url_dict:
@@ -6631,10 +6906,10 @@ class AsyncModels(_api_module.BaseModule):
6631
6906
  response_dict = {} if not response.body else json.loads(response.body)
6632
6907
 
6633
6908
  if self._api_client.vertexai:
6634
- response_dict = _Model_from_vertex(response_dict)
6909
+ response_dict = _Model_from_vertex(response_dict, None, parameter_model)
6635
6910
 
6636
6911
  if not self._api_client.vertexai:
6637
- response_dict = _Model_from_mldev(response_dict)
6912
+ response_dict = _Model_from_mldev(response_dict, None, parameter_model)
6638
6913
 
6639
6914
  return_value = types.Model._from_response(
6640
6915
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6658,7 +6933,7 @@ class AsyncModels(_api_module.BaseModule):
6658
6933
 
6659
6934
  if self._api_client.vertexai:
6660
6935
  request_dict = _DeleteModelParameters_to_vertex(
6661
- self._api_client, parameter_model
6936
+ self._api_client, parameter_model, None, parameter_model
6662
6937
  )
6663
6938
  request_url_dict = request_dict.get('_url')
6664
6939
  if request_url_dict:
@@ -6667,7 +6942,7 @@ class AsyncModels(_api_module.BaseModule):
6667
6942
  path = '{name}'
6668
6943
  else:
6669
6944
  request_dict = _DeleteModelParameters_to_mldev(
6670
- self._api_client, parameter_model
6945
+ self._api_client, parameter_model, None, parameter_model
6671
6946
  )
6672
6947
  request_url_dict = request_dict.get('_url')
6673
6948
  if request_url_dict:
@@ -6697,10 +6972,14 @@ class AsyncModels(_api_module.BaseModule):
6697
6972
  response_dict = {} if not response.body else json.loads(response.body)
6698
6973
 
6699
6974
  if self._api_client.vertexai:
6700
- response_dict = _DeleteModelResponse_from_vertex(response_dict)
6975
+ response_dict = _DeleteModelResponse_from_vertex(
6976
+ response_dict, None, parameter_model
6977
+ )
6701
6978
 
6702
6979
  if not self._api_client.vertexai:
6703
- response_dict = _DeleteModelResponse_from_mldev(response_dict)
6980
+ response_dict = _DeleteModelResponse_from_mldev(
6981
+ response_dict, None, parameter_model
6982
+ )
6704
6983
 
6705
6984
  return_value = types.DeleteModelResponse._from_response(
6706
6985
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6749,7 +7028,7 @@ class AsyncModels(_api_module.BaseModule):
6749
7028
 
6750
7029
  if self._api_client.vertexai:
6751
7030
  request_dict = _CountTokensParameters_to_vertex(
6752
- self._api_client, parameter_model
7031
+ self._api_client, parameter_model, None, parameter_model
6753
7032
  )
6754
7033
  request_url_dict = request_dict.get('_url')
6755
7034
  if request_url_dict:
@@ -6758,7 +7037,7 @@ class AsyncModels(_api_module.BaseModule):
6758
7037
  path = '{model}:countTokens'
6759
7038
  else:
6760
7039
  request_dict = _CountTokensParameters_to_mldev(
6761
- self._api_client, parameter_model
7040
+ self._api_client, parameter_model, None, parameter_model
6762
7041
  )
6763
7042
  request_url_dict = request_dict.get('_url')
6764
7043
  if request_url_dict:
@@ -6788,10 +7067,14 @@ class AsyncModels(_api_module.BaseModule):
6788
7067
  response_dict = {} if not response.body else json.loads(response.body)
6789
7068
 
6790
7069
  if self._api_client.vertexai:
6791
- response_dict = _CountTokensResponse_from_vertex(response_dict)
7070
+ response_dict = _CountTokensResponse_from_vertex(
7071
+ response_dict, None, parameter_model
7072
+ )
6792
7073
 
6793
7074
  if not self._api_client.vertexai:
6794
- response_dict = _CountTokensResponse_from_mldev(response_dict)
7075
+ response_dict = _CountTokensResponse_from_mldev(
7076
+ response_dict, None, parameter_model
7077
+ )
6795
7078
 
6796
7079
  return_value = types.CountTokensResponse._from_response(
6797
7080
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6842,7 +7125,7 @@ class AsyncModels(_api_module.BaseModule):
6842
7125
  raise ValueError('This method is only supported in the Vertex AI client.')
6843
7126
  else:
6844
7127
  request_dict = _ComputeTokensParameters_to_vertex(
6845
- self._api_client, parameter_model
7128
+ self._api_client, parameter_model, None, parameter_model
6846
7129
  )
6847
7130
  request_url_dict = request_dict.get('_url')
6848
7131
  if request_url_dict:
@@ -6873,7 +7156,9 @@ class AsyncModels(_api_module.BaseModule):
6873
7156
  response_dict = {} if not response.body else json.loads(response.body)
6874
7157
 
6875
7158
  if self._api_client.vertexai:
6876
- response_dict = _ComputeTokensResponse_from_vertex(response_dict)
7159
+ response_dict = _ComputeTokensResponse_from_vertex(
7160
+ response_dict, None, parameter_model
7161
+ )
6877
7162
 
6878
7163
  return_value = types.ComputeTokensResponse._from_response(
6879
7164
  response=response_dict, kwargs=parameter_model.model_dump()
@@ -6909,7 +7194,7 @@ class AsyncModels(_api_module.BaseModule):
6909
7194
 
6910
7195
  if self._api_client.vertexai:
6911
7196
  request_dict = _GenerateVideosParameters_to_vertex(
6912
- self._api_client, parameter_model
7197
+ self._api_client, parameter_model, None, parameter_model
6913
7198
  )
6914
7199
  request_url_dict = request_dict.get('_url')
6915
7200
  if request_url_dict:
@@ -6918,7 +7203,7 @@ class AsyncModels(_api_module.BaseModule):
6918
7203
  path = '{model}:predictLongRunning'
6919
7204
  else:
6920
7205
  request_dict = _GenerateVideosParameters_to_mldev(
6921
- self._api_client, parameter_model
7206
+ self._api_client, parameter_model, None, parameter_model
6922
7207
  )
6923
7208
  request_url_dict = request_dict.get('_url')
6924
7209
  if request_url_dict:
@@ -6948,10 +7233,14 @@ class AsyncModels(_api_module.BaseModule):
6948
7233
  response_dict = {} if not response.body else json.loads(response.body)
6949
7234
 
6950
7235
  if self._api_client.vertexai:
6951
- response_dict = _GenerateVideosOperation_from_vertex(response_dict)
7236
+ response_dict = _GenerateVideosOperation_from_vertex(
7237
+ response_dict, None, parameter_model
7238
+ )
6952
7239
 
6953
7240
  if not self._api_client.vertexai:
6954
- response_dict = _GenerateVideosOperation_from_mldev(response_dict)
7241
+ response_dict = _GenerateVideosOperation_from_mldev(
7242
+ response_dict, None, parameter_model
7243
+ )
6955
7244
 
6956
7245
  return_value = types.GenerateVideosOperation._from_response(
6957
7246
  response=response_dict, kwargs=parameter_model.model_dump()