package cmd
|
|
import (
|
"encoding/json"
|
"fmt"
|
"os"
|
"strings"
|
|
"github.com/aisim/kb-cli/internal/index"
|
"github.com/aisim/kb-cli/internal/llm"
|
"github.com/aisim/kb-cli/internal/search"
|
"github.com/spf13/cobra"
|
)
|
|
var (
|
exploreBudget int
|
exploreTopN int
|
exploreJSON bool
|
)
|
|
var exploreCmd = &cobra.Command{
|
Use: "explore <问题>",
|
Short: "一次调用获取相关文档原文 + 关联清单（供 agent 使用）",
|
Long: `# explore - 精准上下文
|
kb-cli explore <问题> [--budget 字节] [--top N] [--json] # 按字节预算返回相关文档原文、关联文档、悬空链接`,
|
Args: cobra.MinimumNArgs(1),
|
RunE: runExplore,
|
}
|
|
func init() {
|
rootCmd.AddCommand(exploreCmd)
|
exploreCmd.Flags().IntVar(&exploreBudget, "budget", 0, "字节预算（0=配置默认 16000）")
|
exploreCmd.Flags().IntVar(&exploreTopN, "top", 0, "文档数（0=配置默认 5）")
|
exploreCmd.Flags().BoolVar(&exploreJSON, "json", false, "JSON 输出")
|
|
// 参数不足时显示 help（与 search 命令同款模式）
|
exploreCmd.SetUsageTemplate(exploreCmd.Long)
|
}
|
|
func runExplore(cmd *cobra.Command, args []string) error {
|
// 打开索引
|
store, err := index.Open(dbPath)
|
if err != nil {
|
return fmt.Errorf("打开索引失败: %w", err)
|
}
|
defer store.Close()
|
|
// pre-flight：快速检查索引是否有变更（只 stat 比对，不读内容）
|
dirty, err := index.QuickCheck(store, vaultPath)
|
if err != nil {
|
return fmt.Errorf("检查索引状态失败: %w", err)
|
}
|
|
if dirty {
|
fmt.Fprintln(os.Stderr, "索引有变更，正在增量同步...")
|
if err := syncIndex(store); err != nil {
|
return fmt.Errorf("增量同步失败: %w", err)
|
}
|
}
|
|
// 关键词按词拆分（与 search 一致）：长复合词走 LIKE 整串匹配，文档里
|
// "电子秤补气失败" 通常不是连续子串，拆成 电子秤/补气失败 才能命中
|
var allKeywords []string
|
for _, arg := range args {
|
allKeywords = append(allKeywords, strings.Fields(arg)...)
|
}
|
if len(allKeywords) == 0 {
|
return fmt.Errorf("请提供至少一个关键词")
|
}
|
|
// 默认值优先读配置（config.yaml 的 explore 节），读不到再用代码缺省
|
budget := exploreBudget
|
topN := exploreTopN
|
cfg := llm.LoadConfig()
|
if budget <= 0 && cfg.Explore.DefaultBudget > 0 {
|
budget = cfg.Explore.DefaultBudget
|
}
|
if topN <= 0 && cfg.Explore.TopN > 0 {
|
topN = cfg.Explore.TopN
|
}
|
hardBudget := cfg.Explore.HardBudget // 0 = Explore 内部用缺省 32000
|
|
res, err := search.Explore(store, allKeywords, search.ExploreOptions{Budget: budget, TopN: topN, HardBudget: hardBudget})
|
if err != nil {
|
return fmt.Errorf("explore 失败: %w", err)
|
}
|
if len(res.Docs) == 0 {
|
fmt.Println("未找到相关文档")
|
return nil
|
}
|
|
if exploreJSON {
|
b, _ := json.MarshalIndent(res, "", " ")
|
fmt.Println(string(b))
|
return nil
|
}
|
for _, d := range res.Docs {
|
fmt.Printf("## %s（%s，score %d）\n", d.Title, d.Path, d.Score)
|
fmt.Print(d.Body)
|
fmt.Println()
|
if rel, ok := res.Related[d.Path]; ok && len(rel) > 0 {
|
fmt.Printf("关联: %s\n", strings.Join(rel, ", "))
|
}
|
fmt.Println()
|
}
|
if len(res.UnresolvedLinks) > 0 {
|
fmt.Printf("⚠️ 悬空链接: %s\n", strings.Join(res.UnresolvedLinks, ", "))
|
}
|
fmt.Println("以上为文档原文直出，agent 无需再读文件")
|
return nil
|
}
|
