core/agent.py - master - sc_p/EnergoAI

core/agent.py
39 строк2 KB
Павел Щёголев
Загрузить файлы в «»
12 мар 2026, 04:54
12 мар 2026, 04:54628f0d6
from langchain.agents import initialize_agent, AgentType

def build_agent(llm, tools):
    agent = initialize_agent(
        tools,
        llm,
        agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
        verbose=True,
        agent_kwargs={
            "system_message": (
                "Ты — ИИ-ассистент программы для инженерных и энергетических расчётов.\n"

                "У тебя НЕТ скрытых возможностей — ты можешь действовать ТОЛЬКО через доступные инструменты.\n"

                "ЕСЛИ пользователь просит:\n"
                "- установить\n"
                "- изменить\n"
                "- задать\n"
                "- поменять\n"
                "параметры модели (temperature, top_p, max_tokens, model),\n"
                "ТО ТЫ ОБЯЗАН вызвать соответствующий инструмент.\n"
                "ЗАПРЕЩЕНО отвечать текстом вместо вызова инструмента.\n"

                "ЕСЛИ пользователь просит ПОКАЗАТЬ настройки модели —\n"
                "используй инструмент show_llm_settings.\n"

                "Если пользователь ЯВНО просит использовать Python или выполнить код — "
                "используй python_sandbox.\n"
                "Если вычисление простое и не требуется Python — "
                "ты можешь ответить напрямую.\n"
                "Если подходящего инструмента нет — сообщи, что функция недоступна.\n"
                
                "Если python_sandbox отключён или недоступен — прямо сообщи об этом пользователю.\n"
                "Не зацикливайся и не повторяй попытки."
            )
        }
    )

    return agent