/// <summary>
/// Enables the usage of SQLite and local storage
/// </summary>
/// <param name="builder">MLOpsBuilder to add Azure Storage providers to</param>
/// <param name="destinationFolder">Destination folder, default location is .mlops under the current user</param>
/// <returns>Provided MLOpsBuilder for chaining</returns>
public static MLOpsBuilder UseSQLite(this MLOpsBuilder builder, string destinationFolder = null)
{
builder.UseMetaDataStore(new SQLiteMetaDataStore());
builder.UseModelRepository(new LocalFileModelRepository(new FileSystem(), destinationFolder));
return(builder);
}
/// <summary>
/// Enables the usage of Azure Blobstorage and TableStorage as a storage provider
/// </summary>
/// <param name="builder">MLOpsBuilder to add Azure Storage providers to</param>
/// <param name="connectionString">The connection string for the azure storage account</param>
/// <returns>Provided MLOpsBuilder for chaining</returns>
public static MLOpsBuilder UseAzureStorage(this MLOpsBuilder builder, string connectionString)
{
builder.UseMetaDataStore(new StorageAccountMetaDataStore(connectionString));
builder.UseModelRepository(new StorageAccountModelRepository(connectionString));
return(builder);
}
/// <summary>
///
/// </summary>
/// <param name="builder"></param>
/// <param name="awsAccessKeyId"></param>
/// <param name="awsSecretAccessKey"></param>
/// <param name="regionName"></param>
/// <returns></returns>
public static MLOpsBuilder UseAWSS3ModelRepository(this MLOpsBuilder builder, string awsAccessKeyId, string awsSecretAccessKey, string regionName)
{
var region = RegionEndpoint.GetBySystemName(regionName);
var amazonS3Client = new AmazonS3Client(awsAccessKeyId, awsSecretAccessKey, region);
builder.UseModelRepository(new S3BucketModelRepository(amazonS3Client, new ModelPathGenerator()));
return(builder);
}
/// <summary>
/// Enables the usage of Azure Blobstorage as a storage provider for the models
/// </summary>
/// <param name="builder">MLOpsBuilder to add Azure Storage providers to</param>
/// <param name="connectionString">The connection string for the azure storage account</param>
/// <returns>Provided MLOpsBuilder for chaining</returns>
public static MLOpsBuilder UseAzureBlobModelRepository(this MLOpsBuilder builder, string connectionString)
{
var modelRepositoryClient = new BlobContainerClient(connectionString, "model-repository");
var deploymentClient = new BlobContainerClient(connectionString, "deployment");
modelRepositoryClient.CreateIfNotExists(PublicAccessType.None);
deploymentClient.CreateIfNotExists(PublicAccessType.Blob);
builder.UseModelRepository(new StorageAccountModelRepository(modelRepositoryClient, deploymentClient, new ModelPathGenerator()));
return(builder);
}
/// <summary>
/// Enables usage of local file share for model storage
/// </summary>
/// <param name="builder"></param>
/// <param name="destinationFolder">Destination folder, default location is .mlops under the current user</param>
/// <returns>Provided MLOpsBuilder for chaining</returns>
public static MLOpsBuilder UseLocalFileModelRepository(this MLOpsBuilder builder, string destinationFolder = null)
{
builder.UseModelRepository(new LocalFileModelRepository(new FileSystem(), destinationFolder));
return(builder);
}