Vamshi868876/ZeroTrust-GenAI-Architect

GitHub: Vamshi868876/ZeroTrust-GenAI-Architect

基于图机器学习和 RAG 的多云零信任态势管理系统,能自动检测有毒云配置并生成修复补丁。

Stars: 0 | Forks: 0

Architecture Cloud Security

🛡️ 零信任 GenAI 架构师 (RAG-Sec)

一个由图机器学习和检索增强生成 (RAG) 驱动的自主、多云 (AWS/Azure) 资产态势管理系统。

## 👨‍💻 作者 **设计与开发:** Vamshi Batthula **邮箱:** [batthulavamshi740@gmail.com](mailto:batthulavamshi740@gmail.com) ## 🚀 价值千万的 FAANG 架构 本项目从根本上重新架构了企业云安全的工作方式。它以解耦的、异步的 **Graph ML & GenAI 编排流水线**取代了手动安全审计,能够在有毒的云配置被利用之前,对其进行预测并自动修补。 ### 云安全面临的难题 手动扫描成千上万个 JSON/Terraform 文件是不可能的。将海量的基础设施即代码 状态直接发送给 LLM(如 GPT-4)会导致严重的幻觉,并产生巨额的 API token 成本。 ### RAG 与 Graph ML 解决方案 此架构通过拆分工作负载来实现**预测性零信任**: ``` graph TD A[Cloud Infrastructure JSON/Terraform] -->|Ingestion| B(Graph ML Engine) B -->|NetworkX Edge Traversal| C{Detects Toxic Combination?} C -->|No| D[Drop Event] C -->|Yes| E[Alert Triggered] E --> F[RAG Vector Database] F -->|Retrieves Enterprise Policy| G[GenAI Auto-Patcher] G -->|Generates IaC Fix| H[Deploy Zero-Trust Patch] H --> I((Slack Webhook Alert)) classDef secure fill:#10b981,stroke:#047857,stroke-width:2px,color:#fff; classDef alert fill:#ef4444,stroke:#b91c1c,stroke-width:2px,color:#fff; classDef ai fill:#8b5cf6,stroke:#6d28d9,stroke-width:2px,color:#fff; class H secure; class E alert; class B,G,F ai; ``` 1. 🕸️ **Graph ML 扫描器 (NetworkX / Scikit-Learn)** - 解析 AWS 和 Azure JSON payload,并将云基础设施转换为数学化的**有向图**。 - 瞬间遍历边缘以检测**毒性组合**(例如,一个直接挂载到 Admin IAM Role 的 Public S3 bucket)。 - 在本地 CPU 上免费过滤掉 99.9% 的正常网络配置。 2. 🧠 **RAG 安全引擎 (Vector Database)** - *仅在* Graph ML 标记出毒性组合时触发。 - 使用**检索增强生成 (RAG)** 查询本地 Vector DB,该数据库包含严格的企业级 AWS/Azure 安全基线。 - 通过将 prompt 锚定在已批准的企业策略上,来防止 LLM 幻觉。 3. ⚡ **GenAI 自动补丁器 (Langchain)** - 动态编写精确的基础设施即代码 补丁,以消除威胁。 - 自动生成 AWS Boto3 Lambda 函数或 Azure Management SDK 脚本,以强制执行**最小权限原则**。 ## 🖥️ 威胁图 UI 整个流水线通过 **FastAPI** 异步运行,并通过 **WebSockets** 将实时遥测数据推送到 **React** dashboard。这创建了一个“实时遥测”UI,安全团队可以在其中观看 Graph ML 检测漏洞以及 GenAI 部署补丁的实时过程。 ## 📁 仓库结构 * `main.py`:异步 FastAPI 网关、Graph ML 和 RAG 逻辑。 * `simulate_scans.py`:CI/CD 流水线模拟器(生成模拟的 AWS/Azure 基础设施 JSON)。 * `/dashboard`:React + Vite WebSockets 遥测 dashboard。 * `/terraform`:AWS ECS Fargate 部署脚本(`main.tf` 和 `Dockerfile`)。 * `INTERVIEW_MASTERCLASS.md`:一份详尽的指南,介绍如何在 Google、Meta、Microsoft、Accenture 和 Infosys 面试 AI 工程师、ML 工程师和云安全岗位时解释此架构。 ## ⚙️ 如何在本地运行 ### 1. 启动 FastAPI 后端 ``` python -m venv venv source venv/bin/activate # (or venv\Scripts\activate on Windows) pip install -r requirements.txt uvicorn main:app --port 8000 ``` ### 2. 启动 React Dashboard ``` cd dashboard npm install npm run dev -- --port 5174 ``` ### 3. 运行基础设施模拟器 ``` python simulate_scans.py ``` 打开 `http://localhost:5174` 即可观看 AI 引擎实时检测和修补漏洞。
标签:云安全态势管理, 前端应用, 图机器学习, 检索增强生成, 特权检测, 自动修复, 请求拦截, 逆向工具