init

assistant_bot/application.py

@@ -0,0 +1,112 @@
+import logging
+
+from telegram import Update
+from telegram.ext import (
+    Application,
+    CommandHandler,
+    InlineQueryHandler,
+    MessageHandler,
+    filters,
+)
+
+from .ai import AIClient
+from .config import (
+    get_assistant_mode,
+    get_bot_token,
+    get_db_path,
+    get_local_timezone,
+    get_ollama_base_url,
+    get_voice_max_duration_seconds,
+    get_whisper_compute_type,
+    get_whisper_device,
+    get_whisper_language,
+    get_whisper_model,
+    get_yandex_cloud_folder,
+    get_yandex_stt_language,
+    get_yandex_stt_model,
+)
+from .handlers import (
+    ask_command,
+    help_command,
+    history_command,
+    inline_query,
+    model_command,
+    models_command,
+    new_conversation_command,
+    private_text,
+    private_voice,
+    start,
+)
+from .jobs import post_init
+from .ollama import OllamaClient
+from .storage import AssistantStorage
+from .speech import SpeechRecognizer, SpeechTranscriber
+from .yandex_ai import YandexAIClient, YandexSpeechRecognizer
+
+
+def configure_logging() -> None:
+    logging.basicConfig(
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        level=logging.INFO,
+    )
+    logging.getLogger("httpx").setLevel(logging.WARNING)
+
+
+def create_application() -> Application:
+    storage = AssistantStorage(get_db_path())
+    mode = get_assistant_mode()
+    application = (
+        Application.builder()
+        .token(get_bot_token())
+        .post_init(post_init)
+        .build()
+    )
+    application.bot_data["storage"] = storage
+    ai_client: AIClient
+    speech_recognizer: SpeechTranscriber
+    if mode == "local":
+        ai_client = OllamaClient(get_ollama_base_url())
+        speech_recognizer = SpeechRecognizer(
+            model_name=get_whisper_model(),
+            device=get_whisper_device(),
+            compute_type=get_whisper_compute_type(),
+            language=get_whisper_language(),
+        )
+    else:
+        folder_id = get_yandex_cloud_folder()
+        ai_client = YandexAIClient(folder_id=folder_id)
+        speech_recognizer = YandexSpeechRecognizer(
+            folder_id=folder_id,
+            language=get_yandex_stt_language(),
+            model=get_yandex_stt_model(),
+        )
+
+    application.bot_data["ai_client"] = ai_client
+    application.bot_data["assistant_mode"] = mode
+    application.bot_data["timezone"] = get_local_timezone()
+    application.bot_data["speech_recognizer"] = speech_recognizer
+    application.bot_data["voice_max_duration"] = get_voice_max_duration_seconds()
+
+    application.add_handler(CommandHandler("start", start))
+    application.add_handler(CommandHandler("help", help_command))
+    application.add_handler(CommandHandler("ask", ask_command))
+    application.add_handler(CommandHandler("models", models_command))
+    application.add_handler(CommandHandler("model", model_command))
+    application.add_handler(CommandHandler("new", new_conversation_command))
+    application.add_handler(CommandHandler("history", history_command))
+    application.add_handler(InlineQueryHandler(inline_query))
+    application.add_handler(
+        MessageHandler(filters.VOICE & filters.ChatType.PRIVATE, private_voice)
+    )
+    application.add_handler(
+        MessageHandler(
+            filters.TEXT & filters.ChatType.PRIVATE & ~filters.COMMAND,
+            private_text,
+        )
+    )
+    return application
+
+
+def run() -> None:
+    configure_logging()
+    create_application().run_polling(allowed_updates=Update.ALL_TYPES)