Редактировать other.py

This commit is contained in:
Markov Andrey
2026-06-30 21:41:19 +00:00
parent 04262a2f2e
commit 0a2d01ceae

View File

@@ -3,6 +3,7 @@
Прочие команды: !reset сброс истории диалога. Прочие команды: !reset сброс истории диалога.
Сбрасывает историю в зависимости от контекста: в личном чате личную историю, Сбрасывает историю в зависимости от контекста: в личном чате личную историю,
в групповом историю комнаты. в групповом историю комнаты.
Использует RAGClient.
""" """
import logging import logging
@@ -16,20 +17,17 @@ class ResetCommand(Command):
aliases = ["/reset", "!сброс", "/сброс"] aliases = ["/reset", "!сброс", "/сброс"]
async def execute(self, msg, args: str, user_jid: str, room_jid: str = None): async def execute(self, msg, args: str, user_jid: str, room_jid: str = None):
if room_jid: try:
await self.bot.db.clear_room_history(room_jid) result = await self.bot.rag_client.reset_history(user_jid, room_jid)
reply = msg.reply(f"🧹 История диалога комнаты {room_jid} сброшена.") if result.get('status') == 'ok':
reply = msg.reply("🧹 История сброшена.")
else:
reply = msg.reply(f"❌ Ошибка сброса: {result}")
except Exception as e:
logger.exception(f"Ошибка в !reset: {e}")
reply = msg.reply(f"❌ Ошибка: {e}")
if reply: if reply:
reply.send() # исправлено reply.send()
else:
logger.error("Не удалось отправить подтверждение сброса истории комнаты")
else:
await self.bot.db.clear_user_history(user_jid)
reply = msg.reply("🧹 Ваша личная история диалога сброшена.")
if reply:
reply.send() # исправлено
else:
logger.error("Не удалось отправить подтверждение сброса личной истории")
def get_help(self) -> str: def get_help(self) -> str:
return "Сбрасывает историю диалога (личную или комнатную)." return "Сбрасывает историю диалога (личную или комнатную)."