Gabi-comm/Network-Log-RAG-Chatbot

GitHub: Gabi-comm/Network-Log-RAG-Chatbot

基于 RAG 架构的网络日志智能分析系统,通过向量数据库和本地 LLM 实现自然语言式的安全日志查询与异常检测。

Stars: 0 | Forks: 0

# Network-Log-RAG-Chatbot 一个智能的日志分析系统,使用 embeddings 将网络日志索引到向量数据库(ChromaDB)中,允许本地语言模型聊天机器人(Ollama)进行查询、分析和检测安全异常。 # 结合 LLM 聊天机器人的嵌入式网络日志模型 一个智能的、隐私优先的网络日志分析系统。本项目实现了一个检索增强生成(RAG)pipeline,将原始的网络安全日志进行 embedding,存储在向量数据库中,并结合本地的大型语言模型(LLM)聊天机器人,以进行交互式的、自然语言的安全分析。 ## 功能 * **日志 Embedding 与摄取:** 使用 Hugging Face embedding 模型将非结构化的网络日志转换为密集的向量表示。 * **向量存储:** 将 embedded 的日志数据存储在本地,以便进行快速的语义相似度搜索。 * **本地 LLM 集成:** 使用开源的本地语言模型查询 embedded 的日志,以确保数据的隐私和安全。 * **交互式聊天界面:** 允许安全分析师使用自然语言查询复杂的网络行为、追踪异常并生成事件摘要。 ## 技术栈 * **语言:** Python 3.14+ * **框架:** LangChain * **Embeddings:** Hugging Face (`langchain-huggingface`) * **向量数据库:** ChromaDB * **本地 LLM 引擎:** Ollama ## 项目结构 本项目被组织成模块化的脚本,代表了检索增强生成(RAG)pipeline 的不同阶段: * **`HR_Policies.txt`**:包含模拟策略信息和隐藏 PII 的原始文本数据集。 * **`redact.py`**:预处理脚本,用于扫描原始文本、检测敏感信息(如 SSN 或电子邮件)并将其清除。 * **`embed.py`**:加载脱敏后的数据,将其分割成易于处理的文本块,生成向量 embeddings,并将它们存储在 ChromaDB 中。 * **`retriever.py`**:处理语义搜索逻辑——接收用户查询,并从向量数据库中获取最相关的文档块。 * **`chatbot.py`**:最终的执行脚本,封装了 retriever 并将其与 Ollama 相连,提供一个交互式的自然语言聊天界面。
标签:AI风险缓解, DLL 劫持, IaC 扫描, 向量数据库, 大语言模型, 检索增强生成, 逆向工具