Skip to content

Commit 76a2737

Browse files
sararobcopybara-github
authored andcommitted
chore: enable converters for nested dictionary fields
PiperOrigin-RevId: 877487964
1 parent e9f1c88 commit 76a2737

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
@@ -384,7 +384,11 @@ def _EvaluationRun_from_vertex(
384384
)
385385

386386
if getv(from_object, ["inferenceConfigs"]) is not None:
387-
setv(to_object, ["inference_configs"], getv(from_object, ["inferenceConfigs"]))
387+
setv(
388+
to_object,
389+
["inference_configs"],
390+
{k: v for k, v in getv(from_object, ["inferenceConfigs"]).items()},
391+
)
388392

389393
if getv(from_object, ["labels"]) is not None:
390394
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)