@@ -643,7 +643,7 @@ private static bool TryUpdateConfiguredDataSourceOptions(
643643 DatabaseType dbType = runtimeConfig . DataSource . DatabaseType ;
644644 string dataSourceConnectionString = runtimeConfig . DataSource . ConnectionString ;
645645 DatasourceHealthCheckConfig ? datasourceHealthCheckConfig = runtimeConfig . DataSource . Health ;
646- UserDelegatedAuthConfig ? userDelegatedAuthConfig = runtimeConfig . DataSource . UserDelegatedAuth ;
646+ UserDelegatedAuthOptions ? userDelegatedAuthConfig = runtimeConfig . DataSource . UserDelegatedAuth ;
647647
648648 if ( options . DataSourceDatabaseType is not null )
649649 {
@@ -735,16 +735,21 @@ private static bool TryUpdateConfiguredDataSourceOptions(
735735 string ? databaseAudience = options . DataSourceUserDelegatedAuthDatabaseAudience
736736 ?? userDelegatedAuthConfig ? . DatabaseAudience ;
737737
738+ // Get provider: preserve existing or use default "EntraId"
739+ string ? provider = userDelegatedAuthConfig ? . Provider ?? "EntraId" ;
740+
738741 // Create or update user-delegated-auth config
739- userDelegatedAuthConfig = new UserDelegatedAuthConfig (
742+ userDelegatedAuthConfig = new UserDelegatedAuthOptions (
740743 Enabled : enabled ,
741- DatabaseAudience : databaseAudience ,
742- DisableConnectionPooling : userDelegatedAuthConfig ? . DisableConnectionPooling ,
743- TokenCacheDurationMinutes : userDelegatedAuthConfig ? . TokenCacheDurationMinutes ) ;
744+ Provider : provider ,
745+ DatabaseAudience : databaseAudience ) ;
744746 }
745747
746748 dbOptions = EnumerableUtilities . IsNullOrEmpty ( dbOptions ) ? null : dbOptions ;
747- DataSource dataSource = new ( dbType , dataSourceConnectionString , dbOptions , datasourceHealthCheckConfig , userDelegatedAuthConfig ) ;
749+ DataSource dataSource = new ( dbType , dataSourceConnectionString , dbOptions , datasourceHealthCheckConfig )
750+ {
751+ UserDelegatedAuth = userDelegatedAuthConfig
752+ } ;
748753 runtimeConfig = runtimeConfig with { DataSource = dataSource } ;
749754
750755 return runtimeConfig != null ;
0 commit comments