You've already forked DataMate
feature: multiple ratio configurations can be set for the data set. (#103)
feature: multiple ratio configurations can be set for the data set.
This commit is contained in:
@@ -27,15 +27,15 @@ class PagedDatasetFileResponse(BaseModel):
|
||||
size: int = Field(..., description="每页大小")
|
||||
|
||||
class DatasetFileTag(BaseModel):
|
||||
id: str = Field(..., description="标签ID")
|
||||
type: str = Field(..., description="类型")
|
||||
from_name: str = Field(..., description="标签名称")
|
||||
value: dict = Field(..., description="标签值")
|
||||
id: str = Field(None, description="标签ID")
|
||||
type: str = Field(None, description="类型")
|
||||
from_name: str = Field(None, description="标签名称")
|
||||
values: dict = Field(None, description="标签值")
|
||||
|
||||
def get_tags(self) -> List[str]:
|
||||
tags = []
|
||||
# 如果 value 是字典类型,根据 type 获取对应的值
|
||||
tag_values = self.value.get(self.type, [])
|
||||
# 如果 values 是字典类型,根据 type 获取对应的值
|
||||
tag_values = self.values.get(self.type, [])
|
||||
|
||||
# 处理标签值
|
||||
if isinstance(tag_values, list):
|
||||
@@ -55,7 +55,7 @@ class FileTagUpdate(BaseModel):
|
||||
"""单个文件的标签更新请求"""
|
||||
file_id: str = Field(..., alias="fileId", description="文件ID")
|
||||
tags: List[Dict[str, Any]] = Field(..., description="要更新的标签列表(部分更新)")
|
||||
|
||||
|
||||
class Config:
|
||||
populate_by_name = True
|
||||
|
||||
@@ -63,7 +63,7 @@ class FileTagUpdate(BaseModel):
|
||||
class BatchUpdateFileTagsRequest(BaseModel):
|
||||
"""批量更新文件标签请求"""
|
||||
updates: List[FileTagUpdate] = Field(..., description="文件标签更新列表", min_length=1)
|
||||
|
||||
|
||||
class Config:
|
||||
populate_by_name = True
|
||||
|
||||
@@ -74,7 +74,7 @@ class FileTagUpdateResult(BaseModel):
|
||||
success: bool = Field(..., description="是否更新成功")
|
||||
message: Optional[str] = Field(None, description="结果信息")
|
||||
tags_updated_at: Optional[datetime] = Field(None, alias="tagsUpdatedAt", description="标签更新时间")
|
||||
|
||||
|
||||
class Config:
|
||||
populate_by_name = True
|
||||
|
||||
@@ -85,6 +85,6 @@ class BatchUpdateFileTagsResponse(BaseModel):
|
||||
total: int = Field(..., description="总更新数量")
|
||||
success_count: int = Field(..., alias="successCount", description="成功数量")
|
||||
failure_count: int = Field(..., alias="failureCount", description="失败数量")
|
||||
|
||||
|
||||
class Config:
|
||||
populate_by_name = True
|
||||
|
||||
Reference in New Issue
Block a user