Skip to content

Commit 6537db7

Browse files
jscudcopybara-github
authored andcommitted
feat: Update data types from discovery doc.
PiperOrigin-RevId: 865450418
1 parent 6ac28a5 commit 6537db7

File tree

3 files changed

+323
-194
lines changed

3 files changed

+323
-194
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: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -398,6 +398,7 @@
398398
from .common import ExecuteSandboxEnvironmentResponse
399399
from .common import ExecuteSandboxEnvironmentResponseDict
400400
from .common import ExecuteSandboxEnvironmentResponseOrDict
401+
from .common import Framework
401402
from .common import GcsSource
402403
from .common import GcsSourceDict
403404
from .common import GcsSourceOrDict
@@ -886,6 +887,12 @@
886887
from .common import SchemaPromptInstanceVariableValue
887888
from .common import SchemaPromptInstanceVariableValueDict
888889
from .common import SchemaPromptInstanceVariableValueOrDict
890+
from .common import SchemaPromptSpecAppBuilderData
891+
from .common import SchemaPromptSpecAppBuilderDataDict
892+
from .common import SchemaPromptSpecAppBuilderDataLinkedResource
893+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceDict
894+
from .common import SchemaPromptSpecAppBuilderDataLinkedResourceOrDict
895+
from .common import SchemaPromptSpecAppBuilderDataOrDict
889896
from .common import SchemaPromptSpecMultimodalPrompt
890897
from .common import SchemaPromptSpecMultimodalPromptDict
891898
from .common import SchemaPromptSpecMultimodalPromptOrDict
@@ -1422,9 +1429,6 @@
14221429
"ReasoningEngineSpecSourceCodeSpecPythonSpec",
14231430
"ReasoningEngineSpecSourceCodeSpecPythonSpecDict",
14241431
"ReasoningEngineSpecSourceCodeSpecPythonSpecOrDict",
1425-
"ReasoningEngineSpecSourceCodeSpecImageSpec",
1426-
"ReasoningEngineSpecSourceCodeSpecImageSpecDict",
1427-
"ReasoningEngineSpecSourceCodeSpecImageSpecOrDict",
14281432
"ReasoningEngineSpecSourceCodeSpec",
14291433
"ReasoningEngineSpecSourceCodeSpecDict",
14301434
"ReasoningEngineSpecSourceCodeSpecOrDict",
@@ -1788,6 +1792,12 @@
17881792
"SchemaPromptSpecMultimodalPrompt",
17891793
"SchemaPromptSpecMultimodalPromptDict",
17901794
"SchemaPromptSpecMultimodalPromptOrDict",
1795+
"SchemaPromptSpecAppBuilderDataLinkedResource",
1796+
"SchemaPromptSpecAppBuilderDataLinkedResourceDict",
1797+
"SchemaPromptSpecAppBuilderDataLinkedResourceOrDict",
1798+
"SchemaPromptSpecAppBuilderData",
1799+
"SchemaPromptSpecAppBuilderDataDict",
1800+
"SchemaPromptSpecAppBuilderDataOrDict",
17911801
"SchemaPromptSpecPartList",
17921802
"SchemaPromptSpecPartListDict",
17931803
"SchemaPromptSpecPartListOrDict",
@@ -1908,6 +1918,9 @@
19081918
"AgentEngine",
19091919
"AgentEngineDict",
19101920
"AgentEngineOrDict",
1921+
"ReasoningEngineSpecSourceCodeSpecImageSpec",
1922+
"ReasoningEngineSpecSourceCodeSpecImageSpecDict",
1923+
"ReasoningEngineSpecSourceCodeSpecImageSpecOrDict",
19111924
"AgentEngineConfig",
19121925
"AgentEngineConfigDict",
19131926
"AgentEngineConfigOrDict",
@@ -1950,6 +1963,7 @@
19501963
"OptimizeJobConfig",
19511964
"OptimizeJobConfigDict",
19521965
"OptimizeJobConfigOrDict",
1966+
"ComputationBasedMetricType",
19531967
"PairwiseChoice",
19541968
"Strategy",
19551969
"AcceleratorType",
@@ -1962,10 +1976,10 @@
19621976
"Language",
19631977
"MachineConfig",
19641978
"State",
1979+
"Framework",
19651980
"EvaluationItemType",
19661981
"SamplingMethod",
19671982
"RubricContentType",
1968-
"ComputationBasedMetricType",
19691983
"EvaluationRunState",
19701984
"OptimizeTarget",
19711985
"MemoryMetadataMergeStrategy",

0 commit comments

Comments
 (0)