feat: PgvectorDocumentStore now supports delete_all_documents and delete_all_documents_async (#2394)

* feat: `PgVectorDocumentStore` now supports `delete_all_documents` and `delete_all_documents_async`

* style: fix typo

* test: Remove comments and verify I can write document after deletion

Author: cmnemoi

integrations/pgvector/src/haystack_integrations/document_stores/pgvector/document_store.py

@@ -987,6 +987,38 @@ async def delete_documents_async(self, document_ids: List[str]) -> None:
             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]:

integrations/pgvector/tests/test_document_store.py

@@ -53,6 +53,13 @@ def test_connection_check_and_recreation(self, document_store: PgvectorDocumentS
             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):

integrations/pgvector/tests/test_document_store_async.py

@@ -94,6 +94,13 @@ async def test_connection_check_and_recreation(self, document_store: PgvectorDoc
             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