442 lines
12 KiB
JavaScript
442 lines
12 KiB
JavaScript
|
|
const express = require('express');
|
|
|
|
const Rag_documentsService = require('../services/rag_documents');
|
|
const Rag_documentsDBApi = require('../db/api/rag_documents');
|
|
const wrapAsync = require('../helpers').wrapAsync;
|
|
|
|
|
|
const router = express.Router();
|
|
|
|
const { parse } = require('json2csv');
|
|
|
|
|
|
const {
|
|
checkCrudPermissions,
|
|
} = require('../middlewares/check-permissions');
|
|
|
|
router.use(checkCrudPermissions('rag_documents'));
|
|
|
|
|
|
/**
|
|
* @swagger
|
|
* components:
|
|
* schemas:
|
|
* Rag_documents:
|
|
* type: object
|
|
* properties:
|
|
|
|
* title:
|
|
* type: string
|
|
* default: title
|
|
* language:
|
|
* type: string
|
|
* default: language
|
|
* content_text:
|
|
* type: string
|
|
* default: content_text
|
|
* content_hash:
|
|
* type: string
|
|
* default: content_hash
|
|
|
|
* token_count:
|
|
* type: integer
|
|
* format: int64
|
|
|
|
|
|
*/
|
|
|
|
/**
|
|
* @swagger
|
|
* tags:
|
|
* name: Rag_documents
|
|
* description: The Rag_documents managing API
|
|
*/
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents:
|
|
* post:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Add new item
|
|
* description: Add new item
|
|
* requestBody:
|
|
* required: true
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* properties:
|
|
* data:
|
|
* description: Data of the updated item
|
|
* type: object
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* responses:
|
|
* 200:
|
|
* description: The item was successfully added
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 405:
|
|
* description: Invalid input data
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.post('/', wrapAsync(async (req, res) => {
|
|
const referer = req.headers.referer || `${req.protocol}://${req.hostname}${req.originalUrl}`;
|
|
const link = new URL(referer);
|
|
await Rag_documentsService.create(req.body.data, req.currentUser, true, link.host);
|
|
const payload = true;
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/budgets/bulk-import:
|
|
* post:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Bulk import items
|
|
* description: Bulk import items
|
|
* requestBody:
|
|
* required: true
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* properties:
|
|
* data:
|
|
* description: Data of the updated items
|
|
* type: array
|
|
* items:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* responses:
|
|
* 200:
|
|
* description: The items were successfully imported
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 405:
|
|
* description: Invalid input data
|
|
* 500:
|
|
* description: Some server error
|
|
*
|
|
*/
|
|
router.post('/bulk-import', wrapAsync(async (req, res) => {
|
|
const referer = req.headers.referer || `${req.protocol}://${req.hostname}${req.originalUrl}`;
|
|
const link = new URL(referer);
|
|
await Rag_documentsService.bulkImport(req, res, true, link.host);
|
|
const payload = true;
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/{id}:
|
|
* put:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Update the data of the selected item
|
|
* description: Update the data of the selected item
|
|
* parameters:
|
|
* - in: path
|
|
* name: id
|
|
* description: Item ID to update
|
|
* required: true
|
|
* schema:
|
|
* type: string
|
|
* requestBody:
|
|
* description: Set new item data
|
|
* required: true
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* properties:
|
|
* id:
|
|
* description: ID of the updated item
|
|
* type: string
|
|
* data:
|
|
* description: Data of the updated item
|
|
* type: object
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* required:
|
|
* - id
|
|
* responses:
|
|
* 200:
|
|
* description: The item data was successfully updated
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 400:
|
|
* description: Invalid ID supplied
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Item not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.put('/:id', wrapAsync(async (req, res) => {
|
|
await Rag_documentsService.update(req.body.data, req.body.id, req.currentUser);
|
|
const payload = true;
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/{id}:
|
|
* delete:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Delete the selected item
|
|
* description: Delete the selected item
|
|
* parameters:
|
|
* - in: path
|
|
* name: id
|
|
* description: Item ID to delete
|
|
* required: true
|
|
* schema:
|
|
* type: string
|
|
* responses:
|
|
* 200:
|
|
* description: The item was successfully deleted
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 400:
|
|
* description: Invalid ID supplied
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Item not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.delete('/:id', wrapAsync(async (req, res) => {
|
|
await Rag_documentsService.remove(req.params.id, req.currentUser);
|
|
const payload = true;
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/deleteByIds:
|
|
* post:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Delete the selected item list
|
|
* description: Delete the selected item list
|
|
* requestBody:
|
|
* required: true
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* properties:
|
|
* ids:
|
|
* description: IDs of the updated items
|
|
* type: array
|
|
* responses:
|
|
* 200:
|
|
* description: The items was successfully deleted
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Items not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.post('/deleteByIds', wrapAsync(async (req, res) => {
|
|
await Rag_documentsService.deleteByIds(req.body.data, req.currentUser);
|
|
const payload = true;
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents:
|
|
* get:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Get all rag_documents
|
|
* description: Get all rag_documents
|
|
* responses:
|
|
* 200:
|
|
* description: Rag_documents list successfully received
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: array
|
|
* items:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Data not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.get('/', wrapAsync(async (req, res) => {
|
|
const filetype = req.query.filetype
|
|
|
|
const currentUser = req.currentUser;
|
|
const payload = await Rag_documentsDBApi.findAll(
|
|
req.query, { currentUser }
|
|
);
|
|
if (filetype && filetype === 'csv') {
|
|
const fields = ['id','title','language','content_text','content_hash',
|
|
'token_count',
|
|
|
|
'published_at',
|
|
];
|
|
const opts = { fields };
|
|
try {
|
|
const csv = parse(payload.rows, opts);
|
|
res.status(200).attachment(csv);
|
|
res.send(csv)
|
|
|
|
} catch (err) {
|
|
console.error(err);
|
|
}
|
|
} else {
|
|
res.status(200).send(payload);
|
|
}
|
|
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/count:
|
|
* get:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Count all rag_documents
|
|
* description: Count all rag_documents
|
|
* responses:
|
|
* 200:
|
|
* description: Rag_documents count successfully received
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: array
|
|
* items:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Data not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.get('/count', wrapAsync(async (req, res) => {
|
|
|
|
const currentUser = req.currentUser;
|
|
const payload = await Rag_documentsDBApi.findAll(
|
|
req.query,
|
|
null,
|
|
{ countOnly: true, currentUser }
|
|
);
|
|
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/autocomplete:
|
|
* get:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Find all rag_documents that match search criteria
|
|
* description: Find all rag_documents that match search criteria
|
|
* responses:
|
|
* 200:
|
|
* description: Rag_documents list successfully received
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: array
|
|
* items:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Data not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.get('/autocomplete', async (req, res) => {
|
|
|
|
const payload = await Rag_documentsDBApi.findAllAutocomplete(
|
|
req.query.query,
|
|
req.query.limit,
|
|
req.query.offset,
|
|
|
|
);
|
|
|
|
res.status(200).send(payload);
|
|
});
|
|
|
|
/**
|
|
* @swagger
|
|
* /api/rag_documents/{id}:
|
|
* get:
|
|
* security:
|
|
* - bearerAuth: []
|
|
* tags: [Rag_documents]
|
|
* summary: Get selected item
|
|
* description: Get selected item
|
|
* parameters:
|
|
* - in: path
|
|
* name: id
|
|
* description: ID of item to get
|
|
* required: true
|
|
* schema:
|
|
* type: string
|
|
* responses:
|
|
* 200:
|
|
* description: Selected item successfully received
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: "#/components/schemas/Rag_documents"
|
|
* 400:
|
|
* description: Invalid ID supplied
|
|
* 401:
|
|
* $ref: "#/components/responses/UnauthorizedError"
|
|
* 404:
|
|
* description: Item not found
|
|
* 500:
|
|
* description: Some server error
|
|
*/
|
|
router.get('/:id', wrapAsync(async (req, res) => {
|
|
const payload = await Rag_documentsDBApi.findBy(
|
|
{ id: req.params.id },
|
|
);
|
|
|
|
|
|
|
|
res.status(200).send(payload);
|
|
}));
|
|
|
|
router.use('/', require('../helpers').commonErrorHandler);
|
|
|
|
module.exports = router;
|