mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-26 20:33:15 +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 树浏览（前端真正用起来，而非留一个没人调的端点）
面板加可折叠的分类树：逐级点选即把搜索范围切到该子树（原先是让人
手打分类名）。当前范围有可见标签与「全库」复位。
126 lines
2.8 KiB
Go
126 lines
2.8 KiB
Go
package sdk
|
|
|
|
import "gitcode.com/JianFeeeee/HomeAgent/internal/knowledge"
|
|
|
|
type knowledgeImpl struct{ ks *knowledge.Store }
|
|
|
|
func NewKnowledge(ks *knowledge.Store) KnowledgeAPI { return &knowledgeImpl{ks: ks} }
|
|
|
|
func (k *knowledgeImpl) Search(query string, topK int) ([]*Knowledge, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
got := k.ks.Search(query, topK)
|
|
out := make([]*Knowledge, len(got))
|
|
for i, item := range got {
|
|
out[i] = &Knowledge{Name: item.Name, Content: item.Content}
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) Add(name, content string) error {
|
|
if k.ks == nil {
|
|
return nil
|
|
}
|
|
return k.ks.Add(name, content)
|
|
}
|
|
|
|
func (k *knowledgeImpl) SearchIn(query, category string, topK int) ([]*Knowledge, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
got := k.ks.SearchIn(query, category, topK)
|
|
out := make([]*Knowledge, len(got))
|
|
for i, item := range got {
|
|
out[i] = &Knowledge{Name: item.Name, Content: item.Content}
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) Tree(opt KnowledgeTreeOptions) (*KnowledgeTreeView, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
return k.ks.Tree(knowledge.TreeOptions{
|
|
MaxDepth: opt.MaxDepth,
|
|
IncludeItems: opt.IncludeItems,
|
|
PreviewLimit: opt.PreviewLimit,
|
|
}), nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) Subtree(category string, opt KnowledgeTreeOptions) (*KnowledgeTreeView, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
return k.ks.Subtree(category, knowledge.TreeOptions{
|
|
MaxDepth: opt.MaxDepth,
|
|
IncludeItems: opt.IncludeItems,
|
|
PreviewLimit: opt.PreviewLimit,
|
|
}), nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) Categories() ([]string, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
return k.ks.Categories(), nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) CategoryCounts() ([]KnowledgeCategoryCount, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
return k.ks.CategoryCounts(), nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) AddWithMedia(name, content string, media []KnowledgeMediaRef) error {
|
|
if k.ks == nil {
|
|
return nil
|
|
}
|
|
return k.ks.AddWithMedia(name, content, media)
|
|
}
|
|
|
|
func (k *knowledgeImpl) AttachMedia(name string, media ...KnowledgeMediaRef) error {
|
|
if k.ks == nil {
|
|
return nil
|
|
}
|
|
return k.ks.AttachMedia(name, media...)
|
|
}
|
|
|
|
func (k *knowledgeImpl) ReindexDense() (int, int) {
|
|
if k.ks == nil {
|
|
return 0, 0
|
|
}
|
|
return k.ks.ReindexDense()
|
|
}
|
|
|
|
func (k *knowledgeImpl) DenseStats() map[string]interface{} {
|
|
if k.ks == nil {
|
|
return map[string]interface{}{}
|
|
}
|
|
return k.ks.DenseStats()
|
|
}
|
|
|
|
func (k *knowledgeImpl) List() ([]string, error) {
|
|
if k.ks == nil {
|
|
return nil, nil
|
|
}
|
|
return k.ks.List(), nil
|
|
}
|
|
|
|
func (k *knowledgeImpl) Stats() map[string]interface{} {
|
|
if k.ks == nil {
|
|
return map[string]interface{}{}
|
|
}
|
|
return k.ks.Stats()
|
|
}
|
|
|
|
func (k *knowledgeImpl) Remove(name string) error {
|
|
if k.ks == nil {
|
|
return nil
|
|
}
|
|
return k.ks.Remove(name)
|
|
}
|
|
|
|
var _ KnowledgeAPI = (*knowledgeImpl)(nil)
|