src/api/routes/system.py

# src/api/routes/system.py

import time

from fastapi import APIRouter, Depends

from src.core.state import AppState, get_state
from src.data.connection import Collections

router = APIRouter(prefix="/system", tags=["System"])

@router.get("/health")
async def health_check(state: AppState = Depends(get_state)):
	"""Health check endpoint"""
	return {
		"status": "healthy",
		"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
		"components": {
			"memory_system": "operational",
			"embedding_client": "operational" if state.embedding_client.is_available() else "fallback_mode",
			"api_rotator": "operational",
			"gemini_keys_available": len([k for k in state.gemini_rotator.keys if k]) > 0,
			"nvidia_keys_available": len([k for k in state.nvidia_rotator.keys if k]) > 0
		}
	}

@router.get("/database")
async def get_database():
	"""List meta information about all collections in the database."""
	from src.data.connection import get_collection

	collections = [
		Collections.ACCOUNT,
		Collections.PATIENT,
		Collections.SESSION,
		Collections.MEDICAL_RECORDS,
		Collections.MEDICAL_MEMORY
	]

	result = {}
	for name in collections:
		collection = get_collection(name)
		indexes = []
		for idx in collection.list_indexes():
			indexes.append({
				"name": idx["name"],
				"keys": list(idx["key"].items())
			})

		stats = collection.estimated_document_count()
		# Get sample document to extract field names
		sample = collection.find_one()
		fields = list(sample.keys()) if sample else []

		result[name] = {
			"document_count": stats,
			"indexes": indexes,
			"fields": fields
		}

	return {
		"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
		"collections": result
	}