feat: add cli kb

Author: yaozheng-fang

docs/content/90.cli/1.overview.md

@@ -14,3 +14,4 @@ VeADK 提供如下命令便捷您的操作：
 | `veadk prompt` | 优化智能体系统提示词 | 借助火山引擎 PromptPilot 产品 |
 | `veadk web` | 支持长短期记忆、知识库的前端调试界面 | 兼容 Google ADK web |
 | `veadk eval` | 支持不同后端的评测 | 评测后端包括 `adk` 与 `deepeval`，评测数据集源包括 Google ADK 评测集格式文件，以及 Tracing 文件 |
+| `veadk kb` | 知识库相关操作 | 向知识库添加本地文件或目录 |

docs/content/90.cli/2.commands.md

@@ -84,3 +84,45 @@ veadk web --session_service_uri="mysql+pymysql://{user}:{password}@{host}/{datab
   火山引擎 Secret Key
   ::
 ::
+
+## 知识库操作
+
+您可以通过 `veadk kb add` 命令来向您的知识库添加本地文件或者目录。例如，准备如下目录：
+
+::code-tree{default-value="knowledges/id.txt"}
+
+```txt [knowledges/id.txt]
+My id is 20250101.
+```
+
+::
+
+之后，将 `knowledges/id.txt` 中的内容存入到 OpenSearch 知识库中：
+
+```bash
+# 知识库的相关配置将会从环境变量中读取
+veadk kb add --backend opensearch --app_name=cmd_app --path=./knowledges
+```
+
+可使用如下代码测试：
+
+```python
+import asyncio
+
+from veadk import Agent, Runner
+from veadk.knowledgebase import KnowledgeBase
+
+app_name = "cmd_app"
+
+knowledgebase = KnowledgeBase(backend="opensearch", app_name=app_name)
+
+agent = Agent(knowledgebase=knowledgebase)
+
+runner = Runner(agent=agent, app_name=app_name)
+
+response = asyncio.run(
+    runner.run(messages="Please help me to fetch my ID from my knowledgebase")
+)
+
+print(response) # Your ID is 20250101.
+```

veadk/cli/cli.py

@@ -18,6 +18,7 @@
 from veadk.cli.cli_deploy import deploy
 from veadk.cli.cli_eval import eval
 from veadk.cli.cli_init import init
+from veadk.cli.cli_kb import kb
 from veadk.cli.cli_pipeline import pipeline
 from veadk.cli.cli_prompt import prompt
 from veadk.cli.cli_web import web
@@ -39,6 +40,7 @@ def veadk():
 veadk.add_command(web)
 veadk.add_command(pipeline)
 veadk.add_command(eval)
+veadk.add_command(kb)
 
 if __name__ == "__main__":
     veadk()

veadk/cli/cli_kb.py

@@ -0,0 +1,75 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pathlib import Path
+from typing import Literal
+
+import click
+
+
+@click.command()
+@click.option(
+    "--backend",
+    type=click.Choice(
+        ["local", "opensearch", "viking", "redis"],
+        case_sensitive=False,
+    ),
+    required=True,
+)
+@click.option(
+    "--app_name",
+    default="",
+    help="`app_name` for init your knowledgebase",
+)
+@click.option(
+    "--index",
+    default="",
+    help="Knowledgebase index",
+)
+@click.option(
+    "--path",
+    required=True,
+    help="Knowledge file or directory path",
+)
+def add(
+    backend: Literal["local", "opensearch", "viking", "redis"],
+    app_name: str,
+    index: str,
+    path: str,
+):
+    """Add files to knowledgebase"""
+    _path = Path(path)
+    assert _path.exists(), f"Path {path} not exists. Please check your input."
+
+    from veadk.knowledgebase import KnowledgeBase
+
+    knowledgebase = KnowledgeBase(backend=backend, app_name=app_name, index=index)
+
+    if _path.is_file():
+        knowledgebase.add_from_files(files=[path])
+    elif _path.is_dir():
+        knowledgebase.add_from_directory(directory=path)
+    else:
+        raise RuntimeError(
+            "Unsupported knowledgebase file type, only support a single file and a directory."
+        )
+
+
+@click.group()
+def kb():
+    """VeADK Knowledgebase management"""
+    pass
+
+
+kb.add_command(add)