mirror of
https://github.com/Nezumi-2711/9router.git
synced 2026-09-22 13:38:31 +00:00
Ngô Tấn TàiandCursor
b33cbb0280
Extend Vercel AI Gateway beyond chat: add OpenAI-compatible embeddings and image generation endpoints, credit balance fetch on the usage dashboard, retry on 429, and models catalog fetcher. Thinking/reasoning mapping is omitted pending a project-wide refactor. Co-authored-by: Ngô Tấn Tài <tantai@newnol.io.vn> Co-authored-by: Cursor <cursoragent@cursor.com>
25 lines
779 B
JavaScript
25 lines
779 B
JavaScript
// Embeddings provider adapter registry
|
|
import createOpenAIEmbeddingAdapter from "./openai.js";
|
|
import gemini from "./gemini.js";
|
|
import openaiCompatNode from "./openaiCompatNode.js";
|
|
|
|
const OPENAI_COMPAT_PROVIDERS = [
|
|
"openai", "openrouter", "mistral", "voyage-ai", "fireworks",
|
|
"together", "nebius", "github", "nvidia", "jina-ai",
|
|
"vercel-ai-gateway",
|
|
];
|
|
|
|
const ADAPTERS = {
|
|
...Object.fromEntries(OPENAI_COMPAT_PROVIDERS.map((id) => [id, createOpenAIEmbeddingAdapter(id)])),
|
|
gemini,
|
|
google_ai_studio: gemini,
|
|
};
|
|
|
|
export function getEmbeddingAdapter(provider) {
|
|
if (ADAPTERS[provider]) return ADAPTERS[provider];
|
|
if (provider?.startsWith?.("openai-compatible-") || provider?.startsWith?.("custom-embedding-")) {
|
|
return openaiCompatNode;
|
|
}
|
|
return null;
|
|
}
|