🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
googleapis · May 23, 2024 · 4cc1a84 · 4cc1a84
1 parent 27b4b3d
commit 4cc1a84
Show file tree

Hide file tree

Showing 2,355 changed files with 40 additions and 1,655,165 deletions.
diff --git a/google/cloud/aiplatform_v1beta1/services/migration_service/client.py b/google/cloud/aiplatform_v1beta1/services/migration_service/client.py
@@ -216,40 +216,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

diff --git a/google/cloud/aiplatform_v1beta1/types/content.py b/google/cloud/aiplatform_v1beta1/types/content.py
@@ -19,6 +19,7 @@
 
 import proto  # type: ignore
 
+from google.cloud.aiplatform_v1beta1.types import openapi
 from google.cloud.aiplatform_v1beta1.types import tool
 from google.protobuf import duration_pb2  # type: ignore
 from google.type import date_pb2  # type: ignore
@@ -309,6 +310,17 @@ class GenerationConfig(proto.Message):
                The model needs to be prompted to output the appropriate
                response type, otherwise the behavior is undefined. This
                is a preview feature.
+        response_schema (google.cloud.aiplatform_v1beta1.types.Schema):
+            Optional. The ``Schema`` object allows the definition of
+            input and output data types. These types can be objects, but
+            also primitives and arrays. Represents a select subset of an
+            `OpenAPI 3.0 schema
+            object <https://spec.openapis.org/oas/v3.0.3#schema>`__. If
+            set, a compatible response_mime_type must also be set.
+            Compatible mimetypes: ``application/json``: Schema for JSON
+            response.
+
+            This field is a member of `oneof`_ ``_response_schema``.
     """
 
     temperature: float = proto.Field(
@@ -354,6 +366,12 @@ class GenerationConfig(proto.Message):
         proto.STRING,
         number=13,
     )
+    response_schema: openapi.Schema = proto.Field(
+        proto.MESSAGE,
+        number=16,
+        optional=True,
+        message=openapi.Schema,
+    )
 
 
 class SafetySetting(proto.Message):

diff --git a/owl-bot-staging/v1/.coveragerc b/owl-bot-staging/v1/.coveragerc
diff --git a/owl-bot-staging/v1/.flake8 b/owl-bot-staging/v1/.flake8
diff --git a/owl-bot-staging/v1/MANIFEST.in b/owl-bot-staging/v1/MANIFEST.in
diff --git a/owl-bot-staging/v1/README.rst b/owl-bot-staging/v1/README.rst
diff --git a/owl-bot-staging/v1/docs/_static/custom.css b/owl-bot-staging/v1/docs/_static/custom.css
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/dataset_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/dataset_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/deployment_resource_pool_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/deployment_resource_pool_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/endpoint_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/endpoint_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_admin_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_admin_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/feature_registry_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/feature_registry_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/featurestore_online_serving_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/featurestore_online_serving_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/featurestore_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/featurestore_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/gen_ai_tuning_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/gen_ai_tuning_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/index_endpoint_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/index_endpoint_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/index_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/index_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/job_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/job_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/llm_utility_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/llm_utility_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/match_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/match_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/metadata_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/metadata_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/migration_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/migration_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/model_garden_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/model_garden_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/model_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/model_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/notebook_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/notebook_service.rst
diff --git a/owl-bot-staging/v1/docs/aiplatform_v1/persistent_resource_service.rst b/owl-bot-staging/v1/docs/aiplatform_v1/persistent_resource_service.rst