Skip to content

Commit a203b7f

Browse files
jscudcopybara-github
authored andcommitted
feat: Update data types from discovery doc.
PiperOrigin-RevId: 859221083
1 parent 0179aa5 commit a203b7f

3 files changed

Lines changed: 285 additions & 138 deletions

File tree

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
@@ -392,6 +392,7 @@
392392
from .common import ExecuteSandboxEnvironmentResponse
393393
from .common import ExecuteSandboxEnvironmentResponseDict
394394
from .common import ExecuteSandboxEnvironmentResponseOrDict
395+
from .common import Framework
395396
from .common import GcsSource
396397
from .common import GcsSourceDict
397398
from .common import GcsSourceOrDict
@@ -873,6 +874,12 @@
873874
from .common import SchemaPromptInstanceVariableValue
874875
from .common import SchemaPromptInstanceVariableValueDict
875876
from .common import SchemaPromptInstanceVariableValueOrDict
877+
from .common import SchemaPromptSpecAppBuilderData
878+
from .common import SchemaPromptSpecAppBuilderDataDict
879+
from .common import SchemaPromptSpecAppBuilderDataLinkedResource
880+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceDict
881+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceOrDict
882+
from .common import SchemaPromptSpecAppBuilderDataOrDict
876883
from .common import SchemaPromptSpecMultimodalPrompt
877884
from .common import SchemaPromptSpecMultimodalPromptDict
878885
from .common import SchemaPromptSpecMultimodalPromptOrDict
@@ -1772,6 +1779,12 @@
17721779
"SchemaPromptSpecMultimodalPrompt",
17731780
"SchemaPromptSpecMultimodalPromptDict",
17741781
"SchemaPromptSpecMultimodalPromptOrDict",
1782+
"SchemaPromptSpecAppBuilderDataLinkedResource",
1783+
"SchemaPromptSpecAppBuilderDataLinkedResourceDict",
1784+
"SchemaPromptSpecAppBuilderDataLinkedResourceOrDict",
1785+
"SchemaPromptSpecAppBuilderData",
1786+
"SchemaPromptSpecAppBuilderDataDict",
1787+
"SchemaPromptSpecAppBuilderDataOrDict",
17751788
"SchemaPromptSpecPartList",
17761789
"SchemaPromptSpecPartListDict",
17771790
"SchemaPromptSpecPartListOrDict",
@@ -1940,6 +1953,7 @@
19401953
"Language",
19411954
"MachineConfig",
19421955
"State",
1956+
"Framework",
19431957
"EvaluationItemType",
19441958
"SamplingMethod",
19451959
"RubricContentType",

0 commit comments

Comments
 (0)