Skip to content

perf: add hnsw vector index for pgvector #76

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ademboukabes merged 1 commit into from
Aug 5, 2026
Merged

perf: add hnsw vector index for pgvector #76

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 30 additions & 0 deletions migrations/versions/9ec59aeb192b_add_hnsw_indexes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
"""add hnsw indexes

Revision ID: 9ec59aeb192b
Revises: 46bceaf84bd8
Create Date: 2026-08-05 17:25:20.815182

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '9ec59aeb192b'
down_revision: Union[str, Sequence[str], None] = '46bceaf84bd8'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
"""Upgrade schema."""
op.execute("CREATE INDEX IF NOT EXISTS idx_users_face_embedding ON users USING hnsw (face_embedding vector_cosine_ops);")
op.execute("CREATE INDEX IF NOT EXISTS idx_photo_faces_embedding ON photo_faces USING hnsw (embedding vector_cosine_ops);")


def downgrade() -> None:
"""Downgrade schema."""
op.execute("DROP INDEX IF EXISTS idx_photo_faces_embedding;")
op.execute("DROP INDEX IF EXISTS idx_users_face_embedding;")