Add support for AzureSQLDatabase in ContentSourceServiceFactory

solliancenet · Jan 29, 2024 · b7504e8 · b7504e8
1 parent d1db600
commit b7504e8
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/...dotnet/Vectorization/Models/Configuration/AzureSQLDatabaseContentSourceServiceSettings.cs b/...dotnet/Vectorization/Models/Configuration/AzureSQLDatabaseContentSourceServiceSettings.cs
@@ -8,6 +8,6 @@ public class AzureSQLDatabaseContentSourceServiceSettings
         /// <summary>
         /// The connection string used for authentication.
         /// </summary>
-        public required string ConnectionString { get; set; }
+        public string? ConnectionString { get; set; }
     }
 }
diff --git a/src/dotnet/Vectorization/Models/Resources/ContentSourceType.cs b/src/dotnet/Vectorization/Models/Resources/ContentSourceType.cs
@@ -19,6 +19,11 @@ public enum ContentSourceType
         /// <summary>
         /// SharePoint Online document library.
         /// </summary>
-        SharePointOnline
+        SharePointOnline,
+
+        /// <summary>
+        /// Azure SQL Database.
+        /// </summary>
+        AzureSQLDatabase
     }
 }
diff --git a/src/dotnet/Vectorization/Services/ContentSources/ContentSourceServiceFactory.cs b/src/dotnet/Vectorization/Services/ContentSources/ContentSourceServiceFactory.cs
@@ -41,6 +41,7 @@ public IContentSourceService GetService(string serviceName)
             {
                 ContentSourceType.AzureDataLake => CreateAzureDataLakeContentSourceService(serviceName),
                 ContentSourceType.SharePointOnline => CreateSharePointOnlineContentSourceService(serviceName),
+                ContentSourceType.AzureSQLDatabase => CreateAzureSQLDatabaseContentSourceService(serviceName),
                 _ => throw new VectorizationException($"The content source type {contentSourceProfile.Type} is not supported."),
             };
         }
@@ -55,6 +56,7 @@ public IContentSourceService GetService(string serviceName)
             {
                 ContentSourceType.AzureDataLake => (CreateAzureDataLakeContentSourceService(serviceName), contentSourceProfile),
                 ContentSourceType.SharePointOnline => (CreateSharePointOnlineContentSourceService(serviceName), contentSourceProfile),
+                ContentSourceType.AzureSQLDatabase => (CreateAzureSQLDatabaseContentSourceService(serviceName), contentSourceProfile),
                 _ => throw new VectorizationException($"The content source type {contentSourceProfile.Type} is not supported."),
             };
         }
@@ -83,5 +85,17 @@ private SharePointOnlineContentSourceService CreateSharePointOnlineContentSource
                 sharePointOnlineContentSourceServiceSettings,
                 _loggerFactory);
         }
+
+        private AzureSQLDatabaseContentSourceService CreateAzureSQLDatabaseContentSourceService(string serviceName)
+        {
+            var azureSQLDatabaseContentSourceServiceSettings = new AzureSQLDatabaseContentSourceServiceSettings();
+            _configuration.Bind(
+                $"{AppConfigurationKeySections.FoundationaLLM_Vectorization_ContentSources}:{serviceName}",
+                azureSQLDatabaseContentSourceServiceSettings);
+
+            return new AzureSQLDatabaseContentSourceService(
+                azureSQLDatabaseContentSourceServiceSettings,
+                _loggerFactory);
+        }
     }
 }