From 1ba86b102ecc01de3191c3861592d9d4d22dfd81 Mon Sep 17 00:00:00 2001
From: ai_xiaopei <xiaopei@aisim.cn>
Date: Sun, 26 Jul 2026 15:17:48 +0800
Subject: [PATCH] feat: 添加 classify 命令和实体提取功能
---
cmd/root.go | 17 ++++++++++++++---
1 files changed, 14 insertions(+), 3 deletions(-)
diff --git a/cmd/root.go b/cmd/root.go
index fca90d1..c23baff 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -35,9 +35,9 @@
}
var rootCmd = &cobra.Command{
- Use: "kb",
+ Use: "kb-cli",
Short: "知识库搜索与管理工具",
- Long: `kb - 知识库搜索与管理工具
+ Long: `kb-cli - 知识库搜索与管理工具
Global: [--vault=<路径>] [--db=<路径>]`,
}
@@ -63,7 +63,18 @@
rootCmd.Long = sb.String()
if err := rootCmd.Execute(); err != nil {
- cobra.CheckErr(err)
+ // 如果是参数不足的错误，显示对应命令的 help
+ errMsg := err.Error()
+ if strings.Contains(errMsg, "arg(s)") || strings.Contains(errMsg, "received") || strings.Contains(errMsg, "required flag") {
+ // 找到当前命令并显示其 help
+ cmd, _, findErr := rootCmd.Find(os.Args[1:])
+ if findErr == nil {
+ cmd.Help()
+ }
+ fmt.Println() // 空行
+ } else {
+ cobra.CheckErr(err)
+ }
}
}
--
Gitblit v1.9.1