Spaces:
Running
Running
| import time | |
| from fastapi import APIRouter, HTTPException, Depends | |
| from typing import Optional, List, Dict, Any | |
| from pydantic import BaseModel | |
| from core.dependencies import get_memory_service, get_storage, get_client_id | |
| from core.storage import MemoryStorage | |
| from services.memory_service import MemoryService | |
| from utils.logger import app_logger | |
| router = APIRouter(prefix="/memory", tags=["Memory & Knowledge"]) | |
| async def get_global_knowledge( | |
| client_id: str = Depends(get_client_id), | |
| memory_service: MemoryService = Depends(get_memory_service), | |
| ): | |
| """ | |
| Retrieve workspace collective knowledge summary and per-document knowledge dossiers for this client. | |
| Provides cross-document intelligence and long-term memory across sessions. | |
| """ | |
| return memory_service.get_global_knowledge(owner_id=client_id) | |
| async def refresh_global_knowledge( | |
| client_id: str = Depends(get_client_id), | |
| memory_service: MemoryService = Depends(get_memory_service), | |
| ): | |
| """ | |
| Synthesize and rebuild the global workspace knowledge memory across all indexed documents for this client. | |
| """ | |
| app_logger.info(f"Triggered manual refresh of global knowledge memory for client {client_id}") | |
| result = await memory_service.synthesize_global_knowledge(owner_id=client_id) | |
| dossiers = memory_service.list_knowledge_dossiers(owner_id=client_id) | |
| return { | |
| "message": "Global knowledge memory refreshed successfully", | |
| "collective_summary": result.get("collective_summary"), | |
| "key_insights": result.get("key_insights", []), | |
| "total_documents": result.get("total_documents", len(dossiers)), | |
| "dossiers": dossiers, | |
| } | |
| async def get_document_dossier( | |
| doc_id: str, | |
| client_id: str = Depends(get_client_id), | |
| memory_service: MemoryService = Depends(get_memory_service), | |
| ): | |
| """ | |
| Retrieve structured knowledge dossier for an individual document belonging to this client. | |
| """ | |
| dossier = memory_service.get_document_dossier(doc_id, owner_id=client_id) | |
| if not dossier: | |
| raise HTTPException(status_code=404, detail="Knowledge dossier not found for this document") | |
| return dossier | |
| async def get_chat_memory_context( | |
| chat_id: str, | |
| client_id: str = Depends(get_client_id), | |
| memory_service: MemoryService = Depends(get_memory_service), | |
| storage: MemoryStorage = Depends(get_storage), | |
| ): | |
| """ | |
| Inspect the short-term and long-term isolated memory for a specific chat session belonging to this client. | |
| """ | |
| if not storage.chat_exists(chat_id, owner_id=client_id): | |
| raise HTTPException(status_code=404, detail="Chat session not found") | |
| context = memory_service.get_session_context(chat_id, owner_id=client_id) | |
| attached_docs = storage.get_chat_docs(chat_id, owner_id=client_id) | |
| return { | |
| "chat_id": chat_id, | |
| "session_summary": context.get("session_summary"), | |
| "recent_turns_count": len(context.get("recent_turns", [])), | |
| "recent_turns": context.get("recent_turns", []), | |
| "attached_documents": attached_docs, | |
| } | |
| async def force_summarize_chat( | |
| chat_id: str, | |
| client_id: str = Depends(get_client_id), | |
| memory_service: MemoryService = Depends(get_memory_service), | |
| storage: MemoryStorage = Depends(get_storage), | |
| ): | |
| """ | |
| Force-generate or update the rolling long-term memory summary for this chat session belonging to this client. | |
| """ | |
| if not storage.chat_exists(chat_id, owner_id=client_id): | |
| raise HTTPException(status_code=404, detail="Chat session not found") | |
| summary = await memory_service.update_session_summary_if_needed(chat_id, owner_id=client_id, force=True) | |
| return { | |
| "message": "Chat session summary updated", | |
| "chat_id": chat_id, | |
| "summary": summary, | |
| } | |