Skip to content

Commit a451fa3

Browse files
jscudcopybara-github
authored andcommitted
feat: Update data types from discovery doc.
PiperOrigin-RevId: 857645392
1 parent 53532eb commit a451fa3

File tree

3 files changed

+285
-138
lines changed

3 files changed

+285
-138
lines changed

vertexai/_genai/evals.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,13 @@ def _CustomCodeExecutionSpec_from_vertex(
130130
getv(from_object, ["evaluation_function"]),
131131
)
132132

133+
if getv(from_object, ["evaluationFunction"]) is not None:
134+
setv(
135+
to_object,
136+
["evaluation_function"],
137+
getv(from_object, ["evaluationFunction"]),
138+
)
139+
133140
return to_object
134141

135142

@@ -145,6 +152,13 @@ def _CustomCodeExecutionSpec_to_vertex(
145152
getv(from_object, ["remote_custom_function"]),
146153
)
147154

155+
if getv(from_object, ["evaluation_function"]) is not None:
156+
setv(
157+
to_object,
158+
["evaluationFunction"],
159+
getv(from_object, ["evaluation_function"]),
160+
)
161+
148162
return to_object
149163

150164

vertexai/_genai/types/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,7 @@
388388
from .common import ExecuteSandboxEnvironmentResponse
389389
from .common import ExecuteSandboxEnvironmentResponseDict
390390
from .common import ExecuteSandboxEnvironmentResponseOrDict
391+
from .common import Framework
391392
from .common import GcsSource
392393
from .common import GcsSourceDict
393394
from .common import GcsSourceOrDict
@@ -869,6 +870,12 @@
869870
from .common import SchemaPromptInstanceVariableValue
870871
from .common import SchemaPromptInstanceVariableValueDict
871872
from .common import SchemaPromptInstanceVariableValueOrDict
873+
from .common import SchemaPromptSpecAppBuilderData
874+
from .common import SchemaPromptSpecAppBuilderDataDict
875+
from .common import SchemaPromptSpecAppBuilderDataLinkedResource
876+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceDict
877+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceOrDict
878+
from .common import SchemaPromptSpecAppBuilderDataOrDict
872879
from .common import SchemaPromptSpecMultimodalPrompt
873880
from .common import SchemaPromptSpecMultimodalPromptDict
874881
from .common import SchemaPromptSpecMultimodalPromptOrDict
@@ -1753,6 +1760,12 @@
17531760
"SchemaPromptSpecMultimodalPrompt",
17541761
"SchemaPromptSpecMultimodalPromptDict",
17551762
"SchemaPromptSpecMultimodalPromptOrDict",
1763+
"SchemaPromptSpecAppBuilderDataLinkedResource",
1764+
"SchemaPromptSpecAppBuilderDataLinkedResourceDict",
1765+
"SchemaPromptSpecAppBuilderDataLinkedResourceOrDict",
1766+
"SchemaPromptSpecAppBuilderData",
1767+
"SchemaPromptSpecAppBuilderDataDict",
1768+
"SchemaPromptSpecAppBuilderDataOrDict",
17561769
"SchemaPromptSpecPartList",
17571770
"SchemaPromptSpecPartListDict",
17581771
"SchemaPromptSpecPartListOrDict",
@@ -1933,6 +1946,7 @@
19331946
"Language",
19341947
"MachineConfig",
19351948
"State",
1949+
"Framework",
19361950
"EvaluationItemType",
19371951
"SamplingMethod",
19381952
"RubricContentType",

0 commit comments

Comments
 (0)