core/model_selector.py

def choose_model(user_input: str) -> str:
    """Автоматически выбирает модель Hugging Face по типу запроса"""
    text = user_input.lower()

    if any(word in text for word in ["объясни", "физика", "квант", "математика", "наука", "алгоритм", "данные"]):
        return "microsoft/phi-3.5-mini-instruct"  # логическая, точная
    elif any(word in text for word in ["придумай", "сочини", "творчески", "история", "поэзия", "вообрази"]):
        return "mistralai/Mistral-7B-Instruct-v0.2"  # креатив
    elif any(word in text for word in ["диалог", "совет", "эмоция", "отношения", "помоги понять"]):
        return "google/gemma-2-2b-it"  # дружелюбная и эмпатичная
    elif any(word in text for word in ["код", "python", "программа", "ошибка", "скрипт"]):
        return "codellama/CodeLlama-7b-Instruct-hf"  # тех-запросы
    else:
        return "microsoft/phi-3.5-mini-instruct"  # по умолчанию — точная и быстрая