From 11a39e31939cdfe064fe82a74ad0216fdaaa2580 Mon Sep 17 00:00:00 2001 From: Qiao Wang Date: Fri, 16 Aug 2024 10:32:19 -0700 Subject: [PATCH] BREAKING_CHANGE: Remove disable_attribution from SDK interfaces PiperOrigin-RevId: 663776030 --- .../generative_models/_generative_models.py | 31 +------------------ 1 file changed, 1 insertion(+), 30 deletions(-) diff --git a/vertexai/generative_models/_generative_models.py b/vertexai/generative_models/_generative_models.py index 5a9d82ee79..d24d77a781 100644 --- a/vertexai/generative_models/_generative_models.py +++ b/vertexai/generative_models/_generative_models.py @@ -2296,36 +2296,7 @@ def __init__( datastore=datastore, ) - class GoogleSearchRetrieval: - r"""Tool to retrieve public web data for grounding, powered by - Google Search. - - Attributes: - disable_attribution (bool): - Optional. Disable using the result from this - tool in detecting grounding attribution. This - does not affect how the result is given to the - model for generation. - """ - - def __init__( - self, - disable_attribution: Optional[ - bool - ] = None, # pylint: disable=unused-argument - ): - """Initializes a Google Search Retrieval tool. - - Args: - disable_attribution (bool): - Optional. This field is Deprecated. Disable using the result - from this tool in detecting grounding attribution. This - does not affect how the result is given to the - model for generation. - """ - if disable_attribution is not None: - warnings.warn("disable_attribution is deprecated.") - self._raw_google_search_retrieval = gapic_tool_types.GoogleSearchRetrieval() + GoogleSearchRetrieval = grounding.GoogleSearchRetrieval def _to_content(