fix: remove SpeechHelpers from STT V2 client library (#452)

* fix: remove SpeechHelpers from STT V2 client library * chore: update post processing script * chore: clean up post processing script Co-authored-by: Nitsan Shai <[email protected]> Co-authored-by: Anthonios Partheniou <[email protected]>
googleapis · Oct 5, 2022 · 9ef71a1 · 9ef71a1
1 parent e70f7c7
commit 9ef71a1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 14 deletions.
diff --git a/packages/google-cloud-python-speech/google/cloud/speech_v2/__init__.py b/packages/google-cloud-python-speech/google/cloud/speech_v2/__init__.py
@@ -69,13 +69,6 @@
 from .types.cloud_speech import UpdateRecognizerRequest
 from .types.cloud_speech import WordInfo
 
-from google.cloud.speech_v1.helpers import SpeechHelpers
-
-
-class SpeechClient(SpeechHelpers, SpeechClient):
-    __doc__ = SpeechClient.__doc__
-
-
 __all__ = (
     "SpeechAsyncClient",
     "AutoDetectDecodingConfig",

diff --git a/packages/google-cloud-python-speech/owlbot.py b/packages/google-cloud-python-speech/owlbot.py
@@ -23,28 +23,29 @@
 default_version = "v1"
 
 for library in s.get_staging_dirs(default_version):
-    # Add the manually written SpeechHelpers to v1 and v1p1beta1
-    # See google/cloud/speech_v1/helpers.py for details
-    count = s.replace(library / f"google/cloud/speech_{library.name}/__init__.py",
-                        """__all__ = \(""",
-                        """from google.cloud.speech_v1.helpers import SpeechHelpers
+    if "v1" in library.name:
+        # Add the manually written SpeechHelpers to v1 and v1p1beta1
+        # See google/cloud/speech_v1/helpers.py for details
+        count = s.replace(library / f"google/cloud/speech_{library.name}/__init__.py",
+                            """__all__ = \(""",
+                            """from google.cloud.speech_v1.helpers import SpeechHelpers
 
 class SpeechClient(SpeechHelpers, SpeechClient):
     __doc__ = SpeechClient.__doc__
 
 __all__ = (
                         """,
                     )
+        assert count == 1
 
     if library.name == "v1":
         # Import from speech_v1 to get the client with SpeechHelpers
         count = s.replace(library / "google/cloud/speech/__init__.py",
             """from google\.cloud\.speech_v1\.services\.speech\.client import SpeechClient""",
             """from google.cloud.speech_v1 import SpeechClient"""
             )
+        assert count == 1
 
-    # Don't move over __init__.py, as we modify it to make the generated client
-    # use helpers.py.
     s.move(library, excludes=["setup.py"])
 
 s.remove_staging_dirs()