diff --git a/vertexai/_genai/evals.py b/vertexai/_genai/evals.py index f090cf3ac4..0848d3cf64 100644 --- a/vertexai/_genai/evals.py +++ b/vertexai/_genai/evals.py @@ -130,6 +130,13 @@ def _CustomCodeExecutionSpec_from_vertex( getv(from_object, ["evaluation_function"]), ) + if getv(from_object, ["evaluationFunction"]) is not None: + setv( + to_object, + ["evaluation_function"], + getv(from_object, ["evaluationFunction"]), + ) + return to_object @@ -145,6 +152,13 @@ def _CustomCodeExecutionSpec_to_vertex( getv(from_object, ["remote_custom_function"]), ) + if getv(from_object, ["evaluation_function"]) is not None: + setv( + to_object, + ["evaluationFunction"], + getv(from_object, ["evaluation_function"]), + ) + return to_object diff --git a/vertexai/_genai/types/__init__.py b/vertexai/_genai/types/__init__.py index e0ed0ca64d..cb4969b1eb 100644 --- a/vertexai/_genai/types/__init__.py +++ b/vertexai/_genai/types/__init__.py @@ -392,6 +392,7 @@ from .common import ExecuteSandboxEnvironmentResponse from .common import ExecuteSandboxEnvironmentResponseDict from .common import ExecuteSandboxEnvironmentResponseOrDict +from .common import Framework from .common import GcsSource from .common import GcsSourceDict from .common import GcsSourceOrDict @@ -873,6 +874,12 @@ from .common import SchemaPromptInstanceVariableValue from .common import SchemaPromptInstanceVariableValueDict from .common import SchemaPromptInstanceVariableValueOrDict +from .common import SchemaPromptSpecAppBuilderData +from .common import SchemaPromptSpecAppBuilderDataDict +from .common import SchemaPromptSpecAppBuilderDataLinkedResource +from .common import SchemaPromptSpecAppBuilderDataLinkedResourceDict +from .common import SchemaPromptSpecAppBuilderDataLinkedResourceOrDict +from .common import SchemaPromptSpecAppBuilderDataOrDict from .common import SchemaPromptSpecMultimodalPrompt from .common import SchemaPromptSpecMultimodalPromptDict from .common import SchemaPromptSpecMultimodalPromptOrDict @@ -1772,6 +1779,12 @@ "SchemaPromptSpecMultimodalPrompt", "SchemaPromptSpecMultimodalPromptDict", "SchemaPromptSpecMultimodalPromptOrDict", + "SchemaPromptSpecAppBuilderDataLinkedResource", + "SchemaPromptSpecAppBuilderDataLinkedResourceDict", + "SchemaPromptSpecAppBuilderDataLinkedResourceOrDict", + "SchemaPromptSpecAppBuilderData", + "SchemaPromptSpecAppBuilderDataDict", + "SchemaPromptSpecAppBuilderDataOrDict", "SchemaPromptSpecPartList", "SchemaPromptSpecPartListDict", "SchemaPromptSpecPartListOrDict", @@ -1940,6 +1953,7 @@ "Language", "MachineConfig", "State", + "Framework", "EvaluationItemType", "SamplingMethod", "RubricContentType",