smart-bot-factory 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

smart_bot_factory/creation/bot_builder.py

@@ -46,6 +46,14 @@ class BotBuilder:
         self.prompt_loader: Optional[PromptLoader] = None
         self.router_manager: Optional[RouterManager] = None
         self._telegram_routers: List = []  # Список Telegram роутеров
+        self._start_handlers: List = []  # Список обработчиков on_start
+        
+        # Хуки для кастомизации process_user_message
+        self._message_validators: List = []  # Валидация ДО обработки
+        self._prompt_enrichers: List = []    # Обогащение системного промпта
+        self._context_enrichers: List = []   # Обогащение контекста для AI
+        self._response_processors: List = [] # Обработка ответа AI
+        self._send_filters: List = []        # Фильтры перед отправкой пользователю
         
         # Флаги инициализации
         self._initialized = False
@@ -326,6 +334,165 @@ class BotBuilder:
         
         logger.info(f"✅ Зарегистрировано {len(telegram_routers)} Telegram роутеров")
     
+    def on_start(self, handler):
+        """
+        Регистрирует обработчик, который вызывается после стандартной логики /start
+        
+        Обработчик получает доступ к:
+        - user_id: int - ID пользователя Telegram
+        - session_id: str - ID созданной сессии
+        - message: Message - Объект сообщения от aiogram
+        - state: FSMContext - Контекст состояния
+        
+        Args:
+            handler: Async функция с сигнатурой:
+                     async def handler(user_id: int, session_id: str, message: Message, state: FSMContext)
+        
+        Example:
+            @bot_builder.on_start
+            async def my_start_handler(user_id, session_id, message, state):
+                keyboard = InlineKeyboardMarkup(...)
+                await message.answer("Выберите действие:", reply_markup=keyboard)
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._start_handlers.append(handler)
+        logger.info(f"✅ Зарегистрирован обработчик on_start: {handler.__name__}")
+        return handler  # Возвращаем handler для использования как декоратор
+    
+    def get_start_handlers(self) -> List:
+        """Получает список обработчиков on_start"""
+        return self._start_handlers.copy()
+    
+    # ========== ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ ==========
+    
+    def validate_message(self, handler):
+        """
+        Регистрирует валидатор сообщений (вызывается ДО обработки AI)
+        
+        Если валидатор возвращает False, обработка прерывается
+        
+        Args:
+            handler: async def(message: Message, supabase_client) -> bool
+        
+        Example:
+            @bot_builder.validate_message
+            async def check_service_names(message, supabase_client):
+                if "неправильное название" in message.text:
+                    await message.answer("Пожалуйста, уточните название услуги")
+                    return False  # Прерываем обработку
+                return True  # Продолжаем
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._message_validators.append(handler)
+        logger.info(f"✅ Зарегистрирован валидатор сообщений: {handler.__name__}")
+        return handler
+    
+    def enrich_prompt(self, handler):
+        """
+        Регистрирует обогатитель системного промпта
+        
+        Args:
+            handler: async def(system_prompt: str, user_id: int, session_id: str, supabase_client) -> str
+        
+        Example:
+            @bot_builder.enrich_prompt
+            async def add_client_info(system_prompt, user_id, session_id, supabase_client):
+                session = await supabase_client.get_active_session(user_id)
+                phone = session.get('metadata', {}).get('phone')
+                if phone:
+                    return f"{system_prompt}\\n\\nТелефон клиента: {phone}"
+                return system_prompt
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._prompt_enrichers.append(handler)
+        logger.info(f"✅ Зарегистрирован обогатитель промпта: {handler.__name__}")
+        return handler
+    
+    def enrich_context(self, handler):
+        """
+        Регистрирует обогатитель контекста для AI (messages array)
+        
+        Args:
+            handler: async def(messages: List[dict], user_id: int, session_id: str) -> List[dict]
+        
+        Example:
+            @bot_builder.enrich_context
+            async def add_external_data(messages, user_id, session_id):
+                # Добавляем данные из внешнего API
+                messages.append({
+                    "role": "system", 
+                    "content": "Дополнительная информация..."
+                })
+                return messages
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._context_enrichers.append(handler)
+        logger.info(f"✅ Зарегистрирован обогатитель контекста: {handler.__name__}")
+        return handler
+    
+    def process_response(self, handler):
+        """
+        Регистрирует обработчик ответа AI (ПОСЛЕ получения ответа)
+        
+        Args:
+            handler: async def(response_text: str, ai_metadata: dict, user_id: int) -> tuple[str, dict]
+        
+        Example:
+            @bot_builder.process_response
+            async def modify_response(response_text, ai_metadata, user_id):
+                # Модифицируем ответ
+                if "цена" in response_text.lower():
+                    response_text += "\\n\\n💰 Актуальные цены на сайте"
+                return response_text, ai_metadata
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._response_processors.append(handler)
+        logger.info(f"✅ Зарегистрирован обработчик ответа: {handler.__name__}")
+        return handler
+    
+    def filter_send(self, handler):
+        """
+        Регистрирует фильтр отправки (может блокировать отправку пользователю)
+        
+        Если фильтр возвращает False, сообщение НЕ отправляется
+        
+        Args:
+            handler: async def(user_id: int, response_text: str) -> bool
+        
+        Example:
+            @bot_builder.filter_send
+            async def block_during_process(user_id, response_text):
+                if is_processing(user_id):
+                    return False  # Не отправляем
+                return True  # Отправляем
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._send_filters.append(handler)
+        logger.info(f"✅ Зарегистрирован фильтр отправки: {handler.__name__}")
+        return handler
+    
+    def get_message_hooks(self) -> Dict[str, List]:
+        """Получает все хуки для обработки сообщений"""
+        return {
+            'validators': self._message_validators.copy(),
+            'prompt_enrichers': self._prompt_enrichers.copy(),
+            'context_enrichers': self._context_enrichers.copy(),
+            'response_processors': self._response_processors.copy(),
+            'send_filters': self._send_filters.copy()
+        }
+    
     def get_router_manager(self) -> RouterManager:
         """Получает менеджер роутеров событий"""
         return self.router_manager
@@ -374,6 +541,8 @@ class BotBuilder:
                 handlers_module.admin_manager = self.admin_manager
                 handlers_module.analytics_manager = self.analytics_manager
                 handlers_module.conversation_manager = self.conversation_manager
+                handlers_module.start_handlers = self._start_handlers  # Передаем обработчики on_start
+                handlers_module.message_hooks = self.get_message_hooks()  # Передаем хуки для обработки сообщений
                 logger.info("✅ Глобальные переменные установлены в handlers")
             except Exception as e:
                 logger.warning(f"⚠️ Не удалось установить глобальные переменные в handlers: {e}")

smart_bot_factory/handlers/handlers.py

@@ -277,6 +277,23 @@ async def user_start_handler(message: Message, state: FSMContext):
         
         logging.info(f"✅ Приветственное сообщение успешно сохранено в БД для сессии {session_id}")
         
+        # ВЫЗЫВАЕМ ПОЛЬЗОВАТЕЛЬСКИЕ ОБРАБОТЧИКИ on_start
+        start_handlers = get_global_var('start_handlers')
+        if start_handlers:
+            logger.info(f"🔔 Вызов {len(start_handlers)} пользовательских обработчиков on_start")
+            for handler in start_handlers:
+                try:
+                    await handler(
+                        user_id=message.from_user.id,
+                        session_id=session_id,
+                        message=message,
+                        state=state
+                    )
+                    logger.info(f"✅ Обработчик on_start '{handler.__name__}' выполнен успешно")
+                except Exception as handler_error:
+                    logger.error(f"❌ Ошибка в обработчике on_start '{handler.__name__}': {handler_error}")
+                    # Продолжаем выполнение остальных обработчиков
+        
     except Exception as e:
         logger.error(f"Ошибка при обработке user /start: {e}")
         await send_message(message, "Произошла ошибка при инициализации. Попробуйте позже.")
@@ -475,10 +492,22 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
     config = get_global_var('config')
     bot = get_global_var('bot')
     prompt_loader = get_global_var('prompt_loader')
+    message_hooks = get_global_var('message_hooks') or {}
     from datetime import datetime
     import pytz  # Добавляем импорт для работы с временными зонами
     
     try:
+        # ============ ХУК 1: ВАЛИДАЦИЯ СООБЩЕНИЯ ============
+        validators = message_hooks.get('validators', [])
+        for validator in validators:
+            try:
+                should_continue = await validator(message, supabase_client)
+                if not should_continue:
+                    logger.info(f"⛔ Валидатор '{validator.__name__}' прервал обработку")
+                    return  # Прерываем обработку
+            except Exception as e:
+                logger.error(f"❌ Ошибка в валидаторе '{validator.__name__}': {e}")
+        
         # Сохраняем сообщение пользователя
         await supabase_client.add_message(
             session_id=session_id,
@@ -497,13 +526,27 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
         current_time = datetime.now(moscow_tz)
         time_info = current_time.strftime('%H:%M, %d.%m.%Y, %A')
         
-        # Модифицируем системный промпт, добавляя время
+        # Базовый системный промпт с временем
         system_prompt_with_time = f"""
 {system_prompt}
 
 ТЕКУЩЕЕ ВРЕМЯ: {time_info} (московское время)
 """
         
+        # ============ ХУК 2: ОБОГАЩЕНИЕ ПРОМПТА ============
+        prompt_enrichers = message_hooks.get('prompt_enrichers', [])
+        for enricher in prompt_enrichers:
+            try:
+                system_prompt_with_time = await enricher(
+                    system_prompt_with_time,
+                    message.from_user.id,
+                    session_id,
+                    supabase_client
+                )
+                logger.info(f"✅ Промпт обогащен '{enricher.__name__}'")
+            except Exception as e:
+                logger.error(f"❌ Ошибка в обогатителе промпта '{enricher.__name__}': {e}")
+        
         # Формируем контекст для OpenAI с обновленным системным промптом
         messages = [{"role": "system", "content": system_prompt_with_time}]
         
@@ -519,6 +562,19 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
             messages.append({"role": "system", "content": final_instructions})
             logger.info(f"🎯 Добавлены финальные инструкции ({len(final_instructions)} символов)")
         
+        # ============ ХУК 3: ОБОГАЩЕНИЕ КОНТЕКСТА ============
+        context_enrichers = message_hooks.get('context_enrichers', [])
+        for enricher in context_enrichers:
+            try:
+                messages = await enricher(
+                    messages,
+                    message.from_user.id,
+                    session_id
+                )
+                logger.info(f"✅ Контекст обогащен '{enricher.__name__}'")
+            except Exception as e:
+                logger.error(f"❌ Ошибка в обогатителе контекста '{enricher.__name__}': {e}")
+        
         logger.info(f"📝 Контекст сформирован: {len(messages)} сообщений (включая время: {time_info})")
         
         await bot.send_chat_action(message.chat.id, "typing")
@@ -571,82 +627,95 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
                 response_text = ai_response
 
             logger.info(f"✅ Финальный текст для отправки: {len(response_text)} символов")
-
-            # Обновляем этап сессии и качество лида
-            if ai_metadata:
-                logger.info("🔍 Анализ метаданных от ИИ:")
-                
-                # Вывод информации об этапе
-                stage = ai_metadata.get('этап')
-                if stage:
-                    logger.info(f"   📈 Этап диалога: {stage}")
-                
-                # Вывод информации о качестве лида
-                quality = ai_metadata.get('качество')
-                if quality is not None:
-                    quality_emoji = "⭐" * min(quality, 5)  # Максимум 5 звезд
-                    logger.info(f"   {quality_emoji} Качество лида: {quality}/10")
-                
-                # Обновляем в базе данных
-                if stage or quality is not None:
-                    await supabase_client.update_session_stage(session_id, stage, quality)
-                    logger.info(f"   ✅ Этап и качество обновлены в БД")
-                
-                # Обрабатываем события
-                events = ai_metadata.get('события', [])
-                if events:
-                    logger.info(f"\n🔔 События в диалоге ({len(events)}):")
-                    for idx, event in enumerate(events, 1):
-                        event_type = event.get('тип', 'неизвестно')
-                        event_info = event.get('инфо', 'нет информации')
-                        
-                        # Подбираем эмодзи для разных типов событий
-                        event_emoji = {
-                            'телефон': '📱',
-                            'email': '📧',
-                            'встреча': '📅',
-                            'заказ': '🛍️',
-                            'вопрос': '❓',
-                            'консультация': '💬',
-                            'жалоба': '⚠️',
-                            'отзыв': '💭'
-                        }.get(event_type.lower(), '📌')
-                        
-                        logger.info(f"   {idx}. {event_emoji} {event_type}: {event_info}")
-                    
-                    # Обрабатываем события в системе
-                    await process_events(session_id, events, message.from_user.id)
-                    logger.info("   ✅ События обработаны")
+        
+        # ============ ХУК 4: ОБРАБОТКА ОТВЕТА ============
+        response_processors = message_hooks.get('response_processors', [])
+        for processor in response_processors:
+            try:
+                response_text, ai_metadata = await processor(
+                    response_text,
+                    ai_metadata,
+                    message.from_user.id
+                )
+                logger.info(f"✅ Ответ обработан '{processor.__name__}'")
+            except Exception as e:
+                logger.error(f"❌ Ошибка в обработчике ответа '{processor.__name__}': {e}")
+        
+        # Обновляем этап сессии и качество лида
+        if ai_metadata:
+            logger.info("🔍 Анализ метаданных от ИИ:")
             
-            # Обрабатываем файлы и каталоги
-            files_list = ai_metadata.get('файлы', [])
-            directories_list = ai_metadata.get('каталоги', [])
+            # Вывод информации об этапе
+            stage = ai_metadata.get('этап')
+            if stage:
+                logger.info(f"   📈 Этап диалога: {stage}")
             
-            # Форматируем информацию о файлах
+            # Вывод информации о качестве лида
+            quality = ai_metadata.get('качество')
+            if quality is not None:
+                quality_emoji = "⭐" * min(quality, 5)  # Максимум 5 звезд
+                logger.info(f"   {quality_emoji} Качество лида: {quality}/10")
+            
+            # Обновляем в базе данных
+            if stage or quality is not None:
+                await supabase_client.update_session_stage(session_id, stage, quality)
+                logger.info(f"   ✅ Этап и качество обновлены в БД")
+            
+            # Обрабатываем события
+            events = ai_metadata.get('события', [])
+            if events:
+                logger.info(f"\n🔔 События в диалоге ({len(events)}):")
+                for idx, event in enumerate(events, 1):
+                    event_type = event.get('тип', 'неизвестно')
+                    event_info = event.get('инфо', 'нет информации')
+
+                    # Подбираем эмодзи для разных типов событий
+                    event_emoji = {
+                        'телефон': '📱',
+                        'email': '📧',
+                        'встреча': '📅',
+                        'заказ': '🛍️',
+                        'вопрос': '❓',
+                        'консультация': '💬',
+                        'жалоба': '⚠️',
+                        'отзыв': '💭'
+                    }.get(event_type.lower(), '📌')
+
+                    logger.info(f"   {idx}. {event_emoji} {event_type}: {event_info}")
+
+                # Обрабатываем события в системе
+                await process_events(session_id, events, message.from_user.id)
+                logger.info("   ✅ События обработаны")
+        
+        # Обрабатываем файлы и каталоги
+        files_list = ai_metadata.get('файлы', [])
+        directories_list = ai_metadata.get('каталоги', [])
+        
+        # Форматируем информацию о файлах
+        if files_list:
+            logger.info("📎 Найденные файлы:")
+            for idx, file in enumerate(files_list, 1):
+                logger.info(f"   {idx}. 📄 {file}")
+        
+        # Форматируем информацию о каталогах
+        if directories_list:
+            logger.info("📂 Найденные каталоги:")
+            for idx, directory in enumerate(directories_list, 1):
+                logger.info(f"   {idx}. 📁 {directory}")
+        
+        # Добавляем информацию в текст ответа
+        if files_list or directories_list:
+            files_info = []
             if files_list:
-                logger.info("📎 Найденные файлы:")
-                for idx, file in enumerate(files_list, 1):
-                    logger.info(f"   {idx}. 📄 {file}")
+                files_str = "\n".join(f"• {file}" for file in files_list)
+                files_info.append(f"\n\n📎 Доступные файлы:\n{files_str}")
             
-            # Форматируем информацию о каталогах
             if directories_list:
-                logger.info("📂 Найденные каталоги:")
-                for idx, directory in enumerate(directories_list, 1):
-                    logger.info(f"   {idx}. 📁 {directory}")
+                dirs_str = "\n".join(f"• {directory}" for directory in directories_list)
+                files_info.append(f"\n\n📂 Доступные каталоги:\n{dirs_str}")
             
-            # Добавляем информацию в текст ответа
-            if files_list or directories_list:
-                files_info = []
-                if files_list:
-                    files_str = "\n".join(f"• {file}" for file in files_list)
-                    files_info.append(f"\n\n📎 Доступные файлы:\n{files_str}")
-                
-                if directories_list:
-                    dirs_str = "\n".join(f"• {directory}" for directory in directories_list)
-                    files_info.append(f"\n\n📂 Доступные каталоги:\n{dirs_str}")
-                
-            else:
-                logger.info("📎 Файлы и каталоги не указаны")
+        else:
+            logger.info("📎 Файлы и каталоги не указаны")
 
         # Сохраняем ответ ассистента с метаданными
         try:
@@ -680,6 +749,17 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
 
         logger.info(f"📱 Отправляем пользователю: {len(final_response)} символов")
         
+        # ============ ХУК 5: ФИЛЬТРЫ ОТПРАВКИ ============
+        send_filters = message_hooks.get('send_filters', [])
+        for filter_func in send_filters:
+            try:
+                should_send = await filter_func(message.from_user.id, final_response)
+                if not should_send:
+                    logger.info(f"⛔ Фильтр '{filter_func.__name__}' заблокировал отправку")
+                    return  # Не отправляем
+            except Exception as e:
+                logger.error(f"❌ Ошибка в фильтре отправки '{filter_func.__name__}': {e}")
+        
         # Отправляем ответ пользователю
         try:
             await send_message(message, final_response, files_list=files_list, directories_list=directories_list)

{smart_bot_factory-0.2.0.dist-info → smart_bot_factory-0.2.2.dist-info}/METADATA

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

{smart_bot_factory-0.2.0.dist-info → smart_bot_factory-0.2.2.dist-info}/RECORD

@@ -33,12 +33,12 @@ smart_bot_factory/core/router_manager.py,sha256=dUwesog-oHk1U2EDdS8p0e4MTSkwtx5_
 smart_bot_factory/core/states.py,sha256=AOc19_yAsDW_8md-2oiowjBhuW3bwcsmMxszCXWZZTQ,355
 smart_bot_factory/core/telegram_router.py,sha256=LcPLOd87Y4Y4YN6TBXVAtmpSp8gAK2otgMI4YS5f5tk,2091
 smart_bot_factory/creation/__init__.py,sha256=IgDk8GDS3pg7Pw_Et41J33ZmeZIU5dRwQdTmYKXfJfE,128
-smart_bot_factory/creation/bot_builder.py,sha256=xWgWRe4DV3PngiLoQlzm6AdhNvPqn3wOCwcgsIoh-QA,25220
+smart_bot_factory/creation/bot_builder.py,sha256=VjRPlar6xgul7QpG57od--uS-juWrlW9-YYNnFN_Yx4,33828
 smart_bot_factory/creation/bot_testing.py,sha256=JDWXyJfZmbgo-DLdAPk8Sd9FiehtHHa4sLD17lBrTOc,55669
 smart_bot_factory/database/database_structure.sql,sha256=26gFtMC2jdQGQF7Zb_F4Br56rMd4hUDTk9FkNZYneLo,2789
 smart_bot_factory/database/schema.sql,sha256=-6kOmA9QnSkUtmGI2iQRbTvbdiqOhEOQcuz1lJn79mU,28059
 smart_bot_factory/event/__init__.py,sha256=hPL449RULIOB-OXv1ZbGNiHctAYaOMUqhSWGPrDHYBM,212
-smart_bot_factory/handlers/handlers.py,sha256=YH8xG0tDcb7uxZXI4socXURzc-y57_FEQ6GqTgYcM5Q,37603
+smart_bot_factory/handlers/handlers.py,sha256=kfzveBtye0KIrRVWhegNVQxfj3PyL9M3hvQBE-lseaY,41601
 smart_bot_factory/integrations/openai_client.py,sha256=aMcDrKO0GEx3ZSVEOGDeDtFCDWSXs6biUfgrbRK8yTU,23180
 smart_bot_factory/integrations/supabase_client.py,sha256=AfALLZdDYeMWHLJw6POTGiBd-sH3i03oT6tT7m9C28I,44644
 smart_bot_factory/message/__init__.py,sha256=6QvjdfF99venyDB9udZv9WDNjIHJLNuaVhYdTK3a44A,282
@@ -48,8 +48,8 @@ smart_bot_factory/supabase/client.py,sha256=8_-I3kxZQlKQElI4cTUjNGYcqlyIyEkSrUZa
 smart_bot_factory/utils/__init__.py,sha256=5zNjw491bj5VkOhoEAwh2hjmv8nldyDOTrG7pbGpz6A,285
 smart_bot_factory/utils/debug_routing.py,sha256=BOoDhKBg7UXe5uHQxRk3TSfPfLPOFqt0N7lAo6kjCOo,4719
 smart_bot_factory/utils/prompt_loader.py,sha256=JSn7CsWnToSbHYtURdeuZn7ectyDqQGrPGHN2ixIGkw,19930
-smart_bot_factory-0.2.0.dist-info/METADATA,sha256=WEC38O2La8OlQpfCZWkXjhMU70VigduV0wIrferi2w0,28224
-smart_bot_factory-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smart_bot_factory-0.2.0.dist-info/entry_points.txt,sha256=ybKEAI0WSb7WoRiey7QE-HHfn88UGV7nxLDxXq7b7SU,50
-smart_bot_factory-0.2.0.dist-info/licenses/LICENSE,sha256=OrK3cwdUTzNzIhJvSPtJaVMoYIyC_sSx5EFE_FDMvGs,1092
-smart_bot_factory-0.2.0.dist-info/RECORD,,
+smart_bot_factory-0.2.2.dist-info/METADATA,sha256=XB5NPdlGpOpGPQtsbskUmJ0SGI3hp74IfAE5OHHAmq4,28224
+smart_bot_factory-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+smart_bot_factory-0.2.2.dist-info/entry_points.txt,sha256=ybKEAI0WSb7WoRiey7QE-HHfn88UGV7nxLDxXq7b7SU,50
+smart_bot_factory-0.2.2.dist-info/licenses/LICENSE,sha256=OrK3cwdUTzNzIhJvSPtJaVMoYIyC_sSx5EFE_FDMvGs,1092
+smart_bot_factory-0.2.2.dist-info/RECORD,,