You've already forked DataMate
Jerry Yan
0e0782a452
实现功能： - 创建 kg_extraction/interface.py（FastAPI 路由） - 实现 POST /api/kg/extract（单条文本抽取） - 实现 POST /api/kg/extract/batch（批量抽取，最多 50 条） - 集成到 FastAPI 主路由（/api/kg/ 前缀） 技术实现： - 配置管理：从环境变量读取 LLM 配置（API Key、Base URL、Model、Temperature） - 安全性： - API Key 使用 SecretStr 保护 - 错误信息脱敏（使用 trace_id，不暴露原始异常） - 请求文本不写入日志（使用 SHA-256 hash） - 强制要求 X-User-Id 头（鉴权边界） - 超时控制： - kg_llm_timeout_seconds（60秒） - kg_llm_max_retries（2次） - 输入校验： - graph_id 和 source_id 使用 UUID pattern - source_type 使用 Enum（4个值） - allowed_nodes/relationships 元素使用正则约束（ASCII，1-50字符） - 审计日志：记录 caller、trace_id、text_hash 代码审查： - 经过 3 轮 Codex 审查和 2 轮 Claude 修复 - 所有问题已解决（5个 P1/P2 + 3个 P3） - 语法检查通过 API 端点： - POST /api/kg/extract：单条文本抽取 - POST /api/kg/extract/batch：批量抽取（最多 50 条） 配置环境变量： - KG_LLM_API_KEY：LLM API 密钥 - KG_LLM_BASE_URL：自定义端点（可选） - KG_LLM_MODEL：模型名称（默认 gpt-4o-mini） - KG_LLM_TEMPERATURE：生成温度（默认 0.0） - KG_LLM_TIMEOUT_SECONDS：超时时间（默认 60） - KG_LLM_MAX_RETRIES：重试次数（默认 2）
26 lines
890 B
Python
26 lines
890 B
Python
from fastapi import APIRouter
|
|
|
|
from .system.interface import router as system_router
|
|
from .annotation.interface import router as annotation_router
|
|
from .ratio.interface import router as ratio_router
|
|
from .generation.interface import router as generation_router
|
|
from .evaluation.interface import router as evaluation_router
|
|
from .collection.interface import router as collection_route
|
|
from .dataset.interface import router as dataset_router
|
|
from .kg_extraction.interface import router as kg_extraction_router
|
|
|
|
router = APIRouter(
|
|
prefix="/api"
|
|
)
|
|
|
|
router.include_router(system_router)
|
|
router.include_router(annotation_router)
|
|
router.include_router(ratio_router)
|
|
router.include_router(generation_router)
|
|
router.include_router(evaluation_router)
|
|
router.include_router(collection_route)
|
|
router.include_router(dataset_router)
|
|
router.include_router(kg_extraction_router)
|
|
|
|
__all__ = ["router"]
|