mirror of
https://github.com/Laurent2916/nio-llm.git
synced 2024-11-09 23:12:05 +00:00
✨ modify matrix message content to format mentions and newlines
This commit is contained in:
parent
10c7513add
commit
d7a14fd4ee
|
@ -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}")
|
||||
|
|
Loading…
Reference in a new issue