You've already forked DataMate
Jerry Yan
fa9e9d9f68
- 移除段落统计相关的数据结构和缓存逻辑 - 删除段落切换确认对话框和自动保存选项 - 简化段落加载和状态管理流程 - 将段落列表视图替换为简单的进度显示 - 更新API接口以支持单段内容获取 - 重构后端服务实现单段内容查询功能
120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
"""
|
|
Label Studio Editor（前端嵌入式）接口
|
|
|
|
说明：
|
|
- 不依赖 Label Studio Server；仅复用其“编辑器”前端库
|
|
- DataMate 负责提供 tasks/annotations 数据与保存能力
|
|
- 当前支持 dataset_type=TEXT/IMAGE 的项目
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from fastapi import APIRouter, Depends, Query, Path
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.logging import get_logger
|
|
from app.db.session import get_db
|
|
from app.module.annotation.schema.editor import (
|
|
EditorProjectInfo,
|
|
EditorTaskListResponse,
|
|
EditorTaskSegmentResponse,
|
|
EditorTaskResponse,
|
|
UpsertAnnotationRequest,
|
|
UpsertAnnotationResponse,
|
|
)
|
|
from app.module.annotation.service.editor import AnnotationEditorService
|
|
from app.module.shared.schema import StandardResponse
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
router = APIRouter(
|
|
prefix="/editor",
|
|
tags=["annotation/editor"],
|
|
)
|
|
|
|
|
|
@router.get(
|
|
"/projects/{project_id}",
|
|
response_model=StandardResponse[EditorProjectInfo],
|
|
)
|
|
async def get_editor_project_info(
|
|
project_id: str = Path(..., description="标注项目ID（t_dm_labeling_projects.id）"),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = AnnotationEditorService(db)
|
|
info = await service.get_project_info(project_id)
|
|
return StandardResponse(code=200, message="success", data=info)
|
|
|
|
|
|
@router.get(
|
|
"/projects/{project_id}/tasks",
|
|
response_model=StandardResponse[EditorTaskListResponse],
|
|
)
|
|
async def list_editor_tasks(
|
|
project_id: str = Path(..., description="标注项目ID（t_dm_labeling_projects.id）"),
|
|
page: int = Query(0, ge=0, description="页码（从0开始）"),
|
|
size: int = Query(50, ge=1, le=200, description="每页大小"),
|
|
exclude_source_documents: Optional[bool] = Query(
|
|
None,
|
|
alias="excludeSourceDocuments",
|
|
description="是否排除已被转换为TXT的源文档文件（PDF/DOC/DOCX，仅文本数据集生效）",
|
|
),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = AnnotationEditorService(db)
|
|
result = await service.list_tasks(
|
|
project_id,
|
|
page=page,
|
|
size=size,
|
|
exclude_source_documents=exclude_source_documents,
|
|
)
|
|
return StandardResponse(code=200, message="success", data=result)
|
|
|
|
|
|
@router.get(
|
|
"/projects/{project_id}/tasks/{file_id}",
|
|
response_model=StandardResponse[EditorTaskResponse],
|
|
)
|
|
async def get_editor_task(
|
|
project_id: str = Path(..., description="标注项目ID（t_dm_labeling_projects.id）"),
|
|
file_id: str = Path(..., description="文件ID（t_dm_dataset_files.id）"),
|
|
segment_index: Optional[int] = Query(None, alias="segmentIndex", description="段落索引（分段模式下使用）"),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = AnnotationEditorService(db)
|
|
task = await service.get_task(project_id, file_id, segment_index=segment_index)
|
|
return StandardResponse(code=200, message="success", data=task)
|
|
|
|
|
|
@router.get(
|
|
"/projects/{project_id}/tasks/{file_id}/segments",
|
|
response_model=StandardResponse[EditorTaskSegmentResponse],
|
|
)
|
|
async def get_editor_task_segment(
|
|
project_id: str = Path(..., description="标注项目ID（t_dm_labeling_projects.id）"),
|
|
file_id: str = Path(..., description="文件ID（t_dm_dataset_files.id）"),
|
|
segment_index: int = Query(..., ge=0, alias="segmentIndex", description="段落索引（从0开始）"),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = AnnotationEditorService(db)
|
|
result = await service.get_task_segment(project_id, file_id, segment_index)
|
|
return StandardResponse(code=200, message="success", data=result)
|
|
|
|
|
|
@router.put(
|
|
"/projects/{project_id}/tasks/{file_id}/annotation",
|
|
response_model=StandardResponse[UpsertAnnotationResponse],
|
|
)
|
|
async def upsert_editor_annotation(
|
|
request: UpsertAnnotationRequest,
|
|
project_id: str = Path(..., description="标注项目ID（t_dm_labeling_projects.id）"),
|
|
file_id: str = Path(..., description="文件ID（t_dm_dataset_files.id）"),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = AnnotationEditorService(db)
|
|
result = await service.upsert_annotation(project_id, file_id, request)
|
|
return StandardResponse(code=200, message="success", data=result)
|
|
|