🐛 temporarily encapsulate message_callback's inside logic inside a try/catch

src/nio_llm/client.py

@@ -77,12 +77,12 @@ class LLMClient(AsyncClient):
 
     async def typing_loop(
         self,
-        sleep_time: int = 10,
+        sleep_time: int = 15,
     ) -> None:
         """Send typing indicators every `sleep_time` seconds.
 
         Args:
-            sleep_time (`int`, default `10`):
+            sleep_time (`int`, default `15`):
                 The time to sleep between sending typing indicators.
         """
         logging.debug("Started typing indicator.")
@@ -108,6 +108,7 @@ class LLMClient(AsyncClient):
             event (`RoomMessageText`):
                 The message event.
         """
+        try:
             logger.debug(f"New RoomMessageText: {event.source}")
 
             # ignore messages pre-dating our spawn time
@@ -175,7 +176,7 @@ class LLMClient(AsyncClient):
                     },
                     *[
                         {
-                        "content": f"{message.sender}: {message.body}",
+                            "content": f"<{message.sender}>: {message.body}",
                             "role": "assistant" if message.sender == self.uid else "user",
                         }
                         for message in self.history
@@ -187,19 +188,21 @@ class LLMClient(AsyncClient):
             logger.debug(f"Generated response: {response}")
 
             # retreive the response
-        output = response["choices"][0]["message"]["content"]  # type: ignore
-        output = output.strip().removeprefix(f"{self.uid}:").strip()
+            output = response["choices"][0]["message"]["content"].strip()  # type: ignore
 
-        # replace newlines with <br>
-        formatted_output = output.replace("\n", "<br>")
+            # strip the bot's uid from the response
+            output = output.removeprefix(f"<{self.uid}>:").strip()
 
             # detect mentions and replace them with html mentions
             formatted_output = re.sub(
-            r"@[^:]+:[^ :]+",
+                r"@[a-zA-Z-_]+:[^.]+\.[a-zA-Z]+",
                 lambda match: f'<a href="https://matrix.to/#/{match.group(0)}"></a>',
-            formatted_output,
+                output,
             )
 
+            # replace newlines with <br>
+            formatted_output = formatted_output.replace("\n", "<br>")
+
             logger.debug(f"Formatted response: {formatted_output}")
 
             # send the response
@@ -217,6 +220,8 @@ class LLMClient(AsyncClient):
 
             # stop typing indicator loop
             typing_task.cancel()
+        except Exception as e:
+            logger.error(f"Exception in message_callback: {e}")
 
     async def start(
         self,