PomogatorProxy / server.js
Rooni's picture
Update server.js
94d575b verified
raw
history blame
No virus
3.8 kB
const express = require('express');
const rateLimit = require('express-rate-limit');
const axios = require('axios');
const app = express();
app.use(express.json());
const openai_key = process.env.OPENAI_KEY;
const start = `Ты генератор рецептов. В этом чате, тебе будут присылать названия блюд или описывать их или присылать название категории, а ты должен будешь написать идеальный, понятный, подробный, лёгкий или сложный рецепт со всеми пропорциями, количествами и этапами готовки (этапы готовки должны быть разделены, например 1.1., 1.2., 2.1., 2.2. (Обязательно со звёздочками, это жирный шрифт) С эмодзи.
Ответ должен выглядить ТОЛЬКО так (чтобы потом regexp его исправил):
"
+название+Название рецепта, блюда-название-
+время+Примерное время готовки (только время, например: "40 минут", "2 часа" и т.д.)-время-
+кат+Категория: На каждый день, Основные блюда, Супы, Десерты и выпечка, Салаты, Несладкая выпечка, Завтраки, Закуски, Напитки, Ресторанные-кат-
+ингр+Ингридиенты: В именительном падеже - то есть например: "Яйцо -
2 штуки", а не "2 яйца"; Здесь обязательно заменяй ВСЕ новые строки ("
") на "÷"-ингр-
+рец+Рецепт; Здесь обязательно заменяй ВСЕ новые строки ("
") на "÷"-рец-
📙 Рецепт сгенерирова в приложении АромаАрт (yufi.rf.gd/aromaart)
"
Если ты не можешь сгенерировать такой рецепт, то напиши так:
"
+ошибка+Причина-ошибка-
"`
// Определяем лимитер с задержкой 30 секунд на IP
const generateLimiter = rateLimit({
windowMs: 30 * 1000, // 30 секунд
max: 1, // лимит: 1 запрос на IP
handler: function (req, res /*next*/) {
return res.status(429).json({
error: "Слишком много запросов с этого IP, пожалуйста, попробуйте позже. (Лимит 1 генерация в 30 секунд)"
});
}
});
// Применяем лимитер только к определенному маршруту
app.post('/generate', generateLimiter, async (req, res) => {
const prompt = req.body.prompt;
if (!prompt) {
return res.status(400).json({ error: 'Требуется параметр "prompt".' });
}
try {
const response = await axios.post('https://api.openai.com/v1/chat/completions', {
messages: [{'role': 'system', 'content': start}, {'role': 'system', 'content': prompt}],
max_tokens: 2000,
model: "gpt-4-1106-preview",
}, {
headers: {
'Authorization': `Bearer ${openai_key}`,
'Content-Type': 'application/json',
},
});
console.log(data['choices'][0]['message']['content'].strip());
res.json(data['choices'][0]['message']['content'].strip());
} catch (error) {
console.error(error);
res.status(500).json({ error: 'Произошла ошибка сервера при генерации.' });
}
});
const port = 7860;
app.listen(port, () => {
console.log(`API сервер запущен на порту ${port}`);
});