Add memory system session context API

memory_system_api/api.py

@@ -4,7 +4,13 @@ from __future__ import annotations
 from fastapi import APIRouter, Depends, HTTPException, Query, status
 
 from .auth import verify_api_key
-from .schemas import MessageIngestRequest, SearchRequest, SessionUserRequest, UserCreateRequest
+from .schemas import (
+    MessageIngestRequest,
+    SearchRequest,
+    SessionContextRequest,
+    SessionUserRequest,
+    UserCreateRequest,
+)
 from .service import MemorySystemService
 
 
@@ -70,6 +76,34 @@ async def extract_session(
         raise user_auth_error(exc) from exc
 
 
+@router.post("/sessions/{session_id}/context")
+async def get_session_context(
+    session_id: str,
+    request: SessionContextRequest,
+    service: MemorySystemService = Depends(get_service),
+):
+    try:
+        return await service.get_session_context(session_id, request)
+    except PermissionError as exc:
+        raise user_auth_error(exc) from exc
+
+
+@router.get("/sessions/{session_id}/context")
+async def get_session_context_from_query(
+    session_id: str,
+    user_id: str = Query(min_length=1),
+    user_key: str = Query(min_length=1),
+    query: str = Query(min_length=1),
+    limit: int = Query(default=10, ge=1, le=100),
+    service: MemorySystemService = Depends(get_service),
+):
+    try:
+        request = SessionContextRequest(user_id=user_id, user_key=user_key, query=query, limit=limit)
+        return await service.get_session_context(session_id, request)
+    except PermissionError as exc:
+        raise user_auth_error(exc) from exc
+
+
 @router.get("/openviking/tasks/{task_id}")
 async def get_openviking_task(
     task_id: str,