feat: Add restricted Retail Search features for Retail API v2. (#68)

PiperOrigin-RevId: 387366941

Source-Link: googleapis/googleapis@cea896d

Source-Link: googleapis/googleapis-gen@11a99d5

docs/retail_v2/completion_service.rst

@@ -0,0 +1,6 @@
+CompletionService
+-----------------------------------
+
+.. automodule:: google.cloud.retail_v2.services.completion_service
+    :members:
+    :inherited-members:

docs/retail_v2/product_service.rst

@@ -4,3 +4,7 @@ ProductService
 .. automodule:: google.cloud.retail_v2.services.product_service
     :members:
     :inherited-members:
+
+.. automodule:: google.cloud.retail_v2.services.product_service.pagers
+    :members:
+    :inherited-members:

docs/retail_v2/search_service.rst

@@ -0,0 +1,10 @@
+SearchService
+-------------------------------
+
+.. automodule:: google.cloud.retail_v2.services.search_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.retail_v2.services.search_service.pagers
+    :members:
+    :inherited-members:

docs/retail_v2/services.rst

@@ -4,6 +4,8 @@ Services for Google Cloud Retail v2 API
     :maxdepth: 2
 
     catalog_service
+    completion_service
     prediction_service
     product_service
+    search_service
     user_event_service

google/cloud/retail/__init__.py

@@ -18,6 +18,12 @@
 from google.cloud.retail_v2.services.catalog_service.async_client import (
     CatalogServiceAsyncClient,
 )
+from google.cloud.retail_v2.services.completion_service.client import (
+    CompletionServiceClient,
+)
+from google.cloud.retail_v2.services.completion_service.async_client import (
+    CompletionServiceAsyncClient,
+)
 from google.cloud.retail_v2.services.prediction_service.client import (
     PredictionServiceClient,
 )
@@ -28,6 +34,10 @@
 from google.cloud.retail_v2.services.product_service.async_client import (
     ProductServiceAsyncClient,
 )
+from google.cloud.retail_v2.services.search_service.client import SearchServiceClient
+from google.cloud.retail_v2.services.search_service.async_client import (
+    SearchServiceAsyncClient,
+)
 from google.cloud.retail_v2.services.user_event_service.client import (
     UserEventServiceClient,
 )
@@ -37,15 +47,29 @@
 
 from google.cloud.retail_v2.types.catalog import Catalog
 from google.cloud.retail_v2.types.catalog import ProductLevelConfig
+from google.cloud.retail_v2.types.catalog_service import GetDefaultBranchRequest
+from google.cloud.retail_v2.types.catalog_service import GetDefaultBranchResponse
 from google.cloud.retail_v2.types.catalog_service import ListCatalogsRequest
 from google.cloud.retail_v2.types.catalog_service import ListCatalogsResponse
+from google.cloud.retail_v2.types.catalog_service import SetDefaultBranchRequest
 from google.cloud.retail_v2.types.catalog_service import UpdateCatalogRequest
+from google.cloud.retail_v2.types.common import Audience
+from google.cloud.retail_v2.types.common import ColorInfo
 from google.cloud.retail_v2.types.common import CustomAttribute
+from google.cloud.retail_v2.types.common import FulfillmentInfo
 from google.cloud.retail_v2.types.common import Image
+from google.cloud.retail_v2.types.common import Interval
 from google.cloud.retail_v2.types.common import PriceInfo
+from google.cloud.retail_v2.types.common import Promotion
+from google.cloud.retail_v2.types.common import Rating
 from google.cloud.retail_v2.types.common import UserInfo
+from google.cloud.retail_v2.types.completion_service import CompleteQueryRequest
+from google.cloud.retail_v2.types.completion_service import CompleteQueryResponse
 from google.cloud.retail_v2.types.import_config import BigQuerySource
+from google.cloud.retail_v2.types.import_config import CompletionDataInputConfig
 from google.cloud.retail_v2.types.import_config import GcsSource
+from google.cloud.retail_v2.types.import_config import ImportCompletionDataRequest
+from google.cloud.retail_v2.types.import_config import ImportCompletionDataResponse
 from google.cloud.retail_v2.types.import_config import ImportErrorsConfig
 from google.cloud.retail_v2.types.import_config import ImportMetadata
 from google.cloud.retail_v2.types.import_config import ImportProductsRequest
@@ -60,13 +84,27 @@
 from google.cloud.retail_v2.types.prediction_service import PredictRequest
 from google.cloud.retail_v2.types.prediction_service import PredictResponse
 from google.cloud.retail_v2.types.product import Product
+from google.cloud.retail_v2.types.product_service import AddFulfillmentPlacesMetadata
+from google.cloud.retail_v2.types.product_service import AddFulfillmentPlacesRequest
+from google.cloud.retail_v2.types.product_service import AddFulfillmentPlacesResponse
 from google.cloud.retail_v2.types.product_service import CreateProductRequest
 from google.cloud.retail_v2.types.product_service import DeleteProductRequest
 from google.cloud.retail_v2.types.product_service import GetProductRequest
+from google.cloud.retail_v2.types.product_service import ListProductsRequest
+from google.cloud.retail_v2.types.product_service import ListProductsResponse
+from google.cloud.retail_v2.types.product_service import RemoveFulfillmentPlacesMetadata
+from google.cloud.retail_v2.types.product_service import RemoveFulfillmentPlacesRequest
+from google.cloud.retail_v2.types.product_service import RemoveFulfillmentPlacesResponse
+from google.cloud.retail_v2.types.product_service import SetInventoryMetadata
+from google.cloud.retail_v2.types.product_service import SetInventoryRequest
+from google.cloud.retail_v2.types.product_service import SetInventoryResponse
 from google.cloud.retail_v2.types.product_service import UpdateProductRequest
 from google.cloud.retail_v2.types.purge_config import PurgeMetadata
 from google.cloud.retail_v2.types.purge_config import PurgeUserEventsRequest
 from google.cloud.retail_v2.types.purge_config import PurgeUserEventsResponse
+from google.cloud.retail_v2.types.search_service import SearchRequest
+from google.cloud.retail_v2.types.search_service import SearchResponse
+from google.cloud.retail_v2.types.user_event import CompletionDetail
 from google.cloud.retail_v2.types.user_event import ProductDetail
 from google.cloud.retail_v2.types.user_event import PurchaseTransaction
 from google.cloud.retail_v2.types.user_event import UserEvent
@@ -79,23 +117,41 @@
 __all__ = (
     "CatalogServiceClient",
     "CatalogServiceAsyncClient",
+    "CompletionServiceClient",
+    "CompletionServiceAsyncClient",
     "PredictionServiceClient",
     "PredictionServiceAsyncClient",
     "ProductServiceClient",
     "ProductServiceAsyncClient",
+    "SearchServiceClient",
+    "SearchServiceAsyncClient",
     "UserEventServiceClient",
     "UserEventServiceAsyncClient",
     "Catalog",
     "ProductLevelConfig",
+    "GetDefaultBranchRequest",
+    "GetDefaultBranchResponse",
     "ListCatalogsRequest",
     "ListCatalogsResponse",
+    "SetDefaultBranchRequest",
     "UpdateCatalogRequest",
+    "Audience",
+    "ColorInfo",
     "CustomAttribute",
+    "FulfillmentInfo",
     "Image",
+    "Interval",
     "PriceInfo",
+    "Promotion",
+    "Rating",
     "UserInfo",
+    "CompleteQueryRequest",
+    "CompleteQueryResponse",
     "BigQuerySource",
+    "CompletionDataInputConfig",
     "GcsSource",
+    "ImportCompletionDataRequest",
+    "ImportCompletionDataResponse",
     "ImportErrorsConfig",
     "ImportMetadata",
     "ImportProductsRequest",
@@ -110,13 +166,27 @@
     "PredictRequest",
     "PredictResponse",
     "Product",
+    "AddFulfillmentPlacesMetadata",
+    "AddFulfillmentPlacesRequest",
+    "AddFulfillmentPlacesResponse",
     "CreateProductRequest",
     "DeleteProductRequest",
     "GetProductRequest",
+    "ListProductsRequest",
+    "ListProductsResponse",
+    "RemoveFulfillmentPlacesMetadata",
+    "RemoveFulfillmentPlacesRequest",
+    "RemoveFulfillmentPlacesResponse",
+    "SetInventoryMetadata",
+    "SetInventoryRequest",
+    "SetInventoryResponse",
     "UpdateProductRequest",
     "PurgeMetadata",
     "PurgeUserEventsRequest",
     "PurgeUserEventsResponse",
+    "SearchRequest",
+    "SearchResponse",
+    "CompletionDetail",
     "ProductDetail",
     "PurchaseTransaction",
     "UserEvent",

google/cloud/retail_v2/__init__.py

@@ -16,24 +16,42 @@
 
 from .services.catalog_service import CatalogServiceClient
 from .services.catalog_service import CatalogServiceAsyncClient
+from .services.completion_service import CompletionServiceClient
+from .services.completion_service import CompletionServiceAsyncClient
 from .services.prediction_service import PredictionServiceClient
 from .services.prediction_service import PredictionServiceAsyncClient
 from .services.product_service import ProductServiceClient
 from .services.product_service import ProductServiceAsyncClient
+from .services.search_service import SearchServiceClient
+from .services.search_service import SearchServiceAsyncClient
 from .services.user_event_service import UserEventServiceClient
 from .services.user_event_service import UserEventServiceAsyncClient
 
 from .types.catalog import Catalog
 from .types.catalog import ProductLevelConfig
+from .types.catalog_service import GetDefaultBranchRequest
+from .types.catalog_service import GetDefaultBranchResponse
 from .types.catalog_service import ListCatalogsRequest
 from .types.catalog_service import ListCatalogsResponse
+from .types.catalog_service import SetDefaultBranchRequest
 from .types.catalog_service import UpdateCatalogRequest
+from .types.common import Audience
+from .types.common import ColorInfo
 from .types.common import CustomAttribute
+from .types.common import FulfillmentInfo
 from .types.common import Image
+from .types.common import Interval
 from .types.common import PriceInfo
+from .types.common import Promotion
+from .types.common import Rating
 from .types.common import UserInfo
+from .types.completion_service import CompleteQueryRequest
+from .types.completion_service import CompleteQueryResponse
 from .types.import_config import BigQuerySource
+from .types.import_config import CompletionDataInputConfig
 from .types.import_config import GcsSource
+from .types.import_config import ImportCompletionDataRequest
+from .types.import_config import ImportCompletionDataResponse
 from .types.import_config import ImportErrorsConfig
 from .types.import_config import ImportMetadata
 from .types.import_config import ImportProductsRequest
@@ -48,13 +66,27 @@
 from .types.prediction_service import PredictRequest
 from .types.prediction_service import PredictResponse
 from .types.product import Product
+from .types.product_service import AddFulfillmentPlacesMetadata
+from .types.product_service import AddFulfillmentPlacesRequest
+from .types.product_service import AddFulfillmentPlacesResponse
 from .types.product_service import CreateProductRequest
 from .types.product_service import DeleteProductRequest
 from .types.product_service import GetProductRequest
+from .types.product_service import ListProductsRequest
+from .types.product_service import ListProductsResponse
+from .types.product_service import RemoveFulfillmentPlacesMetadata
+from .types.product_service import RemoveFulfillmentPlacesRequest
+from .types.product_service import RemoveFulfillmentPlacesResponse
+from .types.product_service import SetInventoryMetadata
+from .types.product_service import SetInventoryRequest
+from .types.product_service import SetInventoryResponse
 from .types.product_service import UpdateProductRequest
 from .types.purge_config import PurgeMetadata
 from .types.purge_config import PurgeUserEventsRequest
 from .types.purge_config import PurgeUserEventsResponse
+from .types.search_service import SearchRequest
+from .types.search_service import SearchResponse
+from .types.user_event import CompletionDetail
 from .types.user_event import ProductDetail
 from .types.user_event import PurchaseTransaction
 from .types.user_event import UserEvent
@@ -66,27 +98,47 @@
 
 __all__ = (
     "CatalogServiceAsyncClient",
+    "CompletionServiceAsyncClient",
     "PredictionServiceAsyncClient",
     "ProductServiceAsyncClient",
+    "SearchServiceAsyncClient",
     "UserEventServiceAsyncClient",
+    "AddFulfillmentPlacesMetadata",
+    "AddFulfillmentPlacesRequest",
+    "AddFulfillmentPlacesResponse",
+    "Audience",
     "BigQuerySource",
     "Catalog",
     "CatalogServiceClient",
     "CollectUserEventRequest",
+    "ColorInfo",
+    "CompleteQueryRequest",
+    "CompleteQueryResponse",
+    "CompletionDataInputConfig",
+    "CompletionDetail",
+    "CompletionServiceClient",
     "CreateProductRequest",
     "CustomAttribute",
     "DeleteProductRequest",
+    "FulfillmentInfo",
     "GcsSource",
+    "GetDefaultBranchRequest",
+    "GetDefaultBranchResponse",
     "GetProductRequest",
     "Image",
+    "ImportCompletionDataRequest",
+    "ImportCompletionDataResponse",
     "ImportErrorsConfig",
     "ImportMetadata",
     "ImportProductsRequest",
     "ImportProductsResponse",
     "ImportUserEventsRequest",
     "ImportUserEventsResponse",
+    "Interval",
     "ListCatalogsRequest",
     "ListCatalogsResponse",
+    "ListProductsRequest",
+    "ListProductsResponse",
     "PredictRequest",
     "PredictResponse",
     "PredictionServiceClient",
@@ -97,13 +149,25 @@
     "ProductInputConfig",
     "ProductLevelConfig",
     "ProductServiceClient",
+    "Promotion",
     "PurchaseTransaction",
     "PurgeMetadata",
     "PurgeUserEventsRequest",
     "PurgeUserEventsResponse",
+    "Rating",
     "RejoinUserEventsMetadata",
     "RejoinUserEventsRequest",
     "RejoinUserEventsResponse",
+    "RemoveFulfillmentPlacesMetadata",
+    "RemoveFulfillmentPlacesRequest",
+    "RemoveFulfillmentPlacesResponse",
+    "SearchRequest",
+    "SearchResponse",
+    "SearchServiceClient",
+    "SetDefaultBranchRequest",
+    "SetInventoryMetadata",
+    "SetInventoryRequest",
+    "SetInventoryResponse",
     "UpdateCatalogRequest",
     "UpdateProductRequest",
     "UserEvent",