DEV: Migration to remove old embeddings tables~

Author: romanrizzi

db/post_migrate/20250114184356_drop_old_embedding_tables2.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+class DropOldEmbeddingTables2 < ActiveRecord::Migration[7.2]
+  def up
+    # Copy rag embeddings created during deploy.
+    execute <<~SQL
+        INSERT INTO ai_document_fragments_embeddings (rag_document_fragment_id, model_id, model_version, strategy_id, strategy_version, digest, embeddings, created_at, updated_at)
+        (
+          SELECT  old_table.*
+          FROM ai_document_fragment_embeddings old_table
+          LEFT OUTER JOIN ai_document_fragments_embeddings target ON (
+            target.model_id = old_table.model_id AND
+            target.strategy_id = old_table.strategy_id AND
+            target.rag_document_fragment_id = old_table.rag_document_fragment_id
+          )
+          WHERE target.rag_document_fragment_id IS NULL
+        )
+      SQL
+
+    execute <<~SQL
+        DROP INDEX IF EXISTS ai_topic_embeddings_1_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_2_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_3_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_4_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_5_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_6_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_7_1_search_bit;
+        DROP INDEX IF EXISTS ai_topic_embeddings_8_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_1_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_2_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_3_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_4_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_5_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_6_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_7_1_search_bit;
+        DROP INDEX IF EXISTS ai_post_embeddings_8_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_1_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_2_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_3_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_4_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_5_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_6_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_7_1_search_bit;
+        DROP INDEX IF EXISTS ai_document_fragment_embeddings_8_1_search_bit;
+      SQL
+    drop_table :ai_topic_embeddings
+    drop_table :ai_post_embeddings
+    drop_table :ai_document_fragment_embeddings
+  end
+
+  def down
+  end
+end