I have a large codebase with several intensive queries to BigQuery. I'm trying to port these synchronous queries to async to improve performance, but am unable to create an async SQLAlchemy engine. Is there a specific dialect and/or driver I can provide in order to enable async functionality? Or is there an easy workaround?
MWE
A simple script:
from sqlalchemy.ext.asyncio import create_async_engine
conn_path = "bigquery://"
engine = create_async_engine(conn_path)
Resulting in the traceback
Traceback (most recent call last):
File "/Users/eshwen.bhal/Library/Application Support/JetBrains/PyCharm2024.1/scratches/scratch_6.py", line 4, in <module>
engine = create_async_engine(conn_path)
File "/Users/eshwen.bhal/Documents/GitHub/mjolnir/.venv/lib/python3.11/site-packages/sqlalchemy/ext/asyncio/engine.py", line 118, in create_async_engine
return AsyncEngine(sync_engine)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/eshwen.bhal/Documents/GitHub/mjolnir/.venv/lib/python3.11/site-packages/sqlalchemy/ext/asyncio/engine.py", line 1023, in __init__
raise exc.InvalidRequestError(
sqlalchemy.exc.InvalidRequestError: The asyncio extension requires an async driver to be used. The loaded 'bigquery' is not async.
Issue
I have a large codebase with several intensive queries to BigQuery. I'm trying to port these synchronous queries to async to improve performance, but am unable to create an async SQLAlchemy engine. Is there a specific dialect and/or driver I can provide in order to enable async functionality? Or is there an easy workaround?
MWE
A simple script:
Resulting in the traceback
Environment
sqlalchemy-bigquery
: 1.11.0