Skip to content

Commit 82ec696

Browse files
committed
fix for migration
1 parent e5dbd45 commit 82ec696

1 file changed

Lines changed: 8 additions & 10 deletions

File tree

wavefront/server/modules/db_repo_module/db_repo_module/alembic/versions/2026_04_10_1000-e8f2a1c3b5d9_add_hnsw_index_on_embeddings.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def upgrade() -> None:
2626

2727
# HNSW index on embedding_vector for cosine distance (used in text RAG retrieval)
2828
op.execute("""
29-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
29+
CREATE INDEX IF NOT EXISTS
3030
ix_kbe_embedding_vector_hnsw_cosine
3131
ON knowledge_base_embeddings
3232
USING hnsw ((embedding_vector::vector(512)) vector_cosine_ops)
@@ -35,7 +35,7 @@ def upgrade() -> None:
3535

3636
# HNSW index on embedding_vector for L2 distance (used in CLIP image search)
3737
op.execute("""
38-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
38+
CREATE INDEX IF NOT EXISTS
3939
ix_kbe_embedding_vector_hnsw_l2
4040
ON knowledge_base_embeddings
4141
USING hnsw ((embedding_vector::vector(512)) vector_l2_ops)
@@ -44,7 +44,7 @@ def upgrade() -> None:
4444

4545
# HNSW index on embedding_vector_1 for cosine distance (used in DINO image search)
4646
op.execute("""
47-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
47+
CREATE INDEX IF NOT EXISTS
4848
ix_kbe_embedding_vector_1_hnsw_cosine
4949
ON knowledge_base_embeddings
5050
USING hnsw ((embedding_vector_1::vector(1024)) vector_cosine_ops)
@@ -53,17 +53,15 @@ def upgrade() -> None:
5353

5454
# GIN index on token column for fast full-text keyword search
5555
op.execute("""
56-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
56+
CREATE INDEX IF NOT EXISTS
5757
ix_kbe_token_gin
5858
ON knowledge_base_embeddings
5959
USING gin (token)
6060
""")
6161

6262

6363
def downgrade() -> None:
64-
op.execute('DROP INDEX CONCURRENTLY IF EXISTS ix_kbe_embedding_vector_hnsw_cosine')
65-
op.execute('DROP INDEX CONCURRENTLY IF EXISTS ix_kbe_embedding_vector_hnsw_l2')
66-
op.execute(
67-
'DROP INDEX CONCURRENTLY IF EXISTS ix_kbe_embedding_vector_1_hnsw_cosine'
68-
)
69-
op.execute('DROP INDEX CONCURRENTLY IF EXISTS ix_kbe_token_gin')
64+
op.execute('DROP INDEX IF EXISTS ix_kbe_embedding_vector_hnsw_cosine')
65+
op.execute('DROP INDEX IF EXISTS ix_kbe_embedding_vector_hnsw_l2')
66+
op.execute('DROP INDEX IF EXISTS ix_kbe_embedding_vector_1_hnsw_cosine')
67+
op.execute('DROP INDEX IF EXISTS ix_kbe_token_gin')

0 commit comments

Comments
 (0)