通用知识库检索技能 (Retriever Skill)

概述

本技能用于从知识库中检索相关内容，支持：

• 向量检索: 基于Milvus的语义相似度检索
• 查询增强: 根据上下文增强查询效果
• 结果排序: 按相关性评分排序
• 过滤筛选: 支持文档类型、区域、行业等过滤条件

核心功能

1. 基础检索

from src.skills.skill_retriever import RetrieverSkill

skill = RetrieverSkill()
result = await skill.execute(
    query="环境影响评价报告编制要求",
    top_k=5,
)

if result.success:
    for item in result.data["results"]:
        print(f"- {item['title']} (score: {item['score']:.2f})")

2. 带上下文的检索

result = await skill.execute(
    query="大气污染防治措施",
    context={
        "chapter_id": "第七章",
        "keywords": ["煤炭", "矿区", "环评"],
        "project_type": "coal_mining",
    },
    top_k=10,
)

3. 带过滤条件的检索

result = await skill.execute(
    query="排放标准",
    filters={
        "document_type": "Regulation",
        "region": "宁夏",
        "status": "有效",
    },
    top_k=5,
)

参数说明

参数	类型	必需	默认值	说明
query	string	是	-	检索查询文本
context	object	否	{}	上下文信息（章节、关键词等）
top_k	integer	否	5	返回结果数量
filters	object	否	{}	过滤条件
layer	string	否	"all"	检索层级（L1/L2/L3/all）
project_id	string	否	-	项目ID（用于L3私有知识检索）

返回结果

{
  "success": true,
  "data": {
    "results": [
      {
        "id": "clause_001",
        "title": "环境空气质量标准",
        "content": "...",
        "document_name": "GB 3095-2012",
        "document_type": "Regulation",
        "score": 0.92
      }
    ],
    "query": "增强后的查询",
    "count": 5,
    "layer": "L1"
  }
}

相关技能

• skill-regulation-citation-extractor - 法规标准引用提取
• skill_regulatory_compliance_check - 合规性校验