Vamshi868876/ZeroTrust-GenAI-Architect
GitHub: Vamshi868876/ZeroTrust-GenAI-Architect
基于图机器学习和 RAG 的多云零信任态势管理系统,能自动检测有毒云配置并生成修复补丁。
Stars: 0 | Forks: 0
## 👨💻 作者
**设计与开发:** 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 引擎实时检测和修补漏洞。
标签:云安全态势管理, 前端应用, 图机器学习, 检索增强生成, 特权检测, 自动修复, 请求拦截, 逆向工具