pgvector-haystack 5.2.1__tar.gz → 5.3.0__tar.gz

{pgvector_haystack-5.2.1 → pgvector_haystack-5.3.0}/CHANGELOG.md

@@ -1,13 +1,16 @@
 # Changelog
 
-## [integrations/pgvector-v5.2.0] - 2025-06-26
+## [integrations/pgvector-v5.2.1] - 2025-07-03
+
+### 🐛 Bug Fixes
+
+- Pgvector - ensure DB is initialized when calling `delete_table` (#2055)
 
-Note: Version 5.2.0 was released by mistake.
-It is functionally identical to version 3.4.1 and does not introduce any new features or changes.
 
 ### 🧹 Chores
 
 - Remove black (#1985)
+- Pgvector - update docker image in examples (#1995)
 
 
 ## [integrations/pgvector-v3.4.1] - 2025-06-20

{pgvector_haystack-5.2.1 → pgvector_haystack-5.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgvector-haystack
-Version: 5.2.1
+Version: 5.3.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
@@ -29,34 +29,16 @@ Description-Content-Type: text/markdown
 [![PyPI - Version](https://img.shields.io/pypi/v/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
 
----
+- [Integration page](https://haystack.deepset.ai/integrations/pgvector-documentstore)
+- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/pgvector/CHANGELOG.md)
 
-**Table of Contents**
+---
 
-- [pgvector-haystack](#pgvector-haystack)
-  - [Installation](#installation)
-  - [Testing](#testing)
-  - [License](#license)
+Refer to the general [Contribution Guidelines](https://github.com/deepset-ai/haystack-core-integrations/blob/main/CONTRIBUTING.md).
 
-## Installation
+To run integration tests locally, you need to have a PostgreSQL DB running with the `pgvector` extension.
+You can start it using Docker:
 
 ```console
-pip install pgvector-haystack
-```
-
-## Testing
-
-Ensure that you have a PostgreSQL running with the `pgvector` extension. For a quick setup using Docker, run:
-```
 docker run -d -p 5432:5432 -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres -e POSTGRES_DB=postgres pgvector/pgvector:pg17
 ```
-
-then run the tests:
-
-```console
-hatch run test:all
-```
-
-## License
-
-`pgvector-haystack` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.

pgvector_haystack-5.3.0/README.md

@@ -0,0 +1,18 @@
+# pgvector-haystack
+
+[![PyPI - Version](https://img.shields.io/pypi/v/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
+
+- [Integration page](https://haystack.deepset.ai/integrations/pgvector-documentstore)
+- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/pgvector/CHANGELOG.md)
+
+---
+
+Refer to the general [Contribution Guidelines](https://github.com/deepset-ai/haystack-core-integrations/blob/main/CONTRIBUTING.md).
+
+To run integration tests locally, you need to have a PostgreSQL DB running with the `pgvector` extension.
+You can start it using Docker:
+
+```console
+docker run -d -p 5432:5432 -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres -e POSTGRES_DB=postgres pgvector/pgvector:pg17
+```

{pgvector_haystack-5.2.1 → pgvector_haystack-5.3.0}/src/haystack_integrations/document_stores/pgvector/document_store.py

@@ -987,6 +987,38 @@ class PgvectorDocumentStore:
             error_msg="Could not delete documents from PgvectorDocumentStore",
         )
 
+    def delete_all_documents(self) -> None:
+        """
+        Deletes all documents in the document store.
+        """
+        query = SQL("TRUNCATE TABLE {schema_name}.{table_name}").format(
+            schema_name=Identifier(self.schema_name),
+            table_name=Identifier(self.table_name),
+        )
+
+        self._ensure_db_setup()
+        assert self._cursor is not None
+        self._execute_sql(
+            cursor=self._cursor, sql_query=query, error_msg="Could not delete all documents from PgvectorDocumentStore"
+        )
+
+    async def delete_all_documents_async(self) -> None:
+        """
+        Asynchronously deletes all documents in the document store.
+        """
+        query = SQL("TRUNCATE TABLE {schema_name}.{table_name}").format(
+            schema_name=Identifier(self.schema_name),
+            table_name=Identifier(self.table_name),
+        )
+
+        await self._ensure_db_setup_async()
+        assert self._async_cursor is not None
+        await self._execute_sql_async(
+            cursor=self._async_cursor,
+            sql_query=query,
+            error_msg="Could not delete all documents from PgvectorDocumentStore",
+        )
+
     def _build_keyword_retrieval_query(
         self, query: str, top_k: int, filters: Optional[Dict[str, Any]] = None
     ) -> Tuple[Composed, tuple]:

{pgvector_haystack-5.2.1 → pgvector_haystack-5.3.0}/tests/test_document_store.py

@@ -53,6 +53,13 @@ class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsT
             same_connection = document_store._connection
             assert same_connection is document_store._connection
 
+    def test_delete_all_documents(self, document_store: PgvectorDocumentStore) -> None:
+        document_store.write_documents([Document(id=str(i)) for i in range(10)])
+        document_store.delete_all_documents()
+        assert document_store.count_documents() == 0
+        document_store.write_documents([Document(id="1")])
+        assert document_store.count_documents() == 1
+
 
 @pytest.mark.usefixtures("patches_for_unit_tests")
 def test_init(monkeypatch):

{pgvector_haystack-5.2.1 → pgvector_haystack-5.3.0}/tests/test_document_store_async.py

@@ -94,6 +94,13 @@ class TestDocumentStoreAsync:
             same_connection = document_store._async_connection
             assert same_connection is document_store._async_connection
 
+    async def test_delete_all_documents_async(self, document_store: PgvectorDocumentStore) -> None:
+        document_store.write_documents([Document(id=str(i)) for i in range(10)])
+        await document_store.delete_all_documents_async()
+        assert document_store.count_documents() == 0
+        document_store.write_documents([Document(id="1")])
+        assert document_store.count_documents() == 1
+
 
 @pytest.mark.integration
 @pytest.mark.asyncio

pgvector_haystack-5.2.1/README.md

@@ -1,36 +0,0 @@
-# pgvector-haystack
-
-[![PyPI - Version](https://img.shields.io/pypi/v/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
-[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pgvector-haystack.svg)](https://pypi.org/project/pgvector-haystack)
-
----
-
-**Table of Contents**
-
-- [pgvector-haystack](#pgvector-haystack)
-  - [Installation](#installation)
-  - [Testing](#testing)
-  - [License](#license)
-
-## Installation
-
-```console
-pip install pgvector-haystack
-```
-
-## Testing
-
-Ensure that you have a PostgreSQL running with the `pgvector` extension. For a quick setup using Docker, run:
-```
-docker run -d -p 5432:5432 -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres -e POSTGRES_DB=postgres pgvector/pgvector:pg17
-```
-
-then run the tests:
-
-```console
-hatch run test:all
-```
-
-## License
-
-`pgvector-haystack` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.