Record uploaded assets in conversation packages

db.mjs

@@ -346,6 +346,22 @@ export async function migrateSchema(pool) {
     ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
   `);
 
+  const uploadSessionColumns = [
+    ['source_session_id', 'VARCHAR(128) NULL AFTER completed_asset_id'],
+    ['source_message_id', 'VARCHAR(128) NULL AFTER source_session_id'],
+  ];
+  for (const [column, definition] of uploadSessionColumns) {
+    if (!(await columnExists(pool, 'h5_upload_sessions', column))) {
+      await pool.query(`ALTER TABLE h5_upload_sessions ADD COLUMN \`${column}\` ${definition}`);
+    }
+  }
+  if (!(await indexExists(pool, 'h5_upload_sessions', 'idx_h5_upload_source_session'))) {
+    await pool.query(
+      `ALTER TABLE h5_upload_sessions
+       ADD KEY idx_h5_upload_source_session (user_id, source_session_id)`,
+    );
+  }
+
   await pool.query(`
     CREATE TABLE IF NOT EXISTS h5_user_memory_items (
       id CHAR(64) PRIMARY KEY,