Skip to content

Commit 05387da

Browse files
committed
Updated examples [skip ci]
1 parent 2968f25 commit 05387da

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

README.md

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -177,10 +177,10 @@ session.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
177177
Add a vector column
178178

179179
```python
180-
from pgvector.sqlalchemy import Vector
180+
from pgvector.sqlalchemy import VECTOR
181181

182182
class Item(Base):
183-
embedding = mapped_column(Vector(3))
183+
embedding = mapped_column(VECTOR(3))
184184
```
185185

186186
Also supports `HALFVEC`, `BIT`, and `SPARSEVEC`
@@ -306,11 +306,11 @@ session.scalars(select(subquery).order_by(subquery.c.embedding.cosine_distance([
306306
Add an array column
307307

308308
```python
309-
from pgvector.sqlalchemy import Vector
309+
from pgvector.sqlalchemy import VECTOR
310310
from sqlalchemy import ARRAY
311311

312312
class Item(Base):
313-
embeddings = mapped_column(ARRAY(Vector(3)))
313+
embeddings = mapped_column(ARRAY(VECTOR(3)))
314314
```
315315

316316
And register the types with the underlying driver
@@ -359,10 +359,10 @@ session.exec(text('CREATE EXTENSION IF NOT EXISTS vector'))
359359
Add a vector column
360360

361361
```python
362-
from pgvector.sqlalchemy import Vector
362+
from pgvector.sqlalchemy import VECTOR
363363

364364
class Item(SQLModel, table=True):
365-
embedding: Any = Field(sa_type=Vector(3))
365+
embedding: Any = Field(sa_type=VECTOR(3))
366366
```
367367

368368
Also supports `HALFVEC`, `BIT`, and `SPARSEVEC`

0 commit comments

Comments
 (0)