#!/usr/bin/env python3
"""
Clean up collection with wrong dimensions
"""

import vecs

def cleanup_wrong_dimensions():
    """Clean up the collection with wrong dimensions"""
    print("🧹 Cleaning up collection with wrong dimensions...")
    
    connection_string = "postgresql://supabase_admin:CzkaYmRvc26Y@localhost:5435/postgres"
    db = vecs.create_client(connection_string)
    
    try:
        # Delete the collection with wrong dimensions
        db.delete_collection("mem0_working_test")
        print("✅ Deleted mem0_working_test collection")
    except Exception as e:
        print(f"⚠️  Could not delete collection: {e}")
    
    # Verify cleanup
    collections = db.list_collections()
    print(f"Remaining collections: {[c.name for c in collections]}")

if __name__ == "__main__":
    cleanup_wrong_dimensions()