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