Spaces:
Running
Running
| import logging | |
| from fastapi import APIRouter, BackgroundTasks, HTTPException | |
| from pydantic import BaseModel | |
| from typing import Dict, List, Optional | |
| from app.services.note_store import create_note, get_note | |
| from app.models.enums import NoteType, NoteStatus | |
| from app.utils.time import now_ts | |
| router = APIRouter(prefix="/internal/notes", tags=["internal"]) | |
| logger = logging.getLogger(__name__) | |
| class CreateAudioNoteRequest(BaseModel): | |
| note_id: str | |
| raw_text: str | |
| metadata: Dict | |
| generate: List[str] = [] | |
| async def create_audio_note(req: CreateAudioNoteRequest, bg: BackgroundTasks): | |
| """ | |
| Idempotent create: | |
| - Nếu note đã tồn tại → return luôn | |
| - KHÔNG throw 500 vì duplicate | |
| """ | |
| now = now_ts() | |
| allowed = {"normalize", "keywords", "summary", "mindmap"} | |
| if not set(req.generate).issubset(allowed): | |
| raise HTTPException(400, "Invalid generate task") | |
| # 🔒 IDEMPOTENT CHECK | |
| existing = get_note(req.note_id) | |
| if existing: | |
| logger.info("Note already exists, skip create note_id=%s", req.note_id) | |
| return { | |
| "note_id": req.note_id, | |
| "status": existing.get("status"), | |
| } | |
| note = { | |
| "note_id": req.note_id, | |
| "type": NoteType.audio, | |
| "raw_text": req.raw_text, | |
| "metadata": req.metadata, | |
| # 🔥 PLACEHOLDER → CREATED | |
| "status": NoteStatus.created, | |
| "created_at": now, | |
| "updated_at": now, | |
| } | |
| # Chuẩn bị field cho enrich (nhưng CHƯA chạy) | |
| if req.generate: | |
| note.update({ | |
| "normalized_text": None, | |
| "title": None, | |
| "keywords": [], | |
| "summary": None, | |
| "mindmap": None, | |
| }) | |
| try: | |
| create_note(note) | |
| except ValueError: | |
| # double-safe (race condition) | |
| logger.warning("Race create_note, ignore note_id=%s", req.note_id) | |
| # ❌ KHÔNG chạy enrich ở đây | |
| # enrich CHỈ được trigger sau ASR xong (transcribed) | |
| return { | |
| "note_id": req.note_id, | |
| "status": note["status"], | |
| } |