update callsql

Author: CTY-git

patchwork/steps/CallSQL/CallSQL.py

@@ -34,7 +34,12 @@ def __build_engine(self, inputs: dict):
             dialect_plus_driver,
             **kwargs,
         )
-        self.engine = create_engine(connection_url)
+
+        connect_args = None
+        if inputs.get("db_driver_args") is not None:
+            connect_args = inputs.get("db_driver_args")
+
+        self.engine = create_engine(connection_url, connect_args=connect_args)
         with self.engine.connect() as conn:
             conn.execute(text("SELECT 1"))
         return self.engine

patchwork/steps/CallSQL/typed.py

@@ -15,6 +15,8 @@ class CallSQLInputs(__RequiredCallSQLInputs, total=False):
     db_host: str
     db_port: int
     db_name: str
+    db_params: dict[str, Any]
+    db_driver_args: dict[str, Any]
     db_query_template_values: dict[str, Any]