import { EmbeddingProvider, EmbeddingProviderInterface, PluginSettings } from '../types';
import { OllamaEmbeddingProvider } from './ollama';
import { OpenAIEmbeddingProvider } from './openai';

export function createEmbeddingProvider(settings: PluginSettings): EmbeddingProviderInterface {
  switch (settings.embedding.provider) {
    case EmbeddingProvider.OLLAMA:
      return new OllamaEmbeddingProvider(settings.embedding.ollama);
    case EmbeddingProvider.OPENAI:
      return new OpenAIEmbeddingProvider(settings.embedding.openai);
    default:
      throw new Error(`Unsupported embedding provider: ${settings.embedding.provider}`);
  }
}

export { BaseEmbeddingProvider } from './provider';
export { OllamaEmbeddingProvider } from './ollama';
export { OpenAIEmbeddingProvider } from './openai';