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

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

@@ -1,5 +1,18 @@
 # Changelog
 
+## [integrations/pgvector-v5.2.1] - 2025-07-03
+
+### 🐛 Bug Fixes
+
+- Pgvector - ensure DB is initialized when calling `delete_table` (#2055)
+
+
+### 🧹 Chores
+
+- Remove black (#1985)
+- Pgvector - update docker image in examples (#1995)
+
+
 ## [integrations/pgvector-v3.4.1] - 2025-06-20
 
 ### 🐛 Bug Fixes

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgvector-haystack
-Version: 5.2.0
+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.0 → pgvector_haystack-5.3.0}/examples/embedding_retrieval.py

@@ -1,7 +1,7 @@
 # Before running this example, ensure you have PostgreSQL installed with the pgvector extension.
 # For a quick setup using Docker:
 # docker run -d -p 5432:5432 -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres
-# -e POSTGRES_DB=postgres ankane/pgvector
+# -e POSTGRES_DB=postgres pgvector/pgvector:pg17
 
 # Install required packages for this example, including pgvector-haystack and other libraries needed
 # for Markdown conversion and embeddings generation. Use the following command:

{pgvector_haystack-5.2.0 → pgvector_haystack-5.3.0}/examples/hybrid_retrieval.py

@@ -1,7 +1,7 @@
 # Before running this example, ensure you have PostgreSQL installed with the pgvector extension.
 # For a quick setup using Docker:
 # docker run -d -p 5432:5432 -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres
-# -e POSTGRES_DB=postgres ankane/pgvector
+# -e POSTGRES_DB=postgres pgvector/pgvector:pg17
 
 # Install required packages for this example, including pgvector-haystack and other libraries needed
 # for Markdown conversion and embeddings generation. Use the following command:

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

@@ -534,6 +534,7 @@ class PgvectorDocumentStore:
         The name of the schema (`schema_name`) and the name of the table (`table_name`)
         are defined when initializing the `PgvectorDocumentStore`.
         """
+        self._ensure_db_setup()
         delete_sql = SQL("DROP TABLE IF EXISTS {schema_name}.{table_name}").format(
             schema_name=Identifier(self.schema_name),
             table_name=Identifier(self.table_name),
@@ -551,6 +552,7 @@ class PgvectorDocumentStore:
         """
         Async method to delete the table used to store Haystack documents.
         """
+        await self._ensure_db_setup_async()
         delete_sql = SQL("DROP TABLE IF EXISTS {schema_name}.{table_name}").format(
             schema_name=Identifier(self.schema_name),
             table_name=Identifier(self.table_name),
@@ -985,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.0 → 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):
@@ -237,3 +244,12 @@ def test_create_table_if_not_exists():
     # Clean up: drop the schema after the test
     with psycopg.connect(connection_string, autocommit=True) as conn:
         conn.execute(f"DROP SCHEMA IF EXISTS {schema_name} CASCADE")
+
+
+@pytest.mark.integration
+def test_delete_table_first_call(document_store):
+    """
+    Test that delete_table can be executed as the initial operation on the Document Store
+    without triggering errors due to an uninitialized state.
+    """
+    document_store.delete_table()  # if throw error, test fails

{pgvector_haystack-5.2.0 → 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
@@ -206,3 +213,13 @@ async def test_create_table_if_not_exists():
     # Clean up: drop the schema after the test
     async with await psycopg.AsyncConnection.connect(connection_string, autocommit=True) as conn:
         await conn.execute(f"DROP SCHEMA IF EXISTS {schema_name} CASCADE")
+
+
+@pytest.mark.integration
+@pytest.mark.asyncio
+async def test_delete_table_async_first_call(document_store):
+    """
+    Test that delete_table_async can be executed as the initial operation on the Document Store
+    without triggering errors due to an uninitialized state.
+    """
+    await document_store.delete_table_async()  # if throw error, test fails

pgvector_haystack-5.2.0/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.