297 lines
9.5 KiB
Python
297 lines
9.5 KiB
Python
|
"""
|
|||
|
|
Decomposition Template API.
|
|||
|
|
[AC-IDSMETA-21, AC-IDSMETA-22] 拆解模板管理接口，支持文本拆解为结构化数据。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from typing import Annotated, Any
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Depends, Query
|
|||
|
|
from fastapi.responses import JSONResponse
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.core.database import get_session
|
|||
|
|
from app.core.exceptions import MissingTenantIdException
|
|||
|
|
from app.core.tenant import get_tenant_id
|
|||
|
|
from app.models.entities import (
|
|||
|
|
DecompositionRequest,
|
|||
|
|
DecompositionTemplateCreate,
|
|||
|
|
DecompositionTemplateStatus,
|
|||
|
|
DecompositionTemplateUpdate,
|
|||
|
|
)
|
|||
|
|
from app.services.decomposition_template_service import DecompositionTemplateService
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/admin/decomposition-templates", tags=["DecompositionTemplates"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_current_tenant_id() -> str:
|
|||
|
|
"""Get current tenant ID from context."""
|
|||
|
|
tenant_id = get_tenant_id()
|
|||
|
|
if not tenant_id:
|
|||
|
|
raise MissingTenantIdException()
|
|||
|
|
return tenant_id
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get(
|
|||
|
|
"",
|
|||
|
|
operation_id="listDecompositionTemplates",
|
|||
|
|
summary="List decomposition templates",
|
|||
|
|
description="[AC-IDSMETA-22] 获取拆解模板列表，支持按状态过滤",
|
|||
|
|
)
|
|||
|
|
async def list_templates(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_current_tenant_id)],
|
|||
|
|
session: Annotated[AsyncSession, Depends(get_session)],
|
|||
|
|
status: Annotated[str | None, Query(
|
|||
|
|
description="按状态过滤: draft/published/archived"
|
|||
|
|
)] = None,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-IDSMETA-22] 列出拆解模板
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-IDSMETA-22] Listing decomposition templates: "
|
|||
|
|
f"tenant={tenant_id}, status={status}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if status and status not in [s.value for s in DecompositionTemplateStatus]:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=400,
|
|||
|
|
content={
|
|||
|
|
"code": "INVALID_STATUS",
|
|||
|
|
"message": f"Invalid status: {status}",
|
|||
|
|
"details": {
|
|||
|
|
"valid_values": [s.value for s in DecompositionTemplateStatus]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
service = DecompositionTemplateService(session)
|
|||
|
|
templates = await service.list_templates(tenant_id, status)
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"items": [
|
|||
|
|
{
|
|||
|
|
"id": str(t.id),
|
|||
|
|
"name": t.name,
|
|||
|
|
"description": t.description,
|
|||
|
|
"version": t.version,
|
|||
|
|
"status": t.status,
|
|||
|
|
"template_schema": t.template_schema,
|
|||
|
|
"extraction_hints": t.extraction_hints,
|
|||
|
|
"example_input": t.example_input,
|
|||
|
|
"example_output": t.example_output,
|
|||
|
|
"created_at": t.created_at.isoformat(),
|
|||
|
|
"updated_at": t.updated_at.isoformat(),
|
|||
|
|
}
|
|||
|
|
for t in templates
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post(
|
|||
|
|
"",
|
|||
|
|
operation_id="createDecompositionTemplate",
|
|||
|
|
summary="Create decomposition template",
|
|||
|
|
description="[AC-IDSMETA-22] 创建新的拆解模板",
|
|||
|
|
status_code=201,
|
|||
|
|
)
|
|||
|
|
async def create_template(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_current_tenant_id)],
|
|||
|
|
session: Annotated[AsyncSession, Depends(get_session)],
|
|||
|
|
template_create: DecompositionTemplateCreate,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-IDSMETA-22] 创建拆解模板
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-IDSMETA-22] Creating decomposition template: "
|
|||
|
|
f"tenant={tenant_id}, name={template_create.name}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
service = DecompositionTemplateService(session)
|
|||
|
|
template = await service.create_template(tenant_id, template_create)
|
|||
|
|
await session.commit()
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=201,
|
|||
|
|
content={
|
|||
|
|
"id": str(template.id),
|
|||
|
|
"name": template.name,
|
|||
|
|
"description": template.description,
|
|||
|
|
"version": template.version,
|
|||
|
|
"status": template.status,
|
|||
|
|
"template_schema": template.template_schema,
|
|||
|
|
"extraction_hints": template.extraction_hints,
|
|||
|
|
"example_input": template.example_input,
|
|||
|
|
"example_output": template.example_output,
|
|||
|
|
"created_at": template.created_at.isoformat(),
|
|||
|
|
"updated_at": template.updated_at.isoformat(),
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get(
|
|||
|
|
"/latest",
|
|||
|
|
operation_id="getLatestPublishedTemplate",
|
|||
|
|
summary="Get latest published template",
|
|||
|
|
description="[AC-IDSMETA-22] 获取最近生效的发布版本模板",
|
|||
|
|
)
|
|||
|
|
async def get_latest_template(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_current_tenant_id)],
|
|||
|
|
session: Annotated[AsyncSession, Depends(get_session)],
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-IDSMETA-22] 获取最近生效的发布版本模板
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-IDSMETA-22] Getting latest published template: tenant={tenant_id}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
service = DecompositionTemplateService(session)
|
|||
|
|
template = await service.get_latest_published_template(tenant_id)
|
|||
|
|
|
|||
|
|
if not template:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=404,
|
|||
|
|
content={
|
|||
|
|
"code": "NOT_FOUND",
|
|||
|
|
"message": "No published template found",
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"id": str(template.id),
|
|||
|
|
"name": template.name,
|
|||
|
|
"description": template.description,
|
|||
|
|
"version": template.version,
|
|||
|
|
"status": template.status,
|
|||
|
|
"template_schema": template.template_schema,
|
|||
|
|
"extraction_hints": template.extraction_hints,
|
|||
|
|
"example_input": template.example_input,
|
|||
|
|
"example_output": template.example_output,
|
|||
|
|
"created_at": template.created_at.isoformat(),
|
|||
|
|
"updated_at": template.updated_at.isoformat(),
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.put(
|
|||
|
|
"/{id}",
|
|||
|
|
operation_id="updateDecompositionTemplate",
|
|||
|
|
summary="Update decomposition template",
|
|||
|
|
description="[AC-IDSMETA-22] 更新拆解模板，支持状态切换",
|
|||
|
|
)
|
|||
|
|
async def update_template(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_current_tenant_id)],
|
|||
|
|
session: Annotated[AsyncSession, Depends(get_session)],
|
|||
|
|
id: str,
|
|||
|
|
template_update: DecompositionTemplateUpdate,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-IDSMETA-22] 更新拆解模板
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-IDSMETA-22] Updating decomposition template: "
|
|||
|
|
f"tenant={tenant_id}, id={id}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if template_update.status and template_update.status not in [s.value for s in DecompositionTemplateStatus]:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=400,
|
|||
|
|
content={
|
|||
|
|
"code": "INVALID_STATUS",
|
|||
|
|
"message": f"Invalid status: {template_update.status}",
|
|||
|
|
"details": {
|
|||
|
|
"valid_values": [s.value for s in DecompositionTemplateStatus]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
service = DecompositionTemplateService(session)
|
|||
|
|
template = await service.update_template(tenant_id, id, template_update)
|
|||
|
|
|
|||
|
|
if not template:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=404,
|
|||
|
|
content={
|
|||
|
|
"code": "NOT_FOUND",
|
|||
|
|
"message": f"Template {id} not found",
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
await session.commit()
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"id": str(template.id),
|
|||
|
|
"name": template.name,
|
|||
|
|
"description": template.description,
|
|||
|
|
"version": template.version,
|
|||
|
|
"status": template.status,
|
|||
|
|
"template_schema": template.template_schema,
|
|||
|
|
"extraction_hints": template.extraction_hints,
|
|||
|
|
"example_input": template.example_input,
|
|||
|
|
"example_output": template.example_output,
|
|||
|
|
"created_at": template.created_at.isoformat(),
|
|||
|
|
"updated_at": template.updated_at.isoformat(),
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post(
|
|||
|
|
"/decompose",
|
|||
|
|
operation_id="decomposeText",
|
|||
|
|
summary="Decompose text to structured data",
|
|||
|
|
description="[AC-IDSMETA-21] 将待录入文本拆解为固定模板输出",
|
|||
|
|
)
|
|||
|
|
async def decompose_text(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_current_tenant_id)],
|
|||
|
|
session: Annotated[AsyncSession, Depends(get_session)],
|
|||
|
|
request: DecompositionRequest,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-IDSMETA-21] 将待录入文本拆解为固定模板输出
|
|||
|
|
|
|||
|
|
如果不指定 template_id，则使用最近生效的发布版本模板
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-IDSMETA-21] Decomposing text: tenant={tenant_id}, "
|
|||
|
|
f"template_id={request.template_id}, text_length={len(request.text)}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
from app.services.llm import get_llm_client
|
|||
|
|
llm_client = get_llm_client()
|
|||
|
|
|
|||
|
|
service = DecompositionTemplateService(session, llm_client)
|
|||
|
|
result = await service.decompose_text(tenant_id, request)
|
|||
|
|
|
|||
|
|
if not result.success:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=400,
|
|||
|
|
content={
|
|||
|
|
"code": "DECOMPOSITION_FAILED",
|
|||
|
|
"message": result.error,
|
|||
|
|
"details": {
|
|||
|
|
"template_id": result.template_id,
|
|||
|
|
"template_version": result.template_version,
|
|||
|
|
"latency_ms": result.latency_ms,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"success": result.success,
|
|||
|
|
"data": result.data,
|
|||
|
|
"template_id": result.template_id,
|
|||
|
|
"template_version": result.template_version,
|
|||
|
|
"confidence": result.confidence,
|
|||
|
|
"latency_ms": result.latency_ms,
|
|||
|
|
}
|
|||
|
|
)
|