34 lines
1.6 KiB
TypeScript
34 lines
1.6 KiB
TypeScript
import { EmbeddingsProviderName } from "../../index.js";
|
|
import BaseEmbeddingsProvider from "./BaseEmbeddingsProvider.js";
|
|
import CohereEmbeddingsProvider from "./CohereEmbeddingsProvider.js";
|
|
import ContinueProxyEmbeddingsProvider from "./ContinueProxyEmbeddingsProvider.js";
|
|
import DeepInfraEmbeddingsProvider from "./DeepInfraEmbeddingsProvider.js";
|
|
import FreeTrialEmbeddingsProvider from "./FreeTrialEmbeddingsProvider.js";
|
|
import GeminiEmbeddingsProvider from "./GeminiEmbeddingsProvider.js";
|
|
import HuggingFaceTEIEmbeddingsProvider from "./HuggingFaceTEIEmbeddingsProvider.js";
|
|
import OllamaEmbeddingsProvider from "./OllamaEmbeddingsProvider.js";
|
|
import OpenAIEmbeddingsProvider from "./OpenAIEmbeddingsProvider.js";
|
|
import TransformersJsEmbeddingsProvider from "./TransformersJsEmbeddingsProvider.js";
|
|
|
|
type EmbeddingsProviderConstructor = new (
|
|
...args: any[]
|
|
) => BaseEmbeddingsProvider;
|
|
|
|
export const allEmbeddingsProviders: Record<
|
|
EmbeddingsProviderName,
|
|
EmbeddingsProviderConstructor
|
|
> = {
|
|
ollama: OllamaEmbeddingsProvider,
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
"transformers.js": TransformersJsEmbeddingsProvider,
|
|
openai: OpenAIEmbeddingsProvider,
|
|
cohere: CohereEmbeddingsProvider,
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
"free-trial": FreeTrialEmbeddingsProvider,
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
"huggingface-tei": HuggingFaceTEIEmbeddingsProvider,
|
|
gemini: GeminiEmbeddingsProvider,
|
|
"continue-proxy": ContinueProxyEmbeddingsProvider,
|
|
deepinfra: DeepInfraEmbeddingsProvider,
|
|
};
|