Update server.js

server.js

@@ -1,7 +1,7 @@
 require('dotenv').config();
 const express = require('express');
 const rateLimit = require('express-rate-limit');
-const { HfInference } = require('@huggingface/inference');
+const { Configuration, OpenAIApi } = require('openai');
 
 const app = express();
 app.use(express.json());
@@ -36,35 +36,41 @@ app.post('/update', async (req, res) => {
 });
 
 async function sendRequest(prompt, prs) {
-    const hf_api_key = getRandomApiKey();
-    const client = new HfInference(hf_api_key);
-  
-    try {
-      const chatCompletion = await client.chatCompletion({
-        model: "Qwen/Qwen2.5-Coder-32B-Instruct",
-        messages: [
-          {
-            role: "system",
-            content: prs
-          },
-          {
-            role: "user",
-            content: prompt
-          }
-        ],
-        max_tokens: 1200,
-      });
-  
-      if (chatCompletion && chatCompletion.choices && chatCompletion.choices.length > 0 && chatCompletion.choices[0].message) {
-        return chatCompletion.choices[0].message.content.trim();
-      } else {
-        throw new Error("Ошибка прочтения ответа");
-      }
-    } catch (error) {
-      console.error("Ошибка при обращении к Hugging Face:", error);
-      throw new Error("Ошибка при генерации");
+  const apiKey = getRandomApiKey();
+
+  const configuration = new Configuration({
+    apiKey: apiKey,
+    basePath: process.env.BASE_URL || undefined,
+  });
+
+  const openai = new OpenAIApi(configuration);
+
+  try {
+    const response = await openai.createChatCompletion({
+      model: process.env.MODEL_NAME || 'gpt-3.5-turbo',
+      messages: [
+        {
+          role: 'system',
+          content: prs
+        },
+        {
+          role: 'user',
+          content: prompt
+        }
+      ],
+      max_tokens: 1200,
+    });
+
+    if (response.data && response.data.choices && response.data.choices.length > 0) {
+      return response.data.choices[0].message.content.trim();
+    } else {
+      throw new Error('Ошибка прочтения ответа');
     }
+  } catch (error) {
+    console.error('Ошибка при обращении к OpenAI:', error.message);
+    throw new Error('Ошибка при генерации');
   }
+}
 
 app.post('/pl', async (req, res) => {
   const prompt = req.body.prompt;
@@ -84,7 +90,7 @@ app.post('/pl', async (req, res) => {
     const content = await sendRequest(prompt, prs);
     res.json({ content });
   } catch (error) {
-        res.json({ content: `{"error":"", "title":"Ошибка", "text":"Произошла ошибка на сервере. (${error.message})", "okb":"Ок", "oklink":"", "cancelable":"true"}` });
+    res.json({ content: `{"error":"", "title":"Ошибка", "text":"Произошла ошибка на сервере. (${error.message})", "okb":"Ок", "oklink":"", "cancelable":"true"}` });
   }
 });
 
@@ -114,4 +120,4 @@ app.post('/plbeta', async (req, res) => {
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);
-});
+});