SuperAPI AI 工具集

byted-viking-search-knowledgebase

编程与开发

查看项目

更新于 10-15 17:30

技能介绍

用途:通过 APIG 网关调用火山引擎 Viking 知识库 API,检索与 query 相关的知识库数据,适用于查询知识库内容、获取相关文档、检索特定信息等需要搜索数据回答问题的场景。

主要能力:info 查看知识库详情,返回 collection_name、description 等元数据,仅用于连接检查或路由决策;search 对已确定知识库做语义检索,返回切片列表、相关度分数、文档信息;auto 对多个有权限知识库并行轻量检索,是知识问答的默认入口。

用法:脚本为 scripts/viking_search.py,支持三种动作,如 python scripts/viking_search.py --action info --resource-id <collection_resource_id>、python scripts/viking_search.py --action search --resource-id <resource_id> --query "关键词1 关键词2 关键词3" --limit 10、python scripts/viking_search.py --action auto --query "关键词1 关键词2"。检索前应把长任务描述拆分为多个独立子意图,每个子意图压缩为 2~5 个关键词的短 query,并在同一轮内并行发起;不要把任务原文或多个子意图拼成一个长 query,不应用完整疑问句。

适用场景:基于知识库回答具体问题默认走 auto;任务已指定具体 resource_id 或 name 时用 search;任务意图为连接检查或列出知识库时用 info;任务无主题关键词时可先 info 列出再请求澄清。

限制:脚本需 Python 3.7+ 与 requests 库,API 地址与鉴权 Key 在执行环境预配置,不应对用户暴露。可访问范围由环境变量 DATABASE_VIKING_COLLECTION 限定,查询列表外的库会报无权限;collection_name 一般为英文/拼音/ID,不要用中文名或飞书 Wiki 目录名当 --name。召回不足时应重写关键词、进一步拆分或换库,不要盲目放大 --limit。环境配置缺失时需向上级回报,建议联系管理员。