Update server.js

server.js

@@ -10,53 +10,26 @@ app.post('/chat', async (req, res) => {
     const { messages, temperature, max_tokens } = req.body;
 
     try {
-        const { default: fetch } = await import('node-fetch');
-
-        const response = await fetch('https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf', {
-            method: 'POST',
+        const response = await axios.post('https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf', {
+            inputs: messages,
+            parameters: {
+                temperature: temperature || 0.7,
+                max_new_tokens: max_tokens || 100
+            }
+        }, {
             headers: {
                 'Content-Type': 'application/json',
                 'Authorization': `Bearer ${apiToken}`
-            },
-            body: JSON.stringify({
-                "inputs": messages,
-                "parameters": {
-                    "temperature": temperature || 0.7,
-                    "max_new_tokens": max_tokens || 100
-                }
-            })
-        });
-
-        const text = await response.text(); // Получаем текст ответа
-        console.log("Response text:", text); // Логируем ответ
-
-        // Проверяем, если ответ содержит JSON
-        if (response.headers.get('content-type')?.includes('application/json')) {
-            let data;
-            try {
-                data = JSON.parse(text); // Пытаемся распарсить JSON
-            } catch (jsonError) {
-                console.error("JSON parsing error:", jsonError);
-                return res.status(500).json({ error: `Ошибка парсинга JSON: ${jsonError.message}` });
             }
+        });
 
-            const generatedText = data.generated_text;
-
-            // Добавляем сгенерированное сообщение в конец массива messages
-            messages.push({ role: 'assistant', content: generatedText });
-            res.json({ messages });
-        } else {
-            console.error("Received non-JSON response:", text);
-
-            const generatedText = text;
+        const data = response.data;
+        const generatedText = data.generated_text;
 
-            // Добавляем сгенерированное сообщение в конец массива messages
-            messages.push({ role: 'assistant', content: generatedText });
-            res.json({ generatedText });
-        }
+        res.json({ generated_text: generatedText });
     } catch (error) {
-        console.error("Error during text generation:", error);
-        res.status(500).json({ error: `Произошла ошибка при генерации текста: ${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}` });
     }
 });