AI Security Agent:告诉它你想查抄什么,剩下的交给 AI

[复制链接]
发表于 2026-3-19 16:57:48 | 显示全部楼层 |阅读模式
AI Security Agent:用天然语言做安全巡检,AI 自己跑下令

做应急相应的时间,有多少时间是花在反复敲下令上的?
ps aux、netstat -antp、cat /var/log/secure、find / -perm -4000……每次排查都是这一套,纯熟的时间呆板,不纯熟的时间还容易漏。
AI Security Agent 想办理的就是这个题目。把任务用天然语言说给它,它自己规划步调、实行下令、看结果、再调解,循环下去直到完成,末了给你一个结论。
工具先容

基于大语言模子的智能主机安全巡检体系,焦点是一个 ReAct Agent Loop:

  • AI 每一步先思索该干什么,再选工具实行,再看结果决定下一步
  • 前端及时展示每步的思索过程和下令输出,全程可观测
  • 支持 Linux(bash)和 Windows(cmd/PowerShell),主动切换语法

除了根本的 Agent 实行,尚有几个比力实用的功能:
工具知识库:AI 实行下令失败时,会主动把错误缘故原由和精确用法记下来,下次不会再踩同一个坑。也可以手动让 AI 去学一个工具的完备用法,学完永世生存。
AI 对话助手:不知道怎么形貌任务?先跟 AI 聊,它帮你把暗昧的需求转化成精准的实行指令。
内置快捷指令:30 条常用安全巡检下令,按 Linux/Windows 分组,点一下填入输入框,支持自界说添加。

体系内置了 whohk 和 Linuxgun 两个 Linux 应急相应工具的知识库,AI 可以直接调用它们做应急排查(工具须要自行下载):

  • whohk:用户账户、汗青下令、定时任务、SSH 登录审计、文件变更检测、WebShell 排查
  • Linuxgun:纯 Shell 编写,历程排查、网络分析、后门检测(SUID/SGID、隧道特性)、K8s 安全查抄
利用方法

环境要求:Python 3.10+,一个兼容 OpenAI 格式的 API Key。
  1. git clone https://github.com/hzhsec/AI-Security-Agent.git
  2. cd AI-Security-Agent
  3. pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
  4. python api.py
复制代码
欣赏器打开 http://localhost:8000,右上角「模子设置」里填入 API Key 即可。
没有 Key 的话,智谱 GLM-4-Flash 完全免费,Groq 免费且速率快,Ollama 可以当地离线运行。
配好之后,直接在输入框里输任务:
  1. 检查系统是否存在异常登录和可疑进程
  2. 检查计划任务和启动项,排查是否有持久化后门
  3. 使用 whohk 对当前系统做全面应急响应检查
复制代码
AI 会主动拆解步调、实行下令、汇总结论,实行中随时可以点克制。
支持 DeepSeek、通义千问、ChatGPT、Claude、Gemini、Kimi、混元等 15+ 提供商,UI 内一键切换。

下载所在

GitHub:https://github.com/hzhsec/AI-Security-Agent

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×
回复

使用道具 举报

登录后关闭弹窗

登录参与点评抽奖  加入IT实名职场社区
去登录
快速回复 返回顶部 返回列表