Fixing the Bidi client to accept None.
This commit is contained in:
@@ -63,8 +63,8 @@ class SageMakerBidiClient:
|
||||
self,
|
||||
endpoint_name: str,
|
||||
region: str,
|
||||
model_invocation_path: str = "",
|
||||
model_query_string: str = "",
|
||||
model_invocation_path: str | None = "",
|
||||
model_query_string: str | None = "",
|
||||
):
|
||||
"""Initialize the SageMaker BiDi client.
|
||||
|
||||
|
||||
@@ -201,8 +201,8 @@ class NvidiaSageMakerWebsocketSTTService(STTService):
|
||||
self._client = SageMakerBidiClient(
|
||||
endpoint_name=self._endpoint_name,
|
||||
region=self._region,
|
||||
model_query_string="",
|
||||
model_invocation_path="",
|
||||
model_query_string=None,
|
||||
model_invocation_path=None,
|
||||
)
|
||||
await self._client.start_session()
|
||||
await self._send_session_config()
|
||||
|
||||
@@ -352,8 +352,8 @@ class NvidiaSageMakerWebsocketTTSService(InterruptibleTTSService):
|
||||
self._client = SageMakerBidiClient(
|
||||
endpoint_name=self._endpoint_name,
|
||||
region=self._region,
|
||||
model_query_string="",
|
||||
model_invocation_path="",
|
||||
model_query_string=None,
|
||||
model_invocation_path=None,
|
||||
)
|
||||
await self._client.start_session()
|
||||
await self._send_session_config()
|
||||
|
||||
Reference in New Issue
Block a user