pgvector-haystack 5.2.0__py3-none-any.whl → 5.3.0__py3-none-any.whl

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.dist-info → pgvector_haystack-5.3.0.dist-info}/METADATA

@@ -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.2.0.dist-info → pgvector_haystack-5.3.0.dist-info}/RECORD

@@ -5,9 +5,9 @@ haystack_integrations/components/retrievers/pgvector/keyword_retriever.py,sha256
 haystack_integrations/document_stores/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 haystack_integrations/document_stores/pgvector/__init__.py,sha256=Ge-7ggarfr3x8UqNdFHUuFGiu52u49sOVe3l16WEwZU,197
 haystack_integrations/document_stores/pgvector/converters.py,sha256=9ORG_Yb8Ss4eGGJB9ZGaDE25z89rLAh3ZisKwRww1bw,2937
-haystack_integrations/document_stores/pgvector/document_store.py,sha256=hztkxq4EM41okcaIQeTOT5GnbYeHr8Yjluebxfmgnpc,49763
+haystack_integrations/document_stores/pgvector/document_store.py,sha256=EKrDg-nezEaRhEYnVhlTH7wmJ_w4sy5IfaaNq6mITaY,51009
 haystack_integrations/document_stores/pgvector/filters.py,sha256=jYdoXJZhAn3hNZTnx1x8K5Ok2LZzh-GB4hOhk6r7ZPk,9021
-pgvector_haystack-5.2.0.dist-info/METADATA,sha256=WQpcPn67Vuy3xKI4Cz2VZ4NKBtQ1Zh0z97_fsr-ZTuQ,2187
-pgvector_haystack-5.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pgvector_haystack-5.2.0.dist-info/licenses/LICENSE.txt,sha256=B05uMshqTA74s-0ltyHKI6yoPfJ3zYgQbvcXfDVGFf8,10280
-pgvector_haystack-5.2.0.dist-info/RECORD,,
+pgvector_haystack-5.3.0.dist-info/METADATA,sha256=O_edb8lqV2SEZqbwb6Va_iC3Pu_p0GtxM_t95SEPNGo,2144
+pgvector_haystack-5.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pgvector_haystack-5.3.0.dist-info/licenses/LICENSE.txt,sha256=B05uMshqTA74s-0ltyHKI6yoPfJ3zYgQbvcXfDVGFf8,10280
+pgvector_haystack-5.3.0.dist-info/RECORD,,