change embedding to None (#7355)

1 year ago · 49b2b0e3c0
parent a2830e3056
commit 49b2b0e3c0
1 changed files with 1 additions and 2 deletions
--- a/langchain/vectorstores/pgvector.py
+++ b/langchain/vectorstores/pgvector.py
@ -19,7 +19,6 @@ from langchain.vectorstores.base import VectorStore
 Base = declarative_base()  # type: Any


-ADA_TOKEN_COUNT = 1536
 _LANGCHAIN_DEFAULT_COLLECTION_NAME = "langchain"


@ -79,7 +78,7 @@ class EmbeddingStore(BaseModel):
    )
    collection = relationship(CollectionStore, back_populates="embeddings")

-    embedding: Vector = sqlalchemy.Column(Vector(ADA_TOKEN_COUNT))
+    embedding: Vector = sqlalchemy.Column(Vector(None))
    document = sqlalchemy.Column(sqlalchemy.String, nullable=True)
    cmetadata = sqlalchemy.Column(JSON, nullable=True)