diff --git a/python-sdk/src/astro/databases/base.py b/python-sdk/src/astro/databases/base.py
index 85045a4c1..93c3ce461 100644
--- a/python-sdk/src/astro/databases/base.py
+++ b/python-sdk/src/astro/databases/base.py
@@ -107,6 +107,7 @@ def run_sql(
         self,
         sql: str | ClauseElement = "",
         parameters: dict | None = None,
+        handler: Callable | None = None,
         **kwargs,
     ) -> CursorResult:
         """