File size: 1,678 Bytes
02117ee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | """
Memory API Routes — Persistent agent memory
"""
import time
from fastapi import APIRouter, HTTPException, Query
from core.models import MemorySaveRequest, MemorySearchRequest
from memory.db import save_memory, search_memory, get_project_memory, get_history
router = APIRouter()
@router.post("/", summary="Save memory")
async def save(req: MemorySaveRequest):
await save_memory(
content=req.content,
memory_type=req.memory_type.value,
session_id=req.session_id,
project_id=req.project_id,
key=req.key,
metadata=req.metadata,
)
return {"status": "saved", "memory_type": req.memory_type, "timestamp": time.time()}
@router.post("/search", summary="Search memory")
async def search(req: MemorySearchRequest):
results = await search_memory(
query=req.query,
session_id=req.session_id,
project_id=req.project_id,
limit=req.limit,
)
return {"results": results, "total": len(results), "query": req.query}
@router.get("/project/{project_id}", summary="Get project memory")
async def project_memory(
project_id: str,
memory_type: str = Query(default=""),
limit: int = Query(default=100, le=500),
):
results = await get_project_memory(project_id, memory_type=memory_type, limit=limit)
return {"project_id": project_id, "memories": results, "total": len(results)}
@router.get("/history/{session_id}", summary="Get conversation history")
async def history(session_id: str, limit: int = Query(default=50, le=200)):
results = await get_history(session_id, limit=limit)
return {"session_id": session_id, "history": results, "total": len(results)}
|