70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""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)
|