diff --git a/src/llama_index_cloud_sql_pg/engine.py b/src/llama_index_cloud_sql_pg/engine.py index 0ec0cf8..e47e41f 100644 --- a/src/llama_index_cloud_sql_pg/engine.py +++ b/src/llama_index_cloud_sql_pg/engine.py @@ -398,7 +398,7 @@ def _run_as_sync(self, coro: Awaitable[T]) -> T: async def close(self) -> None: """Dispose of connection pool""" - await self._pool.dispose() + await self._run_as_async(self._pool.dispose()) async def _ainit_doc_store_table( self,