fast72 commited on
Commit
5ec0431
·
verified ·
1 Parent(s): 6c0bfe7

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +12 -4
server.js CHANGED
@@ -1,3 +1,4 @@
 
1
  const express = require("express");
2
  const axios = require("axios");
3
 
@@ -14,6 +15,8 @@ app.post("/chat", async (req, res) => {
14
  sessions[session] = sessions[session] || [];
15
  sessions[session].push({ role: "user", content: message });
16
 
 
 
17
  try {
18
  const response = await axios.post(
19
  API_URL,
@@ -21,12 +24,17 @@ app.post("/chat", async (req, res) => {
21
  { headers: { Authorization: `Bearer ${process.env.KEY}` }, responseType: "stream" }
22
  );
23
 
24
- let fullResponse = "";
25
- response.data.on("data", (chunk) => (fullResponse += chunk));
26
- response.data.on("end", () => res.send(fullResponse.trim()));
 
 
 
 
 
27
  } catch (err) {
28
  res.status(500).send(err.message);
29
  }
30
  });
31
 
32
- app.listen(7860, () => console.log("Running on http://localhost:3000"));
 
1
+ //require("dotenv").config();
2
  const express = require("express");
3
  const axios = require("axios");
4
 
 
15
  sessions[session] = sessions[session] || [];
16
  sessions[session].push({ role: "user", content: message });
17
 
18
+ res.setHeader("Content-Type", "text/plain");
19
+
20
  try {
21
  const response = await axios.post(
22
  API_URL,
 
24
  { headers: { Authorization: `Bearer ${process.env.KEY}` }, responseType: "stream" }
25
  );
26
 
27
+ response.data.on("data", (chunk) => {
28
+ try {
29
+ const parsed = JSON.parse(chunk.toString());
30
+ if (parsed.response) res.write(parsed.response);
31
+ } catch (err) {}
32
+ });
33
+
34
+ response.data.on("end", () => res.end());
35
  } catch (err) {
36
  res.status(500).send(err.message);
37
  }
38
  });
39
 
40
+ app.listen(7860, () => console.log("Running on http://localhost:7860"));