smart-bot-factory 0.3.7__py3-none-any.whl → 0.3.9__py3-none-any.whl

smart_bot_factory/utils/__init__.py

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

smart_bot_factory/utils/debug_routing.py

@@ -1,103 +1,114 @@
 # debug_routing.py - Утилиты для отладки маршрутизации сообщений
 
 import logging
+
 from aiogram import Router
-from aiogram.types import Message
 from aiogram.fsm.context import FSMContext
+from aiogram.types import Message
 
 logger = logging.getLogger(__name__)
 
 # Создаем роутер для отладочных обработчиков
 debug_router = Router()
 
+
 def setup_debug_handlers(dp):
     """Настройка отладочных обработчиков"""
     dp.include_router(debug_router)
 
+
 # Функция для получения глобальных переменных
 def get_global_var(var_name):
     """Получает глобальную переменную из модуля debug_routing"""
     import sys
+
     current_module = sys.modules[__name__]
     return getattr(current_module, var_name, None)
 
+
 async def debug_user_state(message: Message, state: FSMContext, context: str):
     """Отладочная функция для логирования состояния пользователя"""
-    conversation_manager = get_global_var('conversation_manager')
-    supabase_client = get_global_var('supabase_client')
-    
+    conversation_manager = get_global_var("conversation_manager")
+    supabase_client = get_global_var("supabase_client")
+
     user_id = message.from_user.id
     current_state = await state.get_state()
     state_data = await state.get_data()
-    
+
     logger.info(f"🔍 DEBUG [{context}] User {user_id}:")
     logger.info(f"   📊 FSM State: {current_state}")
     logger.info(f"   📦 State Data: {list(state_data.keys())}")
     logger.info(f"   💬 Message: '{message.text[:50]}...'")
-    
+
     # Проверяем диалог с админом в БД
     conversation = await conversation_manager.is_user_in_admin_chat(user_id)
     logger.info(f"   🗃️ Admin Chat in DB: {'✅' if conversation else '❌'}")
-    
+
     if conversation:
         logger.info(f"   👑 Admin ID: {conversation['admin_id']}")
         logger.info(f"   🆔 Conversation ID: {conversation['id']}")
-    
+
     # Проверяем активную сессию
     session_info = await supabase_client.get_active_session(user_id)
     logger.info(f"   🎯 Active Session: {'✅' if session_info else '❌'}")
-    
+
     if session_info:
         logger.info(f"   📝 Session ID: {session_info['id']}")
-    
+
     logger.info(f"   {'='*50}")
 
+
 async def debug_admin_conversation_creation(admin_id: int, user_id: int):
     """Отладка создания диалога админа с пользователем"""
-    supabase_client = get_global_var('supabase_client')
-    
-    logger.info(f"🔍 DEBUG CONVERSATION CREATION:")
+    supabase_client = get_global_var("supabase_client")
+
+    logger.info("🔍 DEBUG CONVERSATION CREATION:")
     logger.info(f"   👑 Admin: {admin_id}")
     logger.info(f"   👤 User: {user_id}")
-    
+
     # Проверяем активную сессию пользователя ДО создания диалога
     session_info = await supabase_client.get_active_session(user_id)
     logger.info(f"   🎯 User has active session: {'✅' if session_info else '❌'}")
-    
+
     if session_info:
         logger.info(f"   📝 Session ID: {session_info['id']}")
         logger.info(f"   📅 Session created: {session_info['created_at']}")
-    
+
     # Проверяем существующие диалоги пользователя
     try:
-        existing = supabase_client.client.table('admin_user_conversations').select(
-            '*'
-        ).eq('user_id', user_id).eq('status', 'active').execute()
-        
+        existing = (
+            supabase_client.client.table("admin_user_conversations")
+            .select("*")
+            .eq("user_id", user_id)
+            .eq("status", "active")
+            .execute()
+        )
+
         logger.info(f"   💬 Existing active conversations: {len(existing.data)}")
         for conv in existing.data:
             logger.info(f"      - ID: {conv['id']}, Admin: {conv['admin_id']}")
     except Exception as e:
         logger.error(f"   ❌ Error checking existing conversations: {e}")
 
+
 async def test_message_routing(user_id: int, test_message: str):
     """Тестирует маршрутизацию сообщения без отправки через Telegram"""
-    conversation_manager = get_global_var('conversation_manager')
-    
-    logger.info(f"🧪 TESTING MESSAGE ROUTING:")
+    conversation_manager = get_global_var("conversation_manager")
+
+    logger.info("🧪 TESTING MESSAGE ROUTING:")
     logger.info(f"   👤 User: {user_id}")
     logger.info(f"   💬 Message: '{test_message}'")
-    
+
     # Проверяем есть ли диалог с админом
     conversation = await conversation_manager.is_user_in_admin_chat(user_id)
     logger.info(f"   🗃️ Admin conversation exists: {'✅' if conversation else '❌'}")
-    
+
     if conversation:
         logger.info(f"   👑 Admin: {conversation['admin_id']}")
         logger.info(f"   🆔 Conv ID: {conversation['id']}")
         logger.info(f"   📅 Started: {conversation['started_at']}")
-        
+
         # Тестируем должен ли этот пользователь быть в admin_chat
         return "admin_chat"
     else:
-        return "bot_chat"
+        return "bot_chat"

smart_bot_factory/utils/prompt_loader.py

@@ -1,44 +1,52 @@
 # Обновленный prompt_loader.py с поддержкой финальных инструкций
 
 import logging
-import aiofiles
 from pathlib import Path
-from typing import List, Dict
+from typing import Dict
+
+import aiofiles
 
 logger = logging.getLogger(__name__)
 
+
 class PromptLoader:
     """Класс для загрузки промптов из локального каталога"""
-    
+
     def __init__(self, prompts_dir: str):
-            self.prompts_dir = Path(prompts_dir)
-            self.welcome_file = self.prompts_dir / 'welcome_message.txt'
-            self.help_file = self.prompts_dir / 'help_message.txt'
-            self.final_instructions_file = self.prompts_dir / 'final_instructions.txt'
-            
-            # Автоматически находим все .txt файлы промптов (кроме специальных)
-            all_txt_files = list(self.prompts_dir.glob('*.txt'))
-            special_files = {'welcome_message.txt', 'help_message.txt', 'final_instructions.txt'}
-            self.prompt_files = [f.name for f in all_txt_files if f.name not in special_files]
-            
-            logger.info(f"Инициализирован загрузчик промптов: {self.prompts_dir}")
-            logger.info(f"Найдено файлов промптов: {len(self.prompt_files)}")
-            logger.info(f"Файлы промптов: {self.prompt_files}")
+        self.prompts_dir = Path(prompts_dir)
+        self.welcome_file = self.prompts_dir / "welcome_message.txt"
+        self.help_file = self.prompts_dir / "help_message.txt"
+        self.final_instructions_file = self.prompts_dir / "final_instructions.txt"
+
+        # Автоматически находим все .txt файлы промптов (кроме специальных)
+        all_txt_files = list(self.prompts_dir.glob("*.txt"))
+        special_files = {
+            "welcome_message.txt",
+            "help_message.txt",
+            "final_instructions.txt",
+        }
+        self.prompt_files = [
+            f.name for f in all_txt_files if f.name not in special_files
+        ]
+
+        logger.info(f"Инициализирован загрузчик промптов: {self.prompts_dir}")
+        logger.info(f"Найдено файлов промптов: {len(self.prompt_files)}")
+        logger.info(f"Файлы промптов: {self.prompt_files}")
 
     async def load_system_prompt(self) -> str:
         """
         Загружает и объединяет все файлы промптов в один системный промпт
-        
+
         Returns:
             Объединенный системный промпт с инструкциями по JSON
         """
         try:
             prompt_parts = []
-            
+
             for filename in self.prompt_files:
                 logger.debug(f"Загружаем промпт из {filename}")
                 content = await self._load_file(filename)
-                
+
                 if content:
                     # Добавляем заголовок секции
                     section_name = self._get_section_name(filename)
@@ -47,23 +55,25 @@ class PromptLoader:
                     prompt_parts.append("\n")
                 else:
                     logger.warning(f"Файл {filename} пуст")
-            
+
             if not prompt_parts:
                 error_msg = "Не удалось загрузить ни одного промпт файла"
                 logger.error(error_msg)
                 raise ValueError(error_msg)
-            
+
             # Добавляем инструкции по JSON метаданным
             json_instructions = self._get_json_instructions()
             prompt_parts.append("\n")
             prompt_parts.append(json_instructions)
-            
+
             # Объединяем все части
             full_prompt = "".join(prompt_parts).strip()
-            
-            logger.info(f"Системный промпт загружен успешно ({len(full_prompt)} символов)")
+
+            logger.info(
+                f"Системный промпт загружен успешно ({len(full_prompt)} символов)"
+            )
             return full_prompt
-            
+
         except Exception as e:
             logger.error(f"Ошибка при загрузке системного промпта: {e}")
             raise
@@ -71,32 +81,40 @@ class PromptLoader:
     async def load_final_instructions(self) -> str:
         """
         Загружает финальные инструкции из final_instructions.txt
-        
+
         Returns:
             Финальные инструкции или пустая строка если файла нет
         """
         try:
-            logger.debug(f"Загружаем финальные инструкции из {self.final_instructions_file.name}")
-            
+            logger.debug(
+                f"Загружаем финальные инструкции из {self.final_instructions_file.name}"
+            )
+
             if not self.final_instructions_file.exists():
-                logger.debug(f"Файл {self.final_instructions_file.name} не найден - пропускаем")
+                logger.debug(
+                    f"Файл {self.final_instructions_file.name} не найден - пропускаем"
+                )
                 return ""
-            
-            async with aiofiles.open(self.final_instructions_file, 'r', encoding='utf-8') as f:
+
+            async with aiofiles.open(
+                self.final_instructions_file, "r", encoding="utf-8"
+            ) as f:
                 content = await f.read()
-            
+
             if not content.strip():
-                logger.debug(f"Файл {self.final_instructions_file.name} пуст - пропускаем")
+                logger.debug(
+                    f"Файл {self.final_instructions_file.name} пуст - пропускаем"
+                )
                 return ""
-            
+
             logger.info(f"Финальные инструкции загружены ({len(content)} символов)")
             return content.strip()
-                
+
         except Exception as e:
             logger.error(f"Ошибка при загрузке финальных инструкций: {e}")
             # Не прерываем работу - финальные инструкции опциональны
             return ""
-    
+
     def _get_json_instructions(self) -> str:
         """Возвращает инструкции по JSON метаданным для ИИ"""
         return """
@@ -223,59 +241,65 @@ class PromptLoader:
 
 ПОМНИ: Этот JSON критически важен для работы системы администрирования и аналитики!
 """
-    
+
     async def load_welcome_message(self) -> str:
         """
         Загружает приветственное сообщение из welcome_message.txt
-        
+
         Returns:
             Текст приветственного сообщения
         """
         try:
-            logger.debug(f"Загружаем приветственное сообщение из {self.welcome_file.name}")
-            
+            logger.debug(
+                f"Загружаем приветственное сообщение из {self.welcome_file.name}"
+            )
+
             if not self.welcome_file.exists():
                 error_msg = f"Файл приветствия не найден: {self.welcome_file}"
                 logger.error(error_msg)
                 raise FileNotFoundError(error_msg)
-            
-            async with aiofiles.open(self.welcome_file, 'r', encoding='utf-8') as f:
+
+            async with aiofiles.open(self.welcome_file, "r", encoding="utf-8") as f:
                 content = await f.read()
-            
+
             if not content.strip():
                 error_msg = f"Файл приветствия пуст: {self.welcome_file}"
                 logger.error(error_msg)
                 raise ValueError(error_msg)
-            
+
             logger.info(f"Приветственное сообщение загружено ({len(content)} символов)")
             return content.strip()
-                
+
         except Exception as e:
             logger.error(f"Ошибка при загрузке приветственного сообщения: {e}")
             raise
-            
+
     async def load_help_message(self) -> str:
         """
         Загружает справочное сообщение из help_message.txt
-        
+
         Returns:
             Текст справочного сообщения
         """
         try:
             logger.debug(f"Загружаем справочное сообщение из {self.help_file.name}")
-            
+
             if self.help_file.exists():
-                async with aiofiles.open(self.help_file, 'r', encoding='utf-8') as f:
+                async with aiofiles.open(self.help_file, "r", encoding="utf-8") as f:
                     content = await f.read()
-                
+
                 if content.strip():
-                    logger.info(f"Справочное сообщение загружено ({len(content)} символов)")
+                    logger.info(
+                        f"Справочное сообщение загружено ({len(content)} символов)"
+                    )
                     return content.strip()
-            
+
             # Fallback если файл не найден или пуст
-            logger.warning("Файл help_message.txt не найден или пуст, используем дефолтную справку")
+            logger.warning(
+                "Файл help_message.txt не найден или пуст, используем дефолтную справку"
+            )
             return "🤖 **Ваш помощник готов к работе!**\n\n**Команды:**\n/start - Начать диалог\n/help - Показать справку\n/status - Проверить статус"
-                
+
         except Exception as e:
             logger.error(f"Ошибка при загрузке справочного сообщения: {e}")
             # Возвращаем простую справку в случае ошибки
@@ -284,148 +308,157 @@ class PromptLoader:
     async def _load_file(self, filename: str) -> str:
         """Загружает содержимое файла из каталога промптов"""
         file_path = self.prompts_dir / filename
-        
+
         try:
             if not file_path.exists():
                 error_msg = f"Файл промпта не найден: {file_path}"
                 logger.error(error_msg)
                 raise FileNotFoundError(error_msg)
-            
-            async with aiofiles.open(file_path, 'r', encoding='utf-8') as f:
+
+            async with aiofiles.open(file_path, "r", encoding="utf-8") as f:
                 content = await f.read()
-            
+
             if not content.strip():
                 logger.warning(f"Файл {filename} пуст")
                 return ""
-            
+
             logger.debug(f"Загружен файл {filename} ({len(content)} символов)")
             return content
-            
+
         except Exception as e:
             logger.error(f"Ошибка чтения файла {file_path}: {e}")
             raise
-    
+
     def _get_section_name(self, filename: str) -> str:
         """Получает название секции по имени файла"""
         name_mapping = {
-            'system_prompt.txt': 'СИСТЕМНЫЙ ПРОМПТ',
-            'sales_context.txt': 'КОНТЕКСТ ПРОДАЖ', 
-            'product_info.txt': 'ИНФОРМАЦИЯ О ПРОДУКТЕ',
-            'objection_handling.txt': 'ОБРАБОТКА ВОЗРАЖЕНИЙ',
-            '1sales_context.txt': 'КОНТЕКСТ ПРОДАЖ',
-            '2product_info.txt': 'ИНФОРМАЦИЯ О ПРОДУКТЕ', 
-            '3objection_handling.txt': 'ОБРАБОТКА ВОЗРАЖЕНИЙ',
-            'final_instructions.txt': 'ФИНАЛЬНЫЕ ИНСТРУКЦИИ'  # 🆕
+            "system_prompt.txt": "СИСТЕМНЫЙ ПРОМПТ",
+            "sales_context.txt": "КОНТЕКСТ ПРОДАЖ",
+            "product_info.txt": "ИНФОРМАЦИЯ О ПРОДУКТЕ",
+            "objection_handling.txt": "ОБРАБОТКА ВОЗРАЖЕНИЙ",
+            "1sales_context.txt": "КОНТЕКСТ ПРОДАЖ",
+            "2product_info.txt": "ИНФОРМАЦИЯ О ПРОДУКТЕ",
+            "3objection_handling.txt": "ОБРАБОТКА ВОЗРАЖЕНИЙ",
+            "final_instructions.txt": "ФИНАЛЬНЫЕ ИНСТРУКЦИИ",  # 🆕
         }
-        
-        return name_mapping.get(filename, filename.replace('.txt', '').upper())
-    
+
+        return name_mapping.get(filename, filename.replace(".txt", "").upper())
+
     async def reload_prompts(self) -> str:
         """Перезагружает промпты (для обновления без перезапуска бота)"""
         logger.info("Перезагрузка промптов...")
         return await self.load_system_prompt()
-    
+
     async def validate_prompts(self) -> Dict[str, bool]:
         """Проверяет доступность всех файлов промптов и приветственного сообщения"""
         results = {}
-        
+
         # Проверяем файлы промптов
         for filename in self.prompt_files:
             file_path = self.prompts_dir / filename
             try:
                 if file_path.exists():
-                    async with aiofiles.open(file_path, 'r', encoding='utf-8') as f:
+                    async with aiofiles.open(file_path, "r", encoding="utf-8") as f:
                         content = await f.read()
-                    results[filename] = bool(content.strip() and len(content.strip()) > 10)
+                    results[filename] = bool(
+                        content.strip() and len(content.strip()) > 10
+                    )
                 else:
                     results[filename] = False
             except Exception:
                 results[filename] = False
-        
+
         # Проверяем файл приветственного сообщения
         try:
             if self.welcome_file.exists():
-                async with aiofiles.open(self.welcome_file, 'r', encoding='utf-8') as f:
+                async with aiofiles.open(self.welcome_file, "r", encoding="utf-8") as f:
                     content = await f.read()
-                results['welcome_message.txt'] = bool(content.strip() and len(content.strip()) > 5)
+                results["welcome_message.txt"] = bool(
+                    content.strip() and len(content.strip()) > 5
+                )
             else:
-                results['welcome_message.txt'] = False
+                results["welcome_message.txt"] = False
         except Exception:
-            results['welcome_message.txt'] = False
-        
+            results["welcome_message.txt"] = False
+
         # Проверяем файл справки (опционально)
         try:
             if self.help_file.exists():
-                async with aiofiles.open(self.help_file, 'r', encoding='utf-8') as f:
+                async with aiofiles.open(self.help_file, "r", encoding="utf-8") as f:
                     content = await f.read()
-                results['help_message.txt'] = bool(content.strip() and len(content.strip()) > 5)
+                results["help_message.txt"] = bool(
+                    content.strip() and len(content.strip()) > 5
+                )
             else:
-                results['help_message.txt'] = False  # Не критично
+                results["help_message.txt"] = False  # Не критично
         except Exception:
-            results['help_message.txt'] = False
-        
+            results["help_message.txt"] = False
+
         # 🆕 Проверяем финальные инструкции (опционально)
         try:
             if self.final_instructions_file.exists():
-                async with aiofiles.open(self.final_instructions_file, 'r', encoding='utf-8') as f:
+                async with aiofiles.open(
+                    self.final_instructions_file, "r", encoding="utf-8"
+                ) as f:
                     content = await f.read()
-                results['final_instructions.txt'] = bool(content.strip() and len(content.strip()) > 5)
+                results["final_instructions.txt"] = bool(
+                    content.strip() and len(content.strip()) > 5
+                )
             else:
-                results['final_instructions.txt'] = False  # Не критично - опциональный файл
+                results["final_instructions.txt"] = (
+                    False  # Не критично - опциональный файл
+                )
         except Exception:
-            results['final_instructions.txt'] = False
-        
+            results["final_instructions.txt"] = False
+
         return results
-    
+
     def get_prompt_info(self) -> Dict[str, any]:
         """Возвращает информацию о конфигурации промптов"""
         return {
-            'prompts_dir': str(self.prompts_dir),
-            'prompt_files': self.prompt_files,
-            'welcome_file': 'welcome_message.txt',
-            'help_file': 'help_message.txt',
-            'final_instructions_file': 'final_instructions.txt',  # 🆕
-            'total_files': len(self.prompt_files) + 1,  # +1 для welcome message
-            'json_instructions_included': True
+            "prompts_dir": str(self.prompts_dir),
+            "prompt_files": self.prompt_files,
+            "welcome_file": "welcome_message.txt",
+            "help_file": "help_message.txt",
+            "final_instructions_file": "final_instructions.txt",  # 🆕
+            "total_files": len(self.prompt_files) + 1,  # +1 для welcome message
+            "json_instructions_included": True,
         }
-    
+
     async def test_json_parsing(self, test_response: str) -> Dict[str, any]:
         """Тестирует парсинг JSON из ответа ИИ (для отладки)"""
         import json
         import re
-        
+
         try:
             # Используем тот же алгоритм что и в main.py
             json_pattern = r'\{[^{}]*"этап"[^{}]*\}$'
             match = re.search(json_pattern, test_response.strip())
-            
+
             if match:
                 json_str = match.group(0)
-                response_text = test_response[:match.start()].strip()
-                
+                response_text = test_response[: match.start()].strip()
+
                 try:
                     metadata = json.loads(json_str)
                     return {
-                        'success': True,
-                        'response_text': response_text,
-                        'metadata': metadata,
-                        'json_str': json_str
+                        "success": True,
+                        "response_text": response_text,
+                        "metadata": metadata,
+                        "json_str": json_str,
                     }
                 except json.JSONDecodeError as e:
                     return {
-                        'success': False,
-                        'error': f"JSON decode error: {e}",
-                        'json_str': json_str
+                        "success": False,
+                        "error": f"JSON decode error: {e}",
+                        "json_str": json_str,
                     }
             else:
                 return {
-                    'success': False,
-                    'error': "JSON pattern not found",
-                    'response_text': test_response
+                    "success": False,
+                    "error": "JSON pattern not found",
+                    "response_text": test_response,
                 }
-                
+
         except Exception as e:
-            return {
-                'success': False,
-                'error': f"Parse error: {e}"
-            }
+            return {"success": False, "error": f"Parse error: {e}"}