AI-Studio/app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingProviders.cs

using System.Diagnostics.CodeAnalysis;

using AIStudio.Provider;
using AIStudio.Settings;

namespace AIStudio.Tools.Services;

internal static class DataSourceEmbeddingProviders
{
    public static bool TryResolve(SettingsManager settingsManager, IDataSourceBase dataSource, [NotNullWhen(true)] out EmbeddingProvider? embeddingProvider)
    {
        embeddingProvider = settingsManager.ConfigurationData.EmbeddingProviders.FirstOrDefault(provider =>
            dataSource is IIndexedDataSource indexedDataSource &&
            provider.Id.Equals(indexedDataSource.EmbeddingId, StringComparison.OrdinalIgnoreCase));

        return embeddingProvider != default && embeddingProvider.UsedLLMProvider is not LLMProviders.NONE;
    }
}