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 asyncio
|
||||||
import logging
|
import logging
|
||||||
|
import re
|
||||||
import time
|
import time
|
||||||
from collections import deque
|
from collections import deque
|
||||||
|
|
||||||
|
@ -190,6 +191,18 @@ class LLMClient(AsyncClient):
|
||||||
output = response["choices"][0]["message"]["content"] # type: ignore
|
output = response["choices"][0]["message"]["content"] # type: ignore
|
||||||
output = output.strip().removeprefix(f"{self.uid}:").strip()
|
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
|
# send the response
|
||||||
await self.room_send(
|
await self.room_send(
|
||||||
room_id=self.room,
|
room_id=self.room,
|
||||||
|
@ -197,6 +210,8 @@ class LLMClient(AsyncClient):
|
||||||
content={
|
content={
|
||||||
"msgtype": "m.text",
|
"msgtype": "m.text",
|
||||||
"body": output,
|
"body": output,
|
||||||
|
"format": "org.matrix.custom.html",
|
||||||
|
"formatted_body": formatted_output,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
logger.debug(f"Sent response: {output}")
|
logger.debug(f"Sent response: {output}")
|
||||||
|
|
Loading…
Reference in a new issue