Fix queue await error

Author: jason810496

pgmq_sqlalchemy/queue.py

@@ -127,12 +127,24 @@ def _check_pgmq_ext(self) -> None:
         """Check if the pgmq extension exists."""
         self._execute_operation(PGMQOperation.check_pgmq_ext, session=None, commit=True)
 
+    async def _check_pgmq_ext_async(self) -> None:
+        """Check if the pgmq extension exists (async version)."""
+        await self._execute_async_operation(
+            PGMQOperation.check_pgmq_ext_async, session=None, commit=True
+        )
+
     def _check_pg_partman_ext(self) -> None:
         """Check if the pg_partman extension exists."""
         self._execute_operation(
             PGMQOperation.check_pg_partman_ext, session=None, commit=True
         )
 
+    async def _check_pg_partman_ext_async(self) -> None:
+        """Check if the pg_partman extension exists (async version)."""
+        await self._execute_async_operation(
+            PGMQOperation.check_pg_partman_ext_async, session=None, commit=True
+        )
+
     def _execute_operation(
         self,
         op_sync,
@@ -342,7 +354,7 @@ async def create_partitioned_queue_async(
 
         """
         # check if the pg_partman extension exists before creating a partitioned queue at runtime
-        self._check_pg_partman_ext()
+        await self._check_pg_partman_ext_async()
 
         return await self._execute_async_operation(
             PGMQOperation.create_partitioned_queue_async,
@@ -451,7 +463,7 @@ async def drop_queue_async(
         """
         # check if the pg_partman extension exists before dropping a partitioned queue at runtime
         if partitioned:
-            self._check_pg_partman_ext()
+            await self._check_pg_partman_ext_async()
 
         return await self._execute_async_operation(
             PGMQOperation.drop_queue_async,