Merge pull request #13232

* PGVector needs to close its connection if its garbage collected
langchain-ai · Nov 15, 2023 · 1726d5d · 1726d5d
1 parent 85a77d2
commit 1726d5d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/langchain/langchain/vectorstores/pgvector.py b/libs/langchain/langchain/vectorstores/pgvector.py
@@ -147,6 +147,10 @@ def __post_init__(
         self.create_tables_if_not_exists()
         self.create_collection()
 
+    def __del__(self) -> None:
+        if self._conn:
+            self._conn.close()
+
     @property
     def embeddings(self) -> Embeddings:
         return self.embedding_function