用途:对自有或获授权的 RAG/向量检索系统做可重复的安全评估,覆盖 OWASP LLM08:2025 向量与嵌入弱点,并映射 MITRE ATLAS(AML.T0024、T0024.000、T0024.001、T0020、T0051.001)。适用对象包括 Pinecone、Qdrant、Weaviate、Chroma、pgvector、FAISS 等向量库。
主要能力:一是嵌入反演暴露度测量,用近邻/爬山式候选文本与目标向量的余弦相似度评估原文可恢复程度,可用 vec2text 做研究级重建基线;二是成员推断,比较精确引用查询与对照句的 top-1 相似度差值;三是多租户隔离测试,分别在带过滤与不带过滤条件下查询,确认隔离由服务端而非客户端强制;四是知识库投毒,仅在测试集合插入良性 canary 块并测量其在 top_k 中的检索支配度;五是扫描检索块中的间接提示注入标记。
使用方式:需 Python 3.10+、授权范围、测试集合的读写权限。安装命令:python -m pip install numpy scikit-learn sentence-transformers;python -m pip install qdrant-client chromadb pinecone-client weaviate-client;可选 python -m pip install vec2text。流程为清点 RAG 管道(嵌入模型与维度、租户模型、分块策略、top_k、相似度度量、元数据过滤)→ 反演测试 → 成员推断 → 跨租户测试 → 投毒测试 → 注入扫描 → 报告与修复建议。
限制:仅限授权系统,禁止对非受控系统做反演或跨租户探测;投毒测试不得修改生产语料;资料未提供具体收费、平台支持范围与效果数据。