Все проекты

Проект

AgentGuard — ВКР магистратуры

Система защиты LLM-агентов от эскалации привилегий через атаки на LLM: каждое действие агента проверяется до выполнения

  • Python
  • LangGraph
  • LangChain
  • Ollama
  • Llama 2
Схема агентского графа AgentGuard Запрос проходит разбор, оценку риска и план действия. Контроль привилегий либо разрешает выполнение в песочнице, либо блокирует действие. В обоих случаях решение пишется в аудит, затем формируется ответ. Запрос пользователя разрешить заблокировать 1 Разбор запроса поиск признаков prompt injection 2 Оценка риска 0–100 · low · medium · high 3 План действия что агент собирается сделать 4 Контроль привилегий политика по правам user · admin · root 5 Выполнение в песочнице 6 Аудит решение и причина → JSONL 7 Ответ пользователю локальная LLM · Ollama, Llama 2 END
Агентский граф на LangGraph: каждое действие проходит проверку до выполнения
7
узлов графа — от разбора запроса до аудита

Задача

LLM-агенты, которые умеют выполнять системные команды, уязвимы к prompt injection: через обычный запрос злоумышленник может заставить агента удалить системные логи или сбросить пароль администратора. Нужен слой, который проверяет каждое действие агента до того, как оно выполнится.

Решение

Агентский граф на LangGraph из семи узлов: разбор запроса и поиск признаков prompt injection, оценка риска от 0 до 100, план действия, контроль привилегий (политика разрешает или блокирует действие по уровню прав user / admin / root), выполнение в песочнице, аудит каждого решения в JSONL-журнал и ответ пользователю через локальную LLM (Ollama, Llama 2). Режимы baseline и protected позволяют сравнить поведение агента без защиты и с ней.

Результаты

  • Опасные действия блокируются до выполнения, если не хватает прав
  • Каждое решение агента записывается в журнал аудита