[HOLD] Reintroduce TTS WS

deepgram · Aug 5, 2024 · 5e637af · 5e637af
1 parent 78ec6d9
commit 5e637af
Show file tree

Hide file tree

Showing 14 changed files with 1,723 additions and 97 deletions.
diff --git a/deepgram/__init__.py b/deepgram/__init__.py
@@ -96,7 +96,7 @@
 from .client import (
     SpeakOptions,
     SpeakRESTOptions,
-    # SpeakWebSocketOptions,
+    SpeakWebSocketOptions,
     # FileSource,
     SpeakRestSource,
     SpeakSource,
@@ -115,21 +115,21 @@
     SpeakRESTResponse,
 )
 
-# ## speak WebSocket
-# from .client import (
-#     SpeakWebSocketClient,
-#     AsyncSpeakWebSocketClient,
-# )
-# from .client import (
-#     SpeakWebSocketResponse,
-#     # OpenResponse,
-#     # MetadataResponse,
-#     FlushedResponse,
-#     # CloseResponse,
-#     # UnhandledResponse,
-#     WarningResponse,
-#     # ErrorResponse,
-# )
+## speak WebSocket
+from .client import (
+    SpeakWebSocketClient,
+    AsyncSpeakWebSocketClient,
+)
+from .client import (
+    SpeakWebSocketResponse,
+    # OpenResponse,
+    # MetadataResponse,
+    FlushedResponse,
+    # CloseResponse,
+    # UnhandledResponse,
+    WarningResponse,
+    # ErrorResponse,
+)
 
 # manage
 from .client import ManageClient, AsyncManageClient

diff --git a/deepgram/client.py b/deepgram/client.py
@@ -100,7 +100,7 @@
 from .clients import (
     SpeakOptions,
     SpeakRESTOptions,
-    # SpeakWebSocketOptions,
+    SpeakWebSocketOptions,
     # FileSource,
     SpeakRestSource,
     SpeakSource,
@@ -119,21 +119,21 @@
     SpeakRESTResponse,
 )
 
-# ## speak WebSocket
-# from .clients import (
-#     SpeakWebSocketClient,
-#     AsyncSpeakWebSocketClient,
-# )
-# from .clients import (
-#     SpeakWebSocketResponse,
-#     # OpenResponse,
-#     # MetadataResponse,
-#     FlushedResponse,
-#     # CloseResponse,
-#     # UnhandledResponse,
-#     WarningResponse,
-#     # ErrorResponse,
-# )
+## speak WebSocket
+from .clients import (
+    SpeakWebSocketClient,
+    AsyncSpeakWebSocketClient,
+)
+from .clients import (
+    SpeakWebSocketResponse,
+    # OpenResponse,
+    # MetadataResponse,
+    FlushedResponse,
+    # CloseResponse,
+    # UnhandledResponse,
+    WarningResponse,
+    # ErrorResponse,
+)
 
 # manage client classes/input
 from .clients import ManageClient, AsyncManageClient

diff --git a/deepgram/clients/__init__.py b/deepgram/clients/__init__.py
@@ -105,7 +105,7 @@
 from .speak import (
     SpeakOptions,
     SpeakRESTOptions,
-    # SpeakWebSocketOptions,
+    SpeakWebSocketOptions,
     # FileSource,
     SpeakRestSource,
     SpeakSource,
@@ -125,21 +125,21 @@
     SpeakRESTResponse,
 )
 
-# ## text-to-speech WebSocket
-# from .speak import (
-#     SpeakWebSocketClient,
-#     AsyncSpeakWebSocketClient,
-# )
-# from .speak import (
-#     SpeakWebSocketResponse,
-#     # OpenResponse,
-#     # MetadataResponse,
-#     FlushedResponse,
-#     # CloseResponse,
-#     # UnhandledResponse,
-#     WarningResponse,
-#     # ErrorResponse,
-# )
+## text-to-speech WebSocket
+from .speak import (
+    SpeakWebSocketClient,
+    AsyncSpeakWebSocketClient,
+)
+from .speak import (
+    SpeakWebSocketResponse,
+    # OpenResponse,
+    # MetadataResponse,
+    FlushedResponse,
+    # CloseResponse,
+    # UnhandledResponse,
+    WarningResponse,
+    # ErrorResponse,
+)
 
 # manage
 from .manage import ManageClient, AsyncManageClient

diff --git a/deepgram/clients/speak/__init__.py b/deepgram/clients/speak/__init__.py
@@ -9,26 +9,26 @@
     SpeakClient,  # backward compat
     SpeakRESTClient,
     AsyncSpeakRESTClient,
-    # SpeakWebSocketClient,
-    # AsyncSpeakWebSocketClient,
+    SpeakWebSocketClient,
+    AsyncSpeakWebSocketClient,
 )
 from .client import (
     SpeakOptions,
     SpeakRESTOptions,
-    # SpeakWebSocketOptions,
+    SpeakWebSocketOptions,
     FileSource,
     SpeakRestSource,
     SpeakSource,
 )
 from .client import (
     SpeakResponse,  # backward compat
     SpeakRESTResponse,
-    # SpeakWebSocketResponse,
-    # OpenResponse,
-    # MetadataResponse,
-    # FlushedResponse,
-    # CloseResponse,
-    # UnhandledResponse,
-    # WarningResponse,
-    # ErrorResponse,
+    SpeakWebSocketResponse,
+    OpenResponse,
+    MetadataResponse,
+    FlushedResponse,
+    CloseResponse,
+    UnhandledResponse,
+    WarningResponse,
+    ErrorResponse,
 )
diff --git a/deepgram/clients/speak/client.py b/deepgram/clients/speak/client.py
@@ -5,27 +5,27 @@
 from .v1 import (
     SpeakRESTClient as SpeakRESTClientLatest,
     AsyncSpeakRESTClient as AsyncSpeakRESTClientLatest,
-    # SpeakWebSocketClient as SpeakWebSocketClientLatest,
-    # AsyncSpeakWebSocketClient as AsyncSpeakWebSocketClientLatest,
+    SpeakWebSocketClient as SpeakWebSocketClientLatest,
+    AsyncSpeakWebSocketClient as AsyncSpeakWebSocketClientLatest,
 )
 from .v1 import (
     SpeakOptions as SpeakOptionsLatest,
     SpeakRESTOptions as SpeakRESTOptionsLatest,
-    # SpeakWebSocketOptions as SpeakWebSocketOptionsLatest,
+    SpeakWebSocketOptions as SpeakWebSocketOptionsLatest,
     FileSource as FileSourceLatest,
     SpeakRestSource as SpeakRestSourceLatest,
     SpeakSource as SpeakSourceLatest,
 )
 from .v1 import (
     SpeakRESTResponse as SpeakRESTResponseLatest,
-    # SpeakWebSocketResponse as SpeakWebSocketResponseLatest,
-    # OpenResponse as OpenResponseLatest,
-    # MetadataResponse as MetadataResponseLatest,
-    # FlushedResponse as FlushedResponseLatest,
-    # CloseResponse as CloseResponseLatest,
-    # UnhandledResponse as UnhandledResponseLatest,
-    # WarningResponse as WarningResponseLatest,
-    # ErrorResponse as ErrorResponseLatest,
+    SpeakWebSocketResponse as SpeakWebSocketResponseLatest,
+    OpenResponse as OpenResponseLatest,
+    MetadataResponse as MetadataResponseLatest,
+    FlushedResponse as FlushedResponseLatest,
+    CloseResponse as CloseResponseLatest,
+    UnhandledResponse as UnhandledResponseLatest,
+    WarningResponse as WarningResponseLatest,
+    ErrorResponse as ErrorResponseLatest,
 )
 
 # The client.py points to the current supported version in the SDK.
@@ -35,21 +35,21 @@
 # input
 SpeakOptions = SpeakOptionsLatest
 SpeakRESTOptions = SpeakRESTOptionsLatest
-# SpeakWebSocketOptions = SpeakWebSocketOptionsLatest
+SpeakWebSocketOptions = SpeakWebSocketOptionsLatest
 SpeakRestSource = SpeakRestSourceLatest
 FileSource = FileSourceLatest
 SpeakSource = SpeakSourceLatest
 
 # output
 SpeakRESTResponse = SpeakRESTResponseLatest
-# SpeakWebSocketResponse = SpeakWebSocketResponseLatest
-# OpenResponse = OpenResponseLatest
-# MetadataResponse = MetadataResponseLatest
-# FlushedResponse = FlushedResponseLatest
-# CloseResponse = CloseResponseLatest
-# UnhandledResponse = UnhandledResponseLatest
-# WarningResponse = WarningResponseLatest
-# ErrorResponse = ErrorResponseLatest
+SpeakWebSocketResponse = SpeakWebSocketResponseLatest
+OpenResponse = OpenResponseLatest
+MetadataResponse = MetadataResponseLatest
+FlushedResponse = FlushedResponseLatest
+CloseResponse = CloseResponseLatest
+UnhandledResponse = UnhandledResponseLatest
+WarningResponse = WarningResponseLatest
+ErrorResponse = ErrorResponseLatest
 
 
 # backward compatibility
@@ -59,5 +59,5 @@
 # clients
 SpeakRESTClient = SpeakRESTClientLatest
 AsyncSpeakRESTClient = AsyncSpeakRESTClientLatest
-# SpeakWebSocketClient = SpeakWebSocketClientLatest
-# AsyncSpeakWebSocketClient = AsyncSpeakWebSocketClientLatest
+SpeakWebSocketClient = SpeakWebSocketClientLatest
+AsyncSpeakWebSocketClient = AsyncSpeakWebSocketClientLatest
diff --git a/deepgram/clients/speak/v1/__init__.py b/deepgram/clients/speak/v1/__init__.py
@@ -11,26 +11,25 @@
     SpeakSource,
     FileSource,
 )
-
-# from .websocket import (
-#     SpeakWebSocketOptions,
-# )
+from .websocket import (
+    SpeakWebSocketOptions,
+)
 from ....options import DeepgramClientOptions, ClientOptionsFromEnv
 
 # rest
 from .rest import SpeakRESTClient, AsyncSpeakRESTClient
 
 from .rest import SpeakRESTResponse
 
-# # websocket
-# from .websocket import SpeakWebSocketClient, AsyncSpeakWebSocketClient
-# from .websocket import (
-#     SpeakWebSocketResponse,
-#     OpenResponse,
-#     MetadataResponse,
-#     FlushedResponse,
-#     CloseResponse,
-#     UnhandledResponse,
-#     WarningResponse,
-#     ErrorResponse,
-# )
+# websocket
+from .websocket import SpeakWebSocketClient, AsyncSpeakWebSocketClient
+from .websocket import (
+    SpeakWebSocketResponse,
+    OpenResponse,
+    MetadataResponse,
+    FlushedResponse,
+    CloseResponse,
+    UnhandledResponse,
+    WarningResponse,
+    ErrorResponse,
+)
diff --git a/deepgram/clients/speak/v1/websocket/__init__.py b/deepgram/clients/speak/v1/websocket/__init__.py
@@ -0,0 +1,17 @@
+# Copyright 2024 Deepgram SDK contributors. All Rights Reserved.
+# Use of this source code is governed by a MIT license that can be found in the LICENSE file.
+# SPDX-License-Identifier: MIT
+
+from .client import SpeakWebSocketClient
+from .async_client import AsyncSpeakWebSocketClient
+from .response import (
+    SpeakWebSocketResponse,
+    OpenResponse,
+    MetadataResponse,
+    FlushedResponse,
+    CloseResponse,
+    UnhandledResponse,
+    WarningResponse,
+    ErrorResponse,
+)
+from .options import SpeakWebSocketOptions