inference works

test_db_migration.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+"""
+Test script to verify database migration works correctly
+"""
+
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from utils.database_manager import get_database_manager, reset_database_manager
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def test_migration():
+    """Test the database migration"""
+    try:
+        logger.info("Testing database migration...")
+        
+        # Reset the database manager to force re-initialization
+        reset_database_manager()
+        
+        # Get a new instance (this will trigger migration)
+        db_manager = get_database_manager()
+        
+        # Test if we can access the input_features_blob column
+        with db_manager._get_connection() as conn:
+            cursor = conn.execute("PRAGMA table_info(inference_records)")
+            columns = [row[1] for row in cursor.fetchall()]
+            
+            if 'input_features_blob' in columns:
+                logger.info("✅ input_features_blob column exists - migration successful!")
+                return True
+            else:
+                logger.error("❌ input_features_blob column missing - migration failed!")
+                return False
+                
+    except Exception as e:
+        logger.error(f"❌ Migration test failed: {e}")
+        return False
+
+if __name__ == "__main__":
+    success = test_migration()
+    sys.exit(0 if success else 1)