brunoandClaude
842fa8dbe5
本次提交包含完整的AI SOC（Security Operations Center）分析师平台，支持本地大模型和Claude API集成，实现端到端的安全威胁分析。 ## 🚀 主要功能 ### 后端架构 (FastAPI) - **多源数据接入**: 支持JSON/CSV格式安全告警数据上传 - **智能预处理**: 基于YAML配置的字段映射和实体提取 - **图分析引擎**: NetworkX驱动的攻击链图构建和分析 - **评分系统**: 实体和子图的多维度风险评分算法 - **LLM集成**: 支持多提供商(Claude、OpenAI、本地模型)的智能报告生成 - **API接口**: 完整的RESTful API (Swagger文档: /docs) ### 前端界面 (Vue 3 + TypeScript) - **拖拽上传**: 支持拖拽上传和实时预览 - **管道分析**: 可视化的数据处理管道状态 - **智能报告**: 动态生成的威胁分析报告和可视化 - **响应式设计**: 适配桌面和移动端的现代UI ### LLM智能分析 - **多模型支持**: qwen3(8B/32B)、deepseek-r1(70B)、Claude 4等 - **超时优化**: 分层超时配置(120s-600s)适配不同模型规模 - **内容清理**: 自动过滤模型思考过程，输出专业分析报告 - **中文优化**: 针对中文安全分析场景的prompt工程 ## 🛠️ 技术栈 **后端**: FastAPI + SQLAlchemy + NetworkX + 多LLM提供商 **前端**: Vue 3 + TypeScript + Naive UI + Tailwind CSS + Vite **数据处理**: YAML配置 + JSONPath表达式 **部署**: Docker支持 + 环境变量配置 ## 🐛 修复的关键问题 1. **LLM超时问题**: 实现前后端分层超时配置，支持大模型长时间推理 2. **报告格式化**: 修复Markdown渲染和换行符显示问题 3. **数据映射**: 统一前后端字段命名和数据结构映射 4. **时间格式**: 修复时间戳解析和中文本地化显示 5. **思考标签清理**: 自动过滤qwen模型的<think>标签 ## 📊 测试数据 包含完整的多阶段APT攻击场景测试数据: - 钓鱼邮件 → 恶意软件执行 → C2通信 - 云凭证窃取 → 资源探测 → 横向渗透 - 后门用户创建 → 权限提升 → 容器控制 ## 🔧 开发工具 - **模型切换**: switch_model.py 支持快速切换本地LLM模型 - **超时测试**: test_llm_timeout.py 验证不同模型性能 - **依赖安装**: install_llm_deps.py 自动安装LLM依赖 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
42 lines
1.3 KiB
Bash
42 lines
1.3 KiB
Bash
# AI SOC Analyst 环境变量配置示例
|
|
|
|
# =============================================================================
|
|
# LLM API Keys - 根据你使用的模型提供商配置相应的API Key
|
|
# =============================================================================
|
|
|
|
# Claude 4 API Key
|
|
CLAUDE_API_KEY=sk-ant-api03-your-claude-api-key-here
|
|
|
|
# OpenAI API Key
|
|
OPENAI_API_KEY=sk-proj-your-openai-api-key-here
|
|
|
|
# 自定义API Key (可选)
|
|
CUSTOM_API_KEY=your-custom-api-key
|
|
|
|
# =============================================================================
|
|
# 本地模型配置
|
|
# =============================================================================
|
|
|
|
# 如果你使用Ollama运行本地模型，确保Ollama服务正在运行
|
|
# ollama serve (默认端口11434)
|
|
|
|
# 常用的本地模型名称示例：
|
|
# - qwen2.5:7b
|
|
# - llama3.1:8b
|
|
# - mistral:7b
|
|
# - gemma2:9b
|
|
# - deepseek-coder:6.7b
|
|
|
|
# =============================================================================
|
|
# 数据库配置
|
|
# =============================================================================
|
|
|
|
DATABASE_URL=sqlite:///./ai_soc.db
|
|
|
|
# =============================================================================
|
|
# 服务器配置
|
|
# =============================================================================
|
|
|
|
HOST=0.0.0.0
|
|
PORT=8000
|
|
DEBUG=true |