You've already forked DataMate
Jerry Yan
3f566a0b08
- 移除 TemplateConfigurationForm 组件并引入 TemplateConfigurationTreeEditor - 使用 useTagConfig Hook 获取标签配置 - 将自定义XML状态 customXml 替换为 labelConfig - 删除模板编辑标签页和选择模板状态管理 - 更新XML解析逻辑支持更多对象和标注控件类型 - 添加配置验证功能确保至少包含数据对象和标注控件 - 在模板详情页面使用树形编辑器显示配置详情 - 更新任务创建页面集成新的树形配置编辑器 - 调整预览数据生成功能适配新的XML解析方式
96 lines
4.7 KiB
Python
96 lines
4.7 KiB
Python
"""
|
|
Annotation Template Schemas
|
|
"""
|
|
from typing import List, Dict, Any, Optional
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
|
|
|
|
class LabelDefinition(BaseModel):
|
|
"""标签定义"""
|
|
from_name: str = Field(alias="fromName", description="控件名称")
|
|
to_name: str = Field(alias="toName", description="目标对象名称")
|
|
type: str = Field(description="控件类型: choices/rectanglelabels/polygonlabels/textarea/etc")
|
|
options: Optional[List[str]] = Field(None, description="选项列表（用于choices类型）")
|
|
labels: Optional[List[str]] = Field(None, description="标签列表（用于rectanglelabels等类型）")
|
|
required: bool = Field(False, description="是否必填")
|
|
choice: Optional[str] = Field(None, description="Choices 选择模式: single/multiple")
|
|
show_inline: Optional[bool] = Field(None, alias="showInline", description="Choices 是否行内显示")
|
|
description: Optional[str] = Field(None, description="标签描述")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class ObjectDefinition(BaseModel):
|
|
"""对象定义"""
|
|
name: str = Field(description="对象标识符")
|
|
type: str = Field(description="对象类型: Image/Text/Audio/Video/etc")
|
|
value: str = Field(description="变量名，如$image")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class TemplateConfiguration(BaseModel):
|
|
"""模板配置结构"""
|
|
labels: List[LabelDefinition] = Field(description="标签定义列表")
|
|
objects: List[ObjectDefinition] = Field(description="对象定义列表")
|
|
metadata: Optional[Dict[str, Any]] = Field(None, description="额外元数据")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class CreateAnnotationTemplateRequest(BaseModel):
|
|
"""创建标注模板请求"""
|
|
name: str = Field(..., min_length=1, max_length=100, description="模板名称")
|
|
description: Optional[str] = Field(None, max_length=500, description="模板描述")
|
|
data_type: str = Field(alias="dataType", description="数据类型")
|
|
labeling_type: str = Field(alias="labelingType", description="标注类型")
|
|
label_config: str = Field(alias="labelConfig", description="Label Studio XML 配置")
|
|
style: str = Field(default="horizontal", description="样式配置")
|
|
category: str = Field(default="custom", description="模板分类")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class UpdateAnnotationTemplateRequest(BaseModel):
|
|
"""更新标注模板请求"""
|
|
name: Optional[str] = Field(None, min_length=1, max_length=100, description="模板名称")
|
|
description: Optional[str] = Field(None, max_length=500, description="模板描述")
|
|
data_type: Optional[str] = Field(None, alias="dataType", description="数据类型")
|
|
labeling_type: Optional[str] = Field(None, alias="labelingType", description="标注类型")
|
|
label_config: Optional[str] = Field(None, alias="labelConfig", description="Label Studio XML 配置")
|
|
style: Optional[str] = Field(None, description="样式配置")
|
|
category: Optional[str] = Field(None, description="模板分类")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class AnnotationTemplateResponse(BaseModel):
|
|
"""标注模板响应"""
|
|
id: str = Field(..., description="模板ID")
|
|
name: str = Field(..., description="模板名称")
|
|
description: Optional[str] = Field(None, description="模板描述")
|
|
data_type: str = Field(alias="dataType", description="数据类型")
|
|
labeling_type: str = Field(alias="labelingType", description="标注类型")
|
|
configuration: Optional[TemplateConfiguration] = Field(None, description="标注配置")
|
|
label_config: Optional[str] = Field(None, alias="labelConfig", description="Label Studio XML配置")
|
|
style: str = Field(..., description="样式配置")
|
|
category: str = Field(..., description="模板分类")
|
|
built_in: bool = Field(alias="builtIn", description="是否内置模板")
|
|
version: str = Field(..., description="版本号")
|
|
created_at: datetime = Field(alias="createdAt", description="创建时间")
|
|
updated_at: Optional[datetime] = Field(None, alias="updatedAt", description="更新时间")
|
|
|
|
model_config = ConfigDict(populate_by_name=True, from_attributes=True)
|
|
|
|
|
|
class AnnotationTemplateListResponse(BaseModel):
|
|
"""模板列表响应"""
|
|
content: List[AnnotationTemplateResponse] = Field(..., description="模板列表")
|
|
total: int = Field(..., description="总数")
|
|
page: int = Field(..., description="当前页")
|
|
size: int = Field(..., description="每页大小")
|
|
total_pages: int = Field(alias="totalPages", description="总页数")
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|