Copybara import of the project:

--
e8b4a67 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

feat: RagMetadata and RagDataSchema concepts and Batch API definitions added for Preview
docs: Updated comments for `rag_file_metadata_config` across ImportRagFilesRequest and UploadRagFileRequest

PiperOrigin-RevId: 881618214

Source-Link: googleapis/googleapis@1a30906

Source-Link: googleapis/googleapis-gen@7b30aa4
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiN2IzMGFhNDkwYzE4YWUxMmJlNmExODk2ZWU2ZDZhMzA3ZDA1NDU4NCJ9

feat: upgrade protobuf from v25.7 to v31.0

PiperOrigin-RevId: 881595129

Source-Link: googleapis/googleapis@d90d23f

Source-Link: googleapis/googleapis-gen@f1a9443
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZjFhOTQ0MzM2NjRmODY1MGE0YWNiNzNkNjA5Yjk0M2QyZWNiYjZlMCJ9

feat: Add custom session id field to create session v1
feat: Add raw_event field to event proto v1

PiperOrigin-RevId: 880968942

Source-Link: googleapis/googleapis@39d9d44

Source-Link: googleapis/googleapis-gen@f5a7f05
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZjVhN2YwNTdmNjFhZWRkMjVlOWNhYjYyYTgxNmQ3ZWQ3YzEyNjcwNCJ9

feat: Add custom session id field to create session v1beta1
feat: Add custom memory id field to create memory v1beta1
feat: Add raw_event field to event proto v1beta1

PiperOrigin-RevId: 879826477

Source-Link: googleapis/googleapis@052914c

Source-Link: googleapis/googleapis-gen@1fb8d69
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWZiOGQ2OTg0ZmY5MjkwOTk4Yzg5NGRmNDE3Y2I4NDlmODg0NTdmNiJ9

--
1c7ecd8 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

--
b87fa82 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

feat: update the schema proto for the RagChunk with file_id and chunk_id to align with the vertex_rag_data.proto

PiperOrigin-RevId: 885147156

Source-Link: googleapis/googleapis@76b63dd

Source-Link: googleapis/googleapis-gen@1dd7bdb
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWRkN2JkYjEzYWJjMjE4MjkxZWU3MDZiYzQ3ZDc4MDA0MTdkNDg2MSJ9

--
24f55ce by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

--
cefe822 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

feat: Add new `embed_content_config` to message `EmbedContentRequest`

PiperOrigin-RevId: 885256653

Source-Link: googleapis/googleapis@9d5f10d

Source-Link: googleapis/googleapis-gen@c11bedb
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYzExYmVkYjVkY2QxYzNkMDQxMjA5NjBmYzJlNjY3M2EwNzhlZjdkNyJ9

--
94d148c by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

--
c8eca83 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

feat: Add new `embed_content_config` to message `EmbedContentRequest`

PiperOrigin-RevId: 885289577

Source-Link: googleapis/googleapis@7feb582

Source-Link: googleapis/googleapis-gen@e6e60da
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZTZlNjBkYWNlNTA2ZjEyNTQ0MjYxOGM1YzY1MGZjOWE0ZDc5NTdkNCJ9

--
53bfff2 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

--
da643df by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

feat: add `VALIDATED` Function Calling mode to v1

PiperOrigin-RevId: 885373782

Source-Link: googleapis/googleapis@ebfdba3

Source-Link: googleapis/googleapis-gen@0055864
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMDA1NTg2NDVjNjQ0ZWNjMWM3YTlhYzMzNzM3NjAxMmZhYmM0MjNjYyJ9

--
513f200 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>:

🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

COPYBARA_INTEGRATE_REVIEW=#6396 from googleapis:owl-bot-copy c0f517b
PiperOrigin-RevId: 885719789

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1/services/migration_service/client.py

@@ -293,40 +293,40 @@ def parse_dataset_path(path: str) -> Dict[str, str]:
     @staticmethod
     def dataset_path(
         project: str,
+        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/datasets/{dataset}".format(
+        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
             project=project,
+            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
+        m = re.match(
+            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
+            path,
+        )
         return m.groupdict() if m else {}
 
     @staticmethod
     def dataset_path(
         project: str,
-        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
+        return "projects/{project}/datasets/{dataset}".format(
             project=project,
-            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(
-            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
-            path,
-        )
+        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
         return m.groupdict() if m else {}
 
     @staticmethod

google/cloud/aiplatform_v1/services/prediction_service/async_client.py

@@ -1990,7 +1990,7 @@ async def sample_embed_content():
                 should not be set.
             content (:class:`google.cloud.aiplatform_v1.types.Content`):
                 Required. Input content to be
-                embedded. Required.
+                embedded.
 
                 This corresponds to the ``content`` field
                 on the ``request`` instance; if ``request`` is provided, this

google/cloud/aiplatform_v1/services/prediction_service/client.py

@@ -2513,7 +2513,7 @@ def sample_embed_content():
                 should not be set.
             content (google.cloud.aiplatform_v1.types.Content):
                 Required. Input content to be
-                embedded. Required.
+                embedded.
 
                 This corresponds to the ``content`` field
                 on the ``request`` instance; if ``request`` is provided, this

google/cloud/aiplatform_v1/services/session_service/async_client.py

@@ -323,6 +323,7 @@ async def create_session(
         *,
         parent: Optional[str] = None,
         session: Optional[gca_session.Session] = None,
+        session_id: Optional[str] = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
@@ -380,6 +381,20 @@ async def sample_create_session():
                 This corresponds to the ``session`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            session_id (:class:`str`):
+                Optional. The user defined ID to use for session, which
+                will become the final component of the session resource
+                name. If not provided, Vertex AI will generate a value
+                for this ID.
+
+                This value may be up to 63 characters, and valid
+                characters are ``[a-z0-9-]``. The first character must
+                be a letter, and the last character must be a letter or
+                number.
+
+                This corresponds to the ``session_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -401,7 +416,7 @@ async def sample_create_session():
         # Create or coerce a protobuf request object.
         # - Quick check: If we got a request object, we should *not* have
         #   gotten any keyword arguments that map to the request.
-        flattened_params = [parent, session]
+        flattened_params = [parent, session, session_id]
         has_flattened_params = (
             len([param for param in flattened_params if param is not None]) > 0
         )
@@ -422,6 +437,8 @@ async def sample_create_session():
             request.parent = parent
         if session is not None:
             request.session = session
+        if session_id is not None:
+            request.session_id = session_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.

google/cloud/aiplatform_v1/services/session_service/client.py

@@ -838,6 +838,7 @@ def create_session(
         *,
         parent: Optional[str] = None,
         session: Optional[gca_session.Session] = None,
+        session_id: Optional[str] = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
@@ -895,6 +896,20 @@ def sample_create_session():
                 This corresponds to the ``session`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            session_id (str):
+                Optional. The user defined ID to use for session, which
+                will become the final component of the session resource
+                name. If not provided, Vertex AI will generate a value
+                for this ID.
+
+                This value may be up to 63 characters, and valid
+                characters are ``[a-z0-9-]``. The first character must
+                be a letter, and the last character must be a letter or
+                number.
+
+                This corresponds to the ``session_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -916,7 +931,7 @@ def sample_create_session():
         # Create or coerce a protobuf request object.
         # - Quick check: If we got a request object, we should *not* have
         #   gotten any keyword arguments that map to the request.
-        flattened_params = [parent, session]
+        flattened_params = [parent, session, session_id]
         has_flattened_params = (
             len([param for param in flattened_params if param is not None]) > 0
         )
@@ -936,6 +951,8 @@ def sample_create_session():
                 request.parent = parent
             if session is not None:
                 request.session = session
+            if session_id is not None:
+                request.session_id = session_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.

google/cloud/aiplatform_v1/types/prediction_service.py

@@ -1162,29 +1162,39 @@ class EmbedContentRequest(proto.Message):
             This field is a member of `oneof`_ ``_model``.
         content (google.cloud.aiplatform_v1.types.Content):
             Required. Input content to be embedded.
-            Required.
 
             This field is a member of `oneof`_ ``_content``.
         title (str):
-            Optional. An optional title for the text.
+            Optional. Deprecated: Please use
+            EmbedContentConfig.title instead. The title for
+            the text.
 
             This field is a member of `oneof`_ ``_title``.
         task_type (google.cloud.aiplatform_v1.types.EmbedContentRequest.EmbeddingTaskType):
-            Optional. The task type of the embedding.
+            Optional. Deprecated: Please use
+            EmbedContentConfig.task_type instead. The task type of the
+            embedding.
 
             This field is a member of `oneof`_ ``_task_type``.
         output_dimensionality (int):
-            Optional. Optional reduced dimension for the
-            output embedding. If set, excessive values in
-            the output embedding are truncated from the end.
+            Optional. Deprecated: Please use
+            EmbedContentConfig.output_dimensionality instead. Reduced
+            dimension for the output embedding. If set, excessive values
+            in the output embedding are truncated from the end.
 
             This field is a member of `oneof`_ ``_output_dimensionality``.
         auto_truncate (bool):
-            Optional. Whether to silently truncate the
-            input content if it's longer than the maximum
-            sequence length.
+            Optional. Deprecated: Please use
+            EmbedContentConfig.auto_truncate instead. Whether to
+            silently truncate the input content if it's longer than the
+            maximum sequence length.
 
             This field is a member of `oneof`_ ``_auto_truncate``.
+        embed_content_config (google.cloud.aiplatform_v1.types.EmbedContentRequest.EmbedContentConfig):
+            Optional. Configuration for the EmbedContent
+            request.
+
+            This field is a member of `oneof`_ ``_embed_content_config``.
     """
 
     class EmbeddingTaskType(proto.Enum):
@@ -1230,6 +1240,82 @@ class EmbeddingTaskType(proto.Enum):
         FACT_VERIFICATION = 8
         CODE_RETRIEVAL_QUERY = 9
 
+    class EmbedContentConfig(proto.Message):
+        r"""Configurations for the EmbedContent API.
+
+        .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+        Attributes:
+            title (str):
+                Optional. The title for the text.
+
+                Only applicable to text-only embedding models.
+
+                This field is a member of `oneof`_ ``_title``.
+            task_type (google.cloud.aiplatform_v1.types.EmbedContentRequest.EmbeddingTaskType):
+                Optional. The task type of the embedding.
+
+                Only applicable to text-only embedding models.
+
+                This field is a member of `oneof`_ ``_task_type``.
+            auto_truncate (bool):
+                Optional. Whether to silently truncate the
+                input content if it's longer than the maximum
+                sequence length.
+
+                Only applicable to text-only embedding models.
+
+                This field is a member of `oneof`_ ``_auto_truncate``.
+            output_dimensionality (int):
+                Optional. Reduced dimension for the output
+                embedding. If set, excessive values in the
+                output embedding are truncated from the end.
+
+                This field is a member of `oneof`_ ``_output_dimensionality``.
+            document_ocr (bool):
+                Optional. Whether to enable OCR for document
+                content.
+
+                This field is a member of `oneof`_ ``_document_ocr``.
+            audio_track_extraction (bool):
+                Optional. Whether to extract audio from video
+                content.
+
+                This field is a member of `oneof`_ ``_audio_track_extraction``.
+        """
+
+        title: str = proto.Field(
+            proto.STRING,
+            number=1,
+            optional=True,
+        )
+        task_type: "EmbedContentRequest.EmbeddingTaskType" = proto.Field(
+            proto.ENUM,
+            number=2,
+            optional=True,
+            enum="EmbedContentRequest.EmbeddingTaskType",
+        )
+        auto_truncate: bool = proto.Field(
+            proto.BOOL,
+            number=3,
+            optional=True,
+        )
+        output_dimensionality: int = proto.Field(
+            proto.INT32,
+            number=4,
+            optional=True,
+        )
+        document_ocr: bool = proto.Field(
+            proto.BOOL,
+            number=5,
+            optional=True,
+        )
+        audio_track_extraction: bool = proto.Field(
+            proto.BOOL,
+            number=6,
+            optional=True,
+        )
+
     model: str = proto.Field(
         proto.STRING,
         number=1,
@@ -1262,6 +1348,12 @@ class EmbeddingTaskType(proto.Enum):
         number=7,
         optional=True,
     )
+    embed_content_config: EmbedContentConfig = proto.Field(
+        proto.MESSAGE,
+        number=8,
+        optional=True,
+        message=EmbedContentConfig,
+    )
 
 
 class EmbedContentResponse(proto.Message):
@@ -1273,7 +1365,7 @@ class EmbedContentResponse(proto.Message):
             The embedding generated from the input
             content.
         usage_metadata (google.cloud.aiplatform_v1.types.UsageMetadata):
-            Metadata about the response(s).
+            Usage metadata about the response(s).
         truncated (bool):
             Whether the input content was truncated
             before generating the embedding.

google/cloud/aiplatform_v1/types/session.py

@@ -166,6 +166,9 @@ class SessionEvent(proto.Message):
             error.
         event_metadata (google.cloud.aiplatform_v1.types.EventMetadata):
             Optional. Metadata relating to this event.
+        raw_event (google.protobuf.struct_pb2.Struct):
+            Optional. Weakly typed raw event data in
+            proto struct format.
     """
 
     name: str = proto.Field(
@@ -208,6 +211,11 @@ class SessionEvent(proto.Message):
         number=11,
         message="EventMetadata",
     )
+    raw_event: struct_pb2.Struct = proto.Field(
+        proto.MESSAGE,
+        number=12,
+        message=struct_pb2.Struct,
+    )
 
 
 class EventMetadata(proto.Message):

google/cloud/aiplatform_v1/types/session_service.py

@@ -53,6 +53,14 @@ class CreateSessionRequest(proto.Message):
             ``projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}``
         session (google.cloud.aiplatform_v1.types.Session):
             Required. The session to create.
+        session_id (str):
+            Optional. The user defined ID to use for session, which will
+            become the final component of the session resource name. If
+            not provided, Vertex AI will generate a value for this ID.
+
+            This value may be up to 63 characters, and valid characters
+            are ``[a-z0-9-]``. The first character must be a letter, and
+            the last character must be a letter or number.
     """
 
     parent: str = proto.Field(
@@ -64,6 +72,10 @@ class CreateSessionRequest(proto.Message):
         number=2,
         message=gca_session.Session,
     )
+    session_id: str = proto.Field(
+        proto.STRING,
+        number=3,
+    )
 
 
 class CreateSessionOperationMetadata(proto.Message):

google/cloud/aiplatform_v1/types/tool.py

@@ -1107,12 +1107,21 @@ class Mode(proto.Enum):
                 Model will not predict any function calls.
                 Model behavior is same as when not passing any
                 function declarations.
+            VALIDATED (5):
+                Model is constrained to predict either function calls or
+                natural language response. If
+                [allowed_function_names][FunctionCallingConfig.allowed_function_names]
+                are set, the predicted function calls will be limited to any
+                one of ``allowed_function_names``, else the predicted
+                function calls will be any one of the provided
+                [FunctionDeclaration].
         """
 
         MODE_UNSPECIFIED = 0
         AUTO = 1
         ANY = 2
         NONE = 3
+        VALIDATED = 5
 
     mode: Mode = proto.Field(
         proto.ENUM,