2026-05-04 11:29:02 +07:00
|
|
|
// Embeddings provider adapter registry
|
|
|
|
|
import createOpenAIEmbeddingAdapter from "./openai.js";
|
|
|
|
|
import gemini from "./gemini.js";
|
|
|
|
|
import openaiCompatNode from "./openaiCompatNode.js";
|
2026-08-05 13:33:28 +07:00
|
|
|
import selfhostedEmbedding from "./selfhostedEmbedding.js";
|
2026-05-04 11:29:02 +07:00
|
|
|
|
|
|
|
|
const OPENAI_COMPAT_PROVIDERS = [
|
|
|
|
|
"openai", "openrouter", "mistral", "voyage-ai", "fireworks",
|
|
|
|
|
"together", "nebius", "github", "nvidia", "jina-ai",
|
2026-06-13 10:54:51 +07:00
|
|
|
"vercel-ai-gateway",
|
2026-05-04 11:29:02 +07:00
|
|
|
];
|
|
|
|
|
|
|
|
|
|
const ADAPTERS = {
|
|
|
|
|
...Object.fromEntries(OPENAI_COMPAT_PROVIDERS.map((id) => [id, createOpenAIEmbeddingAdapter(id)])),
|
|
|
|
|
gemini,
|
|
|
|
|
google_ai_studio: gemini,
|
2026-08-05 13:33:28 +07:00
|
|
|
// Self-hosted reads creds.providerSpecificData.baseUrl (one provider, many
|
|
|
|
|
// servers) — but via its OWN adapter, not openaiCompatNode: that one falls back
|
|
|
|
|
// to api.openai.com when no baseUrl is set, which under a provider called
|
|
|
|
|
// "Self-hosted Embedding" means silently shipping the input and API key to
|
|
|
|
|
// OpenAI. selfhostedEmbedding refuses instead.
|
|
|
|
|
"selfhosted-embedding": selfhostedEmbedding,
|
2026-05-04 11:29:02 +07:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
export function getEmbeddingAdapter(provider) {
|
|
|
|
|
if (ADAPTERS[provider]) return ADAPTERS[provider];
|
|
|
|
|
if (provider?.startsWith?.("openai-compatible-") || provider?.startsWith?.("custom-embedding-")) {
|
|
|
|
|
return openaiCompatNode;
|
|
|
|
|
}
|
|
|
|
|
return null;
|
|
|
|
|
}
|