Skip to content

corrected lint #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Tushar-ml merged 1 commit into from
Jan 7, 2026
Merged

corrected lint #2

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 13 additions & 34 deletions livekit-plugins/livekit-plugins-simplismart/livekit/plugins/simplismart/stt.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -95,10 +95,9 @@ def __init__(
params: dict[str, Any] | SimplismartSTTOptions | None = None,
http_session: aiohttp.ClientSession | None = None,
):

assert (
base_url is not None or streaming_url is not None
), "base_url or streaming_url are required"
assert base_url is not None or streaming_url is not None, (
"base_url or streaming_url are required"
)
super().__init__(
capabilities=stt.STTCapabilities(
streaming=True if streaming_url is not None else False,
Expand Down Expand Up @@ -142,9 +141,7 @@ async def _recognize_impl(
language: NotGivenOr[str] = NOT_GIVEN,
conn_options: APIConnectOptions = DEFAULT_API_CONNECT_OPTIONS,
) -> stt.SpeechEvent:
language = (
self._opts.language if isinstance(language, type(NOT_GIVEN)) else language
)
language = self._opts.language if isinstance(language, type(NOT_GIVEN)) else language
wav_bytes = rtc.combine_audio_frames(buffer).to_wav_bytes()

audio_b64 = base64.b64encode(wav_bytes).decode("utf-8")
Expand All @@ -165,9 +162,7 @@ async def _recognize_impl(
) as res:
if res.status != 200:
error_text = await res.text()
self._logger.error(
f"Simplismart API error: {res.status} - {error_text}"
)
self._logger.error(f"Simplismart API error: {res.status} - {error_text}")
raise APIStatusError(
message=f"Simplismart API Error: {error_text}",
status_code=res.status,
Expand Down Expand Up @@ -261,9 +256,7 @@ def __init__(
http_session: aiohttp.ClientSession,
) -> None:
self._opts = opts
super().__init__(
stt=stt, conn_options=conn_options, sample_rate=self._SAMPLE_RATE
)
super().__init__(stt=stt, conn_options=conn_options, sample_rate=self._SAMPLE_RATE)
self._api_key = api_key
self._session = http_session
self._logger = logger.getChild(self.__class__.__name__)
Expand All @@ -290,9 +283,7 @@ def __init__(

async def aclose(self) -> None:
"""Close the stream and clean up resources."""
self._logger.debug(
"Starting stream cleanup", extra={"session_id": self._session_id}
)
self._logger.debug("Starting stream cleanup", extra={"session_id": self._session_id})

async with self._connection_lock:
self._connection_state = ConnectionState.DISCONNECTED
Expand All @@ -317,9 +308,7 @@ async def aclose(self) -> None:
try:
if self._ws and not self._ws.closed:
await self._ws.close()
self._logger.debug(
"WebSocket closed", extra={"session_id": self._session_id}
)
self._logger.debug("WebSocket closed", extra={"session_id": self._session_id})
except Exception as e:
self._logger.warning(
f"Error closing WebSocket: {e}", extra={"session_id": self._session_id}
Expand All @@ -339,9 +328,7 @@ async def aclose(self) -> None:
try:
if self._session and not self._session.closed:
await self._session.close()
self._logger.debug(
"HTTP session closed", extra={"session_id": self._session_id}
)
self._logger.debug("HTTP session closed", extra={"session_id": self._session_id})
except Exception as e:
self._logger.warning(
f"Error closing session: {e}", extra={"session_id": self._session_id}
Expand Down Expand Up @@ -419,9 +406,7 @@ async def _run_connection(self) -> None:
"""Run a single WebSocket connection attempt."""
# Check if session is still valid
if self._session.closed:
raise APIConnectionError(
"Session is closed, cannot establish WebSocket connection"
)
raise APIConnectionError("Session is closed, cannot establish WebSocket connection")

async with self._connection_lock:
self._connection_state = ConnectionState.CONNECTING
Expand Down Expand Up @@ -493,9 +478,7 @@ async def _run_connection(self) -> None:
if isinstance(exc, BaseException):
raise exc
else:
raise RuntimeError(
f"Task failed with non-BaseException: {exc}"
)
raise RuntimeError(f"Task failed with non-BaseException: {exc}")

finally:
# Clean up tasks
Expand Down Expand Up @@ -540,9 +523,7 @@ async def _process_audio(self, ws: aiohttp.ClientWebSocketResponse) -> None:
# Check if we have enough data for a chunk
while len(audio_buffer) >= chunk_size:
# Convert to Int16Array
chunk_data = np.array(
audio_buffer[:chunk_size], dtype=np.int16
)
chunk_data = np.array(audio_buffer[:chunk_size], dtype=np.int16)
await ws.send_bytes(chunk_data.tobytes())
chunks_sent += 1

Expand Down Expand Up @@ -631,9 +612,7 @@ async def _process_messages(self, ws: aiohttp.ClientWebSocketResponse) -> None:

elif msg.type == aiohttp.WSMsgType.ERROR:
error_msg = f"WebSocket error: {ws.exception()}"
self._logger.error(
error_msg, extra={"session_id": self._session_id}
)
self._logger.error(error_msg, extra={"session_id": self._session_id})
raise APIConnectionError(error_msg)

else:
Expand Down