#!/usr/bin/env python3
import os
from pathlib import Path

# Remove old checkpoints
files_to_remove = [
    "storage/monika_fresh.pt",
    "storage/proto_lm/proto_lm.json",
]

for f in files_to_remove:
    p = Path(f)
    if p.exists():
        p.unlink()
        print(f"Deleted: {f}")
    else:
        print(f"Not found: {f}")

print("Cleanup complete - MCP will now use storage/proto_lm/checkpoint.pt")