Update server.js

server.js

@@ -9,45 +9,50 @@ app.use(express.json());
 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',
-      headers: {
-        'Content-Type': `application/json`,
-        'Authorization': `Bearer ${apiToken}` // Замените на ваш токен Hugging Face API
-      },
-      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); // Логируем ответ
-
-    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 { default: fetch } = await import('node-fetch');
+
+        const response = await fetch('https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf', {
+            method: 'POST',
+            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);
+            res.status(500).json({ error: `Получен ответ, не являющийся JSON: ${text}` });
+        }
+    } catch (error) {
+        console.error("Error during text generation:", error);
+        res.status(500).json({ error: `Произошла ошибка при генерации текста: ${error.message}` });
     }
-
-    const generatedText = data.generated_text;
-
-    // Добавляем сгенерированное сообщение в конец массива messages
-    messages.push({ role: 'assistant', content: generatedText });
-
-    res.json({ messages });
-  } catch (error) {
-    console.error("Error during text generation:", error);
-    res.status(500).json({ error: `Произошла ошибка при генерации текста: ${error.message}` });
-  }
 });
 
 app.listen(PORT, () => {