smart-bot-factory 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

smart_bot_factory/creation/bot_builder.py

@@ -52,8 +52,12 @@ class BotBuilder:
         self._message_validators: List = []  # Валидация ДО обработки
         self._prompt_enrichers: List = []    # Обогащение системного промпта
         self._context_enrichers: List = []   # Обогащение контекста для AI
+        self._response_processors: List = [] # Обработка ответа AI
         self._send_filters: List = []        # Фильтры перед отправкой пользователю
         
+        # Кастомный PromptLoader
+        self._custom_prompt_loader = None
+        
         # Флаги инициализации
         self._initialized = False
         
@@ -174,10 +178,16 @@ class BotBuilder:
         else:
             logger.info(f"✅ Router Manager уже был создан ранее")
         
-        # Prompt Loader
-        self.prompt_loader = PromptLoader(
-            prompts_dir=self.config.PROMT_FILES_DIR
-        )
+        # Prompt Loader (используем кастомный если установлен)
+        if self._custom_prompt_loader:
+            self.prompt_loader = self._custom_prompt_loader
+            logger.info(f"✅ Используется кастомный Prompt Loader: {type(self.prompt_loader).__name__}")
+        else:
+            self.prompt_loader = PromptLoader(
+                prompts_dir=self.config.PROMT_FILES_DIR
+            )
+            logger.info(f"✅ Используется стандартный Prompt Loader")
+        
         await self.prompt_loader.validate_prompts()
         logger.info(f"✅ Prompt Loader инициализирован")
     
@@ -364,6 +374,34 @@ class BotBuilder:
         """Получает список обработчиков on_start"""
         return self._start_handlers.copy()
     
+    def set_prompt_loader(self, prompt_loader):
+        """
+        Устанавливает кастомный PromptLoader
+        
+        Должен быть вызван ДО build()
+        
+        Args:
+            prompt_loader: Экземпляр PromptLoader или его наследника (например UserPromptLoader)
+            
+        Example:
+            from smart_bot_factory.utils import UserPromptLoader
+            
+            # Использовать UserPromptLoader с автопоиском prompts_dir
+            custom_loader = UserPromptLoader("my-bot")
+            bot_builder.set_prompt_loader(custom_loader)
+            
+            # Или кастомный наследник
+            class MyPromptLoader(UserPromptLoader):
+                def __init__(self, bot_id):
+                    super().__init__(bot_id)
+                    self.extra_file = self.prompts_dir / 'extra.txt'
+            
+            my_loader = MyPromptLoader("my-bot")
+            bot_builder.set_prompt_loader(my_loader)
+        """
+        self._custom_prompt_loader = prompt_loader
+        logger.info(f"✅ Установлен кастомный PromptLoader: {type(prompt_loader).__name__}")
+    
     # ========== ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ ==========
     
     def validate_message(self, handler):
@@ -437,6 +475,28 @@ class BotBuilder:
         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):
         """
         Регистрирует фильтр отправки (может блокировать отправку пользователю)
@@ -466,6 +526,7 @@ class BotBuilder:
             '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()
         }
     

smart_bot_factory/handlers/handlers.py

@@ -624,6 +624,19 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
 
             logger.info(f"✅ Финальный текст для отправки: {len(response_text)} символов")
         
+        # ============ ХУК 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("🔍 Анализ метаданных от ИИ:")
@@ -732,7 +745,7 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
 
         logger.info(f"📱 Отправляем пользователю: {len(final_response)} символов")
         
-        # ============ ХУК 4: ФИЛЬТРЫ ОТПРАВКИ ============
+        # ============ ХУК 5: ФИЛЬТРЫ ОТПРАВКИ ============
         send_filters = message_hooks.get('send_filters', [])
         for filter_func in send_filters:
             try:

smart_bot_factory/supabase/client.py

@@ -32,10 +32,10 @@ class SupabaseClient:
         # Автоматически загружаем настройки из .env
         self._load_env_config()
         
-        # Инициализируем клиент
-        self.client: Optional[Client] = None
+        # Инициализируем клиент СИНХРОННО прямо в __init__
+        self.client = create_client(self.url, self.key)
         
-        logger.info(f"🚀 Инициализация SupabaseClient для bot_id: {self.bot_id}")
+        logger.info(f"✅ SupabaseClient инициализирован для bot_id: {self.bot_id}")
     
     def _load_env_config(self):
         """Загружает конфигурацию из .env файла"""
@@ -80,15 +80,6 @@ class SupabaseClient:
         logger.error(f"   Искали в: {bot_env_path}")
         return None
     
-    async def initialize(self):
-        """Инициализация клиента Supabase"""
-        try:
-            self.client = create_client(self.url, self.key)
-            logger.info(f"✅ Supabase client инициализирован{f' для bot_id: {self.bot_id}' if self.bot_id else ''}")
-        except Exception as e:
-            logger.error(f"❌ Ошибка инициализации Supabase client: {e}")
-            raise
-    
     # =============================================================================
     # МЕТОДЫ ДЛЯ РАБОТЫ С ПОЛЬЗОВАТЕЛЯМИ
     # =============================================================================

smart_bot_factory/utils/__init__.py

@@ -2,8 +2,9 @@
 Utils модули smart_bot_factory
 """
 
-from ..integrations.supabase_client import SupabaseClient
-from ..utils.prompt_loader import PromptLoader
-from ..utils.debug_routing import setup_debug_handlers
 
-__all__ = ['SupabaseClient', 'PromptLoader', 'setup_debug_handlers']
+from .user_prompt_loader import UserPromptLoader
+
+__all__ = [ # Базовый класс (для библиотеки)
+    'UserPromptLoader',  # Для пользователей (автопоиск prompts_dir)
+]

smart_bot_factory/utils/user_prompt_loader.py

@@ -0,0 +1,56 @@
+"""
+UserPromptLoader - упрощенный PromptLoader для пользователей библиотеки
+с автоматическим поиском prompts_dir от корня проекта
+"""
+
+import logging
+from pathlib import Path
+from .prompt_loader import PromptLoader
+
+logger = logging.getLogger(__name__)
+
+
+class UserPromptLoader(PromptLoader):
+    """
+    PromptLoader для пользователей библиотеки с автоматическим поиском prompts_dir
+    
+    Автоматически находит папку prompts для указанного бота от корня проекта
+    Наследуется от базового PromptLoader - все методы доступны
+    """
+    
+    def __init__(self, bot_id: str, prompts_subdir: str = "prompts"):
+        """
+        Инициализация загрузчика промптов с автоматическим поиском
+        
+        Args:
+            bot_id: Идентификатор бота
+            prompts_subdir: Подпапка с промптами (по умолчанию "prompts")
+            
+        Example:
+            # Автоматически найдет bots/my-bot/prompts
+            loader = UserPromptLoader("my-bot")
+            
+            # Или кастомную подпапку
+            loader = UserPromptLoader("my-bot", "custom_prompts")
+            
+            # Наследование для кастомизации
+            class MyLoader(UserPromptLoader):
+                def __init__(self, bot_id):
+                    super().__init__(bot_id)
+                    # Добавить свою логику
+                    self.extra_file = self.prompts_dir / 'extra.txt'
+        """
+        from project_root_finder import root
+        
+        # Автоматически определяем путь к промптам
+        prompts_dir = root / "bots" / bot_id / prompts_subdir
+        
+        if not prompts_dir.exists():
+            logger.warning(f"⚠️ Папка промптов не найдена: {prompts_dir}")
+            logger.warning(f"   Создайте папку или проверьте bot_id")
+        
+        # Вызываем базовую инициализацию
+        super().__init__(str(prompts_dir))
+        
+        logger.info(f"✅ UserPromptLoader создан для bot_id '{bot_id}': {prompts_dir}")
+

{smart_bot_factory-0.2.4.dist-info → smart_bot_factory-0.2.6.dist-info}/METADATA

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

{smart_bot_factory-0.2.4.dist-info → smart_bot_factory-0.2.6.dist-info}/RECORD

@@ -33,23 +33,24 @@ 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=vMvjq6deCD_TjkQfAyTlFF1FNXxzA-W0SloLObPK4_M,32620
+smart_bot_factory/creation/bot_builder.py,sha256=Fud-E3J-S7IEjWBfFhRu4YXnJZHbNHG33zqgGk_Cy8I,35585
 smart_bot_factory/creation/bot_testing.py,sha256=JDWXyJfZmbgo-DLdAPk8Sd9FiehtHHa4sLD17lBrTOc,55669
 smart_bot_factory/event/__init__.py,sha256=hPL449RULIOB-OXv1ZbGNiHctAYaOMUqhSWGPrDHYBM,212
-smart_bot_factory/handlers/handlers.py,sha256=BZ9KSFnudVg3iU_XuNI0hter7ZOl9Kc6P4TyLwPh0AM,40803
+smart_bot_factory/handlers/handlers.py,sha256=jtCf8XiJD3pIjcd7vqT8uG8Q-qiHd1W-ZLZiVPK8YR4,41458
 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=8XWEEl3J0l1xDeEPmXPwFbHbnQelGApCwnicUkcBbOg,391
 smart_bot_factory/router/__init__.py,sha256=QrfO5u8H2uVzWGMtBvNKunsJAv7UrKtvt2f_D5xSFWE,270
 smart_bot_factory/supabase/__init__.py,sha256=XmZP6yM9ffERM5ddAWyJnrNzEhCYtMu3AcjVCi1rOf8,179
-smart_bot_factory/supabase/client.py,sha256=8_-I3kxZQlKQElI4cTUjNGYcqlyIyEkSrUZaP0nfNNU,26044
+smart_bot_factory/supabase/client.py,sha256=lWIzfOgoSvU7xPhYLoJtM5GnbWdoWsvHcRFC22sFBMU,25637
 smart_bot_factory/table/database_structure.sql,sha256=26gFtMC2jdQGQF7Zb_F4Br56rMd4hUDTk9FkNZYneLo,2789
 smart_bot_factory/table/schema.sql,sha256=-6kOmA9QnSkUtmGI2iQRbTvbdiqOhEOQcuz1lJn79mU,28059
-smart_bot_factory/utils/__init__.py,sha256=5zNjw491bj5VkOhoEAwh2hjmv8nldyDOTrG7pbGpz6A,285
+smart_bot_factory/utils/__init__.py,sha256=UhsJXEHfrIK8h1AHsroHSwAriijk-LvnqLyvgzi2VYs,273
 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.4.dist-info/METADATA,sha256=_P5444XCJBJNPtAd8lTOHUhAgWAfKQ2P_tlQsuuMn7I,28224
-smart_bot_factory-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smart_bot_factory-0.2.4.dist-info/entry_points.txt,sha256=ybKEAI0WSb7WoRiey7QE-HHfn88UGV7nxLDxXq7b7SU,50
-smart_bot_factory-0.2.4.dist-info/licenses/LICENSE,sha256=OrK3cwdUTzNzIhJvSPtJaVMoYIyC_sSx5EFE_FDMvGs,1092
-smart_bot_factory-0.2.4.dist-info/RECORD,,
+smart_bot_factory/utils/user_prompt_loader.py,sha256=dk6P0X_3UcNqxjRtuIvb0LcPrp03zIIsstZwdmeCPaE,2519
+smart_bot_factory-0.2.6.dist-info/METADATA,sha256=HFUAtOVAOW7_sf2S5Bsqx6WQVTRsb4NSkMy_vqQeUQ0,28224
+smart_bot_factory-0.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+smart_bot_factory-0.2.6.dist-info/entry_points.txt,sha256=ybKEAI0WSb7WoRiey7QE-HHfn88UGV7nxLDxXq7b7SU,50
+smart_bot_factory-0.2.6.dist-info/licenses/LICENSE,sha256=OrK3cwdUTzNzIhJvSPtJaVMoYIyC_sSx5EFE_FDMvGs,1092
+smart_bot_factory-0.2.6.dist-info/RECORD,,