app/core/fact_management.py

from app.core.logging_setup import logger
from app.core.db_setup import user_memory_collection

def save_user_fact(key, value):
    """Save a user-specific fact to memory."""
    try:
        user_memory_collection.update_one(
            {"key": key},
            {"$set": {"value": value}},
            upsert=True
        )
        logger.info(f"Saved user fact: {key} = {value}")
    except Exception as e:
        logger.error(f"Unexpected error: {e}")

def get_user_fact(key):
    """Retrieve a user-specific fact from memory."""
    try:
        user_fact = user_memory_collection.find_one({"key": key})
        return user_fact["value"] if user_fact else None
    except Exception as e:
        logger.error(f"Unexpected error: {e}")

    return None