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

View File

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

View File

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