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-24 02:26:08 +03:00
parent 555b4bc0b5
commit 66fdbc3360
2 changed files with 23 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tg/handlers/private.py
View file

@ -1,12 +1,13 @@
from functools import partial
from aiogram import Router, F
from aiogram.enums import ChatType
from aiogram.enums import ChatType, ContentType
from aiogram.filters import Command, CommandObject, CommandStart
from aiogram.types import Message
import ai_agent
import utils
from messages import *
import tg.tg_database as database
from .default import ACCEPTED_CONTENT_TYPES
@ -43,11 +44,16 @@ async def reset_context_handler(message: Message):
@router.message(F.content_type.in_(ACCEPTED_CONTENT_TYPES), F.chat.type == ChatType.PRIVATE)
async def any_message_handler(message: Message):
chat_id = message.chat.id
chat = database.DB.create_chat_if_not_exists(chat_id)
chat_prompt = chat['ai_prompt']
await message.answer(
await utils.run_with_progress(
partial(ai_agent.agent.get_private_chat_reply, chat_id, chat_prompt, message.text),
partial(message.bot.send_chat_action, chat_id, 'typing'),
interval=4))
if message.content_type != ContentType.TEXT:
await message.answer(MESSAGE_NOT_TEXT)
return
answer, success = await utils.run_with_progress(
partial(ai_agent.agent.get_private_chat_reply, chat_id, message.text, message.message_id),
partial(message.bot.send_chat_action, chat_id, 'typing'),
interval=4)
answer_id = (await message.answer(answer)).message_id
if success:
ai_agent.agent.set_last_response_id(chat_id, answer_id)

16
vk/handlers/private.py
View file

@ -6,6 +6,7 @@ from vkbottle.framework.labeler import BotLabeler
import ai_agent
import utils
from messages import *
import vk.vk_database as database
@ -41,15 +42,16 @@ async def reset_context_handler(message: Message):
@labeler.private_message()
async def any_message_handler(message: Message):
chat_id = message.peer_id
chat = database.DB.create_chat_if_not_exists(chat_id)
if len(message.text) == 0:
await message.answer(MESSAGE_NOT_TEXT)
return
chat_prompt = chat['ai_prompt']
answer, success = await utils.run_with_progress(
partial(ai_agent.agent.get_private_chat_reply, chat_id, message.text, message.message_id),
partial(message.ctx_api.messages.set_activity, peer_id=chat_id, type='typing'),
interval=4)
await message.answer(
await utils.run_with_progress(
partial(ai_agent.agent.get_private_chat_reply, chat_id, chat_prompt, message.text),
partial(message.ctx_api.messages.set_activity, peer_id=chat_id, type='typing'),
interval=4))
answer_id = (await message.answer(answer)).message_id
if success:
ai_agent.agent.set_last_response_id(chat_id, answer_id)