Spaces:
Running
Running
File size: 2,163 Bytes
dd8f00c 2fa91ca dd8f00c 2fa91ca dd8f00c 2fa91ca dd8f00c 2fa91ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
from fastapi import APIRouter, HTTPException, BackgroundTasks
from pydantic import BaseModel, Field
from typing import Optional, List, Dict, Any
from app.services.note_store import get_note, update_note
from app.jobs.async_enrichment_job import enrich_note
from app.utils.time import now_ts
router = APIRouter(
prefix="/internal/notes",
tags=["internal-notes"],
)
class InternalUpdateNoteRequest(BaseModel):
status: Optional[str] = Field(
None,
description="processing | transcribed | error | failed",
)
raw_text: Optional[str] = None
metadata: Optional[Dict[str, Any]] = None
# pipeline triggers
generate: Optional[List[str]] = None
@router.patch("/{note_id}")
def update_note_internal(note_id: str, req: InternalUpdateNoteRequest, bg: BackgroundTasks):
note = get_note(note_id)
if not note:
raise HTTPException(status_code=404, detail="Note not found")
updates = req.dict(exclude_unset=True)
# cập nhật timestamp hệ thống
updates["updated_at"] = now_ts()
update_note(note_id, updates)
# Decide whether to trigger enrichment:
# - If client supplied `generate` in payload → run those tasks
# - Or if status switched to `transcribed` and the existing note has a `generate` list → run that
try:
allowed = {"normalize", "keywords", "summary", "mindmap"}
tasks = None
if req.generate is not None:
if not set(req.generate).issubset(allowed):
raise HTTPException(400, "Invalid generate task")
tasks = req.generate
elif updates.get("status") == "transcribed":
existing_generate = note.get("generate")
if existing_generate and set(existing_generate).issubset(allowed):
tasks = existing_generate
if tasks:
bg.add_task(enrich_note, note_id, tasks)
except HTTPException:
raise
except Exception:
# Don't fail the update if scheduling enrichment fails; just log in real app
pass
return {
"note_id": note_id,
"updated": True,
"fields": list(updates.keys()),
}
|