github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 06:13:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: fix pgvector sqlalchemy (#14726)

Browse Source 
Fixes #14699
This commit is contained in:
Erick Friis 2023-12-14 13:27:30 -08:00 committed by GitHub
parent 1cec0afc62
commit 9fb26a2a71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/community/langchain_community/vectorstores/pgvector.py
View File

@ -60,7 +60,14 @@ class BaseModel(Base):
uuid = sqlalchemy.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) uuid = sqlalchemy.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
_classes: Any = None
def _get_embedding_collection_store() -> Any: def _get_embedding_collection_store() -> Any:
global _classes
if _classes is not None:
return _classes
from pgvector.sqlalchemy import Vector from pgvector.sqlalchemy import Vector
class CollectionStore(BaseModel): class CollectionStore(BaseModel):
@ -126,7 +133,9 @@ def _get_embedding_collection_store() -> Any:
# custom_id : any user defined id # custom_id : any user defined id
custom_id = sqlalchemy.Column(sqlalchemy.String, nullable=True) custom_id = sqlalchemy.Column(sqlalchemy.String, nullable=True)
return EmbeddingStore, CollectionStore _classes = (EmbeddingStore, CollectionStore)
return _classes
def _results_to_docs(docs_and_scores: Any) -> List[Document]: def _results_to_docs(docs_and_scores: Any) -> List[Document]: