Mnemex

"""Touch memory tool - reinforce a memory by updating its access time.""" import time from typing import Any from ..context import db, mcp from ..core.decay import calculate_score from ..security.validators import validate_uuid @mcp.tool() def touch_memory(memory_id: str, boost_strength: bool = False) -> dict[str, Any]: """ Reinforce a memory by updating its last accessed time and use count. This resets the temporal decay and increases the memory's resistance to being forgotten. Optionally can boost the memory's base strength. Args: memory_id: ID of the memory to reinforce (valid UUID). boost_strength: Whether to boost the base strength. Returns: Updated memory statistics including old and new scores. Raises: ValueError: If memory_id is invalid. """ # Input validation memory_id = validate_uuid(memory_id, "memory_id") memory = db.get_memory(memory_id) if memory is None: return {"success": False, "message": f"Memory not found: {memory_id}"} now = int(time.time()) old_score = calculate_score( use_count=memory.use_count, last_used=memory.last_used, strength=memory.strength, now=now, ) new_use_count = memory.use_count + 1 new_strength = memory.strength if boost_strength: new_strength = min(2.0, memory.strength + 0.1) db.update_memory( memory_id=memory_id, last_used=now, use_count=new_use_count, strength=new_strength, ) new_score = calculate_score( use_count=new_use_count, last_used=now, strength=new_strength, now=now, ) return { "success": True, "memory_id": memory_id, "old_score": round(old_score, 4), "new_score": round(new_score, 4), "use_count": new_use_count, "strength": round(new_strength, 4), "message": f"Memory reinforced. Score: {old_score:.2f} -> {new_score:.2f}", }