diff --git a/src/moai/bot/main.py b/src/moai/bot/main.py index 603447b..6c771fe 100644 --- a/src/moai/bot/main.py +++ b/src/moai/bot/main.py @@ -13,6 +13,7 @@ from telegram.ext import Application, ApplicationBuilder from moai.bot.config import BotConfig from moai.bot.handlers import register_handlers +from moai.core.ai_client import init_ai_client from moai.core.database import close_db, create_tables, init_db # Module-level config reference for post_init callback @@ -34,6 +35,9 @@ async def post_init(application: Application) -> None: await create_tables() logger.info("Database initialized") + init_ai_client(_config) + logger.info("AI client initialized with %s", _config.ai_router) + async def post_shutdown(application: Application) -> None: """Clean up database on bot shutdown.