feat: Add image_spec to Reasoning Engine public protos

docs/aiplatform_v1/services_.rst

@@ -31,6 +31,7 @@ Services for Google Cloud Aiplatform v1 API
     reasoning_engine_execution_service
     reasoning_engine_service
     schedule_service
+    session_service
     specialist_pool_service
     tensorboard_service
     vertex_rag_data_service

docs/aiplatform_v1/session_service.rst

@@ -0,0 +1,10 @@
+SessionService
+--------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.session_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.session_service.pagers
+    :members:
+    :inherited-members:

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/image_object_detection.py

@@ -19,7 +19,7 @@
 
 import proto  # type: ignore
 
-from google.protobuf import struct_pb2  # type: ignore
+import google.protobuf.struct_pb2 as struct_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_action_recognition.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_classification.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/video_object_tracking.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/image_object_detection.py

@@ -19,7 +19,7 @@
 
 import proto  # type: ignore
 
-from google.protobuf import struct_pb2  # type: ignore
+import google.protobuf.struct_pb2 as struct_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/video_action_recognition.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/video_classification.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/video_object_tracking.py

@@ -19,8 +19,8 @@
 
 import proto  # type: ignore
 
-from google.protobuf import duration_pb2  # type: ignore
-from google.protobuf import wrappers_pb2  # type: ignore
+import google.protobuf.duration_pb2 as duration_pb2  # type: ignore
+import google.protobuf.wrappers_pb2 as wrappers_pb2  # type: ignore
 
 
 __protobuf__ = proto.module(

google/cloud/aiplatform_v1/__init__.py

@@ -100,6 +100,8 @@
 from .services.reasoning_engine_service import ReasoningEngineServiceAsyncClient
 from .services.schedule_service import ScheduleServiceClient
 from .services.schedule_service import ScheduleServiceAsyncClient
+from .services.session_service import SessionServiceClient
+from .services.session_service import SessionServiceAsyncClient
 from .services.specialist_pool_service import SpecialistPoolServiceClient
 from .services.specialist_pool_service import SpecialistPoolServiceAsyncClient
 from .services.tensorboard_service import TensorboardServiceClient
@@ -563,6 +565,8 @@
 from .types.io import SharePointSources
 from .types.io import SlackSource
 from .types.io import TFRecordDestination
+from .types.io import VertexMultimodalDatasetDestination
+from .types.io import VertexMultimodalDatasetSource
 from .types.job_service import CancelBatchPredictionJobRequest
 from .types.job_service import CancelCustomJobRequest
 from .types.job_service import CancelDataLabelingJobRequest
@@ -903,6 +907,22 @@
 from .types.service_networking import PSCAutomationConfig
 from .types.service_networking import PscInterfaceConfig
 from .types.service_networking import PSCAutomationState
+from .types.session import EventActions
+from .types.session import EventMetadata
+from .types.session import Session
+from .types.session import SessionEvent
+from .types.session import Transcription
+from .types.session_service import AppendEventRequest
+from .types.session_service import AppendEventResponse
+from .types.session_service import CreateSessionOperationMetadata
+from .types.session_service import CreateSessionRequest
+from .types.session_service import DeleteSessionRequest
+from .types.session_service import GetSessionRequest
+from .types.session_service import ListEventsRequest
+from .types.session_service import ListEventsResponse
+from .types.session_service import ListSessionsRequest
+from .types.session_service import ListSessionsResponse
+from .types.session_service import UpdateSessionRequest
 from .types.specialist_pool import SpecialistPool
 from .types.specialist_pool_service import CreateSpecialistPoolOperationMetadata
 from .types.specialist_pool_service import CreateSpecialistPoolRequest
@@ -1213,6 +1233,7 @@ def _get_version(dependency_name):
     "ReasoningEngineExecutionServiceAsyncClient",
     "ReasoningEngineServiceAsyncClient",
     "ScheduleServiceAsyncClient",
+    "SessionServiceAsyncClient",
     "SpecialistPoolServiceAsyncClient",
     "TensorboardServiceAsyncClient",
     "VertexRagDataServiceAsyncClient",
@@ -1230,6 +1251,8 @@ def _get_version(dependency_name):
     "Annotation",
     "AnnotationSpec",
     "ApiAuth",
+    "AppendEventRequest",
+    "AppendEventResponse",
     "Artifact",
     "AssignNotebookRuntimeOperationMetadata",
     "AssignNotebookRuntimeRequest",
@@ -1369,6 +1392,8 @@ def _get_version(dependency_name):
     "CreateReasoningEngineRequest",
     "CreateRegistryFeatureOperationMetadata",
     "CreateScheduleRequest",
+    "CreateSessionOperationMetadata",
+    "CreateSessionRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
     "CreateStudyRequest",
@@ -1432,6 +1457,7 @@ def _get_version(dependency_name):
     "DeleteReasoningEngineRequest",
     "DeleteSavedQueryRequest",
     "DeleteScheduleRequest",
+    "DeleteSessionRequest",
     "DeleteSpecialistPoolRequest",
     "DeleteStudyRequest",
     "DeleteTensorboardExperimentRequest",
@@ -1483,6 +1509,8 @@ def _get_version(dependency_name):
     "EvaluatedAnnotationExplanation",
     "EvaluationServiceClient",
     "Event",
+    "EventActions",
+    "EventMetadata",
     "ExactMatchInput",
     "ExactMatchInstance",
     "ExactMatchMetricValue",
@@ -1617,6 +1645,7 @@ def _get_version(dependency_name):
     "GetRagFileRequest",
     "GetReasoningEngineRequest",
     "GetScheduleRequest",
+    "GetSessionRequest",
     "GetSpecialistPoolRequest",
     "GetStudyRequest",
     "GetTensorboardExperimentRequest",
@@ -1693,6 +1722,8 @@ def _get_version(dependency_name):
     "ListEndpointsResponse",
     "ListEntityTypesRequest",
     "ListEntityTypesResponse",
+    "ListEventsRequest",
+    "ListEventsResponse",
     "ListExecutionsRequest",
     "ListExecutionsResponse",
     "ListFeatureGroupsRequest",
@@ -1755,6 +1786,8 @@ def _get_version(dependency_name):
     "ListSavedQueriesResponse",
     "ListSchedulesRequest",
     "ListSchedulesResponse",
+    "ListSessionsRequest",
+    "ListSessionsResponse",
     "ListSpecialistPoolsRequest",
     "ListSpecialistPoolsResponse",
     "ListStudiesRequest",
@@ -2019,6 +2052,9 @@ def _get_version(dependency_name):
     "SecretRef",
     "Segment",
     "ServiceAccountSpec",
+    "Session",
+    "SessionEvent",
+    "SessionServiceClient",
     "SharePointSources",
     "ShieldedVmConfig",
     "SlackSource",
@@ -2116,6 +2152,7 @@ def _get_version(dependency_name):
     "ToolParameterKeyMatchSpec",
     "TrainingConfig",
     "TrainingPipeline",
+    "Transcription",
     "Trial",
     "TrialContext",
     "TunedModel",
@@ -2172,6 +2209,7 @@ def _get_version(dependency_name):
     "UpdateReasoningEngineOperationMetadata",
     "UpdateReasoningEngineRequest",
     "UpdateScheduleRequest",
+    "UpdateSessionRequest",
     "UpdateSpecialistPoolOperationMetadata",
     "UpdateSpecialistPoolRequest",
     "UpdateTensorboardExperimentRequest",
@@ -2198,6 +2236,8 @@ def _get_version(dependency_name):
     "Value",
     "VertexAISearch",
     "VertexAiSearchConfig",
+    "VertexMultimodalDatasetDestination",
+    "VertexMultimodalDatasetSource",
     "VertexRagDataServiceClient",
     "VertexRagServiceClient",
     "VertexRagStore",

google/cloud/aiplatform_v1/gapic_metadata.json

@@ -4572,6 +4572,130 @@
         }
       }
     },
+    "SessionService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "SessionServiceClient",
+          "rpcs": {
+            "AppendEvent": {
+              "methods": [
+                "append_event"
+              ]
+            },
+            "CreateSession": {
+              "methods": [
+                "create_session"
+              ]
+            },
+            "DeleteSession": {
+              "methods": [
+                "delete_session"
+              ]
+            },
+            "GetSession": {
+              "methods": [
+                "get_session"
+              ]
+            },
+            "ListEvents": {
+              "methods": [
+                "list_events"
+              ]
+            },
+            "ListSessions": {
+              "methods": [
+                "list_sessions"
+              ]
+            },
+            "UpdateSession": {
+              "methods": [
+                "update_session"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "SessionServiceAsyncClient",
+          "rpcs": {
+            "AppendEvent": {
+              "methods": [
+                "append_event"
+              ]
+            },
+            "CreateSession": {
+              "methods": [
+                "create_session"
+              ]
+            },
+            "DeleteSession": {
+              "methods": [
+                "delete_session"
+              ]
+            },
+            "GetSession": {
+              "methods": [
+                "get_session"
+              ]
+            },
+            "ListEvents": {
+              "methods": [
+                "list_events"
+              ]
+            },
+            "ListSessions": {
+              "methods": [
+                "list_sessions"
+              ]
+            },
+            "UpdateSession": {
+              "methods": [
+                "update_session"
+              ]
+            }
+          }
+        },
+        "rest": {
+          "libraryClient": "SessionServiceClient",
+          "rpcs": {
+            "AppendEvent": {
+              "methods": [
+                "append_event"
+              ]
+            },
+            "CreateSession": {
+              "methods": [
+                "create_session"
+              ]
+            },
+            "DeleteSession": {
+              "methods": [
+                "delete_session"
+              ]
+            },
+            "GetSession": {
+              "methods": [
+                "get_session"
+              ]
+            },
+            "ListEvents": {
+              "methods": [
+                "list_events"
+              ]
+            },
+            "ListSessions": {
+              "methods": [
+                "list_sessions"
+              ]
+            },
+            "UpdateSession": {
+              "methods": [
+                "update_session"
+              ]
+            }
+          }
+        }
+      }
+    },
     "SpecialistPoolService": {
       "clients": {
         "grpc": {