kb_create fail
This commit is contained in:
@@ -26,11 +26,12 @@ class CreateKnowledgeBaseRequest(BaseModel):
|
||||
- parse_type=2: 使用自定义 pipeline,需要 pipeline_id,parser_id 为空
|
||||
"""
|
||||
name: str
|
||||
parse_type: Literal[1, 2] = Field(..., description="解析类型:1=内置解析器,2=自定义pipeline")
|
||||
parse_type: Literal[1, 2] = Field(default=1, description="解析类型:1=内置解析器,2=自定义pipeline")
|
||||
embd_id: str = Field(..., description="嵌入模型ID")
|
||||
parser_id: Optional[str] = Field(default="", description="解析器ID,parse_type=1时必需")
|
||||
pipeline_id: Optional[str] = Field(default="", description="流水线ID,parse_type=2时必需")
|
||||
description: Optional[str] = None
|
||||
permission: Optional[str] = Field(default="me", description="权限:me|team")
|
||||
pagerank: Optional[int] = None
|
||||
|
||||
@model_validator(mode='after')
|
||||
@@ -41,8 +42,9 @@ class CreateKnowledgeBaseRequest(BaseModel):
|
||||
parser_id_val = self.parser_id or ""
|
||||
pipeline_id_val = self.pipeline_id or ""
|
||||
|
||||
# 如果 parser_id 为空,自动设置为 "naive"
|
||||
if parser_id_val.strip() == "":
|
||||
raise ValueError("parse_type=1时,parser_id不能为空")
|
||||
self.parser_id = "naive"
|
||||
if pipeline_id_val.strip() != "":
|
||||
raise ValueError("parse_type=1时,pipeline_id必须为空")
|
||||
elif self.parse_type == 2:
|
||||
|
||||
Reference in New Issue
Block a user