mirror of
https://github.com/Laurent2916/nio-llm.git
synced 2024-11-09 23:12:05 +00:00
🐛 ignore messages comming from threads
This commit is contained in:
parent
f1b274f5a1
commit
012d6e0eaf
|
@ -69,6 +69,14 @@ class LLMClient(AsyncClient):
|
||||||
logger.debug("Ignoring edited message.")
|
logger.debug("Ignoring edited message.")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# ignore thread messages
|
||||||
|
if (
|
||||||
|
"m.relates_to" in event.source["content"]
|
||||||
|
and event.source["content"]["m.relates_to"]["rel_type"] == "m.thread"
|
||||||
|
):
|
||||||
|
logger.debug("Ignoring thread message.")
|
||||||
|
return
|
||||||
|
|
||||||
# update history
|
# update history
|
||||||
self.history.append(event)
|
self.history.append(event)
|
||||||
|
|
||||||
|
@ -95,6 +103,7 @@ class LLMClient(AsyncClient):
|
||||||
logger.debug(f"Prompt:\n{prompt}")
|
logger.debug(f"Prompt:\n{prompt}")
|
||||||
logger.debug(f"Tokens: {len(tokens)}")
|
logger.debug(f"Tokens: {len(tokens)}")
|
||||||
|
|
||||||
|
# ignore prompts that are too long
|
||||||
if len(tokens) > 512:
|
if len(tokens) > 512:
|
||||||
logger.debug("Prompt too long, skipping.")
|
logger.debug("Prompt too long, skipping.")
|
||||||
await self.room_send(
|
await self.room_send(
|
||||||
|
|
Loading…
Reference in a new issue