Skip to content
This repository was archived by the owner on Jul 22, 2025. It is now read-only.

Commit cd03874

Browse files
authored
FIX: Missing table check in post_migration (#1068)
1 parent 65456c8 commit cd03874

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

db/post_migrate/20250114184356_drop_old_embedding_tables2.rb

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,22 @@
11
# frozen_string_literal: true
22
class DropOldEmbeddingTables2 < ActiveRecord::Migration[7.2]
33
def up
4-
# Copy rag embeddings created during deploy.
5-
execute <<~SQL
6-
INSERT INTO ai_document_fragments_embeddings (rag_document_fragment_id, model_id, model_version, strategy_id, strategy_version, digest, embeddings, created_at, updated_at)
7-
(
8-
SELECT old_table.*
9-
FROM ai_document_fragment_embeddings old_table
10-
LEFT OUTER JOIN ai_document_fragments_embeddings target ON (
11-
target.model_id = old_table.model_id AND
12-
target.strategy_id = old_table.strategy_id AND
13-
target.rag_document_fragment_id = old_table.rag_document_fragment_id
4+
if table_exists?(:ai_document_fragment_embeddings)
5+
# Copy rag embeddings created during deploy.
6+
execute <<~SQL
7+
INSERT INTO ai_document_fragments_embeddings (rag_document_fragment_id, model_id, model_version, strategy_id, strategy_version, digest, embeddings, created_at, updated_at)
8+
(
9+
SELECT old_table.*
10+
FROM ai_document_fragment_embeddings old_table
11+
LEFT OUTER JOIN ai_document_fragments_embeddings target ON (
12+
target.model_id = old_table.model_id AND
13+
target.strategy_id = old_table.strategy_id AND
14+
target.rag_document_fragment_id = old_table.rag_document_fragment_id
15+
)
16+
WHERE target.rag_document_fragment_id IS NULL
1417
)
15-
WHERE target.rag_document_fragment_id IS NULL
16-
)
17-
SQL
18+
SQL
19+
end
1820

1921
execute <<~SQL
2022
DROP INDEX IF EXISTS ai_topic_embeddings_1_1_search_bit;

0 commit comments

Comments
 (0)