From 49b2b0e3c0089645f577389330d350bf16012f53 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Fri, 7 Jul 2023 12:33:03 -0400 Subject: [PATCH] change embedding to None (#7355) --- langchain/vectorstores/pgvector.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/langchain/vectorstores/pgvector.py b/langchain/vectorstores/pgvector.py index d124808eed..2825fc16c8 100644 --- 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)