56 lines
2.7 KiB
Ruby
56 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
class DropOldEmbeddingTables2 < ActiveRecord::Migration[7.2]
|
|
def up
|
|
if table_exists?(:ai_document_fragment_embeddings)
|
|
# 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
|
|
end
|
|
|
|
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, if_exists: true
|
|
drop_table :ai_post_embeddings, if_exists: true
|
|
drop_table :ai_document_fragment_embeddings, if_exists: true
|
|
end
|
|
|
|
def down
|
|
end
|
|
end
|