🦉 Updates from OwlBot post-processor

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

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1/__init__.py

@@ -976,6 +976,7 @@
 from .types.tool import FunctionResponsePart
 from .types.tool import GoogleMaps
 from .types.tool import GoogleSearchRetrieval
+from .types.tool import PartialArg
 from .types.tool import RagRetrievalConfig
 from .types.tool import Retrieval
 from .types.tool import RetrievalConfig
@@ -1752,6 +1753,7 @@
     "PairwiseSummarizationQualityResult",
     "PairwiseSummarizationQualitySpec",
     "Part",
+    "PartialArg",
     "PauseModelDeploymentMonitoringJobRequest",
     "PauseScheduleRequest",
     "PersistentDiskSpec",

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

@@ -243,40 +243,40 @@ def parse_annotated_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/types/__init__.py

@@ -1093,6 +1093,7 @@
     FunctionResponsePart,
     GoogleMaps,
     GoogleSearchRetrieval,
+    PartialArg,
     RagRetrievalConfig,
     Retrieval,
     RetrievalConfig,
@@ -2078,6 +2079,7 @@
     "FunctionResponsePart",
     "GoogleMaps",
     "GoogleSearchRetrieval",
+    "PartialArg",
     "RagRetrievalConfig",
     "Retrieval",
     "RetrievalConfig",

google/cloud/aiplatform_v1/types/tool.py

@@ -31,6 +31,7 @@
         "UrlContext",
         "FunctionDeclaration",
         "FunctionCall",
+        "PartialArg",
         "FunctionResponsePart",
         "FunctionResponseBlob",
         "FunctionResponseFileData",
@@ -391,12 +392,22 @@ class FunctionCall(proto.Message):
 
     Attributes:
         name (str):
-            Required. The name of the function to call. Matches
+            Optional. The name of the function to call. Matches
             [FunctionDeclaration.name].
         args (google.protobuf.struct_pb2.Struct):
-            Optional. Required. The function parameters and values in
-            JSON object format. See [FunctionDeclaration.parameters] for
-            parameter details.
+            Optional. The function parameters and values in JSON object
+            format. See [FunctionDeclaration.parameters] for parameter
+            details.
+        partial_args (MutableSequence[google.cloud.aiplatform_v1.types.PartialArg]):
+            Optional. The partial argument value of the
+            function call. If provided, represents the
+            arguments/fields that are streamed
+            incrementally.
+        will_continue (bool):
+            Optional. Whether this is the last part of
+            the FunctionCall. If true, another partial
+            message for the current FunctionCall is expected
+            to follow.
     """
 
     name: str = proto.Field(
@@ -408,6 +419,83 @@ class FunctionCall(proto.Message):
         number=2,
         message=struct_pb2.Struct,
     )
+    partial_args: MutableSequence["PartialArg"] = proto.RepeatedField(
+        proto.MESSAGE,
+        number=4,
+        message="PartialArg",
+    )
+    will_continue: bool = proto.Field(
+        proto.BOOL,
+        number=5,
+    )
+
+
+class PartialArg(proto.Message):
+    r"""Partial argument value of the function call.
+
+    This message has `oneof`_ fields (mutually exclusive fields).
+    For each oneof, at most one member field can be set at the same time.
+    Setting any member of the oneof automatically clears all other
+    members.
+
+    .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+    Attributes:
+        null_value (google.protobuf.struct_pb2.NullValue):
+            Optional. Represents a null value.
+
+            This field is a member of `oneof`_ ``delta``.
+        number_value (float):
+            Optional. Represents a double value.
+
+            This field is a member of `oneof`_ ``delta``.
+        string_value (str):
+            Optional. Represents a string value.
+
+            This field is a member of `oneof`_ ``delta``.
+        bool_value (bool):
+            Optional. Represents a boolean value.
+
+            This field is a member of `oneof`_ ``delta``.
+        json_path (str):
+            Required. A JSON Path (RFC 9535) to the argument being
+            streamed. https://datatracker.ietf.org/doc/html/rfc9535.
+            e.g. "$.foo.bar[0].data".
+        will_continue (bool):
+            Optional. Whether this is not the last part of the same
+            json_path. If true, another PartialArg message for the
+            current json_path is expected to follow.
+    """
+
+    null_value: struct_pb2.NullValue = proto.Field(
+        proto.ENUM,
+        number=2,
+        oneof="delta",
+        enum=struct_pb2.NullValue,
+    )
+    number_value: float = proto.Field(
+        proto.DOUBLE,
+        number=3,
+        oneof="delta",
+    )
+    string_value: str = proto.Field(
+        proto.STRING,
+        number=4,
+        oneof="delta",
+    )
+    bool_value: bool = proto.Field(
+        proto.BOOL,
+        number=5,
+        oneof="delta",
+    )
+    json_path: str = proto.Field(
+        proto.STRING,
+        number=1,
+    )
+    will_continue: bool = proto.Field(
+        proto.BOOL,
+        number=6,
+    )
 
 
 class FunctionResponsePart(proto.Message):
@@ -991,6 +1079,11 @@ class FunctionCallingConfig(proto.Message):
             ANY. Function names should match [FunctionDeclaration.name].
             With mode set to ANY, model will predict a function call
             from the set of function names provided.
+        stream_function_call_arguments (bool):
+            Optional. When set to true, arguments of a single function
+            call will be streamed out in multiple
+            parts/contents/responses. Partial parameter results will be
+            returned in the [FunctionCall.partial_args] field.
     """
 
     class Mode(proto.Enum):
@@ -1030,6 +1123,10 @@ class Mode(proto.Enum):
         proto.STRING,
         number=2,
     )
+    stream_function_call_arguments: bool = proto.Field(
+        proto.BOOL,
+        number=4,
+    )
 
 
 class RetrievalConfig(proto.Message):

google/cloud/aiplatform_v1beta1/__init__.py

@@ -743,6 +743,7 @@
 from .types.machine_resources import DiskSpec
 from .types.machine_resources import FlexStart
 from .types.machine_resources import FullFineTunedResources
+from .types.machine_resources import LustreMount
 from .types.machine_resources import MachineSpec
 from .types.machine_resources import NfsMount
 from .types.machine_resources import PersistentDiskSpec
@@ -2028,6 +2029,7 @@
     "LlmUtilityServiceClient",
     "LogprobsResult",
     "LookupStudyRequest",
+    "LustreMount",
     "MachineSpec",
     "ManualBatchTuningParameters",
     "MatchServiceClient",

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

@@ -265,40 +265,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_v1beta1/types/__init__.py

@@ -770,6 +770,7 @@
     DiskSpec,
     FlexStart,
     FullFineTunedResources,
+    LustreMount,
     MachineSpec,
     NfsMount,
     PersistentDiskSpec,
@@ -2102,6 +2103,7 @@
     "DiskSpec",
     "FlexStart",
     "FullFineTunedResources",
+    "LustreMount",
     "MachineSpec",
     "NfsMount",
     "PersistentDiskSpec",

google/cloud/aiplatform_v1beta1/types/custom_job.py

@@ -420,6 +420,8 @@ class WorkerPoolSpec(proto.Message):
             use for this worker pool.
         nfs_mounts (MutableSequence[google.cloud.aiplatform_v1beta1.types.NfsMount]):
             Optional. List of NFS mount spec.
+        lustre_mounts (MutableSequence[google.cloud.aiplatform_v1beta1.types.LustreMount]):
+            Optional. List of Lustre mounts.
         disk_spec (google.cloud.aiplatform_v1beta1.types.DiskSpec):
             Disk spec.
     """
@@ -450,6 +452,11 @@ class WorkerPoolSpec(proto.Message):
         number=4,
         message=machine_resources.NfsMount,
     )
+    lustre_mounts: MutableSequence[machine_resources.LustreMount] = proto.RepeatedField(
+        proto.MESSAGE,
+        number=9,
+        message=machine_resources.LustreMount,
+    )
     disk_spec: machine_resources.DiskSpec = proto.Field(
         proto.MESSAGE,
         number=5,
@@ -553,13 +560,13 @@ class Scheduling(proto.Message):
 
     Attributes:
         timeout (google.protobuf.duration_pb2.Duration):
-            The maximum job running time. The default is
-            7 days.
+            Optional. The maximum job running time. The
+            default is 7 days.
         restart_job_on_worker_restart (bool):
-            Restarts the entire CustomJob if a worker
-            gets restarted. This feature can be used by
-            distributed training jobs that are not resilient
-            to workers leaving and joining a job.
+            Optional. Restarts the entire CustomJob if a
+            worker gets restarted. This feature can be used
+            by distributed training jobs that are not
+            resilient to workers leaving and joining a job.
         strategy (google.cloud.aiplatform_v1beta1.types.Scheduling.Strategy):
             Optional. This determines which type of
             scheduling strategy to use.

google/cloud/aiplatform_v1beta1/types/machine_resources.py

@@ -40,6 +40,7 @@
         "DiskSpec",
         "PersistentDiskSpec",
         "NfsMount",
+        "LustreMount",
         "AutoscalingMetricSpec",
         "ShieldedVmConfig",
         "FlexStart",
@@ -569,6 +570,40 @@ class NfsMount(proto.Message):
     )
 
 
+class LustreMount(proto.Message):
+    r"""Represents a mount configuration for Lustre file system.
+
+    Attributes:
+        instance_ip (str):
+            Required. IP address of the Lustre instance.
+        volume_handle (str):
+            Required. The unique identifier of the Lustre
+            volume.
+        filesystem (str):
+            Required. The name of the Lustre filesystem.
+        mount_point (str):
+            Required. Destination mount path. The Lustre file system
+            will be mounted for the user under /mnt/lustre/<mount_point>
+    """
+
+    instance_ip: str = proto.Field(
+        proto.STRING,
+        number=1,
+    )
+    volume_handle: str = proto.Field(
+        proto.STRING,
+        number=2,
+    )
+    filesystem: str = proto.Field(
+        proto.STRING,
+        number=3,
+    )
+    mount_point: str = proto.Field(
+        proto.STRING,
+        number=4,
+    )
+
+
 class AutoscalingMetricSpec(proto.Message):
     r"""The metric specification that defines the target resource
     utilization (CPU utilization, accelerator's duty cycle, and so