diff --git a/src/functions/delta_scan.cpp b/src/functions/delta_scan.cpp index 4ac6826..df5ad92 100644 --- a/src/functions/delta_scan.cpp +++ b/src/functions/delta_scan.cpp @@ -100,6 +100,7 @@ static ffi::EngineBuilder* CreateBuilder(ClientContext &context, const string &p auto key_id = kv_secret.TryGetValue("key_id").ToString(); auto secret = kv_secret.TryGetValue("secret").ToString(); + auto session_token = kv_secret.TryGetValue("session_token").ToString(); auto region = kv_secret.TryGetValue("region").ToString(); if (key_id.empty() && secret.empty()) { @@ -112,6 +113,9 @@ static ffi::EngineBuilder* CreateBuilder(ClientContext &context, const string &p if (!secret.empty()) { ffi::set_builder_option(builder, KernelUtils::ToDeltaString("aws_secret_access_key"), KernelUtils::ToDeltaString(secret)); } + if (!session_token.empty()) { + ffi::set_builder_option(builder, KernelUtils::ToDeltaString("aws_session_token"), KernelUtils::ToDeltaString(session_token)); + } ffi::set_builder_option(builder, KernelUtils::ToDeltaString("aws_region"), KernelUtils::ToDeltaString(region)); return builder;