Add files via upload

This commit is contained in:
公明
2026-07-07 14:04:18 +08:00
committed by GitHub
parent 3dfcde7c88
commit 00802a275c
61 changed files with 9460 additions and 24 deletions
+272
View File
@@ -0,0 +1,272 @@
# 知识库
知识库用于把本地安全知识、漏洞手册、测试方法和组织经验转成可检索上下文,供 Agent 在任务中按需引用。
## 启用
```yaml
knowledge:
enabled: true
base_path: knowledge_base
embedding:
provider: openai
model: text-embedding-v4
base_url: ""
api_key: ""
database:
knowledge_db_path: data/knowledge.db
```
`embedding.base_url/api_key` 留空时会复用 `openai` 配置。建议知识库数据库独立保存,便于迁移和复用。
## 内容目录
默认目录是 `knowledge_base/`。项目中已有示例:
```text
knowledge_base/
SQL Injection/
README.md
MySQL Injection.md
Prompt Injection/
README.md
```
推荐用一级目录表示风险类型或知识域,如:
- `SQL Injection`
- `XSS`
- `File Upload`
- `Cloud Security`
- `Incident Response`
## 管理流程
常见流程:
1. 把 Markdown 知识文件放到 `knowledge_base/`
2. 在 Web 知识库页面扫描目录。
3. 重建索引。
4. 用搜索功能验证召回效果。
5. 在角色或任务中要求 Agent 优先查询知识库。
接口入口包括:
- `GET /api/knowledge/categories`
- `GET /api/knowledge/items`
- `POST /api/knowledge/scan`
- `POST /api/knowledge/index`
- `POST /api/knowledge/search`
- `GET /api/knowledge/index-status`
- `GET /api/knowledge/retrieval-logs`
## 索引
索引配置:
```yaml
knowledge:
indexing:
chunk_size: 512
chunk_overlap: 50
max_chunks_per_item: 0
max_rpm: 0
rate_limit_delay_ms: 300
max_retries: 3
retry_delay_ms: 1000
chunk_strategy: markdown_then_recursive
request_timeout_seconds: 120
prefer_source_file: false
batch_size: 10
sub_indexes: []
```
建议:
- 文档结构清晰时用 `markdown_then_recursive`
- 嵌入接口限制严格时降低 `batch_size`,增加 `rate_limit_delay_ms`
- 单篇超长文档可设置 `max_chunks_per_item` 控制成本。
- 需要按业务域隔离时使用 `sub_indexes``sub_index_filter`
## 检索
```yaml
knowledge:
retrieval:
top_k: 5
similarity_threshold: 0.4
multi_query:
max_queries: 4
post_retrieve:
prefetch_top_k: 20
max_context_chars: 0
max_context_tokens: 0
```
检索链路大致为:
1. 用户查询或 Agent 查询。
2. MultiQuery 改写出多个语义变体。
3. 向量检索获取候选块。
4. rerank 精排。
5. 后处理去重、限长。
6. 返回给 Agent 或 API 调用方。
`similarity_threshold` 太高会漏召回,太低会带入噪声。初始建议 0.35 到 0.45。
## Rerank
```yaml
knowledge:
retrieval:
rerank:
provider: ""
model: ""
base_url: ""
api_key: ""
```
留空时会根据 `base_url` 推断。DashScope 常用 `gte-rerank`;其他 OpenAI 兼容端点可能走 `/v1/rerank`。如果服务商不支持 rerank,检索质量可能下降,建议降低 `top_k` 并提高知识条目质量。
## MCP 工具
启用知识库后,会注册类似以下能力:
- 列出风险类型。
- 搜索知识库。
- 获取相关知识片段。
角色提示词中可以写明:
```text
遇到漏洞验证、修复建议或检测方法不确定时,先检索知识库,再给出结论。
```
## 内容编写建议
每篇知识建议包含:
- 适用场景。
- 检测方法。
- 验证步骤。
- 常见误报。
- 修复建议。
- 工具命令示例。
- 参考链接或内部标准。
避免把无关主题堆在同一篇长文中。小而清晰的文档更利于 chunk 和召回。
## 排错
索引失败:
- 检查 embedding API Key、模型名、base_url。
- 降低 `batch_size`
- 增大 `request_timeout_seconds`
- 查看服务日志中的 400/401/429/5xx。
检索为空:
- 检查是否已重建索引。
- 降低 `similarity_threshold`
- 查看 `categories` 是否识别到风险类型。
- 搜索时不要使用过窄的 `riskType`
召回不准:
- 优化标题层级。
- 把混杂内容拆成多篇。
- 增加关键术语和同义词。
- 调整 `top_k``prefetch_top_k` 和 rerank 配置。
## 内部数据流
知识库链路不是“全文搜索”,而是一个多阶段检索系统:
```mermaid
flowchart LR
F["Markdown / Web 知识项"] --> M["Manager"]
M --> C["Chunker"]
C --> E["Embedding"]
E --> V["SQLite Vector Index"]
Q["Agent 查询"] --> MQ["MultiQuery 改写"]
MQ --> V
V --> R["Rerank"]
R --> P["Post-process 去重/限长"]
P --> A["Agent 上下文"]
```
因此检索质量取决于四件事:原文结构、chunk 粒度、embedding 质量、rerank 可用性。单纯调 `top_k` 往往不是最有效的办法。
## 知识项写作反例
不好的知识:
```text
SQL注入很危险,可以用sqlmap扫,修复就是过滤。
```
好的知识:
```markdown
# MySQL UNION 注入验证
## 触发条件
- 参数进入 SELECT 查询并直接拼接。
- 页面返回字段数量错误或类型错误。
## 验证步骤
1. 使用 `' order by 1-- -` 递增列数。
2. 使用 `union select null,...` 校验回显位。
3. 用只读函数确认数据库类型,例如 `database()`
## 误报排除
- WAF 注入拦截页可能模拟 SQL 错误。
- 统一错误页不能直接证明注入。
## 修复
- 参数化查询。
- 最小数据库权限。
- 统一错误处理但不吞掉安全日志。
```
第二种写法能给 chunk 足够的标题、术语和步骤信号,Agent 也能直接执行。
## 调参方法
先固定一组测试问题,例如:
```text
MySQL union 注入怎么判断字段数?
SSRF 如何验证云元数据访问?
文件上传黑名单绕过有哪些误报?
```
然后逐项调:
1. 搜索为空:降低 `similarity_threshold`,确认索引完成。
2. 结果主题错:提高文档标题质量,增加风险类型过滤。
3. 结果片段断裂:增大 `chunk_overlap` 或降低 `chunk_size` 后重建索引。
4. 噪声多:提高 `similarity_threshold`,启用/修复 rerank。
5. 成本高:降低 `multi_query.max_queries``prefetch_top_k``top_k`
每次只改一个参数,并记录查询结果,否则无法判断哪个变量有效。
## 检索日志怎么用
检索日志不只是排错用,还可以反向改进知识库:
- 高频无结果查询:说明缺知识或同义词不足。
- 高频低分查询:说明文档标题和术语不匹配。
- 同一问题召回多个重复文档:说明需要合并或加 category。
- Agent 常忽略知识库结果:说明结果太长、太散或缺明确结论。
## 源码锚点
- 知识管理:`internal/knowledge/manager.go`
- 索引流水线:`internal/knowledge/index_pipeline.go`
- Eino chunk`internal/knowledge/chunk_eino.go`
- 检索器:`internal/knowledge/retriever.go`
- Eino 检索链:`internal/knowledge/eino_retrieve_chain.go`
- rerank`internal/knowledge/rerank_http.go`
- MCP 工具:`internal/knowledge/tool.go`