mirror of https://github.com/arc53/DocsGPT
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.9 KiB
JavaScript
96 lines
2.9 KiB
JavaScript
import jsonServer from "json-server";
|
|
import routes from "./mocks/routes.json" assert { type: "json" };
|
|
import { v4 as uuid } from "uuid";
|
|
import cors from 'cors'
|
|
const server = jsonServer.create();
|
|
const router = jsonServer.router("./src/mocks/db.json");
|
|
const middlewares = jsonServer.defaults();
|
|
|
|
const localStorage = [];
|
|
|
|
server.use(middlewares);
|
|
server.use(cors({ origin: '*' }))
|
|
server.use(jsonServer.rewriter(routes));
|
|
|
|
server.use((req, res, next) => {
|
|
if (req.method === "POST") {
|
|
if (req.url.includes("/delete_conversation")) {
|
|
req.method = "DELETE";
|
|
} else if (req.url === "/upload") {
|
|
const taskId = uuid();
|
|
localStorage.push(taskId);
|
|
}
|
|
}
|
|
next();
|
|
});
|
|
|
|
router.render = (req, res) => {
|
|
if (req.url === "/feedback") {
|
|
res.status(200).jsonp({ status: "ok" });
|
|
} else if (req.url === "/upload") {
|
|
res.status(200).jsonp({
|
|
status: "ok",
|
|
task_id: localStorage[localStorage.length - 1],
|
|
});
|
|
} else if (req.url.includes("/task_status")) {
|
|
const taskId = req.query["task_id"];
|
|
const taskIdExists = localStorage.includes(taskId);
|
|
if (taskIdExists) {
|
|
res.status(200).jsonp({
|
|
result: {
|
|
directory: "temp",
|
|
filename: "install.rst",
|
|
formats: [".rst", ".md", ".pdf"],
|
|
name_job: "somename",
|
|
user: "local",
|
|
},
|
|
status: "SUCCESS",
|
|
});
|
|
} else {
|
|
res.status(404).jsonp({});
|
|
}
|
|
} else if (req.url === "/stream" && req.method === "POST") {
|
|
console.log('pinged !')
|
|
res.writeHead(200, {
|
|
'Content-Type': 'text/event-stream',
|
|
'Cache-Control': 'no-cache',
|
|
'Connection': 'keep-alive'
|
|
});
|
|
const message = ('Hi, How are you today?').split(' ');
|
|
let index = 0;
|
|
const interval = setInterval(() => {
|
|
if (index < message.length) {
|
|
res.write(`data: {"answer": "${message[index++]} "}\n`);
|
|
} else {
|
|
res.write(`data: {"type": "id", "id": "65cbc39d11f077b9eeb06d26"}\n`)
|
|
res.write(`data: {"type": "end"}\n`)
|
|
clearInterval(interval); // Stop the interval once the message is fully streamed
|
|
res.end(); // End the response
|
|
}
|
|
}, 500); // Send a word every 1 second
|
|
}
|
|
else if (req.url === '/search' && req.method === 'POST') {
|
|
res.status(200).json(
|
|
[
|
|
{
|
|
"text": "\n\n/api/answer\nIt's a POST request that sends a JSON in body with 4 values. It will receive an answer for a user provided question.\n",
|
|
"title": "API-docs.md"
|
|
},
|
|
{
|
|
"text": "\n\nOur Standards\n\nExamples of behavior that contribute to a positive environment for our\ncommunity include:\n* Demonstrating empathy and kindness towards other people\n",
|
|
"title": "How-to-use-different-LLM.md"
|
|
}
|
|
]
|
|
)
|
|
}
|
|
else {
|
|
res.status(res.statusCode).jsonp(res.locals.data);
|
|
}
|
|
};
|
|
|
|
server.use(router);
|
|
|
|
server.listen(7091, () => {
|
|
console.log("JSON Server is running");
|
|
});
|