Laurent/nio-llm
1
0
Fork 0
mirror of https://github.com/Laurent2916/nio-llm.git synced 2024-09-19 19:45:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 parametrize typing_loop with typing_loop

Browse source
This commit is contained in:
Laureηt 2023-10-19 16:13:05 +00:00
parent 0f312a0a70
commit 5b5a18d73b
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/nio_llm/client.py
View file

@ -74,14 +74,22 @@ class LLMClient(AsyncClient):
# add callbacks
self.add_event_callback(self.message_callback, RoomMessageText) # type: ignore
async def typing_loop(self) -> None:
"""Send typing indicators every 5 seconds."""
async def typing_loop(
self,
sleep_time: int = 10,
) -> None:
"""Send typing indicators every `sleep_time` seconds.
Args:
sleep_time (`int`, default `10`):
The time to sleep between sending typing indicators.
"""
logging.debug("Started typing indicator.")
try:
while True:
logging.debug("Sending typing indicator.")
await self.room_typing(self.room, True)
await asyncio.sleep(5)
await asyncio.sleep(sleep_time)
except asyncio.CancelledError:
await self.room_typing(self.room, False)
logging.debug("Stopped typing indicator.")