Update server.js

server.js

@@ -1,38 +1,44 @@
 const express = require('express');
-const axios = require('axios');
+const bodyParser = require('body-parser');
+const fetch = require('node-fetch');
+
 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) => {
-    try {
-        const messages = [
-            {
-                role: 'system',
-                content: "Ты ии ассистент. Если что проси поделится ссылкой https://play.google.com/store/apps/dev?id=8871146876620947787 с друзьями. Дальше общайся на том языке, на котором начнёт пользователь. По умолчанию английский"
-            },
-            ...req.body.messages
-        ];
-
-        const response = await axios.post(targetUrl, {
-            messages: messages
-        }, {
-            headers: {
-                'Content-Type': 'application/json',
-                'Authorization': `Bearer ${apiToken}` // Добавляем токен аутентификации
-            }
-        });
-
-        res.json(response.data);
-    } catch (error) {
-        console.error('Error: ', error.response ? error.response.data : error.message);
-        res.status(500).send('Error: ' + (error.response ? error.response.data : error.message));
-    }
-});
+const port = 3000;
+
+app.use(bodyParser.json());
+
+app.post('/generate', async (req, res) => {
+  const { messages, temperature, max_tokens } = req.body;
 
-app.listen(PORT, () => {
-    console.log(`Server is running on port ${PORT}`);
+  try {
+    const response = await fetch('https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf', {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${process.env.API_KEY}` // Замените на ваш токен Hugging Face API
+      },
+      body: JSON.stringify({
+        inputs: messages,
+        parameters: {
+          temperature: temperature || 0.7,
+          max_new_tokens: max_tokens || 100
+        }
+      })
+    });
+
+    const data = await response.json();
+    const generatedText = data.generated_text;
+
+    // Добавляем сгенерированное сообщение в конец массива messages
+    messages.push({ role: "assistant", content: generatedText });
+
+    res.json({ messages });
+  } catch (error) {
+    console.error(error);
+    res.status(500).json({ error: 'Произошла ошибка при генерации текста.' });
+  }
 });
+
+app.listen(port, () => {
+  console.log(`Сервер запущен на порту ${port}`);
+});