Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions vertexai/_genai/evals.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand All @@ -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


Expand Down
14 changes: 14 additions & 0 deletions vertexai/_genai/types/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -388,6 +388,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
Expand Down Expand Up @@ -869,6 +870,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
Expand Down Expand Up @@ -1753,6 +1760,12 @@
"SchemaPromptSpecMultimodalPrompt",
"SchemaPromptSpecMultimodalPromptDict",
"SchemaPromptSpecMultimodalPromptOrDict",
"SchemaPromptSpecAppBuilderDataLinkedResource",
"SchemaPromptSpecAppBuilderDataLinkedResourceDict",
"SchemaPromptSpecAppBuilderDataLinkedResourceOrDict",
"SchemaPromptSpecAppBuilderData",
"SchemaPromptSpecAppBuilderDataDict",
"SchemaPromptSpecAppBuilderDataOrDict",
"SchemaPromptSpecPartList",
"SchemaPromptSpecPartListDict",
"SchemaPromptSpecPartListOrDict",
Expand Down Expand Up @@ -1933,6 +1946,7 @@
"Language",
"MachineConfig",
"State",
"Framework",
"EvaluationItemType",
"SamplingMethod",
"RubricContentType",
Expand Down
Loading
Loading