admin
b02d3378fc
Full-stack application for document-to-knowledge-graph pipeline: - Backend: FastAPI + LangGraph ReAct agent + DeepSeek + MinerU parsing - Frontend: React 19 + Vite + D3.js + shadcn/ui - Pipeline: MinerU parsing → LangExtract entity extraction → KG building Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
"""A 组：文档管理（4 个端点）"""
|
|
from fastapi import APIRouter, File, Form, HTTPException, UploadFile
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from models.schemas import APIResponse
|
|
from services import document_service as svc
|
|
|
|
router = APIRouter(prefix="/documents", tags=["Documents"])
|
|
|
|
|
|
@router.post("/upload", status_code=200)
|
|
async def upload_document(
|
|
file: UploadFile = File(...),
|
|
language: str = Form("ch"),
|
|
enable_formula: bool = Form(True),
|
|
enable_table: bool = Form(True),
|
|
):
|
|
content = await file.read()
|
|
ok, code, msg = svc.validate_upload(file.filename or "", len(content))
|
|
if not ok:
|
|
return JSONResponse(
|
|
status_code=400,
|
|
content=APIResponse.err(code, msg).model_dump(),
|
|
)
|
|
doc = svc.save_upload(file.filename or "upload", content, language, enable_formula, enable_table)
|
|
# Remove internal field
|
|
doc.pop("upload_filename", None)
|
|
return APIResponse.ok(doc)
|
|
|
|
|
|
@router.get("/{doc_id}")
|
|
async def get_document(doc_id: str):
|
|
doc = svc.get_document(doc_id)
|
|
if not doc:
|
|
return JSONResponse(
|
|
status_code=404,
|
|
content=APIResponse.err(2001, f"Document '{doc_id}' not found").model_dump(),
|
|
)
|
|
doc.pop("upload_filename", None)
|
|
return APIResponse.ok(doc)
|
|
|
|
|
|
@router.get("")
|
|
async def list_documents(
|
|
page: int = 1,
|
|
page_size: int = 20,
|
|
status: str | None = None,
|
|
format: str | None = None,
|
|
):
|
|
page_size = min(page_size, 100)
|
|
result = svc.list_documents(page, page_size, status, format)
|
|
for item in result["items"]:
|
|
item.pop("upload_filename", None)
|
|
return APIResponse.ok(result)
|
|
|
|
|
|
@router.delete("/{doc_id}")
|
|
async def delete_document(doc_id: str):
|
|
doc = svc.get_document(doc_id)
|
|
if not doc:
|
|
return JSONResponse(
|
|
status_code=404,
|
|
content=APIResponse.err(2001, f"Document '{doc_id}' not found").model_dump(),
|
|
)
|
|
ok, removed_nodes, removed_edges = svc.delete_document(doc_id)
|
|
return APIResponse.ok({
|
|
"deleted": True,
|
|
"doc_id": doc_id,
|
|
"removed_nodes": removed_nodes,
|
|
"removed_edges": removed_edges,
|
|
})
|