life-echo/api/app/features/memory/router.py

"""Memory 策展与内部扩展 API。"""

from fastapi import APIRouter, Depends, HTTPException, status
from pydantic import BaseModel, Field

from app.core.dependencies import get_current_user
from app.features.memory.deps import get_memory_service
from app.features.memory.service import MemoryService
from app.features.user.models import User

router = APIRouter(prefix="/api/memory", tags=["memory"])


class ExcludeBody(BaseModel):
    reason: str = Field(default="", max_length=2000)


class RejectFactBody(BaseModel):
    reason: str = Field(default="", max_length=2000)


@router.post("/chunks/{chunk_id}/exclude", status_code=status.HTTP_204_NO_CONTENT)
async def exclude_chunk(
    chunk_id: str,
    body: ExcludeBody | None = None,
    current_user: User = Depends(get_current_user),
    memory: MemoryService = Depends(get_memory_service),
):
    reason = (body.reason if body else "") or ""
    ok = await memory.exclude_chunk(current_user.id, chunk_id, reason=reason)
    if not ok:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND, detail="chunk 不存在"
        )


@router.post("/chunks/{chunk_id}/restore", status_code=status.HTTP_204_NO_CONTENT)
async def restore_chunk(
    chunk_id: str,
    current_user: User = Depends(get_current_user),
    memory: MemoryService = Depends(get_memory_service),
):
    ok = await memory.restore_chunk(current_user.id, chunk_id)
    if not ok:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND, detail="chunk 不存在"
        )


@router.post("/facts/{fact_id}/confirm", status_code=status.HTTP_204_NO_CONTENT)
async def confirm_fact(
    fact_id: str,
    current_user: User = Depends(get_current_user),
    memory: MemoryService = Depends(get_memory_service),
):
    ok = await memory.confirm_fact(current_user.id, fact_id)
    if not ok:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="fact 不存在")


@router.post("/facts/{fact_id}/reject", status_code=status.HTTP_204_NO_CONTENT)
async def reject_fact(
    fact_id: str,
    body: RejectFactBody | None = None,
    current_user: User = Depends(get_current_user),
    memory: MemoryService = Depends(get_memory_service),
):
    reason = (body.reason if body else "") or ""
    ok = await memory.reject_fact(current_user.id, fact_id, reason=reason)
    if not ok:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="fact 不存在")