pgvector-python
pgvector support for Python
Supports Django, SQLAlchemy, SQLModel, Psycopg 3, Psycopg 2, asyncpg, and Peewee
Installation
Run:
pip install pgvector
And follow the instructions for your database library:
Django
SQLAlchemy
SQLModel
Psycopg 3
Psycopg 2
asyncpg
Peewee
Or check out some examples:
Embeddings with OpenAI
Sentence embeddings with SentenceTransformers
Hybrid search with SentenceTransformers (Reciprocal Rank Fusion)
Hybrid search with SentenceTransformers (cross-encoder)
Sparse search with Transformers
Image search with PyTorch
Image search with perceptual hashing
Implicit feedback recommendations with Implicit
Explicit feedback recommendations with Surprise
Recommendations with LightFM
Horizontal scaling with Citus
Bulk loading with COPY
Django
Create a migration to enable the extension
from pgvector.django import VectorExtension
class Migration(migrations.Migration):
operations = [
VectorExtension()
]
Add a vector field to your model
fro
|