fix: pgvector hang risk if table not exist #3883 (#3884)

1 year ago · 188a7bd653
parent 9acf80fd69
commit 188a7bd653
1 changed files with 2 additions and 0 deletions
--- a/langchain/vectorstores/pgvector.py
+++ b/langchain/vectorstores/pgvector.py
@ -165,9 +165,11 @@ class PGVector(VectorStore):

    def create_tables_if_not_exists(self) -> None:
        Base.metadata.create_all(self._conn)
+        self._conn.commit()

    def drop_tables(self) -> None:
        Base.metadata.drop_all(self._conn)
+        self._conn.commit()

    def create_collection(self) -> None:
        if self.pre_delete_collection: