import sys, subprocess, requests, os

chroma_base = "http://192.168.0.200:8001/api/v1"
name = sys.argv[1]

# 1. Reset Chroma collection
requests.delete(f"{chroma_base}/collections/{name}")
requests.post(f"{chroma_base}/collections", json={"name": name})
print(f"chroma collection reset: {name}")

# 2. Wipe mem0's SQLite so integer IDs don't drift from UUIDs
result = subprocess.run(
    ["docker", "exec", "mem0server", "find", "/", "-name", "*.db"],
    capture_output=True, text=True
)
for path in result.stdout.strip().splitlines():
    if "mem0" in path.lower():
        subprocess.run(["docker", "exec", "mem0server", "rm", "-f", path])
        print(f"removed sqlite: {path}")

# 3. Restart so mem0 reinitialises cleanly
subprocess.run(["docker", "compose", "restart", "mem0server"])
print("mem0server restarted")