smart-bot-factory 0.2.8__tar.gz → 0.2.10__tar.gz

{smart_bot_factory-0.2.8 → smart_bot_factory-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smart-bot-factory
-Version: 0.2.8
+Version: 0.2.10
 Summary: Библиотека для создания умных чат-ботов
 Author-email: Kopatych <kopatych@example.com>
 License: MIT

{smart_bot_factory-0.2.8 → smart_bot_factory-0.2.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "smart-bot-factory"
-version = "0.2.8"
+version = "0.2.10"
 description = "Библиотека для создания умных чат-ботов"
 authors = [
     {name = "Kopatych", email = "kopatych@example.com"}

{smart_bot_factory-0.2.8 → smart_bot_factory-0.2.10}/smart_bot_factory/core/bot_utils.py

@@ -177,6 +177,17 @@ def parse_ai_response_method2(ai_response: str) -> tuple[str, dict]:
 
 async def process_events(session_id: str, events: list, user_id: int):
     """Обрабатывает события из ответа ИИ"""
+    
+    # Проверяем кастомный процессор
+    custom_processor = get_global_var('custom_event_processor')
+    
+    if custom_processor:
+        # Используем кастомную функцию обработки событий
+        logger.info(f"🔄 Используется кастомная обработка событий: {custom_processor.__name__}")
+        await custom_processor(session_id, events, user_id)
+        return
+    
+    # Стандартная обработка
     supabase_client = get_global_var('supabase_client')
     
     for event in events:

{smart_bot_factory-0.2.8 → smart_bot_factory-0.2.10}/smart_bot_factory/creation/bot_builder.py

@@ -58,6 +58,9 @@ class BotBuilder:
         # Кастомный PromptLoader
         self._custom_prompt_loader = None
         
+        # Кастомный процессор событий
+        self._custom_event_processor = None
+        
         # Флаги инициализации
         self._initialized = False
         
@@ -402,6 +405,44 @@ class BotBuilder:
         self._custom_prompt_loader = prompt_loader
         logger.info(f"✅ Установлен кастомный PromptLoader: {type(prompt_loader).__name__}")
     
+    def set_event_processor(self, custom_processor):
+        """
+        Устанавливает кастомную функцию для обработки событий
+        
+        Полностью заменяет стандартную process_events из bot_utils
+        
+        Args:
+            custom_processor: async def(session_id: str, events: list, user_id: int)
+            
+        Example:
+            from smart_bot_factory.message import get_bot
+            from smart_bot_factory.core.decorators import execute_event_handler
+            
+            async def my_process_events(session_id, events, user_id):
+                '''Моя кастомная обработка событий'''
+                bot = get_bot()
+                
+                for event in events:
+                    event_type = event.get('тип')
+                    event_info = event.get('инфо')
+                    
+                    if event_type == 'запись':
+                        # Кастомная логика для бронирования
+                        telegram_user = await bot.get_chat(user_id)
+                        name = telegram_user.first_name or 'Клиент'
+                        # ... ваша обработка
+                    else:
+                        # Для остальных - стандартная обработка
+                        await execute_event_handler(event_type, user_id, event_info)
+            
+            bot_builder.set_event_processor(my_process_events)
+        """
+        if not callable(custom_processor):
+            raise TypeError(f"Процессор должен быть callable, получен {type(custom_processor)}")
+        
+        self._custom_event_processor = custom_processor
+        logger.info(f"✅ Установлена кастомная функция обработки событий: {custom_processor.__name__}")
+    
     # ========== ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ ==========
     
     def validate_message(self, handler):
@@ -580,6 +621,7 @@ class BotBuilder:
                 handlers_module.conversation_manager = self.conversation_manager
                 handlers_module.start_handlers = self._start_handlers  # Передаем обработчики on_start
                 handlers_module.message_hooks = self.get_message_hooks()  # Передаем хуки для обработки сообщений
+                handlers_module.custom_event_processor = self._custom_event_processor  # Передаем кастомный процессор событий
                 logger.info("✅ Глобальные переменные установлены в handlers")
             except Exception as e:
                 logger.warning(f"⚠️ Не удалось установить глобальные переменные в handlers: {e}")
@@ -611,6 +653,7 @@ class BotBuilder:
                 bot_utils_module.admin_manager = self.admin_manager
                 bot_utils_module.analytics_manager = self.analytics_manager
                 bot_utils_module.conversation_manager = self.conversation_manager
+                bot_utils_module.custom_event_processor = self._custom_event_processor  # Передаем кастомный процессор событий
                 logger.info("✅ Глобальные переменные установлены в bot_utils")
             except Exception as e:
                 logger.warning(f"⚠️ Не удалось установить глобальные переменные в bot_utils: {e}")

smart_bot_factory-0.2.10/smart_bot_factory/message/__init__.py

@@ -0,0 +1,60 @@
+"""
+Message модули smart_bot_factory
+"""
+
+
+from ..core.message_sender import (
+    send_message_by_human, 
+    send_message_by_ai,
+    send_message_to_users_by_stage,
+    send_message,
+)
+
+
+def get_bot():
+    """
+    Получает aiogram Bot из глобального контекста
+    
+    Доступен после вызова bot_builder.start()
+    
+    Returns:
+        Bot: aiogram Bot объект
+        
+    Raises:
+        RuntimeError: Если bot еще не инициализирован
+        
+    Example:
+        from smart_bot_factory.message import get_bot
+        
+        @event_router.event_handler("booking")
+        async def handle_booking(user_id: int, event_data: str):
+            bot = get_bot()
+            
+            # Получаем информацию о пользователе из Telegram
+            telegram_user = await bot.get_chat(user_id)
+            name = telegram_user.first_name or 'Клиент'
+            
+            # Используем любые методы aiogram Bot
+            await bot.send_message(user_id, f"Привет, {name}!")
+            await bot.send_photo(user_id, photo=...)
+    """
+    from ..handlers.handlers import get_global_var
+    
+    bot = get_global_var('bot')
+    if not bot:
+        raise RuntimeError(
+            "Bot еще не инициализирован. "
+            "Убедитесь что bot_builder.start() уже вызван. "
+            "Функция get_bot() доступна только внутри обработчиков событий, "
+            "которые выполняются во время работы бота."
+        )
+    return bot
+
+
+__all__ = [
+    'send_message_by_human',
+    'send_message_by_ai',
+    'send_message_to_users_by_stage',
+    'send_message',  # Чистая отправка с файлами и кнопками
+    'get_bot',  # Доступ к aiogram Bot
+]

{smart_bot_factory-0.2.8 → smart_bot_factory-0.2.10}/uv.lock

@@ -1637,7 +1637,7 @@ wheels = [
 
 [[package]]
 name = "smart-bot-factory"
-version = "0.2.7"
+version = "0.2.9"
 source = { editable = "." }
 dependencies = [
     { name = "aiofiles" },

smart_bot_factory-0.2.8/smart_bot_factory/message/__init__.py

@@ -1,18 +0,0 @@
-"""
-Message модули smart_bot_factory
-"""
-
-
-from ..core.message_sender import (
-    send_message_by_human, 
-    send_message_by_ai,
-    send_message_to_users_by_stage,
-    send_message,
-)
-
-__all__ = [
-    'send_message_by_human',
-    'send_message_by_ai',
-    'send_message_to_users_by_stage',
-    'send_message',  # Чистая отправка с файлами и кнопками
-]