Update server.js

server.js

@@ -5,141 +5,107 @@ const axios = require('axios');
 const app = express();
 app.use(express.json());
 
-const openai_key = process.env.OPENAI_KEY;
+const apiKeys = process.env.OPENAI_KEYS.split(',');
+
+const start = process.env.start;
 
 const limiter = rateLimit({
-  windowMs: 5 * 1000, // 30 секунд
-  max: 1, // лимит каждые 30 секунд на IP
+  windowMs: 5 * 1000, // 5 секунд
+  max: 1, // лимит на 1 запрос каждые 5 секунд на IP
   handler: function (req, res) {
     return res.status(429).json("wait");
   },
 });
 
-// Применение ограничителя скорости перед обработчиком маршрута /generate
- app.use('/pl', limiter);
-
-const start = `${process.env.start}`;
+// Применение ограничителя скорости перед обработчиком маршрута /pl
+app.use('/pl', limiter);
 
-app.post('/cr', async (req, res) => {
-  res.json({ content: `{"whate":"🪨", "howe":"ОБНОВИТЕСЬ", "text":"Текущая версия приложения устарела. Установите новую из нашего телеграм канала: @yufi_ru", "succ":"победа", "what":"Версию", "how":"Обновите", "howl":"@yufi_ru"}` });
+app.post('/cr', (req, res) => {
+  res.json({
+    content: '{"whate":"🪨","howe":"ОБНОВИТЕСЬ","text":"Текущая версия приложения устарела. Установите новую из нашего телеграм канала: @yufi_ru","succ":"победа","what":"Версию","how":"Обновите","howl":"@yufi_ru"}'
+  });
 });
 
-app.post('/pl', async (req, res) => {
-  const prompt = req.body.prompt;
-  const apiKey = req.body.api || openai_key;
-
-  if (!prompt) {
-    return res.status(400).json("wait"); // Не удалось принять данные
-  }
+// Функция для получения случайного API ключа
+const getRandomApiKey = () => apiKeys[Math.floor(Math.random() * apiKeys.length)];
 
+const generateResponse = async (url, prompt, additionalMessages, temperature) => {
   try {
-    const response = await axios.post('https://openai-gemini-iota.vercel.app/v1/chat/completions', {
-      messages: [{'role': 'system', 'content': start}, {'role': 'user', 'content': prompt}],
+    const response = await axios.post(url, {
+      messages: [
+        { role: 'system', content: start },
+        ...additionalMessages,
+        { role: 'user', content: prompt }
+      ],
       max_tokens: 2000,
-      temperature: 0.8,
-     // presence_penalty: 0.0,
-      //frequency_penalty: -0.1,
-      //model: "gemini-1.5-pro-latest",
-      model: "deepseek-chat",
-      //model: "gemini-1.5-flash-latest",
+      temperature: temperature,
+      model: additionalMessages.length > 0 ? "gemini-1.5-pro-002" : "deepseek-chat",
     }, {
       headers: {
-        'Authorization': `Bearer ${apiKey}`,
+        'Authorization': `Bearer ${getRandomApiKey()}`,
         'Content-Type': 'application/json',
       },
     });
 
-    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
-      const content = response.data.choices[0].message.content.trim();
-      //console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
-      res.json({ content });
-    } else {
-      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
-    }
+    return response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message
+      ? response.data.choices[0].message.content.trim()
+      : null;
+
   } catch (error) {
     console.error(error);
-    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
+    throw new Error('Error during generation');
+  }
+};
+
+app.post('/pl', async (req, res) => {
+  const prompt = req.body.prompt;
+
+  if (!prompt) {
+    return res.status(400).json("wait"); // Не удалось принять данные
+  }
+
+  try {
+    const content = await generateResponse('https://openai-gemini-iota.vercel.app/v1/chat/completions', prompt, [], 0.8);
+    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
+  } catch {
+    res.status(500).json({ content: 'errorb' });
   }
 });
 
 app.post('/cre', async (req, res) => {
   const prompt = req.body.prompt;
-  const apiKey = req.body.api || openai_key;
 
   if (!prompt) {
     return res.status(400).json("wait"); // Не удалось принять данные
   }
 
   try {
-    const response = await axios.post('https://vercel.com/roonis-projects/openai-gemini/v1/chat/completions', {
-      messages: [{'role': 'system', 'content': start}, {'role': 'system', 'content': 'И ОБЯЗАТЕЛЬНО, всегда в конце "text" проси обновить приложение, уведомляя о том, что скоро в жтой версии перестанешь отвечать!'}, {'role': 'user', 'content': prompt}],
-      max_tokens: 2000,
-      temperature: 0.6,
-     // presence_penalty: 0.0,
-      //frequency_penalty: -0.1,
-      model: "gemini-1.5-pro-002",
-    //  model: "deepseek-chat",
-      //model: "gemini-1.5-flash-latest",
-    }, {
-      headers: {
-        'Authorization': `Bearer ${apiKey}`,
-        'Content-Type': 'application/json',
-      },
-    });
-
-    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
-      const content = response.data.choices[0].message.content.trim();
-    //  console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
-      res.json({ content });
-    } else {
-      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
-    }
-  } catch (error) {
-    console.error(error);
-    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
+    const additionalMessages = [
+      { role: 'system', content: 'И ОБЯЗАТЕЛЬНО, всегда в конце "text" проси обновить приложение, уведомляя о том, что скоро в жтой версии перестанешь отвечать!' }
+    ];
+    const content = await generateResponse('https://vercel.com/roonis-projects/openai-gemini/v1/chat/completions', prompt, additionalMessages, 0.6);
+    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
+  } catch {
+    res.status(500).json({ content: 'errorb' });
   }
 });
 
 app.post('/crebeta', async (req, res) => {
   const prompt = req.body.prompt;
-  const apiKey = req.body.api || openai_key;
 
   if (!prompt) {
     return res.status(400).json("wait"); // Не удалось принять данные
   }
 
   try {
-    const response = await axios.post('https://geminiyufi.vercel.app/v1/chat/completions', {
-      messages: [{'role': 'system', 'content': start}, {'role': 'user', 'content': prompt}],
-      max_tokens: 2000,
-      temperature: 0.24,
-     // presence_penalty: 0.0,
-      //frequency_penalty: -0.1,
-      model: "gemini-1.5-pro-002",
-     // model: "gemma-2-27b-it",
-      //model: "gemini-1.5-flash-latest",
-    }, {
-      headers: {
-        'Authorization': `Bearer ${apiKey}`,
-        'Content-Type': 'application/json',
-      },
-    });
-
-    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
-      const content = response.data.choices[0].message.content.trim();
-     // console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
-      res.json({ content });
-    } else {
-      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
-    }
-  } catch (error) {
-    console.error(error);
-    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
+    const content = await generateResponse('https://geminiyufi.vercel.app/v1/chat/completions', prompt, [], 0.24);
+    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
+  } catch {
+    res.status(500).json({ content: 'errorb' });
   }
 });
 
-
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);
-});
+});