| | |
|---|
| | | "strings" |
|---|
| | | |
|---|
| | | "github.com/spf13/cobra" |
|---|
| | | "github.com/aisim/kb-cli/internal/llm" |
|---|
| | | ) |
|---|
| | | |
|---|
| | | var ( |
|---|
| | |
|---|
| | | |
|---|
| | | func init() { |
|---|
| | | homeDir, _ := os.UserHomeDir() |
|---|
| | | defaultVault := os.Getenv("KB_VAULT") |
|---|
| | | |
|---|
| | | // 从配置文件读取默认值（优先级：配置文件 > 环境变量 > 内置默认值） |
|---|
| | | cfg := llm.LoadConfig() |
|---|
| | | defaultVault := cfg.KnowledgeBase.VaultPath |
|---|
| | | defaultDB := cfg.KnowledgeBase.DBPath |
|---|
| | | |
|---|
| | | // 如果配置文件没有设置，使用环境变量或默认值 |
|---|
| | | if defaultVault == "" { |
|---|
| | | defaultVault = homeDir + "/aisim/note/001/笔记001" |
|---|
| | | defaultVault = os.Getenv("KB_VAULT") |
|---|
| | | if defaultVault == "" { |
|---|
| | | defaultVault = homeDir + "/aisim/note/001/笔记001" |
|---|
| | | } |
|---|
| | | } |
|---|
| | | defaultDB := homeDir + "/.cache/kb-cli/kb.db" |
|---|
| | | if defaultDB == "" { |
|---|
| | | defaultDB = homeDir + "/.cache/kb-cli/kb.db" |
|---|
| | | } |
|---|
| | | |
|---|
| | | rootCmd.PersistentFlags().StringVar(&vaultPath, "vault", defaultVault, "知识库根目录") |
|---|
| | | rootCmd.PersistentFlags().StringVar(&dbPath, "db", defaultDB, "索引数据库路径") |
|---|
