x

src/database/init_db.py

@@ -185,6 +185,18 @@ async def ensure_schema_updates():
             CREATE UNIQUE INDEX IF NOT EXISTS uq_product_material_unique
             ON product_materials (finished_product_id, material_product_id)
         """))
+        await conn.execute(text("ALTER TABLE mold_cavity_data ADD COLUMN IF NOT EXISTS best_scheme_id VARCHAR(64)"))
+        await conn.execute(text("ALTER TABLE mold_cavity_data ADD COLUMN IF NOT EXISTS confidence_score DOUBLE PRECISION"))
+        await conn.execute(text("ALTER TABLE mold_cavity_data ADD COLUMN IF NOT EXISTS is_fallback BOOLEAN"))
+        await conn.execute(text("ALTER TABLE mold_cavity_data ADD COLUMN IF NOT EXISTS fallback_reason TEXT"))
+        await conn.execute(text("""
+            CREATE INDEX IF NOT EXISTS idx_mold_cavity_best_scheme_id
+            ON mold_cavity_data (best_scheme_id)
+        """))
+        await conn.execute(text("""
+            CREATE INDEX IF NOT EXISTS idx_mold_cavity_is_fallback
+            ON mold_cavity_data (is_fallback)
+        """))
 
 
 if __name__ == "__main__":