Skip to content
This repository has been archived by the owner on Dec 17, 2023. It is now read-only.

Commit

Permalink
fix: remove gRPC send/recv limits (#245)
Browse files Browse the repository at this point in the history
  • Loading branch information
@yoshi-automation
yoshi-automation committed Dec 30, 2020
1 parent f3cb980 commit ceba454
Show file tree
Hide file tree
Showing 70 changed files with 817 additions and 126 deletions.
4 changes: 2 additions & 2 deletions google/cloud/dialogflow_v2/__init__.py
View file
Expand Up @@ -139,7 +139,6 @@
"EntityTypeBatch",
"EntityTypesClient",
"Environment",
"EnvironmentsClient",
"EventInput",
"ExportAgentRequest",
"ExportAgentResponse",
Expand Down Expand Up @@ -179,6 +178,7 @@
"SentimentAnalysisResult",
"SessionEntityType",
"SessionEntityTypesClient",
"SessionsClient",
"SetAgentRequest",
"SpeechContext",
"SpeechModelVariant",
Expand All @@ -199,5 +199,5 @@
"VoiceSelectionParams",
"WebhookRequest",
"WebhookResponse",
"SessionsClient",
"EnvironmentsClient",
)
1 change: 0 additions & 1 deletion google/cloud/dialogflow_v2/services/agents/transports/__init__.py
View file
Expand Up @@ -28,7 +28,6 @@
_transport_registry["grpc"] = AgentsGrpcTransport
_transport_registry["grpc_asyncio"] = AgentsGrpcAsyncIOTransport


__all__ = (
"AgentsTransport",
"AgentsGrpcTransport",
Expand Down
19 changes: 13 additions & 6 deletions google/cloud/dialogflow_v2/services/agents/transports/grpc.py
View file
Expand Up @@ -151,6 +151,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -169,9 +173,14 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

self._stubs = {} # type: Dict[str, Callable]
self._operations_client = None

# Run the base constructor.
super().__init__(
Expand All @@ -195,7 +204,7 @@ def create_channel(
) -> grpc.Channel:
"""Create and return a gRPC channel object.
Args:
address (Optionsl[str]): The host for the channel to use.
address (Optional[str]): The host for the channel to use.
credentials (Optional[~.Credentials]): The
authorization credentials to attach to requests. These
credentials identify this application to the service. If
Expand Down Expand Up @@ -242,13 +251,11 @@ def operations_client(self) -> operations_v1.OperationsClient:
client.
"""
# Sanity check: Only create a new client if we do not already have one.
if "operations_client" not in self.__dict__:
self.__dict__["operations_client"] = operations_v1.OperationsClient(
self.grpc_channel
)
if self._operations_client is None:
self._operations_client = operations_v1.OperationsClient(self.grpc_channel)

# Return the client from cache.
return self.__dict__["operations_client"]
return self._operations_client

@property
def get_agent(self) -> Callable[[agent.GetAgentRequest], agent.Agent]:
Expand Down
15 changes: 12 additions & 3 deletions google/cloud/dialogflow_v2/services/agents/transports/grpc_asyncio.py
View file
Expand Up @@ -196,6 +196,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -214,6 +218,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

# Run the base constructor.
Expand All @@ -227,6 +235,7 @@ def __init__(
)

self._stubs = {}
self._operations_client = None

@property
def grpc_channel(self) -> aio.Channel:
Expand All @@ -246,13 +255,13 @@ def operations_client(self) -> operations_v1.OperationsAsyncClient:
client.
"""
# Sanity check: Only create a new client if we do not already have one.
if "operations_client" not in self.__dict__:
self.__dict__["operations_client"] = operations_v1.OperationsAsyncClient(
if self._operations_client is None:
self._operations_client = operations_v1.OperationsAsyncClient(
self.grpc_channel
)

# Return the client from cache.
return self.__dict__["operations_client"]
return self._operations_client

@property
def get_agent(self) -> Callable[[agent.GetAgentRequest], Awaitable[agent.Agent]]:
Expand Down
1 change: 0 additions & 1 deletion google/cloud/dialogflow_v2/services/contexts/transports/__init__.py
View file
Expand Up @@ -28,7 +28,6 @@
_transport_registry["grpc"] = ContextsGrpcTransport
_transport_registry["grpc_asyncio"] = ContextsGrpcAsyncIOTransport


__all__ = (
"ContextsTransport",
"ContextsGrpcTransport",
Expand Down
10 changes: 9 additions & 1 deletion google/cloud/dialogflow_v2/services/contexts/transports/grpc.py
View file
Expand Up @@ -148,6 +148,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -166,6 +170,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

self._stubs = {} # type: Dict[str, Callable]
Expand All @@ -192,7 +200,7 @@ def create_channel(
) -> grpc.Channel:
"""Create and return a gRPC channel object.
Args:
address (Optionsl[str]): The host for the channel to use.
address (Optional[str]): The host for the channel to use.
credentials (Optional[~.Credentials]): The
authorization credentials to attach to requests. These
credentials identify this application to the service. If
Expand Down
8 changes: 8 additions & 0 deletions google/cloud/dialogflow_v2/services/contexts/transports/grpc_asyncio.py
View file
Expand Up @@ -193,6 +193,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -211,6 +215,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

# Run the base constructor.
Expand Down
1 change: 0 additions & 1 deletion google/cloud/dialogflow_v2/services/entity_types/transports/__init__.py
View file
Expand Up @@ -28,7 +28,6 @@
_transport_registry["grpc"] = EntityTypesGrpcTransport
_transport_registry["grpc_asyncio"] = EntityTypesGrpcAsyncIOTransport


__all__ = (
"EntityTypesTransport",
"EntityTypesGrpcTransport",
Expand Down
19 changes: 13 additions & 6 deletions google/cloud/dialogflow_v2/services/entity_types/transports/grpc.py
View file
Expand Up @@ -151,6 +151,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -169,9 +173,14 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

self._stubs = {} # type: Dict[str, Callable]
self._operations_client = None

# Run the base constructor.
super().__init__(
Expand All @@ -195,7 +204,7 @@ def create_channel(
) -> grpc.Channel:
"""Create and return a gRPC channel object.
Args:
address (Optionsl[str]): The host for the channel to use.
address (Optional[str]): The host for the channel to use.
credentials (Optional[~.Credentials]): The
authorization credentials to attach to requests. These
credentials identify this application to the service. If
Expand Down Expand Up @@ -242,13 +251,11 @@ def operations_client(self) -> operations_v1.OperationsClient:
client.
"""
# Sanity check: Only create a new client if we do not already have one.
if "operations_client" not in self.__dict__:
self.__dict__["operations_client"] = operations_v1.OperationsClient(
self.grpc_channel
)
if self._operations_client is None:
self._operations_client = operations_v1.OperationsClient(self.grpc_channel)

# Return the client from cache.
return self.__dict__["operations_client"]
return self._operations_client

@property
def list_entity_types(
Expand Down
15 changes: 12 additions & 3 deletions google/cloud/dialogflow_v2/services/entity_types/transports/grpc_asyncio.py
View file
Expand Up @@ -196,6 +196,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -214,6 +218,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

# Run the base constructor.
Expand All @@ -227,6 +235,7 @@ def __init__(
)

self._stubs = {}
self._operations_client = None

@property
def grpc_channel(self) -> aio.Channel:
Expand All @@ -246,13 +255,13 @@ def operations_client(self) -> operations_v1.OperationsAsyncClient:
client.
"""
# Sanity check: Only create a new client if we do not already have one.
if "operations_client" not in self.__dict__:
self.__dict__["operations_client"] = operations_v1.OperationsAsyncClient(
if self._operations_client is None:
self._operations_client = operations_v1.OperationsAsyncClient(
self.grpc_channel
)

# Return the client from cache.
return self.__dict__["operations_client"]
return self._operations_client

@property
def list_entity_types(
Expand Down
1 change: 0 additions & 1 deletion google/cloud/dialogflow_v2/services/environments/transports/__init__.py
View file
Expand Up @@ -28,7 +28,6 @@
_transport_registry["grpc"] = EnvironmentsGrpcTransport
_transport_registry["grpc_asyncio"] = EnvironmentsGrpcAsyncIOTransport


__all__ = (
"EnvironmentsTransport",
"EnvironmentsGrpcTransport",
Expand Down
10 changes: 9 additions & 1 deletion google/cloud/dialogflow_v2/services/environments/transports/grpc.py
View file
Expand Up @@ -147,6 +147,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -165,6 +169,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

self._stubs = {} # type: Dict[str, Callable]
Expand All @@ -191,7 +199,7 @@ def create_channel(
) -> grpc.Channel:
"""Create and return a gRPC channel object.
Args:
address (Optionsl[str]): The host for the channel to use.
address (Optional[str]): The host for the channel to use.
credentials (Optional[~.Credentials]): The
authorization credentials to attach to requests. These
credentials identify this application to the service. If
Expand Down
8 changes: 8 additions & 0 deletions google/cloud/dialogflow_v2/services/environments/transports/grpc_asyncio.py
View file
Expand Up @@ -192,6 +192,10 @@ def __init__(
ssl_credentials=ssl_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)
self._ssl_channel_credentials = ssl_credentials
else:
Expand All @@ -210,6 +214,10 @@ def __init__(
ssl_credentials=ssl_channel_credentials,
scopes=scopes or self.AUTH_SCOPES,
quota_project_id=quota_project_id,
options=[
("grpc.max_send_message_length", -1),
("grpc.max_receive_message_length", -1),
],
)

# Run the base constructor.
Expand Down
1 change: 0 additions & 1 deletion google/cloud/dialogflow_v2/services/intents/transports/__init__.py
View file
Expand Up @@ -28,7 +28,6 @@
_transport_registry["grpc"] = IntentsGrpcTransport
_transport_registry["grpc_asyncio"] = IntentsGrpcAsyncIOTransport


__all__ = (
"IntentsTransport",
"IntentsGrpcTransport",
Expand Down

0 comments on commit ceba454

Please sign in to comment.