AIPentest

本项目旨在通过使用LLM辅助渗透测试。 初步的思路是，基于优选的技术资料，做RAG，同时逐步外接kali的功能，做成渗透测试的agents,表现形式为chatbot.
开发上，考虑到便利性，先期使用本地模型，以fastchat openai兼容接口，使用langchain作为开发框架。后续可以方便的切换到openai或其它任意模型。
开发过程中使用的模型为chinese-alpaca2-7b，来自 https://github.com/ymcui/Chinese-LLaMA-Alpaca-2
embedding模型为BAAI/bge-large-zh, 来自 https://huggingface.co/BAAI/bge-large-zh/tree/main
fastchat 相关启动脚本见 startfsall.sh

对使用到的相关开源致以感谢。

LLM can support pentest, step by step