smart-bot-factory 0.2.1__tar.gz → 0.2.2__tar.gz

{smart_bot_factory-0.2.1 → smart_bot_factory-0.2.2}/PKG-INFO

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

{smart_bot_factory-0.2.1 → smart_bot_factory-0.2.2}/best-valera.py

@@ -1,10 +1,13 @@
 import asyncio
+import logging
 
 from smart_bot_factory.router import EventRouter, TelegramRouter
 from smart_bot_factory.message import send_message_by_human, send_message_to_users_by_stage
 from smart_bot_factory.supabase import SupabaseClient
 from smart_bot_factory.creation import BotBuilder
 
+logger = logging.getLogger(__name__)
+
 # =============================================================================
 # СОЗДАНИЕ РОУТЕРОВ
 # =============================================================================
@@ -55,6 +58,80 @@ async def custom_start_handler(user_id: int, session_id: str, message: Message,
         reply_markup=keyboard
     )
 
+# =============================================================================
+# ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ
+# =============================================================================
+
+# ХУК 1: Валидация сообщений
+@bot_builder.validate_message
+async def validate_service_names(message, supabase_client):
+    """Проверяем корректность названий услуг"""
+    # Пример: блокируем неправильные названия
+    incorrect_names = ["массаш", "педекюр", "макияш"]
+    
+    if message.text:
+        for incorrect in incorrect_names:
+            if incorrect in message.text.lower():
+                await message.answer(
+                    f"❗ Возможно, вы имели в виду другую услугу?\n"
+                    f"Пожалуйста, уточните название"
+                )
+                return False  # Прерываем обработку AI
+    
+    return True  # Продолжаем обработку
+
+# ХУК 2: Обогащение промпта
+@bot_builder.enrich_prompt
+async def add_client_info_to_prompt(system_prompt, user_id, session_id, supabase_client):
+    """Добавляем информацию о клиенте в промпт"""
+    try:
+        session = await supabase_client.get_active_session(user_id)
+        
+        if session and session.get('metadata'):
+            phone = session['metadata'].get('phone')
+            name = session['metadata'].get('confirmed_name') or session['metadata'].get('telegram_name')
+            
+            if phone:
+                client_info = f"\n\nИНФОРМАЦИЯ О КЛИЕНТЕ:\n- Телефон: {phone}"
+                if name:
+                    client_info += f"\n- Имя: {name}"
+                client_info += "\n\n⚠️ ВАЖНО: При создании записи используй ЭТОТ телефон и имя!"
+                
+                return system_prompt + client_info
+    except Exception as e:
+        logger.error(f"Ошибка обогащения промпта: {e}")
+    
+    return system_prompt
+
+# ХУК 3: Обогащение контекста (например, данные из внешнего API)
+@bot_builder.enrich_context
+async def add_external_api_data(messages, user_id, session_id):
+    """Добавляем данные из внешних систем"""
+    # Пример: можно добавить расписание из YClients, данные из CRM и т.д.
+    # messages.append({
+    #     "role": "system",
+    #     "content": "Реальное доступное время: ..."
+    # })
+    return messages
+
+# ХУК 4: Обработка ответа AI
+@bot_builder.process_response
+async def add_promo_to_price_response(response_text, ai_metadata, user_id):
+    """Добавляем промо-код к ответам о ценах"""
+    if "цен" in response_text.lower() or "стоимост" in response_text.lower():
+        response_text += "\n\n🎁 Промокод FIRST10 для скидки 10% на первый визит!"
+    
+    return response_text, ai_metadata
+
+# ХУК 5: Фильтр отправки
+@bot_builder.filter_send
+async def allow_all_messages(user_id, response_text):
+    """Разрешаем все сообщения (можно добавить блокировку по условию)"""
+    # Пример: блокировка во время обработки
+    # if is_processing(user_id):
+    #     return False
+    return True
+
 @telegram_router_1.router.message(Command("price", "цена"))
 async def handle_price_command(message: Message, state: FSMContext):
     """Обработчик команды /price"""

{smart_bot_factory-0.2.1 → smart_bot_factory-0.2.2}/pyproject.toml

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

{smart_bot_factory-0.2.1 → smart_bot_factory-0.2.2}/smart_bot_factory/creation/bot_builder.py

@@ -48,6 +48,13 @@ class BotBuilder:
         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
         
@@ -358,6 +365,134 @@ class BotBuilder:
         """Получает список обработчиков 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
@@ -407,6 +542,7 @@ class BotBuilder:
                 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-0.2.1 → smart_bot_factory-0.2.2}/smart_bot_factory/handlers/handlers.py

@@ -492,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,
@@ -514,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}]
         
@@ -536,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")
@@ -588,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:
@@ -697,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.1 → smart_bot_factory-0.2.2}/uv.lock

@@ -1259,7 +1259,7 @@ wheels = [
 
 [[package]]
 name = "smart-bot-factory"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "aiofiles" },