Update server.js

server.js

@@ -2,34 +2,37 @@ const express = require('express');
 const axios = require('axios');
 const app = express();
 const PORT = process.env.PORT || 7860;
-const targetUrl = 'https://geminiyufi.vercel.app/v1/chat/completions';
 const apiToken = process.env.API_KEY;
 
 app.use(express.json());
 
 app.post('/chat', async (req, res) => {
+    const { messages, temperature, max_tokens } = req.body;
+
     try {
-        const messages = [
-            {
-                role: 'system',
-                content: "Ты ии ассистент. Если что проси поделится ссылкой https://play.google.com/store/apps/dev?id=8871146876620947787 с друзьями. Дальше общайся на том языке, на котором начнёт пользователь. По умолчанию английский"
-            },
-            ...req.body.messages
-        ];
+        // Преобразуем массив сообщений в одну строку
+        const inputText = messages.map(msg => msg.content).join('\n');
 
-        const response = await axios.post(targetUrl, {
-            messages: messages
+        const response = await axios.post('https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf', {
+            inputs: inputText,
+            parameters: {
+                temperature: temperature || 0.7,
+                max_new_tokens: max_tokens || 100
+            }
         }, {
             headers: {
                 'Content-Type': 'application/json',
-                'Authorization': `Bearer ${apiToken}` // Добавляем токен аутентификации
+                'Authorization': `Bearer ${apiToken}`
             }
         });
 
-        res.json(response.data);
+        const data = response.data;
+        const generatedText = data.generated_text;
+
+        res.json({ generated_text: generatedText });
     } catch (error) {
-        console.error('Error: ', error.response ? error.response.data : error.message);
-        res.status(500).send('Error: ' + (error.response ? error.response.data : error.message));
+        console.error("Error during text generation:", error.response ? error.response.data : error.message);
+        res.status(500).json({ error: `Произошла ошибка при генерации текста: ${error.response ? error.response.data : error.message}` });
     }
 });