Update server.js

server.js

@@ -1,4 +1,3 @@
-//require("dotenv").config();
 const express = require("express");
 const axios = require("axios");
 
@@ -6,6 +5,7 @@ const app = express();
 app.use(express.json());
 
 const API_URL = `https://api.cloudflare.com/client/v4/accounts/${process.env.ID}/ai/run/@cf/deepseek-ai/deepseek-r1-distill-qwen-32b`;
+const API_KEY = process.env.KEY;
 const sessions = {};
 
 app.post("/chat", async (req, res) => {
@@ -15,16 +15,14 @@ app.post("/chat", async (req, res) => {
   sessions[session] = sessions[session] || [];
   sessions[session].push({ role: "user", content: message });
 
-  res.setHeader("Content-Type", "text/plain");
-
   try {
-    const { data } = await axios.post(
+    const response = await axios.post(
       API_URL,
       { messages: sessions[session], temperature },
-      { headers: { Authorization: `Bearer ${process.env.KEY}` }, responseType: "stream" }
+      { headers: { Authorization: `Bearer ${API_KEY}` } }
     );
-
-    res.send(data);
+    const answer = response?.data?.response;
+    res.send((answer.includes('</think>') ?? '<think>\n' + answer)?.trim()));
   } catch (err) {
     res.status(500).send(err.message);
   }