fix: tools imports now that rollout is complete

GoogleCloudPlatform · Jul 10, 2024 · acc1be4 · acc1be4
1 parent 1d56889
commit acc1be4
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/dfcx_scrapi/core/tools.py b/src/dfcx_scrapi/core/tools.py
@@ -17,11 +17,8 @@
 from dfcx_scrapi.core import scrapi_base
 from typing import Dict
 
-# TODO (pmarlow) Pending Bug fix to remove visibility on v3beta1 service
-# from google.cloud.dialogflowcx_v3beta1 import services
-# from google.cloud.dialogflowcx_v3beta1 import types
-from google.cloud.dialogflow_v3alpha1 import services
-from google.cloud.dialogflow_v3alpha1 import types
+from google.cloud.dialogflowcx_v3beta1 import services
+from google.cloud.dialogflowcx_v3beta1 import types
 
 
 class Tools(scrapi_base.ScrapiBase):