check_sqlite_vec_status.py•5.14 kB
#!/usr/bin/env python3
"""
Check the status of SQLite-vec database and identify issues.
"""
import sqlite3
import sys
import os
def check_sqlite_vec_status(db_path):
"""Check the status of the SQLite-vec database."""
print(f"Checking SQLite-vec database: {db_path}")
print("="*60)
if not os.path.exists(db_path):
print(f"❌ Database not found: {db_path}")
return False
conn = sqlite3.connect(db_path)
cursor = conn.cursor()
issues_found = []
try:
# Check basic tables
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name")
tables = [row[0] for row in cursor.fetchall()]
print(f"Tables: {', '.join(tables)}")
if 'memories' not in tables:
issues_found.append("Missing 'memories' table")
else:
cursor.execute("SELECT COUNT(*) FROM memories")
memory_count = cursor.fetchone()[0]
print(f"📝 Memories: {memory_count}")
if 'memory_embeddings' not in tables:
issues_found.append("Missing 'memory_embeddings' table")
else:
# Try to access the embeddings table
try:
cursor.execute("SELECT COUNT(*) FROM memory_embeddings")
embedding_count = cursor.fetchone()[0]
print(f"🧠 Embeddings: {embedding_count}")
# Check if counts match
if 'memories' in tables:
if memory_count != embedding_count:
issues_found.append(f"Count mismatch: {memory_count} memories vs {embedding_count} embeddings")
except Exception as e:
if "no such module: vec0" in str(e):
issues_found.append("sqlite-vec extension not loaded - cannot access embeddings")
else:
issues_found.append(f"Cannot access embeddings table: {e}")
# Check if extension loading is possible
try:
conn.enable_load_extension(True)
extension_support = True
except:
extension_support = False
issues_found.append("Extension loading not supported")
print(f"Extension loading: {'✅ Supported' if extension_support else '❌ Not supported'}")
# Try to load sqlite-vec
if extension_support:
try:
# This will fail if sqlite-vec is not installed
import sqlite_vec
sqlite_vec.load(conn)
print("✅ sqlite-vec extension loaded successfully")
# Now try to access embeddings
try:
cursor.execute("SELECT COUNT(*) FROM memory_embeddings")
embedding_count = cursor.fetchone()[0]
print(f"✅ Can now access embeddings: {embedding_count}")
# Test a simple search
if embedding_count > 0:
cursor.execute("SELECT * FROM memory_embeddings LIMIT 1")
row = cursor.fetchone()
print("✅ Embedding data accessible")
except Exception as e:
issues_found.append(f"Still cannot access embeddings after loading extension: {e}")
except ImportError:
issues_found.append("sqlite-vec Python module not installed")
except Exception as e:
issues_found.append(f"Failed to load sqlite-vec extension: {e}")
except Exception as e:
issues_found.append(f"Database error: {e}")
finally:
conn.close()
print("\n" + "="*60)
if issues_found:
print("⚠️ Issues Found:")
for i, issue in enumerate(issues_found, 1):
print(f" {i}. {issue}")
print("\nRecommendations:")
if "sqlite-vec Python module not installed" in str(issues_found):
print(" • Install sqlite-vec: uv pip install sqlite-vec")
if "sentence-transformers" in str(issues_found) or "embedding" in str(issues_found).lower():
print(" • Install sentence-transformers: uv pip install sentence-transformers torch")
if "Count mismatch" in str(issues_found):
print(" • Run repair script to regenerate missing embeddings")
if "cannot access embeddings" in str(issues_found).lower():
print(" • Database may need migration to fix schema issues")
else:
print("✅ No issues found - database appears healthy!")
return len(issues_found) == 0
def main():
if len(sys.argv) != 2:
print("Usage: python check_sqlite_vec_status.py <database_path>")
sys.exit(1)
db_path = sys.argv[1]
healthy = check_sqlite_vec_status(db_path)
sys.exit(0 if healthy else 1)
if __name__ == "__main__":
main()