You've already forked DataMate
ba0c69086a
* feat: data annotation page adaptation to the backend API. * feat: Implement labeling configuration editor and enhance annotation task creation form
52 lines
2.2 KiB
Python
52 lines
2.2 KiB
Python
from pydantic import Field, BaseModel
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
|
|
from app.module.shared.schema import BaseResponseModel
|
|
from app.module.shared.schema import StandardResponse
|
|
|
|
|
|
class DatasetMappingCreateRequest(BaseModel):
|
|
"""数据集映射 创建 请求模型
|
|
|
|
Accept both snake_case and camelCase field names from frontend JSON by
|
|
declaring explicit aliases. Frontend sends `datasetId`, `name`,
|
|
`description` (camelCase), so provide aliases so pydantic will map them
|
|
to the internal attributes used in the service code (dataset_id, name,
|
|
description).
|
|
"""
|
|
dataset_id: str = Field(..., alias="datasetId", description="源数据集ID")
|
|
name: Optional[str] = Field(None, alias="name", description="标注项目名称")
|
|
description: Optional[str] = Field(None, alias="description", description="标注项目描述")
|
|
|
|
class Config:
|
|
# allow population by field name when constructing model programmatically
|
|
allow_population_by_field_name = True
|
|
|
|
class DatasetMappingCreateResponse(BaseResponseModel):
|
|
"""数据集映射 创建 响应模型"""
|
|
id: str = Field(..., description="映射UUID")
|
|
labeling_project_id: str = Field(..., description="Label Studio项目ID")
|
|
labeling_project_name: str = Field(..., description="Label Studio项目名称")
|
|
|
|
class DatasetMappingUpdateRequest(BaseResponseModel):
|
|
"""数据集映射 更新 请求模型"""
|
|
dataset_id: Optional[str] = Field(None, description="源数据集ID")
|
|
|
|
class DatasetMappingResponse(BaseModel):
|
|
dataset_id: str = Field(..., description="源数据集ID")
|
|
"""数据集映射 查询 响应模型"""
|
|
id: str = Field(..., description="映射UUID")
|
|
labeling_project_id: str = Field(..., description="标注项目ID")
|
|
name: Optional[str] = Field(None, description="标注项目名称")
|
|
created_at: datetime = Field(..., description="创建时间")
|
|
deleted_at: Optional[datetime] = Field(None, description="删除时间")
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
populate_by_name = True
|
|
|
|
class DeleteDatasetResponse(BaseResponseModel):
|
|
"""删除数据集响应模型"""
|
|
id: str = Field(..., description="映射UUID")
|
|
status: str = Field(..., description="删除状态") |