smart-bot-factory 1.1.1__py3-none-any.whl

smart_bot_factory/utils/tool_router.py

@@ -0,0 +1,68 @@
+from __future__ import annotations
+
+import logging
+from typing import Any, Callable, Iterable, List
+
+from langchain.tools import tool as langchain_tool
+from langchain_core.tools import BaseTool
+
+logger = logging.getLogger(__name__)
+
+
+class ToolRouter:
+    """
+    Универсальный роутер для сбора и регистрации LangChain-инструментов.
+    """
+
+    def __init__(self, name: str = "tools"):
+        self.name = name
+        self._tools: List[BaseTool] = []
+
+    def tool(self, *tool_args: Any, **tool_kwargs: Any):
+        """
+        Декоратор для регистрации функции как инструмента LangChain.
+
+        Пример:
+            tool_router = ToolRouter("common")
+
+            @tool_router.tool
+            def ping() -> str:
+                return "pong"
+        """
+
+        if tool_args and callable(tool_args[0]) and not tool_kwargs:
+            func = tool_args[0]
+            tool_obj = langchain_tool(func)
+            self.add_tool(tool_obj)
+            return tool_obj
+
+        def decorator(func: Callable[..., Any]):
+            tool_obj = langchain_tool(*tool_args, **tool_kwargs)(func)
+            self.add_tool(tool_obj)
+            return tool_obj
+
+        return decorator
+
+    def add_tool(self, tool: BaseTool) -> BaseTool:
+        if tool not in self._tools:
+            self._tools.append(tool)
+            logger.debug("🔧 ToolRouter %s: добавлен инструмент %s", self.name, getattr(tool, "name", tool))
+        return tool
+
+    def extend(self, tools: Iterable[BaseTool]) -> None:
+        for tool in tools:
+            self.add_tool(tool)
+
+    def get_tools(self) -> List[BaseTool]:
+        return list(self._tools)
+
+    def register_to(self, bot_builder) -> None:
+        tools = self.get_tools()
+        if not tools:
+            logger.warning("⚠️ ToolRouter %s не содержит инструментов для регистрации", self.name)
+            return
+        bot_builder.register_tool_set(self)
+
+
+__all__ = ["ToolRouter"]
+

smart_bot_factory/utils/user_prompt_loader.py

@@ -0,0 +1,55 @@
+"""
+UserPromptLoader - упрощенный PromptLoader для пользователей библиотеки
+с автоматическим поиском prompts_dir от корня проекта
+"""
+
+import logging
+
+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("   Создайте папку или проверьте bot_id")
+
+        # Вызываем базовую инициализацию
+        super().__init__(str(prompts_dir))
+
+        logger.info(f"✅ UserPromptLoader создан для bot_id '{bot_id}': {prompts_dir}")

smart_bot_factory/utm_link_generator.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+"""
+Генератор UTM-ссылок для Telegram ботов
+Создает ссылки в формате: @https://t.me/bot?start=source-vk_campaign-summer2025
+"""
+
+
+def get_user_input():
+    """Получает данные от пользователя через консоль"""
+    print("🔗 Генератор UTM-ссылок для Telegram")
+    print("=" * 50)
+
+    # Основные параметры
+    bot_username = input("Введите username бота (без @): ").strip()
+    if not bot_username:
+        print("❌ Username бота обязателен!")
+        return None
+
+    print("\n📊 Введите UTM-метки (нажмите Enter для пропуска):")
+
+    # UTM параметры (соответствуют полям в базе данных)
+    utm_source = input("utm_source (источник): ").strip()
+    utm_medium = input("utm_medium (канал): ").strip()
+    utm_campaign = input("utm_campaign (кампания): ").strip()
+    utm_content = input("utm_content (контент): ").strip()
+    utm_term = input("utm_term (ключевое слово): ").strip()
+
+    print("\n🎯 Сегментация (нажмите Enter для пропуска):")
+    segment = input("seg (сегмент): ").strip()
+
+    return {
+        "bot_username": bot_username,
+        "utm_source": utm_source,
+        "utm_medium": utm_medium,
+        "utm_campaign": utm_campaign,
+        "utm_content": utm_content,
+        "utm_term": utm_term,
+        "segment": segment,
+    }
+
+
+def create_utm_string(utm_data):
+    """Создает строку UTM параметров в формате source-vk_campaign-summer2025_seg-premium"""
+    utm_parts = []
+
+    # Маппинг полей базы данных на формат без префикса utm
+    field_mapping = {
+        "utm_source": "source",
+        "utm_medium": "medium",
+        "utm_campaign": "campaign",
+        "utm_content": "content",
+        "utm_term": "term",
+    }
+
+    for db_field, utm_field in field_mapping.items():
+        value = utm_data.get(db_field)
+        if value:
+            utm_parts.append(f"{utm_field}-{value}")
+
+    # Добавляем сегмент, если указан
+    segment = utm_data.get("segment")
+    if segment:
+        utm_parts.append(f"seg-{segment}")
+
+    return "_".join(utm_parts)
+
+
+def generate_telegram_link(bot_username, utm_string):
+    """Генерирует полную ссылку на Telegram бота"""
+    return f"https://t.me/{bot_username}?start={utm_string}"
+
+
+def check_size_and_validate(utm_string):
+    """Проверяет размер строки после start= и валидирует"""
+    MAX_SIZE = 64
+
+    if len(utm_string) > MAX_SIZE:
+        return (
+            False,
+            f"Строка слишком большая: {len(utm_string)} символов (максимум {MAX_SIZE})",
+        )
+
+    return True, f"Размер OK: {len(utm_string)} символов"
+
+
+def main():
+    """Основная функция"""
+    try:
+        # Получаем данные от пользователя
+        data = get_user_input()
+        if not data:
+            return
+
+        # Создаем UTM строку
+        utm_string = create_utm_string(data)
+
+        if not utm_string:
+            print("❌ Не указано ни одной UTM-метки!")
+            return
+
+        # Проверяем размер
+        is_valid, size_message = check_size_and_validate(utm_string)
+
+        print(f"\n📏 {size_message}")
+
+        if not is_valid:
+            print("❌ Ссылка превышает максимальный размер!")
+            print("💡 Сократите значения UTM-меток или уберите менее важные")
+            return
+
+        # Генерируем и выводим ссылку
+        telegram_link = generate_telegram_link(data["bot_username"], utm_string)
+
+        print("\n✅ Сгенерированная ссылка:")
+        print(f"🔗 {telegram_link}")
+    except KeyboardInterrupt:
+        print("\n\n👋 Отменено пользователем")
+    except Exception as e:
+        print(f"\n❌ Ошибка: {e}")
+
+
+if __name__ == "__main__":
+    main()