UltraCoder/vk_chat_bot
1
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages Activity Actions

Улучшена логика активации диалога с ИИ.

Browse source
This commit is contained in:
Kirill Kirilenko 2026-01-20 18:28:49 +03:00
parent 1672d62e47
commit 3ca84006f9
2 changed files with 49 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tg/handlers/default.py
View file

@ -60,24 +60,28 @@ async def any_message_handler(message: Message):
ai_message = ai_agent.Message()
ai_fwd_messages: list[ai_agent.Message] = []
# Ответ на сообщение бота
if message.reply_to_message and message.reply_to_message.from_user.id == bot_user.id:
ai_message.text = message.text
else:
bot_username_mention = '@' + bot_user.username
if message.content_type == ContentType.TEXT and message.text.find(bot_username_mention) != -1:
# Сообщение содержит @bot_username
bot_username_mention = '@' + bot_user.username
if message.content_type == ContentType.TEXT and message.text.find(bot_username_mention) != -1:
ai_message.text = message.text.replace(bot_username_mention, bot_user.first_name)
ai_message.text = message.text.replace(bot_username_mention, bot_user.first_name)
if message.reply_to_message:
# Сообщение является ответом
if message.reply_to_message.content_type == ContentType.TEXT:
ai_fwd_messages = [
ai_agent.Message(user_name=await get_user_name_for_ai(message.reply_to_message.from_user),
text=message.reply_to_message.text)]
else:
await message.reply(MESSAGE_NOT_TEXT)
return
elif message.reply_to_message and message.reply_to_message.from_user.id == bot_user.id:
# Ответ на сообщение бота
if message.content_type == ContentType.TEXT:
ai_message.text = message.text
else:
await message.reply(MESSAGE_NOT_TEXT)
return
if message.reply_to_message and message.reply_to_message.content_type == ContentType.TEXT:
ai_fwd_messages = [
ai_agent.Message(user_name=await get_user_name_for_ai(message.reply_to_message.from_user),
text=message.reply_to_message.text)]
if ai_message.text is None:
await message.reply(MESSAGE_NOT_TEXT)
else:
return
ai_message.user_name = await get_user_name_for_ai(message.from_user)

51
vk/handlers/default.py
View file

@ -46,36 +46,45 @@ async def any_message_handler(message: Message):
ai_message = ai_agent.Message()
ai_fwd_messages: list[ai_agent.Message] = []
# Ответ на сообщение бота
if message.reply_message and message.reply_message.from_id == -bot_user.id:
ai_message.text = message.text
else:
bot_username_mention = '@' + bot_user.screen_name
pattern = r"\[club" + str(bot_user.id) + r"\|(.+)]"
bot_mentioned = False
if len(message.text) > 0 and message.text.find(bot_username_mention) != -1:
# Сообщение содержит @bot_username
bot_username_mention = '@' + bot_user.screen_name
if message.text is not None and message.text.find(bot_username_mention) != -1:
ai_message.text = message.text.replace(bot_username_mention, bot_user.name)
ai_message.text = message.text.replace(bot_username_mention, bot_user.name)
bot_mentioned = True
elif len(message.text) > 0 and re.search(pattern, message.text) is not None:
# Сообщение содержит [club<bot_id>|<some_name>]
ai_message.text = re.sub(pattern, r'\1', message.text)
bot_mentioned = True
elif message.reply_message and message.reply_message.from_id == -bot_user.id:
# Ответ на сообщение бота
if len(message.text) > 0:
ai_message.text = message.text
else:
# Сообщение содержит [club<bot_id>|<some_name>]
pattern = r"\[club" + str(bot_user.id) + r"\|(.+)]"
if message.text is not None and re.search(pattern, message.text) is not None:
ai_message.text = re.sub(pattern, r'\1', message.text)
else:
return
await message.reply(MESSAGE_NOT_TEXT)
return
else:
return
if message.reply_message and len(message.reply_message.text) > 0:
ai_fwd_messages.append(
ai_agent.Message(user_name=await get_user_name_for_ai(message.ctx_api, message.reply_message.from_id),
text=message.reply_message.text))
if bot_mentioned:
if message.reply_message:
if len(message.reply_message.text) > 0:
ai_fwd_messages.append(
ai_agent.Message(user_name=await get_user_name_for_ai(message.ctx_api, message.reply_message.from_id),
text=message.reply_message.text))
else:
await message.reply(MESSAGE_NOT_TEXT)
return
else:
for fwd_message in message.fwd_messages:
if len(fwd_message.text) > 0:
ai_fwd_messages.append(
ai_agent.Message(user_name=await get_user_name_for_ai(message.ctx_api, fwd_message.from_id),
text=fwd_message.text))
if len(ai_message.text) == 0:
await message.reply(MESSAGE_NOT_TEXT)
return
else:
await message.reply(MESSAGE_NOT_TEXT)
return
ai_message.user_name = await get_user_name_for_ai(message.ctx_api, message.from_id)
chat_prompt = chat['ai_prompt']