diff --git a/backend/package/yuxi/knowledge/__init__.py b/backend/package/yuxi/knowledge/__init__.py
index 549a4990..7a215e55 100644
--- a/backend/package/yuxi/knowledge/__init__.py
+++ b/backend/package/yuxi/knowledge/__init__.py
@@ -12,10 +12,10 @@ _SKIP_APP_INIT = os.environ.get("YUXI_SKIP_APP_INIT") == "1"
if not _LITE_MODE:
# 注册知识库类型
- KnowledgeBaseFactory.register("milvus", MilvusKB, {"description": "基于 Milvus 的生产级向量知识库，适合高性能部署"})
+ KnowledgeBaseFactory.register(MilvusKB)
-KnowledgeBaseFactory.register("dify", DifyKB, {"description": "连接 Dify Dataset 的只读检索知识库"})
-KnowledgeBaseFactory.register("notion", NotionKB, {"description": "连接 Notion Data Source 的只读知识库，支持检索、打开页面和页内查找"})
+KnowledgeBaseFactory.register(DifyKB)
+KnowledgeBaseFactory.register(NotionKB)
# 创建知识库管理器
work_dir = os.path.join(config.save_dir, "knowledge_base_data")
diff --git a/backend/package/yuxi/knowledge/base.py b/backend/package/yuxi/knowledge/base.py
index 5cf12f0f..2bb7d03d 100644
--- a/backend/package/yuxi/knowledge/base.py
+++ b/backend/package/yuxi/knowledge/base.py
@@ -43,6 +43,9 @@ class KBOperationError(KnowledgeBaseException):
class KnowledgeBase(ABC):
"""知识库抽象基类，定义统一接口"""
+ kb_type = ""
+ name = ""
+ description = ""
requires_embedding_model = True
supports_documents = True
apply_chunk_defaults = True
@@ -160,12 +163,6 @@ class KnowledgeBase(ABC):
if normalized:
b["updated_at"] = normalized
- @property
- @abstractmethod
- def kb_type(self) -> str:
- """知识库类型标识"""
- pass
-
@classmethod
def get_create_params_config(cls) -> dict[str, Any]:
"""获取创建知识库时的类型特定参数配置。"""
diff --git a/backend/package/yuxi/knowledge/factory.py b/backend/package/yuxi/knowledge/factory.py
index da2bc13d..4e708a94 100644
--- a/backend/package/yuxi/knowledge/factory.py
+++ b/backend/package/yuxi/knowledge/factory.py
@@ -8,25 +8,21 @@ class KnowledgeBaseFactory:
# 注册的知识库类型映射 {kb_type: kb_class}
_kb_types: dict[str, type[KnowledgeBase]] = {}
- # 每种类型的默认配置
- _default_configs: dict[str, dict] = {}
-
@classmethod
- def register(cls, kb_type: str, kb_class: type[KnowledgeBase], default_config: dict = None):
+ def register(cls, kb_class: type[KnowledgeBase]):
"""
注册知识库类型
Args:
- kb_type: 知识库类型标识
kb_class: 知识库类
- default_config: 默认配置
"""
if not issubclass(kb_class, KnowledgeBase):
raise ValueError("Knowledge base class must inherit from KnowledgeBase")
+ if not kb_class.kb_type:
+ raise ValueError("Knowledge base class must define kb_type")
- cls._kb_types[kb_type] = kb_class
- cls._default_configs[kb_type] = default_config or {}
- # logger.info(f"Registered knowledge base type: {kb_type}")
+ cls._kb_types[kb_class.kb_type] = kb_class
+ # logger.info(f"Registered knowledge base type: {kb_class.kb_type}")
@classmethod
def create(cls, kb_type: str, work_dir: str, **kwargs) -> KnowledgeBase:
@@ -50,13 +46,9 @@ class KnowledgeBaseFactory:
kb_class = cls._kb_types[kb_type]
- # 合并默认配置和用户配置
- config = cls._default_configs[kb_type].copy()
- config.update(kwargs)
-
try:
# 创建实例
- instance = kb_class(work_dir, **config)
+ instance = kb_class(work_dir, **kwargs)
logger.info(f"Created {kb_type} knowledge base instance at {work_dir}")
return instance
except Exception as e:
@@ -74,9 +66,8 @@ class KnowledgeBaseFactory:
result = {}
for kb_type, kb_class in cls._kb_types.items():
result[kb_type] = {
- "class_name": kb_class.__name__,
- "description": kb_class.__doc__ or "",
- "default_config": cls._default_configs[kb_type],
+ "name": kb_class.name,
+ "description": kb_class.description,
"requires_embedding_model": kb_class.requires_embedding_model,
"supports_documents": kb_class.supports_documents,
"create_params": kb_class.get_create_params_config(),
@@ -111,16 +102,3 @@ class KnowledgeBaseFactory:
是否支持
"""
return kb_type in cls._kb_types
-
- @classmethod
- def get_default_config(cls, kb_type: str) -> dict:
- """
- 获取指定类型的默认配置
-
- Args:
- kb_type: 知识库类型
-
- Returns:
- 默认配置字典
- """
- return cls._default_configs.get(kb_type, {}).copy()
diff --git a/backend/package/yuxi/knowledge/implementations/dify.py b/backend/package/yuxi/knowledge/implementations/dify.py
index a9c37626..261ffc46 100644
--- a/backend/package/yuxi/knowledge/implementations/dify.py
+++ b/backend/package/yuxi/knowledge/implementations/dify.py
@@ -12,14 +12,14 @@ DIFY_REQUIRED_PARAMS = ("dify_api_url", "dify_token", "dify_dataset_id")
class DifyKB(ReadOnlyConnectors):
"""基于 Dify Dataset Retrieve API 的只读检索知识库实现"""
+ kb_type = "dify"
+ name = "Dify"
+ description = "连接 Dify Dataset 的只读检索知识库"
+
def __init__(self, work_dir: str, **kwargs):
del kwargs
super().__init__(work_dir)
- @property
- def kb_type(self) -> str:
- return "dify"
-
@classmethod
def get_create_params_config(cls) -> dict[str, Any]:
return {
diff --git a/backend/package/yuxi/knowledge/implementations/milvus.py b/backend/package/yuxi/knowledge/implementations/milvus.py
index b0df8a30..5b2ed90b 100644
--- a/backend/package/yuxi/knowledge/implementations/milvus.py
+++ b/backend/package/yuxi/knowledge/implementations/milvus.py
@@ -239,6 +239,10 @@ def _retrieval_config_options() -> list[dict[str, Any]]:
class MilvusKB(KnowledgeBase):
"""基于 Milvus 的生产级向量库"""
+ kb_type = "milvus"
+ name = "Milvus"
+ description = "基于 Milvus 的生产级向量知识库，适合高性能部署"
+
def __init__(self, work_dir: str, **kwargs):
"""
初始化 Milvus 知识库
@@ -273,11 +277,6 @@ class MilvusKB(KnowledgeBase):
logger.info("MilvusKB initialized")
- @property
- def kb_type(self) -> str:
- """知识库类型标识"""
- return "milvus"
-
def _init_connection(self):
"""初始化 Milvus 连接"""
try:
diff --git a/backend/package/yuxi/knowledge/implementations/notion.py b/backend/package/yuxi/knowledge/implementations/notion.py
index 426ebd6a..e3488e61 100644
--- a/backend/package/yuxi/knowledge/implementations/notion.py
+++ b/backend/package/yuxi/knowledge/implementations/notion.py
@@ -170,15 +170,15 @@ class _NotionClient:
class NotionKB(ReadOnlyConnectors):
"""连接 Notion Data Source 的只读知识库实现"""
+ kb_type = "notion"
+ name = "Notion"
+ description = "连接 Notion Data Source 的只读知识库，支持检索、打开页面和页内查找"
+
def __init__(self, work_dir: str, **kwargs):
del kwargs
super().__init__(work_dir)
self._page_markdown_cache: dict[tuple[str, str, str, str], tuple[float, str]] = {}
- @property
- def kb_type(self) -> str:
- return "notion"
-
@classmethod
def get_create_params_config(cls) -> dict[str, Any]:
return {
diff --git a/backend/test/integration/api/test_knowledge_router.py b/backend/test/integration/api/test_knowledge_router.py
index 390392c3..f45ec452 100644
--- a/backend/test/integration/api/test_knowledge_router.py
+++ b/backend/test/integration/api/test_knowledge_router.py
@@ -402,6 +402,9 @@ async def test_get_knowledge_base_types(test_client, admin_headers):
payload = response.json()
assert payload["message"] == "success"
assert "kb_types" in payload
+ assert "default_config" not in payload["kb_types"]["dify"]
+ assert payload["kb_types"]["dify"]["name"] == "Dify"
+ assert payload["kb_types"]["dify"]["description"] == "连接 Dify Dataset 的只读检索知识库"
assert payload["kb_types"]["dify"]["requires_embedding_model"] is False
assert payload["kb_types"]["dify"]["supports_documents"] is False
assert [option["key"] for option in payload["kb_types"]["dify"]["create_params"]["options"]] == [
@@ -409,6 +412,12 @@ async def test_get_knowledge_base_types(test_client, admin_headers):
"dify_token",
"dify_dataset_id",
]
+ assert "default_config" not in payload["kb_types"]["notion"]
+ assert payload["kb_types"]["notion"]["name"] == "Notion"
+ assert (
+ payload["kb_types"]["notion"]["description"]
+ == "连接 Notion Data Source 的只读知识库，支持检索、打开页面和页内查找"
+ )
assert payload["kb_types"]["notion"]["requires_embedding_model"] is False
assert payload["kb_types"]["notion"]["supports_documents"] is False
assert [option["key"] for option in payload["kb_types"]["notion"]["create_params"]["options"]] == [
diff --git a/web/src/views/DataBaseView.vue b/web/src/views/DataBaseView.vue
index f036e710..3982c7b8 100644
--- a/web/src/views/DataBaseView.vue
+++ b/web/src/views/DataBaseView.vue
@@ -25,7 +25,7 @@
-
+
新建知识库
@@ -159,6 +159,7 @@
key="submit"
type="primary"
:loading="dbState.creating"
+ :disabled="!selectedKbTypeInfo"
@click="handleCreateDatabase"
>创建
@@ -175,7 +176,12 @@
暂无知识库
创建您的第一个知识库，开始管理文档和知识
-
+
@@ -276,7 +282,7 @@ const createEmptyDatabaseForm = () => ({
name: '',
description: '',
embedding_model_spec: configStore.config?.embed_model,
- kb_type: 'milvus',
+ kb_type: '',
storage: '',
chunk_preset_id: 'general',
additional_params: {}
@@ -300,8 +306,7 @@ const createParamOptions = computed(
() => selectedKbTypeInfo.value?.create_params?.options || []
)
-const getKbTypeDescription = (typeInfo) =>
- typeInfo?.default_config?.description || typeInfo?.description || ''
+const getKbTypeDescription = (typeInfo) => typeInfo?.description || ''
const resetCreateParamValues = () => {
newDatabase.additional_params = {}
@@ -320,25 +325,21 @@ const resetCreateParamValues = () => {
const loadSupportedKbTypes = async () => {
try {
const data = await typeApi.getKnowledgeBaseTypes()
- supportedKbTypes.value = data.kb_types
+ supportedKbTypes.value = data.kb_types || {}
+ newDatabase.kb_type = kbTypes.value[0] || ''
resetCreateParamValues()
- console.log('支持的知识库类型:', supportedKbTypes.value)
} catch (error) {
console.error('加载知识库类型失败:', error)
- // 如果加载失败，设置默认类型
- supportedKbTypes.value = {
- milvus: {
- description: '基于 Milvus 的生产级向量知识库，支持文档检索和图谱构建',
- class_name: 'MilvusKB',
- requires_embedding_model: true,
- create_params: { options: [] }
- }
- }
+ supportedKbTypes.value = {}
+ newDatabase.kb_type = ''
+ resetCreateParamValues()
+ message.error('加载知识库类型失败，请稍后重试')
}
}
const resetNewDatabase = () => {
Object.assign(newDatabase, createEmptyDatabaseForm())
+ newDatabase.kb_type = kbTypes.value[0] || ''
resetCreateParamValues()
// 重置共享配置
shareConfig.value = {
@@ -430,6 +431,11 @@ const buildRequestData = () => {
// 创建按钮处理
const handleCreateDatabase = async () => {
+ if (!selectedKbTypeInfo.value) {
+ message.error('知识库类型加载失败，无法创建知识库')
+ return
+ }
+
for (const field of createParamOptions.value) {
if (!field.required) continue
const value = newDatabase.additional_params[field.key]