40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
import { DEFAULT_PROVIDER } from "../../agents/defaults.js";
|
|
import { buildAllowedModelSet } from "../../agents/model-selection.js";
|
|
import { loadConfig } from "../../config/config.js";
|
|
import {
|
|
ErrorCodes,
|
|
errorShape,
|
|
formatValidationErrors,
|
|
validateModelsListParams,
|
|
} from "../protocol/index.js";
|
|
import type { GatewayRequestHandlers } from "./types.js";
|
|
|
|
export const modelsHandlers: GatewayRequestHandlers = {
|
|
"models.list": async ({ params, respond, context }) => {
|
|
if (!validateModelsListParams(params)) {
|
|
respond(
|
|
false,
|
|
undefined,
|
|
errorShape(
|
|
ErrorCodes.INVALID_REQUEST,
|
|
`invalid models.list params: ${formatValidationErrors(validateModelsListParams.errors)}`,
|
|
),
|
|
);
|
|
return;
|
|
}
|
|
try {
|
|
const catalog = await context.loadGatewayModelCatalog();
|
|
const cfg = loadConfig();
|
|
const { allowedCatalog } = buildAllowedModelSet({
|
|
cfg,
|
|
catalog,
|
|
defaultProvider: DEFAULT_PROVIDER,
|
|
});
|
|
const models = allowedCatalog.length > 0 ? allowedCatalog : catalog;
|
|
respond(true, { models }, undefined);
|
|
} catch (err) {
|
|
respond(false, undefined, errorShape(ErrorCodes.UNAVAILABLE, String(err)));
|
|
}
|
|
},
|
|
};
|