diff --git a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/client.py b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/client.py index 041b6c3..be21a0c 100644 --- a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/client.py +++ b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -408,7 +408,7 @@ def __init__( def suggest_queries( self, - request: auto_suggestion_service.SuggestQueriesRequest = None, + request: Union[auto_suggestion_service.SuggestQueriesRequest, dict] = None, *, retry: retries.Retry = gapic_v1.method.DEFAULT, timeout: float = None, @@ -418,7 +418,7 @@ def suggest_queries( AutoSuggestion tolerance should be less than 1 second. Args: - request (google.cloud.dataqna_v1alpha.types.SuggestQueriesRequest): + request (Union[google.cloud.dataqna_v1alpha.types.SuggestQueriesRequest, dict]): The request object. Request for query suggestions. retry (google.api_core.retry.Retry): Designation of what errors, if any, should be retried. diff --git a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/base.py b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/base.py index f27152f..afc02ce 100644 --- a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/base.py +++ b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/base.py @@ -115,7 +115,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc.py b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc.py index 6703d90..83ede04 100644 --- a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc.py +++ b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc.py @@ -150,16 +150,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc_asyncio.py b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc_asyncio.py index 02a5d22..f182915 100644 --- a/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc_asyncio.py +++ b/google/cloud/dataqna_v1alpha/services/auto_suggestion_service/transports/grpc_asyncio.py @@ -197,16 +197,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/dataqna_v1alpha/services/question_service/client.py b/google/cloud/dataqna_v1alpha/services/question_service/client.py index 534591b..1d754dd 100644 --- a/google/cloud/dataqna_v1alpha/services/question_service/client.py +++ b/google/cloud/dataqna_v1alpha/services/question_service/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -391,7 +391,7 @@ def __init__( def get_question( self, - request: question_service.GetQuestionRequest = None, + request: Union[question_service.GetQuestionRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -401,7 +401,7 @@ def get_question( r"""Gets a previously created question. Args: - request (google.cloud.dataqna_v1alpha.types.GetQuestionRequest): + request (Union[google.cloud.dataqna_v1alpha.types.GetQuestionRequest, dict]): The request object. A request to get a previously created question. name (str): @@ -465,7 +465,7 @@ def get_question( def create_question( self, - request: question_service.CreateQuestionRequest = None, + request: Union[question_service.CreateQuestionRequest, dict] = None, *, parent: str = None, question: gcd_question.Question = None, @@ -476,7 +476,7 @@ def create_question( r"""Creates a question. Args: - request (google.cloud.dataqna_v1alpha.types.CreateQuestionRequest): + request (Union[google.cloud.dataqna_v1alpha.types.CreateQuestionRequest, dict]): The request object. Request to create a question resource. parent (str): @@ -548,7 +548,7 @@ def create_question( def execute_question( self, - request: question_service.ExecuteQuestionRequest = None, + request: Union[question_service.ExecuteQuestionRequest, dict] = None, *, name: str = None, interpretation_index: int = None, @@ -559,7 +559,7 @@ def execute_question( r"""Executes an interpretation. Args: - request (google.cloud.dataqna_v1alpha.types.ExecuteQuestionRequest): + request (Union[google.cloud.dataqna_v1alpha.types.ExecuteQuestionRequest, dict]): The request object. Request to execute an interpretation. name (str): @@ -632,7 +632,7 @@ def execute_question( def get_user_feedback( self, - request: question_service.GetUserFeedbackRequest = None, + request: Union[question_service.GetUserFeedbackRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -642,7 +642,7 @@ def get_user_feedback( r"""Gets previously created user feedback. Args: - request (google.cloud.dataqna_v1alpha.types.GetUserFeedbackRequest): + request (Union[google.cloud.dataqna_v1alpha.types.GetUserFeedbackRequest, dict]): The request object. Request to get user feedback. name (str): Required. The unique identifier for the user feedback. @@ -702,7 +702,7 @@ def get_user_feedback( def update_user_feedback( self, - request: question_service.UpdateUserFeedbackRequest = None, + request: Union[question_service.UpdateUserFeedbackRequest, dict] = None, *, user_feedback: gcd_user_feedback.UserFeedback = None, update_mask: field_mask_pb2.FieldMask = None, @@ -714,7 +714,7 @@ def update_user_feedback( there was none before (upsert). Args: - request (google.cloud.dataqna_v1alpha.types.UpdateUserFeedbackRequest): + request (Union[google.cloud.dataqna_v1alpha.types.UpdateUserFeedbackRequest, dict]): The request object. Request to updates user feedback. user_feedback (google.cloud.dataqna_v1alpha.types.UserFeedback): Required. The user feedback to diff --git a/google/cloud/dataqna_v1alpha/services/question_service/transports/base.py b/google/cloud/dataqna_v1alpha/services/question_service/transports/base.py index f8c5a87..7677868 100644 --- a/google/cloud/dataqna_v1alpha/services/question_service/transports/base.py +++ b/google/cloud/dataqna_v1alpha/services/question_service/transports/base.py @@ -119,7 +119,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc.py b/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc.py index fe837f1..272bf11 100644 --- a/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc.py +++ b/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc.py @@ -100,16 +100,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc_asyncio.py b/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc_asyncio.py index 083c07d..16c0989 100644 --- a/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc_asyncio.py +++ b/google/cloud/dataqna_v1alpha/services/question_service/transports/grpc_asyncio.py @@ -147,16 +147,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/scripts/fixup_dataqna_v1alpha_keywords.py b/scripts/fixup_dataqna_v1alpha_keywords.py index da5349a..8b84330 100644 --- a/scripts/fixup_dataqna_v1alpha_keywords.py +++ b/scripts/fixup_dataqna_v1alpha_keywords.py @@ -39,12 +39,12 @@ def partition( class dataqnaCallTransformer(cst.CSTTransformer): CTRL_PARAMS: Tuple[str] = ('retry', 'timeout', 'metadata') METHOD_TO_PARAMS: Dict[str, Tuple[str]] = { - 'create_question': ('parent', 'question', ), - 'execute_question': ('name', 'interpretation_index', ), - 'get_question': ('name', 'read_mask', ), - 'get_user_feedback': ('name', ), - 'suggest_queries': ('parent', 'scopes', 'query', 'suggestion_types', ), - 'update_user_feedback': ('user_feedback', 'update_mask', ), + 'create_question': ('parent', 'question', ), + 'execute_question': ('name', 'interpretation_index', ), + 'get_question': ('name', 'read_mask', ), + 'get_user_feedback': ('name', ), + 'suggest_queries': ('parent', 'scopes', 'query', 'suggestion_types', ), + 'update_user_feedback': ('user_feedback', 'update_mask', ), } def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: @@ -63,7 +63,7 @@ def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: return updated kwargs, ctrl_kwargs = partition( - lambda a: not a.keyword.value in self.CTRL_PARAMS, + lambda a: a.keyword.value not in self.CTRL_PARAMS, kwargs )