/
alt3rmann
/
LibreChat
Обзор
Документация
Войти
/
alt3rmann
/
LibreChat
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
api/server/middleware/validateModel.js
47 строк
1 KB
Danny Avila
🤖 feat(Anthropic): Claude 3 & Vision Support (#1984)
06 мар 2024, 08:04
Не верифицирован
06 мар 2024, 08:04
8263ddd
Код
Авторство
О чём код?
const { ViolationTypes } = require('librechat-data-provider'); const { getModelsConfig } = require('~/server/controllers/ModelController'); const { handleError } = require('~/server/utils'); const { logViolation } = require('~/cache'); /** * Validates the model of the request. * * @async * @param {Express.Request} req - The Express request object. * @param {Express.Response} res - The Express response object. * @param {Function} next - The Express next function. */ const validateModel = async (req, res, next) => { const { model, endpoint } = req.body; if (!model) { return handleError(res, { text: 'Model not provided' }); } const modelsConfig = await getModelsConfig(req); if (!modelsConfig) { return handleError(res, { text: 'Models not loaded' }); } const availableModels = modelsConfig[endpoint]; if (!availableModels) { return handleError(res, { text: 'Endpoint models not loaded' }); } let validModel = !!availableModels.find((availableModel) => availableModel === model); if (validModel) { return next(); } const { ILLEGAL_MODEL_REQ_SCORE: score = 5 } = process.env ?? {}; const type = ViolationTypes.ILLEGAL_MODEL_REQUEST; const errorMessage = { type, }; await logViolation(req, res, type, errorMessage, score); return handleError(res, { text: 'Illegal model request' }); }; module.exports = validateModel;