Fixing the Bidi client to accept None.

This commit is contained in:
filipi87
2026-05-12 12:19:30 -03:00
parent a34864d643
commit 6a27ed35b1
3 changed files with 6 additions and 6 deletions

View File

@@ -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.

View File

@@ -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()

View File

@@ -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()