Skip to content

Commit fef8cfa

Browse files
sararobcopybara-github
authored andcommitted
chore: enable converters for nested dictionary fields
PiperOrigin-RevId: 878102157
1 parent 84c5602 commit fef8cfa

2 files changed

Lines changed: 20 additions & 4 deletions

File tree

vertexai/_genai/evals.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,7 +414,11 @@ def _EvaluationRun_from_vertex(
414414
)
415415

416416
if getv(from_object, ["inferenceConfigs"]) is not None:
417-
setv(to_object, ["inference_configs"], getv(from_object, ["inferenceConfigs"]))
417+
setv(
418+
to_object,
419+
["inference_configs"],
420+
{k: v for k, v in getv(from_object, ["inferenceConfigs"]).items()},
421+
)
418422

419423
if getv(from_object, ["labels"]) is not None:
420424
setv(to_object, ["labels"], getv(from_object, ["labels"]))

vertexai/_genai/memories.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,11 @@ def _AgentEngineMemoryConfig_to_vertex(
8484
)
8585

8686
if getv(from_object, ["metadata"]) is not None:
87-
setv(parent_object, ["metadata"], getv(from_object, ["metadata"]))
87+
setv(
88+
parent_object,
89+
["metadata"],
90+
{k: v for k, v in getv(from_object, ["metadata"]).items()},
91+
)
8892

8993
return to_object
9094

@@ -163,7 +167,11 @@ def _GenerateAgentEngineMemoriesConfig_to_vertex(
163167
)
164168

165169
if getv(from_object, ["metadata"]) is not None:
166-
setv(parent_object, ["metadata"], getv(from_object, ["metadata"]))
170+
setv(
171+
parent_object,
172+
["metadata"],
173+
{k: v for k, v in getv(from_object, ["metadata"]).items()},
174+
)
167175

168176
if getv(from_object, ["metadata_merge_strategy"]) is not None:
169177
setv(
@@ -447,7 +455,11 @@ def _UpdateAgentEngineMemoryConfig_to_vertex(
447455
)
448456

449457
if getv(from_object, ["metadata"]) is not None:
450-
setv(parent_object, ["metadata"], getv(from_object, ["metadata"]))
458+
setv(
459+
parent_object,
460+
["metadata"],
461+
{k: v for k, v in getv(from_object, ["metadata"]).items()},
462+
)
451463

452464
if getv(from_object, ["update_mask"]) is not None:
453465
setv(

0 commit comments

Comments
 (0)