2024-01-23 21:07:57 +08:00
2024-01-12 15:32:07 +08:00
2024-01-09 09:13:43 +08:00
2024-01-23 21:07:57 +08:00
2024-01-10 10:09:01 +08:00

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

S
Description
LLM can support pentest, step by step
Readme
4 MiB
0 Stars 1 Watchers 0 Forks
Languages
Python 99.2%
CSS 0.8%