You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: wavefront/server/modules/db_repo_module/db_repo_module/alembic/versions/2026_04_10_1000-e8f2a1c3b5d9_add_hnsw_index_on_embeddings.py
+8-10Lines changed: 8 additions & 10 deletions
Original file line number
Diff line number
Diff line change
@@ -26,7 +26,7 @@ def upgrade() -> None:
26
26
27
27
# HNSW index on embedding_vector for cosine distance (used in text RAG retrieval)
28
28
op.execute("""
29
-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
29
+
CREATE INDEX IF NOT EXISTS
30
30
ix_kbe_embedding_vector_hnsw_cosine
31
31
ON knowledge_base_embeddings
32
32
USING hnsw ((embedding_vector::vector(512)) vector_cosine_ops)
@@ -35,7 +35,7 @@ def upgrade() -> None:
35
35
36
36
# HNSW index on embedding_vector for L2 distance (used in CLIP image search)
37
37
op.execute("""
38
-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
38
+
CREATE INDEX IF NOT EXISTS
39
39
ix_kbe_embedding_vector_hnsw_l2
40
40
ON knowledge_base_embeddings
41
41
USING hnsw ((embedding_vector::vector(512)) vector_l2_ops)
@@ -44,7 +44,7 @@ def upgrade() -> None:
44
44
45
45
# HNSW index on embedding_vector_1 for cosine distance (used in DINO image search)
46
46
op.execute("""
47
-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
47
+
CREATE INDEX IF NOT EXISTS
48
48
ix_kbe_embedding_vector_1_hnsw_cosine
49
49
ON knowledge_base_embeddings
50
50
USING hnsw ((embedding_vector_1::vector(1024)) vector_cosine_ops)
@@ -53,17 +53,15 @@ def upgrade() -> None:
53
53
54
54
# GIN index on token column for fast full-text keyword search
55
55
op.execute("""
56
-
CREATE INDEX CONCURRENTLY IF NOT EXISTS
56
+
CREATE INDEX IF NOT EXISTS
57
57
ix_kbe_token_gin
58
58
ON knowledge_base_embeddings
59
59
USING gin (token)
60
60
""")
61
61
62
62
63
63
defdowngrade() ->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