File size: 4,089 Bytes
c8ae75d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | import { PROVIDER_MODELS } from "open-sse/config/providerModels.js";
import { AI_PROVIDERS, ALIAS_TO_ID } from "@/shared/constants/providers";
import { getModelKind } from "@/shared/constants/models";
const KIND_ENDPOINT = {
llm: "/v1/chat/completions",
image: "/v1/images/generations",
tts: "/v1/audio/speech",
stt: "/v1/audio/transcriptions",
embedding: "/v1/embeddings",
imageToText: "/v1/chat/completions",
webSearch: "/v1/search",
webFetch: "/v1/fetch",
};
const TTS_VOICES_API = new Set(["elevenlabs", "edge-tts", "deepgram", "inworld", "local-device"]);
function buildInfo({ alias, providerId, model, kind, providerInfo }) {
const out = {
id: `${alias}/${model.id}`,
name: model.name || model.id,
kind,
owned_by: alias,
endpoint: KIND_ENDPOINT[kind] || null,
};
if (model.params) out.params = model.params;
if (model.capabilities) out.capabilities = model.capabilities;
if (model.options) out.options = model.options;
if (model.dimensions) out.dimensions = model.dimensions;
if (model.contextWindow) out.contextWindow = model.contextWindow;
if (kind === "tts" && TTS_VOICES_API.has(providerId)) {
out.voicesUrl = `/v1/audio/voices?provider=${providerId}`;
}
if (kind === "webSearch" && providerInfo?.searchConfig) {
const cfg = providerInfo.searchConfig;
if (cfg.searchTypes) out.searchTypes = cfg.searchTypes;
if (cfg.maxMaxResults) out.maxResults = cfg.maxMaxResults;
if (cfg.requiredOptions) out.required = cfg.requiredOptions;
}
return out;
}
// id format: "{alias}/{modelId}" - alias may also be providerId
// requestedKind: optional, disambiguates duplicate ids across kinds (e.g. gemini-2.5-pro llm vs stt)
function lookup(fullId, requestedKind) {
if (!fullId || !fullId.includes("/")) return null;
const slash = fullId.indexOf("/");
const alias = fullId.slice(0, slash);
const modelId = fullId.slice(slash + 1);
const providerId = ALIAS_TO_ID[alias] || alias;
const providerInfo = AI_PROVIDERS[providerId];
// PROVIDER_MODELS lookup (by alias key, fallback to providerId)
const list = PROVIDER_MODELS[alias] || PROVIDER_MODELS[providerId] || [];
const m = requestedKind
? list.find((x) => x.id === modelId && getModelKind(x, "llm") === requestedKind)
: list.find((x) => x.id === modelId);
if (m) {
const kind = getModelKind(m, "llm");
return buildInfo({ alias, providerId, model: m, kind, providerInfo });
}
// Web search/fetch — virtual model id "search" / "fetch"
if (modelId === "search" && providerInfo?.searchConfig) {
return buildInfo({
alias, providerId, kind: "webSearch", providerInfo,
model: { id: "search", name: `${providerInfo.name} Search`, params: ["query", "max_results", "country", "language", "time_range", "domain_filter", "search_type"] },
});
}
if (modelId === "fetch" && providerInfo?.fetchConfig) {
return buildInfo({
alias, providerId, kind: "webFetch", providerInfo,
model: { id: "fetch", name: `${providerInfo.name} Fetch`, params: ["url", "format", "max_characters"] },
});
}
return null;
}
export async function OPTIONS() {
return new Response(null, {
headers: { "Access-Control-Allow-Origin": "*", "Access-Control-Allow-Methods": "GET, OPTIONS" },
});
}
// GET /v1/models/info?id={alias}/{modelId} — metadata for a single model
export async function GET(request) {
const { searchParams } = new URL(request.url);
const id = searchParams.get("id");
const kind = searchParams.get("kind");
if (!id) {
return Response.json(
{ error: { message: "Missing required query param: id (e.g. ?id=openai/dall-e-3)", type: "invalid_request_error" } },
{ status: 400, headers: { "Access-Control-Allow-Origin": "*" } },
);
}
const info = lookup(id, kind);
if (!info) {
return Response.json(
{ error: { message: `Model not found: ${id}`, type: "not_found" } },
{ status: 404, headers: { "Access-Control-Allow-Origin": "*" } },
);
}
return Response.json(info, { headers: { "Access-Control-Allow-Origin": "*" } });
}
|