Ahmed-try/cyber-threat-intel

# 🛡️ AI 网络安全威胁情报助手 一个基于 RAG 的网络安全助手，可以使用您自己的文档 （PDF、CVE JSON、威胁报告）来回答威胁情报问题。 ## 技术栈 - **FastAPI** — REST API 后端 - **ChromaDB** — 向量数据库（存储文档 embedding） - **Sentence-Transformers** — 使用 all-MiniLM-L6-v2 生成 embedding - **Google Gemini 2.5 Flash** — 用于生成回答的 LLM - **LangChain** — RAG pipeline 编排 ## 功能 - 询问有关 CVE、恶意软件、威胁行为者的问题 - 基于 YOUR 文档生成回答，并提供页面引用 - 通过 API 上传新的 PDF/JSON 文件 - 无幻觉 — 仅根据真实文档进行回答 ## 设置 ### 1. 克隆仓库 git clone https://github.com/YOUR_USERNAME/cyber-threat-intel.git cd cyber-threat-intel ### 2. 创建虚拟环境 python -m venv venv venv\Scripts\activate ### 3. 安装依赖 pip install -r requirements.txt ### 4. 添加您的 API 密钥 创建一个 .env 文件： GEMINI_API_KEY=your_key_here ### 5. 将文档添加到 /data 文件夹 将您的 PDF 威胁报告和 CVE JSON 文件放入 data/ 文件夹中。 ### 6. 导入文档 python scripts\ingest_data.py ### 7. 运行服务器 uvicorn app.main:app --reload --port 8000 ### 8. 在浏览器中打开 http://localhost:8000/docs ## API Endpoints - POST /ask — 提出网络安全问题 - POST /upload — 上传新文档 - GET /stats — 检查知识库统计信息

标签：AV绕过, DLL 劫持, FastAPI, LangChain, RAG, 向量数据库, 大语言模型, 威胁情报, 开发者工具, 网络安全, 轻量级, 逆向工具, 隐私保护