Merge branch 'main' into owl-bot-copy

googleapis · Nov 13, 2024 · 7de59c8 · 7de59c8
2 parents 00ba5b5 + 9c3ad83
commit 7de59c8
Show file tree

Hide file tree

Showing 34 changed files with 319 additions and 38 deletions.
diff --git a/.release-please-manifest.json b/.release-please-manifest.json
@@ -1,3 +1,3 @@
 {
-    ".": "1.71.1"
+    ".": "1.72.0"
 }
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,54 @@
 # Changelog
 
+## [1.72.0](https://github.com/googleapis/python-aiplatform/compare/v1.71.1...v1.72.0) (2024-11-12)
+
+
+### Features
+
+* Add AdvancedCompleteQuery API ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add BatchCreateFeatures rpc to feature_registry_service.proto ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* Add BillingEstimation in data store ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add code execution tool API ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add fast_tryout_enabled to FasterDeploymentConfig message in aiplatform v1beta1 endpoint.proto ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add Feature Monitoring API to Feature Store ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* Add field `protobuf_pythonic_types_enabled` to message `ExperimentalFeatures` ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* Add GroundedGenerationService API ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add lite search API to allow public website search with API key ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add new PscInterfaceConfig field to custom_job.proto ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add one_box_page_size on search ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add Sitemap APIs to preview channel ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add StopNotebookRuntime method ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add StopNotebookRuntime method ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Add system labels field to model garden deployments ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* Add UpdateEndpointLongRunning API in v1beta1 version ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Adding fast_tryout_enabled option to Vertex public preview SDK ([98288b8](https://github.com/googleapis/python-aiplatform/commit/98288b8b2d6ab5592e67cca1bd8914df3a1f19e7))
+* Increase the upperbounds on langchain dependencies to support v0.3+ ([1fca9c0](https://github.com/googleapis/python-aiplatform/commit/1fca9c058d7d8993f6d0c28580f6e0b899e4e83a))
+* Support advanced boost search ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Support Google Workspace search ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Support natural language understanding search ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Support query regex in control match rules ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Support search personalization to preview channel ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+
+
+### Bug Fixes
+
+* Add timeout to prediction rawPredict/streamRawPredict ([b7de16a](https://github.com/googleapis/python-aiplatform/commit/b7de16ae21b2934bc9195bf4bfc14d56adaf7701))
+* Convert float values with no decimals to integers in FunctionCall. ([a1857ed](https://github.com/googleapis/python-aiplatform/commit/a1857ed9bc97b9c6e367a8fa06b62883540a6638))
+* Deepcopy error from baseline_model in pairwiseMetric ([ccc5c85](https://github.com/googleapis/python-aiplatform/commit/ccc5c85a8ba39728f710fe64e8afbe98f84bf50e))
+
+
+### Documentation
+
+* A comment for field `feature_group_id` in message `.google.cloud.aiplatform.v1.CreateFeatureGroupRequest` is changed ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* A comment for field `unit` in message `.google.api.QuotaLimit` is changed ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* A comment for message `BatchCreateFeaturesRequest` is modified to call out BatchCreateFeatures ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+* Deprecate asynchronous mode in answer generation ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Deprecate extractive_answers in answer generation ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Keep the API doc up-to-date with recent changes ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* **samples:** Fixed Vertex AI Vector Search Create Index `index_update_method` ([7dff586](https://github.com/googleapis/python-aiplatform/commit/7dff586f38ef53887dca733741f874686d16563d))
+* Update documentation ([169dd44](https://github.com/googleapis/python-aiplatform/commit/169dd447703d7734a0b979cb77a0ab84ef8df68f))
+* Update feature creation message commentary ([acf3113](https://github.com/googleapis/python-aiplatform/commit/acf31138177fbcc29e469c863615cfa70b4ccc19))
+
 ## [1.71.1](https://github.com/googleapis/python-aiplatform/compare/v1.71.0...v1.71.1) (2024-10-31)
 
 

diff --git a/google/cloud/aiplatform/compat/__init__.py b/google/cloud/aiplatform/compat/__init__.py
@@ -94,6 +94,8 @@
     types.explanation_metadata = types.explanation_metadata_v1beta1
     types.feature = types.feature_v1beta1
     types.feature_group = types.feature_group_v1beta1
+    types.feature_monitor = types.feature_monitor_v1beta1
+    types.feature_monitor_job = types.feature_monitor_job_v1beta1
     types.feature_monitoring_stats = types.feature_monitoring_stats_v1beta1
     types.feature_online_store = types.feature_online_store_v1beta1
     types.feature_online_store_admin_service = (

diff --git a/google/cloud/aiplatform/compat/types/__init__.py b/google/cloud/aiplatform/compat/types/__init__.py
@@ -44,6 +44,8 @@
     explanation_metadata as explanation_metadata_v1beta1,
     feature as feature_v1beta1,
     feature_group as feature_group_v1beta1,
+    feature_monitor as feature_monitor_v1beta1,
+    feature_monitor_job as feature_monitor_job_v1beta1,
     feature_monitoring_stats as feature_monitoring_stats_v1beta1,
     feature_online_store as feature_online_store_v1beta1,
     feature_online_store_admin_service as feature_online_store_admin_service_v1beta1,

diff --git a/google/cloud/aiplatform/gapic_version.py b/google/cloud/aiplatform/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -2229,6 +2229,7 @@ def predict(
                 body=json.dumps({"instances": instances, "parameters": parameters}),
                 headers={"Content-Type": "application/json"},
                 use_dedicated_endpoint=use_dedicated_endpoint,
+                timeout=timeout,
             )
             json_response = raw_predict_response.json()
             return Prediction(
@@ -2277,6 +2278,7 @@ def predict(
                     }
                 ),
                 headers=headers,
+                timeout=timeout,
             )
 
             prediction_response = json.loads(response.text)
@@ -2382,6 +2384,7 @@ def raw_predict(
         headers: Dict[str, str],
         *,
         use_dedicated_endpoint: Optional[bool] = False,
+        timeout: Optional[float] = None,
     ) -> requests.models.Response:
         """Makes a prediction request using arbitrary headers.
 
@@ -2408,6 +2411,7 @@ def raw_predict(
             use_dedicated_endpoint (bool):
                 Optional. Default value is False. If set to True, the underlying prediction call will be made
                 using the dedicated endpoint dns.
+            timeout (float): Optional. The timeout for this request in seconds.
 
         Returns:
             A requests.models.Response object containing the status code and prediction results.
@@ -2435,15 +2439,17 @@ def raw_predict(
                     "and model are ready before making a prediction."
                 )
             url = f"https://{self._gca_resource.dedicated_endpoint_dns}/v1/{self.resource_name}:rawPredict"
-
-        return self.authorized_session.post(url=url, data=body, headers=headers)
+        return self.authorized_session.post(
+            url=url, data=body, headers=headers, timeout=timeout
+        )
 
     def stream_raw_predict(
         self,
         body: bytes,
         headers: Dict[str, str],
         *,
         use_dedicated_endpoint: Optional[bool] = False,
+        timeout: Optional[float] = None,
     ) -> Iterator[requests.models.Response]:
         """Makes a streaming prediction request using arbitrary headers.
 
@@ -2480,6 +2486,7 @@ def stream_raw_predict(
             use_dedicated_endpoint (bool):
                 Optional. Default value is False. If set to True, the underlying prediction call will be made
                 using the dedicated endpoint dns.
+            timeout (float): Optional. The timeout for this request in seconds.
 
         Yields:
             predictions (Iterator[requests.models.Response]):
@@ -2513,6 +2520,7 @@ def stream_raw_predict(
             url=url,
             data=body,
             headers=headers,
+            timeout=timeout,
             stream=True,
         ) as resp:
             for line in resp.iter_lines():

diff --git a/google/cloud/aiplatform/v1/schema/predict/instance/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/instance/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/instance_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/instance_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/params/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/params/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/params_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/params_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/prediction/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/prediction/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/prediction_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/prediction_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/trainingjob/definition/gapic_version.py b/google/cloud/aiplatform/v1/schema/trainingjob/definition/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/params/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/version.py b/google/cloud/aiplatform/version.py
@@ -15,4 +15,4 @@
 # limitations under the License.
 #
 
-__version__ = "1.71.1"
+__version__ = "1.72.0"
diff --git a/google/cloud/aiplatform_v1/gapic_version.py b/google/cloud/aiplatform_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform_v1beta1/gapic_version.py b/google/cloud/aiplatform_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.71.1"  # {x-release-please-version}
+__version__ = "1.72.0"  # {x-release-please-version}
diff --git a/pypi/_vertex_ai_placeholder/version.py b/pypi/_vertex_ai_placeholder/version.py
@@ -15,4 +15,4 @@
 # limitations under the License.
 #
 
-__version__ = "1.71.1"
+__version__ = "1.72.0"
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-aiplatform",
-    "version": "0.1.0"
+    "version": "1.72.0"
   },
   "snippets": [
     {

diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-aiplatform",
-    "version": "0.1.0"
+    "version": "1.72.0"
   },
   "snippets": [
     {

diff --git a/samples/model-builder/vector_search/vector_search_create_index_endpoint_sample.py b/samples/model-builder/vector_search/vector_search_create_index_endpoint_sample.py
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from google.cloud import aiplatform
 
 

diff --git a/setup.py b/setup.py
@@ -47,7 +47,7 @@
 ]
 
 profiler_extra_require = [
-    "tensorboard-plugin-profile >= 2.4.0, <3.0.0dev",
+    "tensorboard-plugin-profile >= 2.4.0, <2.18.0",  # <3.0.0dev",
     "werkzeug >= 2.0.0, <2.1.0dev",
     "tensorflow >=2.4.0, <3.0.0dev",
 ]
@@ -152,9 +152,9 @@
 ]
 
 langchain_extra_require = [
-    "langchain >= 0.1.16, < 0.3",
-    "langchain-core < 0.3",
-    "langchain-google-vertexai < 2",
+    "langchain >= 0.1.16, < 0.4",
+    "langchain-core < 0.4",
+    "langchain-google-vertexai < 3",
     "openinference-instrumentation-langchain >= 0.1.19, < 0.2",
 ]