🦉 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

@@ -119,6 +119,7 @@
 from .types.content import GroundingChunk
 from .types.content import GroundingMetadata
 from .types.content import GroundingSupport
+from .types.content import ImageConfig
 from .types.content import LogprobsResult
 from .types.content import ModalityTokenCount
 from .types.content import ModelArmorConfig
@@ -1514,6 +1515,7 @@
     "HarmCategory",
     "HyperparameterTuningJob",
     "IdMatcher",
+    "ImageConfig",
     "ImportDataConfig",
     "ImportDataOperationMetadata",
     "ImportDataRequest",

google/cloud/aiplatform_v1/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_v1/types/__init__.py

@@ -48,6 +48,7 @@
     GroundingChunk,
     GroundingMetadata,
     GroundingSupport,
+    ImageConfig,
     LogprobsResult,
     ModalityTokenCount,
     ModelArmorConfig,
@@ -1221,6 +1222,7 @@
     "GroundingChunk",
     "GroundingMetadata",
     "GroundingSupport",
+    "ImageConfig",
     "LogprobsResult",
     "ModalityTokenCount",
     "ModelArmorConfig",

google/cloud/aiplatform_v1/types/content.py

@@ -37,6 +37,7 @@
         "Blob",
         "FileData",
         "VideoMetadata",
+        "ImageConfig",
         "GenerationConfig",
         "SafetySetting",
         "SafetyRating",
@@ -340,6 +341,34 @@ class VideoMetadata(proto.Message):
     )
 
 
+class ImageConfig(proto.Message):
+    r"""Config for image generation features.
+
+    .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+    Attributes:
+        aspect_ratio (str):
+            Optional. The desired aspect ratio for the
+            generated images. The following aspect ratios
+            are supported:
+
+            "1:1"
+            "2:3", "3:2"
+            "3:4", "4:3"
+            "4:5", "5:4"
+            "9:16", "16:9"
+            "21:9".
+
+            This field is a member of `oneof`_ ``_aspect_ratio``.
+    """
+
+    aspect_ratio: str = proto.Field(
+        proto.STRING,
+        number=2,
+        optional=True,
+    )
+
+
 class GenerationConfig(proto.Message):
     r"""Generation config.
 
@@ -464,6 +493,11 @@ class GenerationConfig(proto.Message):
             Optional. Config for thinking features.
             An error will be returned if this field is set
             for models that don't support thinking.
+        image_config (google.cloud.aiplatform_v1.types.ImageConfig):
+            Optional. Config for image generation
+            features.
+
+            This field is a member of `oneof`_ ``_image_config``.
     """
 
     class RoutingConfig(proto.Message):
@@ -676,6 +710,12 @@ class ThinkingConfig(proto.Message):
         number=25,
         message=ThinkingConfig,
     )
+    image_config: "ImageConfig" = proto.Field(
+        proto.MESSAGE,
+        number=30,
+        optional=True,
+        message="ImageConfig",
+    )
 
 
 class SafetySetting(proto.Message):