mirror of
https://gitflic.ru/project/maks1ms/ocab.git
synced 2025-04-06 17:43:47 +03:00
28 lines
940 B
Python
28 lines
940 B
Python
# flake8: noqa
|
|
import asyncio
|
|
|
|
from aiogram import Bot
|
|
from aiogram.types import Message
|
|
|
|
from ocab_modules.external.yandexgpt.yandexgpt import *
|
|
from ocab_modules.standard.config.config import (
|
|
get_yandexgpt_catalog_id,
|
|
get_yandexgpt_prompt,
|
|
get_yandexgpt_token,
|
|
)
|
|
from ocab_modules.standard.database.db_api import add_message
|
|
|
|
|
|
async def answer_to_message(message: Message, bot: Bot):
|
|
# print("answer_to_message")
|
|
await log("answer_to_message")
|
|
yagpt = YandexGPT(get_yandexgpt_token(), get_yandexgpt_catalog_id())
|
|
text = message.text
|
|
prompt = get_yandexgpt_prompt()
|
|
# response = await yagpt.async_yandexgpt(system_prompt=prompt, input_messages=text)
|
|
response = await yagpt.yandexgpt_request(
|
|
chat_id=message.chat.id, message_id=message.message_id, type="yandexgpt"
|
|
)
|
|
reply = await message.reply(response, parse_mode="Markdown")
|
|
add_message(reply, message_ai_model="yandexgpt")
|