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

✨ modify matrix message content to format mentions and newlines

Browse source
This commit is contained in:
Laureηt 2023-10-19 17:27:27 +00:00
parent 10c7513add
commit d7a14fd4ee
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/nio_llm/client.py
View file

@ -1,5 +1,6 @@
import asyncio
import logging
import re
import time
from collections import deque
@ -190,6 +191,18 @@ class LLMClient(AsyncClient):
output = response["choices"][0]["message"]["content"] # type: ignore
output = output.strip().removeprefix(f"{self.uid}:").strip()
# replace newlines with <br>
formatted_output = output.replace("\n", "<br>")
# detect mentions and replace them with html mentions
formatted_output = re.sub(
r"@[^:]+:[^ :]+",
lambda match: f'<a href="https://matrix.to/#/{match.group(0)}"></a>',
formatted_output,
)
logger.debug(f"Formatted response: {formatted_output}")
# send the response
await self.room_send(
room_id=self.room,
@ -197,6 +210,8 @@ class LLMClient(AsyncClient):
content={
"msgtype": "m.text",
"body": output,
"format": "org.matrix.custom.html",
"formatted_body": formatted_output,
},
)
logger.debug(f"Sent response: {output}")