reset_memory.py 820 B

1234567891011121314151617181920212223
  1. import sys, subprocess, requests, os
  2. chroma_base = "http://192.168.0.200:8001/api/v1"
  3. name = sys.argv[1]
  4. # 1. Reset Chroma collection
  5. requests.delete(f"{chroma_base}/collections/{name}")
  6. requests.post(f"{chroma_base}/collections", json={"name": name})
  7. print(f"chroma collection reset: {name}")
  8. # 2. Wipe mem0's SQLite so integer IDs don't drift from UUIDs
  9. result = subprocess.run(
  10. ["docker", "exec", "mem0server", "find", "/", "-name", "*.db"],
  11. capture_output=True, text=True
  12. )
  13. for path in result.stdout.strip().splitlines():
  14. if "mem0" in path.lower():
  15. subprocess.run(["docker", "exec", "mem0server", "rm", "-f", path])
  16. print(f"removed sqlite: {path}")
  17. # 3. Restart so mem0 reinitialises cleanly
  18. subprocess.run(["docker", "compose", "restart", "mem0server"])
  19. print("mem0server restarted")