feat: add ReasoningEngineRuntimeRevisionService and update related fields in ReasoningEngine

.kokoro/samples/python3.9/common.cfg

@@ -0,0 +1,40 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+# Build logs will be here
+action {
+  define_artifacts {
+    regex: "**/*sponge_log.xml"
+  }
+}
+
+# Specify which tests to run
+env_vars: {
+    key: "RUN_TESTS_SESSION"
+    value: "py-3.9"
+}
+
+# Declare build specific Cloud project.
+env_vars: {
+    key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
+    value: "python-docs-samples-tests-py39"
+}
+
+env_vars: {
+    key: "TRAMPOLINE_BUILD_FILE"
+    value: "github/python-aiplatform/.kokoro/test-samples.sh"
+}
+
+# Configure the docker image for kokoro-trampoline.
+env_vars: {
+    key: "TRAMPOLINE_IMAGE"
+    value: "gcr.io/cloud-devrel-kokoro-resources/python-samples-testing-docker"
+}
+
+# Download secrets for samples
+gfile_resources: "/bigstore/cloud-devrel-kokoro-resources/python-docs-samples"
+
+# Download trampoline resources.
+gfile_resources: "/bigstore/cloud-devrel-kokoro-resources/trampoline"
+
+# Use the trampoline script to run in docker.
+build_file: "python-aiplatform/.kokoro/trampoline_v2.sh"

.kokoro/samples/python3.9/continuous.cfg

@@ -0,0 +1,6 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+    key: "INSTALL_LIBRARY_FROM_SOURCE"
+    value: "True"
+}

.kokoro/samples/python3.9/periodic-head.cfg

@@ -0,0 +1,11 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+    key: "INSTALL_LIBRARY_FROM_SOURCE"
+    value: "True"
+}
+
+env_vars: {
+    key: "TRAMPOLINE_BUILD_FILE"
+    value: "github/python-aiplatform/.kokoro/test-samples-against-head.sh"
+}

.kokoro/samples/python3.9/periodic.cfg

@@ -0,0 +1,6 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+    key: "INSTALL_LIBRARY_FROM_SOURCE"
+    value: "False"
+}

.kokoro/samples/python3.9/presubmit.cfg

@@ -0,0 +1,6 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+    key: "INSTALL_LIBRARY_FROM_SOURCE"
+    value: "True"
+}

docs/aiplatform_v1beta1/reasoning_engine_runtime_revision_service.rst

@@ -0,0 +1,10 @@
+ReasoningEngineRuntimeRevisionService
+-------------------------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.reasoning_engine_runtime_revision_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.reasoning_engine_runtime_revision_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1beta1/services_.rst

@@ -34,6 +34,7 @@ Services for Google Cloud Aiplatform v1beta1 API
     pipeline_service
     prediction_service
     reasoning_engine_execution_service
+    reasoning_engine_runtime_revision_service
     reasoning_engine_service
     schedule_service
     session_service

google/cloud/aiplatform_v1/types/tool.py

@@ -96,6 +96,12 @@ class Tool(proto.Message):
             Optional. Tool to support searching public
             web data, powered by Vertex AI Search and Sec4
             compliance.
+        parallel_ai_search (google.cloud.aiplatform_v1.types.Tool.ParallelAiSearch):
+            Optional. If specified, Vertex AI will use
+            Parallel.ai to search for information to answer
+            user queries. The search results will be
+            grounded on Parallel.ai and presented to the
+            model for response generation
         code_execution (google.cloud.aiplatform_v1.types.Tool.CodeExecution):
             Optional. CodeExecution tool type.
             Enables the model to execute code as part of
@@ -178,6 +184,44 @@ class GoogleSearch(proto.Message):
             enum="Tool.PhishBlockThreshold",
         )
 
+    class ParallelAiSearch(proto.Message):
+        r"""ParallelAiSearch tool type.
+        A tool that uses the Parallel.ai search engine for grounding.
+
+        Attributes:
+            api_key (str):
+                Optional. The API key for ParallelAiSearch.
+                If an API key is not provided, the system will
+                attempt to verify access by checking for an
+                active Parallel.ai subscription through the
+                Google Cloud Marketplace.
+                See
+                https://docs.parallel.ai/search/search-quickstart
+                for more details.
+            custom_configs (google.protobuf.struct_pb2.Struct):
+                Optional. Custom configs for ParallelAiSearch. This field
+                can be used to pass any parameter from the Parallel.ai
+                Search API. See the Parallel.ai documentation for the full
+                list of available parameters and their usage:
+                https://docs.parallel.ai/api-reference/search-beta/search
+                Currently only ``source_policy``, ``excerpts``,
+                ``max_results``, ``mode``, ``fetch_policy`` can be set via
+                this field. For example: { "source_policy": {
+                "include_domains": ["google.com", "wikipedia.org"],
+                "exclude_domains": ["example.com"] }, "fetch_policy": {
+                "max_age_seconds": 3600 } }
+        """
+
+        api_key: str = proto.Field(
+            proto.STRING,
+            number=1,
+        )
+        custom_configs: struct_pb2.Struct = proto.Field(
+            proto.MESSAGE,
+            number=3,
+            message=struct_pb2.Struct,
+        )
+
     class CodeExecution(proto.Message):
         r"""Tool that executes code generated by the model, and automatically
         returns the result to the model.
@@ -259,6 +303,11 @@ class Environment(proto.Enum):
         number=6,
         message="EnterpriseWebSearch",
     )
+    parallel_ai_search: ParallelAiSearch = proto.Field(
+        proto.MESSAGE,
+        number=13,
+        message=ParallelAiSearch,
+    )
     code_execution: CodeExecution = proto.Field(
         proto.MESSAGE,
         number=4,

google/cloud/aiplatform_v1beta1/__init__.py

@@ -106,6 +106,12 @@
 from .services.reasoning_engine_execution_service import (
     ReasoningEngineExecutionServiceAsyncClient,
 )
+from .services.reasoning_engine_runtime_revision_service import (
+    ReasoningEngineRuntimeRevisionServiceClient,
+)
+from .services.reasoning_engine_runtime_revision_service import (
+    ReasoningEngineRuntimeRevisionServiceAsyncClient,
+)
 from .services.reasoning_engine_service import ReasoningEngineServiceClient
 from .services.reasoning_engine_service import ReasoningEngineServiceAsyncClient
 from .services.schedule_service import ScheduleServiceClient
@@ -1125,6 +1131,22 @@
 from .types.reasoning_engine_execution_service import QueryReasoningEngineRequest
 from .types.reasoning_engine_execution_service import QueryReasoningEngineResponse
 from .types.reasoning_engine_execution_service import StreamQueryReasoningEngineRequest
+from .types.reasoning_engine_runtime_revision import ReasoningEngineRuntimeRevision
+from .types.reasoning_engine_runtime_revision_service import (
+    DeleteReasoningEngineRuntimeRevisionOperationMetadata,
+)
+from .types.reasoning_engine_runtime_revision_service import (
+    DeleteReasoningEngineRuntimeRevisionRequest,
+)
+from .types.reasoning_engine_runtime_revision_service import (
+    GetReasoningEngineRuntimeRevisionRequest,
+)
+from .types.reasoning_engine_runtime_revision_service import (
+    ListReasoningEngineRuntimeRevisionsRequest,
+)
+from .types.reasoning_engine_runtime_revision_service import (
+    ListReasoningEngineRuntimeRevisionsResponse,
+)
 from .types.reasoning_engine_service import CreateReasoningEngineOperationMetadata
 from .types.reasoning_engine_service import CreateReasoningEngineRequest
 from .types.reasoning_engine_service import DeleteReasoningEngineRequest
@@ -1524,6 +1546,7 @@ def _get_version(dependency_name):
     "PipelineServiceAsyncClient",
     "PredictionServiceAsyncClient",
     "ReasoningEngineExecutionServiceAsyncClient",
+    "ReasoningEngineRuntimeRevisionServiceAsyncClient",
     "ReasoningEngineServiceAsyncClient",
     "ScheduleServiceAsyncClient",
     "SessionServiceAsyncClient",
@@ -1819,6 +1842,8 @@ def _get_version(dependency_name):
     "DeleteRagFileRequest",
     "DeleteRagMetadataRequest",
     "DeleteReasoningEngineRequest",
+    "DeleteReasoningEngineRuntimeRevisionOperationMetadata",
+    "DeleteReasoningEngineRuntimeRevisionRequest",
     "DeleteSavedQueryRequest",
     "DeleteScheduleRequest",
     "DeleteSessionRequest",
@@ -2064,6 +2089,7 @@ def _get_version(dependency_name):
     "GetRagFileRequest",
     "GetRagMetadataRequest",
     "GetReasoningEngineRequest",
+    "GetReasoningEngineRuntimeRevisionRequest",
     "GetScheduleRequest",
     "GetSessionRequest",
     "GetSpecialistPoolRequest",
@@ -2228,6 +2254,8 @@ def _get_version(dependency_name):
     "ListRagFilesResponse",
     "ListRagMetadataRequest",
     "ListRagMetadataResponse",
+    "ListReasoningEngineRuntimeRevisionsRequest",
+    "ListReasoningEngineRuntimeRevisionsResponse",
     "ListReasoningEnginesRequest",
     "ListReasoningEnginesResponse",
     "ListSavedQueriesRequest",
@@ -2483,6 +2511,8 @@ def _get_version(dependency_name):
     "ReasoningEngine",
     "ReasoningEngineContextSpec",
     "ReasoningEngineExecutionServiceClient",
+    "ReasoningEngineRuntimeRevision",
+    "ReasoningEngineRuntimeRevisionServiceClient",
     "ReasoningEngineServiceClient",
     "ReasoningEngineSpec",
     "RebaseTunedModelOperationMetadata",

google/cloud/aiplatform_v1beta1/gapic_metadata.json

@@ -5409,6 +5409,70 @@
         }
       }
     },
+    "ReasoningEngineRuntimeRevisionService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "ReasoningEngineRuntimeRevisionServiceClient",
+          "rpcs": {
+            "DeleteReasoningEngineRuntimeRevision": {
+              "methods": [
+                "delete_reasoning_engine_runtime_revision"
+              ]
+            },
+            "GetReasoningEngineRuntimeRevision": {
+              "methods": [
+                "get_reasoning_engine_runtime_revision"
+              ]
+            },
+            "ListReasoningEngineRuntimeRevisions": {
+              "methods": [
+                "list_reasoning_engine_runtime_revisions"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "ReasoningEngineRuntimeRevisionServiceAsyncClient",
+          "rpcs": {
+            "DeleteReasoningEngineRuntimeRevision": {
+              "methods": [
+                "delete_reasoning_engine_runtime_revision"
+              ]
+            },
+            "GetReasoningEngineRuntimeRevision": {
+              "methods": [
+                "get_reasoning_engine_runtime_revision"
+              ]
+            },
+            "ListReasoningEngineRuntimeRevisions": {
+              "methods": [
+                "list_reasoning_engine_runtime_revisions"
+              ]
+            }
+          }
+        },
+        "rest": {
+          "libraryClient": "ReasoningEngineRuntimeRevisionServiceClient",
+          "rpcs": {
+            "DeleteReasoningEngineRuntimeRevision": {
+              "methods": [
+                "delete_reasoning_engine_runtime_revision"
+              ]
+            },
+            "GetReasoningEngineRuntimeRevision": {
+              "methods": [
+                "get_reasoning_engine_runtime_revision"
+              ]
+            },
+            "ListReasoningEngineRuntimeRevisions": {
+              "methods": [
+                "list_reasoning_engine_runtime_revisions"
+              ]
+            }
+          }
+        }
+      }
+    },
     "ReasoningEngineService": {
       "clients": {
         "grpc": {

google/cloud/aiplatform_v1beta1/services/dataset_service/transports/rest.py

@@ -2001,6 +2001,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
                         "method": "post",
                         "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/sessions/*/operations/*}:cancel",
                     },
+                    {
+                        "method": "post",
+                        "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/runtimeRevisions/*/operations/*}:cancel",
+                    },
                     {
                         "method": "post",
                         "uri": "/v1beta1/{name=projects/*/locations/*/schedules/*/operations/*}:cancel",
@@ -2403,6 +2407,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
                         "method": "delete",
                         "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/sessions/*/operations/*}",
                     },
+                    {
+                        "method": "delete",
+                        "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/runtimeRevisions/*/operations/*}",
+                    },
                     {
                         "method": "delete",
                         "uri": "/v1beta1/{name=projects/*/locations/*/solvers/*/operations/*}",
@@ -2853,6 +2861,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
                         "method": "get",
                         "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/sessions/*/operations/*}",
                     },
+                    {
+                        "method": "get",
+                        "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/runtimeRevisions/*/operations/*}",
+                    },
                     {
                         "method": "get",
                         "uri": "/v1beta1/{name=projects/*/locations/*/solvers/*/operations/*}",
@@ -3299,6 +3311,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
                         "method": "get",
                         "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/sessions/*}/operations",
                     },
+                    {
+                        "method": "get",
+                        "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/runtimeRevisions/*}/operations",
+                    },
                     {
                         "method": "get",
                         "uri": "/v1beta1/{name=projects/*/locations/*/solvers/*}/operations",
@@ -3745,6 +3761,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
                         "method": "post",
                         "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/sessions/*/operations/*}:wait",
                     },
+                    {
+                        "method": "post",
+                        "uri": "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*/runtimeRevisions/*/operations/*}:wait",
+                    },
                     {
                         "method": "post",
                         "uri": "/v1beta1/{name=projects/*/locations/*/studies/*/operations/*}:wait",