Simplify to memory system api
This commit is contained in:
69
memory_system_api/api.py
Normal file
69
memory_system_api/api.py
Normal file
@ -0,0 +1,69 @@
|
||||
"""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)
|
||||
Reference in New Issue
Block a user