|
|
@ -19,7 +19,6 @@ from langchain.vectorstores.base import VectorStore
|
|
|
|
Base = declarative_base() # type: Any
|
|
|
|
Base = declarative_base() # type: Any
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ADA_TOKEN_COUNT = 1536
|
|
|
|
|
|
|
|
_LANGCHAIN_DEFAULT_COLLECTION_NAME = "langchain"
|
|
|
|
_LANGCHAIN_DEFAULT_COLLECTION_NAME = "langchain"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -79,7 +78,7 @@ class EmbeddingStore(BaseModel):
|
|
|
|
)
|
|
|
|
)
|
|
|
|
collection = relationship(CollectionStore, back_populates="embeddings")
|
|
|
|
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)
|
|
|
|
document = sqlalchemy.Column(sqlalchemy.String, nullable=True)
|
|
|
|
cmetadata = sqlalchemy.Column(JSON, nullable=True)
|
|
|
|
cmetadata = sqlalchemy.Column(JSON, nullable=True)
|
|
|
|
|
|
|
|
|
|
|
|