pgvector-haystack 8.0.0__tar.gz → 8.1.0__tar.gz

{pgvector_haystack-8.0.0 → pgvector_haystack-8.1.0}/CHANGELOG.md

@@ -1,5 +1,17 @@
 # Changelog
 
+## [integrations/pgvector-v8.0.0] - 2026-10-16
+
+### 🧪 Testing
+
+- Test compatible integrations with Python 3.15; update pyproject (#4104)
+
+### 🧹 Chores
+
+- Drop redundant deserialize_secrets_inplace calls in integrations requiring haystack-ai>=2.23 (#4058)
+- [**breaking**] Drop Python 3.10 support (#4105)
+
+
 ## [integrations/pgvector-v7.0.0] - 2026-09-29
 
 ### 🐛 Bug Fixes

{pgvector_haystack-8.0.0 → pgvector_haystack-8.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: pgvector-haystack
-Version: 8.0.0
+Version: 8.1.0
 Summary: An integration of pgvector (vector search extension for Postgres) with Haystack
 Project-URL: Source, https://github.com/deepset-ai/haystack-core-integrations
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/pgvector/README.md

{pgvector_haystack-8.0.0 → pgvector_haystack-8.1.0}/src/haystack_integrations/document_stores/pgvector/document_store.py

@@ -370,6 +370,47 @@ class PgvectorDocumentStore:
 
         return result
 
+    def _create_connection(self) -> Connection:
+        """
+        Opens a new connection to the PostgreSQL database.
+
+        Subclasses can override this method to connect differently (e.g. through a cloud provider's connector).
+        It must only open and return the connection: autocommit, the pgvector extension, vector type
+        registration, cursors, and table initialization are handled by `_ensure_db_setup`.
+
+        :raises DocumentStoreError: If the connection cannot be opened.
+        """
+        conn_str = self.connection_string.resolve_value() or ""
+        try:
+            return Connection.connect(conn_str)
+        except Error as e:
+            msg = (
+                "Failed to connect to PostgreSQL database.  Ensure the connection string follows the "
+                "PostgreSQL connection specification: "
+                "https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING."
+            )
+            raise DocumentStoreError(msg) from e
+
+    async def _create_connection_async(self) -> AsyncConnection:
+        """
+        Opens a new async connection to the PostgreSQL database.
+
+        Async counterpart of `_create_connection`, overridable in the same way. Everything else is handled by
+        `_ensure_db_setup_async`.
+
+        :raises DocumentStoreError: If the connection cannot be opened.
+        """
+        conn_str = self.connection_string.resolve_value() or ""
+        try:
+            return await AsyncConnection.connect(conn_str)
+        except Error as e:
+            msg = (
+                "Failed to connect to PostgreSQL database.  Ensure the connection string follows the "
+                "PostgreSQL connection specification: "
+                "https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING."
+            )
+            raise DocumentStoreError(msg) from e
+
     def _ensure_db_setup(self) -> None:
         """
         Ensures that the connection to the PostgreSQL database exists and is valid.
@@ -387,16 +428,7 @@ class PgvectorDocumentStore:
             except Error as e:
                 logger.debug("Failed to close connection: {e}", e=str(e))
 
-        conn_str = self.connection_string.resolve_value() or ""
-        try:
-            connection = Connection.connect(conn_str)
-        except Error as e:
-            msg = (
-                "Failed to connect to PostgreSQL database.  Ensure the connection string follows the "
-                "PostgreSQL connection specification: "
-                "https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING."
-            )
-            raise DocumentStoreError(msg) from e
+        connection = self._create_connection()
         connection.autocommit = True
         if self.create_extension:
             connection.execute("CREATE EXTENSION IF NOT EXISTS vector")
@@ -430,16 +462,7 @@ class PgvectorDocumentStore:
         if self._async_connection:
             await self._async_connection.close()
 
-        conn_str = self.connection_string.resolve_value() or ""
-        try:
-            async_connection = await AsyncConnection.connect(conn_str)
-        except Error as e:
-            msg = (
-                "Failed to connect to PostgreSQL database.  Ensure the connection string follows the "
-                "PostgreSQL connection specification: "
-                "https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING."
-            )
-            raise DocumentStoreError(msg) from e
+        async_connection = await self._create_connection_async()
         await async_connection.set_autocommit(True)
         if self.create_extension:
             await async_connection.execute("CREATE EXTENSION IF NOT EXISTS vector")

{pgvector_haystack-8.0.0 → pgvector_haystack-8.1.0}/tests/test_document_store.py

@@ -448,6 +448,27 @@ def test_close_is_exception_safe(mock_store):
     assert mock_store._connection is None
 
 
+@pytest.mark.usefixtures("patches_for_unit_tests")
+def test_ensure_db_setup_uses_overridden_create_connection(monkeypatch):
+    monkeypatch.setenv("PG_CONN_STR", "some-connection-string")
+    mock_connection = Mock(spec=Connection)
+
+    class CustomConnectionDocumentStore(PgvectorDocumentStore):
+        def _create_connection(self) -> Connection:
+            return mock_connection
+
+    store = CustomConnectionDocumentStore()
+    with patch.object(CustomConnectionDocumentStore, "_initialize_table") as mock_initialize_table:
+        store._ensure_db_setup()
+
+    assert store._connection is mock_connection
+    assert mock_connection.autocommit is True
+    mock_connection.execute.assert_called_once_with("CREATE EXTENSION IF NOT EXISTS vector")
+    assert store._cursor is not None
+    assert store._dict_cursor is not None
+    mock_initialize_table.assert_called_once()
+
+
 @pytest.mark.integration
 def test_halfvec_hnsw_write_documents(document_store_w_halfvec_hnsw_index: PgvectorDocumentStore):
     documents = [

{pgvector_haystack-8.0.0 → pgvector_haystack-8.1.0}/tests/test_document_store_async.py

@@ -409,3 +409,30 @@ async def test_close_async_is_exception_safe(mock_store):
     await mock_store.close_async()
 
     assert mock_store._async_connection is None
+
+
+@pytest.mark.asyncio
+async def test_ensure_db_setup_async_uses_overridden_create_connection_async(monkeypatch):
+    monkeypatch.setenv("PG_CONN_STR", "some-connection-string")
+    mock_connection = Mock(spec=AsyncConnection)
+    mock_connection.set_autocommit = AsyncMock()
+    mock_connection.execute = AsyncMock()
+
+    class CustomConnectionDocumentStore(PgvectorDocumentStore):
+        async def _create_connection_async(self) -> AsyncConnection:
+            return mock_connection
+
+    store = CustomConnectionDocumentStore()
+    with (
+        patch("haystack_integrations.document_stores.pgvector.document_store.register_vector_async") as mock_register,
+        patch.object(CustomConnectionDocumentStore, "_initialize_table_async") as mock_initialize_table,
+    ):
+        await store._ensure_db_setup_async()
+
+    assert store._async_connection is mock_connection
+    mock_connection.set_autocommit.assert_awaited_once_with(True)
+    mock_connection.execute.assert_awaited_once_with("CREATE EXTENSION IF NOT EXISTS vector")
+    mock_register.assert_awaited_once_with(mock_connection)
+    assert store._async_cursor is not None
+    assert store._async_dict_cursor is not None
+    mock_initialize_table.assert_awaited_once()