mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-03 15:53:56 +00:00
feat(kbtree): 知识库分类树的独立只读服务 + agent 技能 + WebUI 树浏览
让**外部 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 树浏览(前端真正用起来,而非留一个没人调的端点)
面板加可折叠的分类树:逐级点选即把搜索范围切到该子树(原先是让人
手打分类名)。当前范围有可见标签与「全库」复位。
This commit is contained in:
370
internal/plugins/kbtree/plugin.go
Normal file
370
internal/plugins/kbtree/plugin.go
Normal file
@ -0,0 +1,370 @@
|
||||
// Package kbtree 把知识库的**分类树**作为独立 HTTP 服务暴露给外部 agent。
|
||||
//
|
||||
// 定位(与 HomeAgent 内部知识工具的分工):
|
||||
// - HomeAgent 自己的 agent **直接调内部方法**(knowledge_search /
|
||||
// knowledge_create 等内核工具),走的是进程内直调,最快、也最少攻击面。
|
||||
// - 其它 agent(别的进程、别的机器、别的语言写的)走本插件的 HTTP 接口。
|
||||
//
|
||||
// 为何独立成服务而不是复用 WebUI 的 /api/v1/knowledge*:
|
||||
// 1. **不共享 WebUI 的鉴权与端口**。WebUI 的 api_key 是给人操作界面用的,
|
||||
// 把它分发给外部 agent 等于把管理面凭据扩散出去。本服务用**独立 token**
|
||||
// 且**独立端口**,可单独关闭。
|
||||
// 2. **只读**。外部 agent 读知识库就够了;写入要决定分类归属与媒体处理,
|
||||
// 让外部自行拼装反而容易造出越界/重名的条目 —— 写入留给内核工具。
|
||||
// 3. 形状按树组织(分类导航、分类内检索、懒加载),而不是平铺的搜索接口。
|
||||
//
|
||||
// 配套的 assets/skills/knowledge-base/SKILL.md 是给 agent 读的指令文档,
|
||||
// 两者配合即可"别的 agent 也能查这套知识库"。
|
||||
package kbtree
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gitcode.com/JianFeeeee/HomeAgent/internal/plugin"
|
||||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||||
)
|
||||
|
||||
// defaultAddr 默认只监听本机。外部 agent 若在别的机器,改配置显式对外,
|
||||
// 并自行确认 token 的分发方式(见 RegisterDef 里的说明)。
|
||||
const defaultAddr = "127.0.0.1:9892"
|
||||
|
||||
func init() {
|
||||
plugin.RegisterPluginMeta("kbtree", "知识库树服务", "Knowledge Tree Service")
|
||||
plugin.RegisterFactory("kbtree", func(name string, config map[string]interface{}) (sdk.Plugin, error) {
|
||||
return New(name), nil
|
||||
})
|
||||
}
|
||||
|
||||
type Plugin struct {
|
||||
name string
|
||||
sdkRef *sdk.PluginSDK
|
||||
addr string
|
||||
token string
|
||||
server *http.Server
|
||||
mux *http.ServeMux
|
||||
// 启动时未显式配置 token 则自动生成(与 remotedevice 同策略)
|
||||
generatedToken bool
|
||||
}
|
||||
|
||||
func New(name string) *Plugin {
|
||||
return &Plugin{
|
||||
name: name,
|
||||
addr: defaultAddr,
|
||||
mux: http.NewServeMux(),
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Plugin) Name() string { return p.name }
|
||||
|
||||
func (p *Plugin) Start(s *sdk.PluginSDK) error {
|
||||
if s == nil {
|
||||
return fmt.Errorf("kbtree: SDK 不可用")
|
||||
}
|
||||
p.sdkRef = s
|
||||
|
||||
// Settings 缺失时不能直接 RegisterDef —— 那是空接口上的调用,会 panic。
|
||||
// 缺 Settings 意味着宿主装配不完整(测试环境或裁剪版内核),
|
||||
// 此时用默认值把服务跑起来,而不是让整个插件加载失败。
|
||||
if set := s.Settings(); set != nil {
|
||||
set.RegisterDef(sdk.ConfigDef{
|
||||
Key: "listen_addr", Default: defaultAddr, Type: "string",
|
||||
DisplayName: "监听地址", Category: "kbtree",
|
||||
Description: "知识库树服务 HTTP 监听地址(默认 127.0.0.1:9892,仅本机)。改为对外地址前请确认 token 分发方式",
|
||||
})
|
||||
set.RegisterDef(sdk.ConfigDef{
|
||||
Key: "token", Default: "", Type: "password",
|
||||
DisplayName: "访问令牌", Category: "kbtree",
|
||||
Description: "外部 agent 访问本服务所需的令牌;留空则启动时随机生成(仅本次运行有效)",
|
||||
})
|
||||
if v, _ := set.Get("listen_addr"); v != nil {
|
||||
if a, ok := v.(string); ok && strings.TrimSpace(a) != "" {
|
||||
p.addr = strings.TrimSpace(a)
|
||||
}
|
||||
}
|
||||
if v, _ := set.Get("token"); v != nil {
|
||||
if tk, ok := v.(string); ok && strings.TrimSpace(tk) != "" {
|
||||
p.token = strings.TrimSpace(tk)
|
||||
}
|
||||
}
|
||||
}
|
||||
if p.token == "" {
|
||||
p.token = genToken()
|
||||
p.generatedToken = true
|
||||
}
|
||||
|
||||
// 启动前确认知识库可用:不可用就别占着端口
|
||||
kn := s.Knowledge()
|
||||
if kn == nil {
|
||||
return fmt.Errorf("kbtree: 知识库不可用,未启动服务(避免占端口后只回 503)")
|
||||
}
|
||||
|
||||
p.registerRoutes()
|
||||
p.server = &http.Server{
|
||||
Addr: p.addr,
|
||||
Handler: p.mux,
|
||||
ReadHeaderTimeout: 5 * time.Second, // Slowloris 防护
|
||||
WriteTimeout: 30 * time.Second,
|
||||
IdleTimeout: 60 * time.Second,
|
||||
}
|
||||
go func() {
|
||||
if err := p.server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
log.Printf("[kbtree] 服务错误: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
if p.generatedToken {
|
||||
log.Printf("[kbtree] 已启动 http://%s(token 未配置,本次随机生成;重启后失效)", p.addr)
|
||||
} else {
|
||||
log.Printf("[kbtree] 已启动 http://%s", p.addr)
|
||||
}
|
||||
log.Printf("[kbtree] 外部 agent 可用:GET /tree、/categories、/counts、/search?q=&category=")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Plugin) Stop() error {
|
||||
if p.server != nil {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
return p.server.Shutdown(ctx)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// registerRoutes 挂载只读路由。全部需 token。
|
||||
func (p *Plugin) registerRoutes() {
|
||||
p.mux.HandleFunc("/tree", p.requireToken(p.handleTree))
|
||||
p.mux.HandleFunc("/categories", p.requireToken(p.handleCategories))
|
||||
p.mux.HandleFunc("/counts", p.requireToken(p.handleCounts))
|
||||
p.mux.HandleFunc("/search", p.requireToken(p.handleSearch))
|
||||
// 根路径给个自述,便于外部 agent 摸索
|
||||
p.mux.HandleFunc("/", p.requireToken(p.handleRoot))
|
||||
}
|
||||
|
||||
func (p *Plugin) requireToken(next http.HandlerFunc) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
k := r.Header.Get("X-API-Key")
|
||||
if k == "" {
|
||||
k = r.Header.Get("Authorization")
|
||||
k = strings.TrimPrefix(k, "Bearer ")
|
||||
k = strings.TrimSpace(k)
|
||||
}
|
||||
if k == "" {
|
||||
k = r.URL.Query().Get("token")
|
||||
}
|
||||
if k == "" || k != p.token {
|
||||
writeJSON(w, http.StatusUnauthorized, map[string]string{"error": "unauthorized"})
|
||||
return
|
||||
}
|
||||
if r.Method != http.MethodGet {
|
||||
writeJSON(w, http.StatusMethodNotAllowed, map[string]string{"error": "只支持 GET(本服务只读)"})
|
||||
return
|
||||
}
|
||||
next(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
// handleRoot 自述端点:告诉调用方有哪些接口可用。
|
||||
func (p *Plugin) handleRoot(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path != "/" {
|
||||
writeJSON(w, http.StatusNotFound, map[string]string{"error": "not found"})
|
||||
return
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"service": "kbtree",
|
||||
"purpose": "HomeAgent 知识库分类树的只读访问(供外部 agent 使用)",
|
||||
"read_only": true,
|
||||
"endpoints": []string{
|
||||
"GET /tree?category=&depth=&items= 分类树(可指定子树/层数)",
|
||||
"GET /categories 全部分类路径(含中间层)",
|
||||
"GET /counts 各分类条目数(按数量倒序)",
|
||||
"GET /search?q=&category=&limit= 检索(category 前缀匹配子树)",
|
||||
},
|
||||
"auth": "X-API-Key 头 或 Authorization: Bearer <token> 或 ?token=",
|
||||
"notes": "结果按相关度排序,只采用第一条;节点 name 是本级段名,path 是完整路径",
|
||||
})
|
||||
}
|
||||
|
||||
// handleTree 返回分类树。
|
||||
func (p *Plugin) handleTree(w http.ResponseWriter, r *http.Request) {
|
||||
kn := p.knowledge(w)
|
||||
if kn == nil {
|
||||
return
|
||||
}
|
||||
q := r.URL.Query()
|
||||
opt := sdk.KnowledgeTreeOptions{
|
||||
MaxDepth: intQ(q.Get("depth"), 0),
|
||||
IncludeItems: boolQ(q.Get("items"), true),
|
||||
PreviewLimit: intQ(q.Get("preview"), 0),
|
||||
}
|
||||
cat := strings.Trim(strings.TrimSpace(q.Get("category")), "/")
|
||||
view, err := kn.Subtree(cat, opt)
|
||||
if err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
if view == nil {
|
||||
writeJSON(w, http.StatusNotFound, map[string]interface{}{
|
||||
"error": "分类不存在: " + cat,
|
||||
"categories": p.safeCategories(kn),
|
||||
})
|
||||
return
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{"tree": view})
|
||||
}
|
||||
|
||||
// handleCategories 返回平铺分类列表。
|
||||
func (p *Plugin) handleCategories(w http.ResponseWriter, r *http.Request) {
|
||||
kn := p.knowledge(w)
|
||||
if kn == nil {
|
||||
return
|
||||
}
|
||||
names := p.safeCategories(kn)
|
||||
if names == nil {
|
||||
names = []string{}
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{"categories": names})
|
||||
}
|
||||
|
||||
// handleCounts 返回各分类条目数(倒序)。
|
||||
func (p *Plugin) handleCounts(w http.ResponseWriter, r *http.Request) {
|
||||
kn := p.knowledge(w)
|
||||
if kn == nil {
|
||||
return
|
||||
}
|
||||
counts, err := kn.CategoryCounts()
|
||||
if err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
if counts == nil {
|
||||
counts = []sdk.KnowledgeCategoryCount{}
|
||||
}
|
||||
// 附总量:只数叶子分类,避免中间层重复计数
|
||||
total := 0
|
||||
byCat := make(map[string]int, len(counts))
|
||||
for _, c := range counts {
|
||||
byCat[c.Category] = c.Count
|
||||
}
|
||||
for c := range byCat {
|
||||
if !isParentCategory(counts, c) {
|
||||
total += byCat[c]
|
||||
}
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{"counts": counts, "total": total})
|
||||
}
|
||||
|
||||
// handleSearch 按分类 + 关键词检索。
|
||||
func (p *Plugin) handleSearch(w http.ResponseWriter, r *http.Request) {
|
||||
kn := p.knowledge(w)
|
||||
if kn == nil {
|
||||
return
|
||||
}
|
||||
q := r.URL.Query()
|
||||
query := strings.TrimSpace(q.Get("q"))
|
||||
if query == "" {
|
||||
writeJSON(w, http.StatusBadRequest, map[string]string{"error": "q 必填"})
|
||||
return
|
||||
}
|
||||
limit := intQ(q.Get("limit"), 10)
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 10
|
||||
}
|
||||
cat := strings.Trim(strings.TrimSpace(q.Get("category")), "/")
|
||||
results, err := kn.SearchIn(query, cat, limit)
|
||||
if err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
items := make([]map[string]interface{}, 0, len(results))
|
||||
for _, k := range results {
|
||||
items = append(items, map[string]interface{}{
|
||||
"name": k.Name,
|
||||
"content": k.Content,
|
||||
})
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"query": query,
|
||||
"category": cat,
|
||||
"results": items,
|
||||
"hint": "结果按相关度排序;只采用第一条,第一条不相关请换分类或关键词",
|
||||
})
|
||||
}
|
||||
|
||||
// knowledge 取知识库;不可用时写 503 并返回 nil。
|
||||
func (p *Plugin) knowledge(w http.ResponseWriter) sdk.KnowledgeAPI {
|
||||
if p.sdkRef == nil {
|
||||
writeJSON(w, http.StatusServiceUnavailable, map[string]string{"error": "plugin not started"})
|
||||
return nil
|
||||
}
|
||||
kn := p.sdkRef.Knowledge()
|
||||
if kn == nil {
|
||||
writeJSON(w, http.StatusServiceUnavailable, map[string]string{"error": "knowledge not available"})
|
||||
return nil
|
||||
}
|
||||
return kn
|
||||
}
|
||||
|
||||
func (p *Plugin) safeCategories(kn sdk.KnowledgeAPI) []string {
|
||||
names, err := kn.Categories()
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
return names
|
||||
}
|
||||
|
||||
// isParentCategory 判断 c 是否是别的分类的前缀(即中间层)。
|
||||
func isParentCategory(counts []sdk.KnowledgeCategoryCount, c string) bool {
|
||||
for _, o := range counts {
|
||||
if o.Category != c && strings.HasPrefix(o.Category, c+"/") {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func genToken() string {
|
||||
buf := make([]byte, 16)
|
||||
if _, err := rand.Read(buf); err != nil {
|
||||
return fmt.Sprintf("tok-%d", time.Now().UnixNano())
|
||||
}
|
||||
return hex.EncodeToString(buf)
|
||||
}
|
||||
|
||||
func writeJSON(w http.ResponseWriter, status int, v interface{}) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(status)
|
||||
_ = json.NewEncoder(w).Encode(v)
|
||||
}
|
||||
|
||||
func intQ(s string, def int) int {
|
||||
if s == "" {
|
||||
return def
|
||||
}
|
||||
n, err := strconv.Atoi(s)
|
||||
if err != nil {
|
||||
return def
|
||||
}
|
||||
return n
|
||||
}
|
||||
|
||||
func boolQ(s string, def bool) bool {
|
||||
if s == "" {
|
||||
return def
|
||||
}
|
||||
switch strings.ToLower(s) {
|
||||
case "1", "true", "yes", "on":
|
||||
return true
|
||||
case "0", "false", "no", "off":
|
||||
return false
|
||||
}
|
||||
return def
|
||||
}
|
||||
352
internal/plugins/kbtree/plugin_test.go
Normal file
352
internal/plugins/kbtree/plugin_test.go
Normal file
@ -0,0 +1,352 @@
|
||||
package kbtree
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"gitcode.com/JianFeeeee/HomeAgent/internal/knowledge"
|
||||
"gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||||
"gitcode.com/JianFeeeee/HomeAgent/internal/supervisor"
|
||||
"gitcode.com/JianFeeeee/HomeAgent/pkg/types"
|
||||
)
|
||||
|
||||
type rig struct {
|
||||
p *Plugin
|
||||
ks *knowledge.Store
|
||||
sup *supervisor.Daemon
|
||||
srv *httptest.Server
|
||||
tok string
|
||||
}
|
||||
|
||||
func newRig(t *testing.T) *rig {
|
||||
t.Helper()
|
||||
ks := knowledge.NewStore(t.TempDir())
|
||||
if err := ks.Start(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Cleanup(ks.Stop)
|
||||
for _, e := range []struct{ name, body string }{
|
||||
{"tech/go/并发", "goroutine 调度 GMP 抢占 通道"},
|
||||
{"tech/go/context", "context 取消 超时 传播"},
|
||||
{"tech/rust/所有权", "borrow checker move 语义"},
|
||||
{"life/sleep", "作息 褪黑素 深睡"},
|
||||
{"cook/coffee", "手冲 烘焙 水温"},
|
||||
} {
|
||||
if err := ks.Add(e.name, e.body); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
cfg := &types.Config{Daemon: types.DaemonConfig{
|
||||
CheckInterval: time.Minute, HeartbeatInterval: 30 * time.Second,
|
||||
}}
|
||||
sup := supervisor.New(cfg)
|
||||
sup.Start()
|
||||
t.Cleanup(sup.Shutdown)
|
||||
|
||||
s := sdk.New("test", sdk.SDKConfig{
|
||||
Supervisor: supervisor.NewSDKAdapter(sup),
|
||||
Knowledge: sdk.NewKnowledge(ks),
|
||||
Config: sdk.NewConfig(cfg),
|
||||
})
|
||||
// 让 Start 监听 :0(随机空闲端口),这样既走真实启动路径又不撞固定端口
|
||||
p := New("kbtree")
|
||||
p.addr = "127.0.0.1:0"
|
||||
if err := p.Start(s); err != nil {
|
||||
t.Fatalf("Start 失败: %v", err)
|
||||
}
|
||||
t.Cleanup(func() { _ = p.Stop() })
|
||||
|
||||
// 测试直接打插件的 mux(真实端口已由 :0 分配,无需再开 server)
|
||||
srv := httptest.NewServer(p.mux)
|
||||
t.Cleanup(srv.Close)
|
||||
|
||||
return &rig{p: p, ks: ks, sup: sup, srv: srv, tok: p.token}
|
||||
}
|
||||
|
||||
// get 带 token 请求。
|
||||
func (r *rig) get(t *testing.T, path string) (int, []byte) {
|
||||
t.Helper()
|
||||
req, err := http.NewRequest(http.MethodGet, r.srv.URL+path, nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
req.Header.Set("X-API-Key", r.tok)
|
||||
resp, err := r.srv.Client().Do(req)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
b, _ := io.ReadAll(resp.Body)
|
||||
return resp.StatusCode, b
|
||||
}
|
||||
|
||||
func (r *rig) getNoAuth(t *testing.T, path string) int {
|
||||
t.Helper()
|
||||
resp, err := r.srv.Client().Get(r.srv.URL + path)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
return resp.StatusCode
|
||||
}
|
||||
|
||||
// 无 token / 错 token 一律 401。
|
||||
func TestRequiresToken(t *testing.T) {
|
||||
r := newRig(t)
|
||||
if code := r.getNoAuth(t, "/tree"); code != http.StatusUnauthorized {
|
||||
t.Errorf("无 token 应 401,实为 %d", code)
|
||||
}
|
||||
// 错 token
|
||||
resp, err := r.srv.Client().Get(r.srv.URL + "/tree?token=wrong")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusUnauthorized {
|
||||
t.Errorf("错 token 应 401,实为 %d", resp.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
// token 也可用 ?token= 传(外部脚本友好)。
|
||||
func TestTokenViaQuery(t *testing.T) {
|
||||
r := newRig(t)
|
||||
resp, err := r.srv.Client().Get(r.srv.URL + "/tree?token=" + r.tok)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Errorf("?token= 应放行,实为 %d", resp.StatusCode)
|
||||
}
|
||||
// Bearer 也行
|
||||
req, _ := http.NewRequest(http.MethodGet, r.srv.URL+"/tree", nil)
|
||||
req.Header.Set("Authorization", "Bearer "+r.tok)
|
||||
resp2, err := r.srv.Client().Do(req)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer resp2.Body.Close()
|
||||
if resp2.StatusCode != http.StatusOK {
|
||||
t.Errorf("Bearer token 应放行,实为 %d", resp2.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
// 根路径自述端点。
|
||||
func TestRootSelfDescription(t *testing.T) {
|
||||
r := newRig(t)
|
||||
code, body := r.get(t, "/")
|
||||
if code != http.StatusOK {
|
||||
t.Fatalf("应 200,实为 %d", code)
|
||||
}
|
||||
if !strings.Contains(string(body), "read_only") {
|
||||
t.Errorf("自述里应标明只读: %s", body)
|
||||
}
|
||||
if !strings.Contains(string(body), "/tree") {
|
||||
t.Errorf("自述里应列出端点: %s", body)
|
||||
}
|
||||
}
|
||||
|
||||
// 树端点:结构、计数、懒加载。
|
||||
func TestTreeEndpoint(t *testing.T) {
|
||||
r := newRig(t)
|
||||
code, body := r.get(t, "/tree?items=0")
|
||||
if code != http.StatusOK {
|
||||
t.Fatalf("应 200,实为 %d: %s", code, body)
|
||||
}
|
||||
if !strings.Contains(string(body), `"total_count":5`) {
|
||||
t.Errorf("总条目应为 5: %s", body)
|
||||
}
|
||||
if !strings.Contains(string(body), `"tech"`) {
|
||||
t.Errorf("应含 tech 分类: %s", body)
|
||||
}
|
||||
// items=0 时无 preview
|
||||
if strings.Contains(string(body), `"preview"`) {
|
||||
t.Error("items=0 不应返回 preview")
|
||||
}
|
||||
|
||||
// 懒加载:depth=1 仍给出准确 total
|
||||
code, body = r.get(t, "/tree?depth=1&items=0")
|
||||
if code != http.StatusOK {
|
||||
t.Fatalf("depth=1 应 200,实为 %d", code)
|
||||
}
|
||||
if !strings.Contains(string(body), `"total_count":5`) {
|
||||
t.Errorf("剪枝不应改变总数: %s", body)
|
||||
}
|
||||
}
|
||||
|
||||
// 子树端点。
|
||||
func TestTreeSubtreeEndpoint(t *testing.T) {
|
||||
r := newRig(t)
|
||||
code, body := r.get(t, "/tree?category=tech/go&items=1")
|
||||
if code != http.StatusOK {
|
||||
t.Fatalf("应 200,实为 %d: %s", code, body)
|
||||
}
|
||||
var resp struct {
|
||||
Tree struct {
|
||||
Path string `json:"path"`
|
||||
Name string `json:"name"`
|
||||
ItemCount int `json:"item_count"`
|
||||
} `json:"tree"`
|
||||
}
|
||||
json.Unmarshal(body, &resp)
|
||||
if resp.Tree.Path != "tech/go" || resp.Tree.ItemCount != 2 {
|
||||
t.Errorf("tech/go 应 2 条,实为 %q/%d", resp.Tree.Path, resp.Tree.ItemCount)
|
||||
}
|
||||
if resp.Tree.Name != "go" {
|
||||
t.Errorf("name 应为本级段名 go,实为 %q", resp.Tree.Name)
|
||||
}
|
||||
}
|
||||
|
||||
// 不存在的分类:404 + 附现有分类。
|
||||
func TestTreeNotFound(t *testing.T) {
|
||||
r := newRig(t)
|
||||
code, body := r.get(t, "/tree?category=nope/here")
|
||||
if code != http.StatusNotFound {
|
||||
t.Fatalf("应 404,实为 %d", code)
|
||||
}
|
||||
if !strings.Contains(string(body), "categories") {
|
||||
t.Errorf("404 应附现有分类便于自查: %s", body)
|
||||
}
|
||||
}
|
||||
|
||||
// 检索端点:首位最相关、范围外不混入、缺 q 报 400。
|
||||
func TestSearchEndpoint(t *testing.T) {
|
||||
r := newRig(t)
|
||||
code, body := r.get(t, "/search?q=borrow&category=tech")
|
||||
if code != http.StatusOK {
|
||||
t.Fatalf("应 200,实为 %d: %s", code, body)
|
||||
}
|
||||
var resp struct {
|
||||
Results []struct {
|
||||
Name string `json:"name"`
|
||||
} `json:"results"`
|
||||
Hint string `json:"hint"`
|
||||
}
|
||||
json.Unmarshal(body, &resp)
|
||||
if len(resp.Results) == 0 {
|
||||
t.Fatal("应有命中")
|
||||
}
|
||||
if resp.Results[0].Name != "tech/rust/所有权" {
|
||||
t.Errorf("首位应最相关,实为 %q", resp.Results[0].Name)
|
||||
}
|
||||
for _, it := range resp.Results {
|
||||
if !strings.HasPrefix(it.Name, "tech/") {
|
||||
t.Errorf("范围外 %q 混入", it.Name)
|
||||
}
|
||||
}
|
||||
if resp.Hint == "" {
|
||||
t.Error("应带 hint")
|
||||
}
|
||||
// 缺 q
|
||||
if code, _ := r.get(t, "/search"); code != http.StatusBadRequest {
|
||||
t.Errorf("缺 q 应 400,实为 %d", code)
|
||||
}
|
||||
}
|
||||
|
||||
// 分类与计数端点。
|
||||
func TestCategoriesAndCountsEndpoint(t *testing.T) {
|
||||
r := newRig(t)
|
||||
_, body := r.get(t, "/categories")
|
||||
for _, want := range []string{"tech", "tech/go", "tech/rust", "life", "cook"} {
|
||||
if !strings.Contains(string(body), `"`+want+`"`) {
|
||||
t.Errorf("缺少分类 %s: %s", want, body)
|
||||
}
|
||||
}
|
||||
|
||||
_, body = r.get(t, "/counts")
|
||||
var resp struct {
|
||||
Counts []struct {
|
||||
Category string `json:"category"`
|
||||
Count int `json:"count"`
|
||||
} `json:"counts"`
|
||||
Total int `json:"total"`
|
||||
}
|
||||
json.Unmarshal(body, &resp)
|
||||
byCat := map[string]int{}
|
||||
for _, c := range resp.Counts {
|
||||
byCat[c.Category] = c.Count
|
||||
}
|
||||
if byCat["tech"] != 3 {
|
||||
t.Errorf("tech 应聚合 3 条,实为 %d", byCat["tech"])
|
||||
}
|
||||
if resp.Total != 5 {
|
||||
t.Errorf("total 只数叶子应为 5,实为 %d", resp.Total)
|
||||
}
|
||||
}
|
||||
|
||||
// 只读:写方法一律 405,且不改变知识库。
|
||||
func TestReadOnly(t *testing.T) {
|
||||
r := newRig(t)
|
||||
before := len(r.ks.List())
|
||||
for _, m := range []string{http.MethodPost, http.MethodPut, http.MethodDelete, http.MethodPatch} {
|
||||
req, _ := http.NewRequest(m, r.srv.URL+"/tree", nil)
|
||||
req.Header.Set("X-API-Key", r.tok)
|
||||
resp, err := r.srv.Client().Do(req)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusMethodNotAllowed {
|
||||
t.Errorf("%s 应 405,实为 %d", m, resp.StatusCode)
|
||||
}
|
||||
}
|
||||
if len(r.ks.List()) != before {
|
||||
t.Error("只读服务不应改变知识库")
|
||||
}
|
||||
}
|
||||
|
||||
// 知识库不可用时 Start 应失败(不占端口)。
|
||||
func TestStartFailsWithoutKnowledge(t *testing.T) {
|
||||
cfg := &types.Config{Daemon: types.DaemonConfig{CheckInterval: time.Minute, HeartbeatInterval: 30 * time.Second}}
|
||||
sup := supervisor.New(cfg)
|
||||
sup.Start()
|
||||
defer sup.Shutdown()
|
||||
s := sdk.New("test", sdk.SDKConfig{
|
||||
Supervisor: supervisor.NewSDKAdapter(sup),
|
||||
Config: sdk.NewConfig(cfg),
|
||||
})
|
||||
if err := New("kbtree").Start(s); err == nil {
|
||||
t.Error("无知识库时 Start 应失败")
|
||||
}
|
||||
}
|
||||
|
||||
// 技能文档里的端点路径必须与代码一致(防止文档漂移)。
|
||||
func TestSkillDocEndpointsMatchCode(t *testing.T) {
|
||||
doc, err := readSkillDoc()
|
||||
if err != nil {
|
||||
t.Skipf("读不到 skill 文档: %v", err)
|
||||
}
|
||||
for _, ep := range []string{"/tree", "/categories", "/counts", "/search"} {
|
||||
if !strings.Contains(doc, ep) {
|
||||
t.Errorf("SKILL.md 未提及端点 %s(文档与代码漂移)", ep)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func readSkillDoc() (string, error) {
|
||||
// 测试在 internal/plugins/kbtree,skill 在 ../../assets/skills
|
||||
paths := []string{
|
||||
filepath.Join("..", "..", "assets", "skills", "knowledge-base", "SKILL.md"),
|
||||
filepath.Join("..", "..", "..", "assets", "skills", "knowledge-base", "SKILL.md"),
|
||||
}
|
||||
var lastErr error
|
||||
for _, p := range paths {
|
||||
if b, err := readFile(p); err == nil {
|
||||
return string(b), nil
|
||||
} else {
|
||||
lastErr = err
|
||||
}
|
||||
}
|
||||
return "", lastErr
|
||||
}
|
||||
|
||||
func readFile(p string) ([]byte, error) {
|
||||
return os.ReadFile(p)
|
||||
}
|
||||
Reference in New Issue
Block a user