fix: Pgvector - ensure DB is initialized when calling delete_table (#2055)

* fix: delete table first call, ensure_db_setup add and covered with test

* chore: remove unnecessary file

* fix: move  written test to test_document_store and test_document_store_async

* fix: comment line

Author: gulbaki

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

@@ -534,6 +534,7 @@ def delete_table(self):
         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 @@ async def delete_table_async(self):
         """
         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),

integrations/pgvector/tests/test_document_store.py

@@ -237,3 +237,12 @@ def get_table_oid(document_store, schema_name, table_name):
     # 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

integrations/pgvector/tests/test_document_store_async.py

@@ -206,3 +206,13 @@ async def get_table_oid(cursor, document_store, schema_name, table_name):
     # 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