feat: add context manager support in client

google/cloud/dialogflow_v2/services/agents/async_client.py

@@ -915,6 +915,12 @@ async def get_validation_result(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/agents/client.py

@@ -343,10 +343,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def get_agent(
@@ -1110,6 +1107,19 @@ def get_validation_result(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/agents/transports/base.py

@@ -193,6 +193,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def operations_client(self) -> operations_v1.OperationsClient:
         """Return the client designed to process long-running operations."""

google/cloud/dialogflow_v2/services/agents/transports/grpc.py

@@ -565,5 +565,8 @@ def get_validation_result(
             )
         return self._stubs["get_validation_result"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("AgentsGrpcTransport",)

google/cloud/dialogflow_v2/services/agents/transports/grpc_asyncio.py

@@ -575,5 +575,8 @@ def get_validation_result(
             )
         return self._stubs["get_validation_result"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("AgentsGrpcAsyncIOTransport",)

google/cloud/dialogflow_v2/services/answer_records/async_client.py

@@ -361,6 +361,12 @@ async def update_answer_record(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/answer_records/client.py

@@ -345,10 +345,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def list_answer_records(
@@ -545,6 +542,19 @@ def update_answer_record(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/answer_records/transports/base.py

@@ -168,6 +168,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def list_answer_records(
         self,

google/cloud/dialogflow_v2/services/answer_records/transports/grpc.py

@@ -285,5 +285,8 @@ def update_answer_record(
             )
         return self._stubs["update_answer_record"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("AnswerRecordsGrpcTransport",)

google/cloud/dialogflow_v2/services/answer_records/transports/grpc_asyncio.py

@@ -289,5 +289,8 @@ def update_answer_record(
             )
         return self._stubs["update_answer_record"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("AnswerRecordsGrpcAsyncIOTransport",)

google/cloud/dialogflow_v2/services/contexts/async_client.py

@@ -672,6 +672,12 @@ async def delete_all_contexts(
             request, retry=retry, timeout=timeout, metadata=metadata,
         )
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/contexts/client.py

@@ -345,10 +345,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def list_contexts(
@@ -867,6 +864,19 @@ def delete_all_contexts(
             request, retry=retry, timeout=timeout, metadata=metadata,
         )
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/contexts/transports/base.py

@@ -179,6 +179,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def list_contexts(
         self,

google/cloud/dialogflow_v2/services/contexts/transports/grpc.py

@@ -384,5 +384,8 @@ def delete_all_contexts(
             )
         return self._stubs["delete_all_contexts"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("ContextsGrpcTransport",)

google/cloud/dialogflow_v2/services/contexts/transports/grpc_asyncio.py

@@ -391,5 +391,8 @@ def delete_all_contexts(
             )
         return self._stubs["delete_all_contexts"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("ContextsGrpcAsyncIOTransport",)

google/cloud/dialogflow_v2/services/conversation_profiles/async_client.py

@@ -586,6 +586,12 @@ async def delete_conversation_profile(
             request, retry=retry, timeout=timeout, metadata=metadata,
         )
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/conversation_profiles/client.py

@@ -414,10 +414,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def list_conversation_profiles(
@@ -842,6 +839,19 @@ def delete_conversation_profile(
             request, retry=retry, timeout=timeout, metadata=metadata,
         )
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/conversation_profiles/transports/base.py

@@ -188,6 +188,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def list_conversation_profiles(
         self,

google/cloud/dialogflow_v2/services/conversation_profiles/transports/grpc.py

@@ -387,5 +387,8 @@ def delete_conversation_profile(
             )
         return self._stubs["delete_conversation_profile"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("ConversationProfilesGrpcTransport",)

google/cloud/dialogflow_v2/services/conversation_profiles/transports/grpc_asyncio.py

@@ -391,5 +391,8 @@ def delete_conversation_profile(
             )
         return self._stubs["delete_conversation_profile"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("ConversationProfilesGrpcAsyncIOTransport",)

google/cloud/dialogflow_v2/services/conversations/async_client.py

@@ -601,6 +601,12 @@ async def list_messages(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/conversations/client.py

@@ -378,10 +378,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def create_conversation(
@@ -811,6 +808,19 @@ def list_messages(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/dialogflow_v2/services/conversations/transports/base.py

@@ -177,6 +177,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def create_conversation(
         self,

google/cloud/dialogflow_v2/services/conversations/transports/grpc.py

@@ -392,5 +392,8 @@ def list_messages(
             )
         return self._stubs["list_messages"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("ConversationsGrpcTransport",)

google/cloud/dialogflow_v2/services/conversations/transports/grpc_asyncio.py

@@ -399,5 +399,8 @@ def list_messages(
             )
         return self._stubs["list_messages"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("ConversationsGrpcAsyncIOTransport",)