"""FastAPI router for the lightweight Memory System API.""" from __future__ import annotations from fastapi import APIRouter, Depends, HTTPException, Query, status from .auth import verify_api_key from .schemas import MessageIngestRequest, SearchRequest, SessionUserRequest from .service import MemorySystemService router = APIRouter( prefix="/memory-system", tags=["memory-system"], dependencies=[Depends(verify_api_key)], ) def get_service() -> MemorySystemService: return MemorySystemService() @router.get("/health") async def health(service: MemorySystemService = Depends(get_service)): return await service.health() @router.post("/messages") async def ingest_messages(request: MessageIngestRequest, service: MemorySystemService = Depends(get_service)): try: return await service.ingest_messages(request) except ValueError as exc: raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=str(exc)) from exc @router.post("/sessions/{session_id}/commit") async def commit_session( session_id: str, request: SessionUserRequest, service: MemorySystemService = Depends(get_service), ): return await service.commit_session(request.user_id, session_id) @router.post("/sessions/{session_id}/extract") async def extract_session( session_id: str, request: SessionUserRequest, service: MemorySystemService = Depends(get_service), ): return await service.extract_session(request.user_id, session_id) @router.get("/openviking/tasks/{task_id}") async def get_openviking_task( task_id: str, user_id: str = Query(min_length=1), service: MemorySystemService = Depends(get_service), ): return await service.get_openviking_task(user_id, task_id) @router.post("/search") async def search(request: SearchRequest, service: MemorySystemService = Depends(get_service)): return await service.search(request) @router.get("/users/{user_id}/profile") async def get_profile(user_id: str, service: MemorySystemService = Depends(get_service)): return await service.get_profile(user_id)