mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-26 12:23:23 +00:00
41d754334e
让**外部 agent** 也能按分类树用这套知识库。HomeAgent 自己的 agent 仍
直接调内部方法（knowledge_search/create 等），走进程内直调，不经此服务。
一、内核树视图（internal/knowledge/tree.go）
为什么不复用 TreeIndex：那个是**内部导出物**，面向 .index.json 落盘，
每个条目带 top-20 的 TF-IDF 特征向量。直接序列化给外部有三个问题：
体积（200 条时 .index.json 已 246KB 且冗余存了 preview，而正本在
content.md）、泄漏（稀疏特征表 = 分词/IDF 内部表示）、语义错位
（外部要的是"有哪些分类、每类下有什么"）。
新增 TreeView/Subtree/Categories/CategoryCounts：不含向量，带条目数
与可读摘要，支持 MaxDepth 懒加载、IncludeItems 只看结构。
节点 Name 是**本级段名**（"go"）、Path 是完整路径（"tech/go"）——
最初把全路径写进 Name，前端拼层级会得到 "tech/tech/go"，已修。
二、kbtree 插件：独立 HTTP 服务（默认 127.0.0.1:9892）
为何不挂在 WebUI 的 /api/v1/knowledge* 下：
1. 不共享鉴权与端口。WebUI 的 api_key 是给人操作界面用的，把它分发给
外部 agent 等于把管理面凭据扩散出去。本服务用**独立 token** +
独立端口，可单独关闭（token 未配置则启动时随机生成）。
2. 只读。写入要决定分类归属与媒体处理，外部自行拼装容易造出越界/重名
条目 —— 写入留给内核工具。
3. 形状按树组织，而不是平铺搜索接口。
端点：/tree（可指定 category/depth/items）、/categories、/counts、
/search、/ （自述）。全部需 token（X-API-Key / Bearer / ?token=），
非 GET 一律 405。无知识库时 Start 直接失败，不占端口。
鉴权与 Slowloris/超时设置照 remotedevice 范式。
三、agent 技能（assets/skills/knowledge-base/SKILL.md）
指令文档型 skill：教模型"先看树 → 定位分类 → 分类内检索"，并列出
易错点（name 已含分类别再拼、只看第一条、404 附现有分类）。
加载与校验由 internal/plugin/skill_bundled_test.go 守住 —— 这条断言
的由来：非白名单的二级标题会被 extractToolDefs 当成工具定义，报错
"invalid tool name"，而提示与真正原因（标题层级）毫无关联。
kbtree 的测试还会校验文档提到的端点与代码一致，防漂移。
四、WebUI 树浏览（前端真正用起来，而非留一个没人调的端点）
面板加可折叠的分类树：逐级点选即把搜索范围切到该子树（原先是让人
手打分类名）。当前范围有可见标签与「全库」复位。
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package plugin
|
||
|
||
import (
|
||
"path/filepath"
|
||
"testing"
|
||
)
|
||
|
||
// 随发行版分发的 knowledge-base skill 必须能被 LoadSKILL 正确解析。
|
||
//
|
||
// 这条断言的由来：SKILL.md 里**非白名单的二级标题（## ）会被
|
||
// extractToolDefs 当成工具定义**，进而 ValidateSKILLContent 报
|
||
// "invalid tool name"。写文档时用中文小标题（"## 前提：…"）就会踩到，
|
||
// 报错的提示与真正的原因（标题层级）毫无关联，极难自查。
|
||
// 故把「bundled skill 必须可加载且可校验」固化成测试。
|
||
func TestBundledSkillsAreLoadable(t *testing.T) {
|
||
root := filepath.Join("..", "..", "assets", "skills")
|
||
dirs, err := filepath.Glob(filepath.Join(root, "*"))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(dirs) == 0 {
|
||
t.Skip("没有随发行版分发的 skill")
|
||
}
|
||
for _, dir := range dirs {
|
||
name := filepath.Base(dir)
|
||
t.Run(name, func(t *testing.T) {
|
||
sk, err := LoadSKILL(dir)
|
||
if err != nil {
|
||
t.Fatalf("LoadSKILL 失败: %v", err)
|
||
}
|
||
if sk.Name() == "" {
|
||
t.Error("skill 名为空")
|
||
}
|
||
if sk.Description() == "" || sk.Description() == "---" {
|
||
t.Errorf("description 解析失败（frontmatter 未被正确跳过）: %q", sk.Description())
|
||
}
|
||
if err := ValidateSKILLContent(sk.RawContent()); err != nil {
|
||
t.Errorf("ValidateSKILLContent 失败: %v", err)
|
||
}
|
||
})
|
||
}
|
||
}
|