google-genai 1.60.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/_interactions/_base_client.py +5 -2
- google/genai/_interactions/_compat.py +3 -3
- google/genai/_interactions/_utils/_json.py +50 -0
- google/genai/_interactions/resources/interactions.py +50 -28
- google/genai/_interactions/types/__init__.py +2 -1
- google/genai/_interactions/types/content_delta.py +1 -1
- google/genai/_interactions/types/function_result_content.py +2 -1
- google/genai/_interactions/types/function_result_content_param.py +4 -4
- google/genai/_interactions/types/{interaction_event.py → interaction_complete_event.py} +3 -3
- google/genai/_interactions/types/interaction_create_params.py +4 -4
- google/genai/_interactions/types/interaction_get_params.py +3 -0
- google/genai/_interactions/types/interaction_sse_event.py +11 -2
- google/genai/_interactions/types/interaction_start_event.py +36 -0
- google/genai/batches.py +3 -0
- google/genai/errors.py +19 -6
- google/genai/files.py +15 -15
- google/genai/live.py +22 -2
- google/genai/live_music.py +14 -1
- google/genai/models.py +486 -197
- google/genai/tests/batches/test_create_with_inlined_requests.py +31 -15
- google/genai/tests/batches/test_get.py +1 -1
- google/genai/tests/client/test_client_close.py +0 -1
- google/genai/tests/errors/test_api_error.py +38 -0
- google/genai/tests/files/test_register_table.py +1 -1
- google/genai/tests/transformers/test_schema.py +10 -1
- google/genai/tests/tunings/test_tune.py +87 -0
- google/genai/tunings.py +211 -20
- google/genai/types.py +178 -14
- google/genai/version.py +1 -1
- {google_genai-1.60.0.dist-info → google_genai-1.62.0.dist-info}/METADATA +1 -1
- {google_genai-1.60.0.dist-info → google_genai-1.62.0.dist-info}/RECORD +34 -32
- {google_genai-1.60.0.dist-info → google_genai-1.62.0.dist-info}/WHEEL +1 -1
- {google_genai-1.60.0.dist-info → google_genai-1.62.0.dist-info}/licenses/LICENSE +0 -0
- {google_genai-1.60.0.dist-info → google_genai-1.62.0.dist-info}/top_level.txt +0 -0
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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'])),
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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']),
|
|
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']),
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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']),
|
|
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']),
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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']),
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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(
|
|
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()
|