38689-vm/backend/src/db/api/knowledge_articles.js
2026-02-22 15:41:40 +00:00

618 lines
15 KiB
JavaScript

const db = require('../models');
const FileDBApi = require('./file');
const crypto = require('crypto');
const Utils = require('../utils');
const Sequelize = db.Sequelize;
const Op = Sequelize.Op;
module.exports = class Knowledge_articlesDBApi {
static async create(data, options) {
const currentUser = (options && options.currentUser) || { id: null };
const transaction = (options && options.transaction) || undefined;
const knowledge_articles = await db.knowledge_articles.create(
{
id: data.id || undefined,
title: data.title
||
null
,
slug: data.slug
||
null
,
visibility: data.visibility
||
null
,
content: data.content
||
null
,
tags: data.tags
||
null
,
external_ref: data.external_ref
||
null
,
quality_score: data.quality_score
||
null
,
published_at: data.published_at
||
null
,
last_reviewed_at: data.last_reviewed_at
||
null
,
importHash: data.importHash || null,
createdById: currentUser.id,
updatedById: currentUser.id,
},
{ transaction },
);
await knowledge_articles.setKnowledge_source( data.knowledge_source || null, {
transaction,
});
return knowledge_articles;
}
static async bulkImport(data, options) {
const currentUser = (options && options.currentUser) || { id: null };
const transaction = (options && options.transaction) || undefined;
// Prepare data - wrapping individual data transformations in a map() method
const knowledge_articlesData = data.map((item, index) => ({
id: item.id || undefined,
title: item.title
||
null
,
slug: item.slug
||
null
,
visibility: item.visibility
||
null
,
content: item.content
||
null
,
tags: item.tags
||
null
,
external_ref: item.external_ref
||
null
,
quality_score: item.quality_score
||
null
,
published_at: item.published_at
||
null
,
last_reviewed_at: item.last_reviewed_at
||
null
,
importHash: item.importHash || null,
createdById: currentUser.id,
updatedById: currentUser.id,
createdAt: new Date(Date.now() + index * 1000),
}));
// Bulk create items
const knowledge_articles = await db.knowledge_articles.bulkCreate(knowledge_articlesData, { transaction });
// For each item created, replace relation files
return knowledge_articles;
}
static async update(id, data, options) {
const currentUser = (options && options.currentUser) || {id: null};
const transaction = (options && options.transaction) || undefined;
const knowledge_articles = await db.knowledge_articles.findByPk(id, {}, {transaction});
const updatePayload = {};
if (data.title !== undefined) updatePayload.title = data.title;
if (data.slug !== undefined) updatePayload.slug = data.slug;
if (data.visibility !== undefined) updatePayload.visibility = data.visibility;
if (data.content !== undefined) updatePayload.content = data.content;
if (data.tags !== undefined) updatePayload.tags = data.tags;
if (data.external_ref !== undefined) updatePayload.external_ref = data.external_ref;
if (data.quality_score !== undefined) updatePayload.quality_score = data.quality_score;
if (data.published_at !== undefined) updatePayload.published_at = data.published_at;
if (data.last_reviewed_at !== undefined) updatePayload.last_reviewed_at = data.last_reviewed_at;
updatePayload.updatedById = currentUser.id;
await knowledge_articles.update(updatePayload, {transaction});
if (data.knowledge_source !== undefined) {
await knowledge_articles.setKnowledge_source(
data.knowledge_source,
{ transaction }
);
}
return knowledge_articles;
}
static async deleteByIds(ids, options) {
const currentUser = (options && options.currentUser) || { id: null };
const transaction = (options && options.transaction) || undefined;
const knowledge_articles = await db.knowledge_articles.findAll({
where: {
id: {
[Op.in]: ids,
},
},
transaction,
});
await db.sequelize.transaction(async (transaction) => {
for (const record of knowledge_articles) {
await record.update(
{deletedBy: currentUser.id},
{transaction}
);
}
for (const record of knowledge_articles) {
await record.destroy({transaction});
}
});
return knowledge_articles;
}
static async remove(id, options) {
const currentUser = (options && options.currentUser) || {id: null};
const transaction = (options && options.transaction) || undefined;
const knowledge_articles = await db.knowledge_articles.findByPk(id, options);
await knowledge_articles.update({
deletedBy: currentUser.id
}, {
transaction,
});
await knowledge_articles.destroy({
transaction
});
return knowledge_articles;
}
static async findBy(where, options) {
const transaction = (options && options.transaction) || undefined;
const knowledge_articles = await db.knowledge_articles.findOne(
{ where },
{ transaction },
);
if (!knowledge_articles) {
return knowledge_articles;
}
const output = knowledge_articles.get({plain: true});
output.conversation_messages_knowledge_article_used = await knowledge_articles.getConversation_messages_knowledge_article_used({
transaction
});
output.rag_results_knowledge_article = await knowledge_articles.getRag_results_knowledge_article({
transaction
});
output.knowledge_source = await knowledge_articles.getKnowledge_source({
transaction
});
return output;
}
static async findAll(
filter,
options
) {
const limit = filter.limit || 0;
let offset = 0;
let where = {};
const currentPage = +filter.page;
offset = currentPage * limit;
const orderBy = null;
const transaction = (options && options.transaction) || undefined;
let include = [
{
model: db.knowledge_sources,
as: 'knowledge_source',
where: filter.knowledge_source ? {
[Op.or]: [
{ id: { [Op.in]: filter.knowledge_source.split('|').map(term => Utils.uuid(term)) } },
{
source_name: {
[Op.or]: filter.knowledge_source.split('|').map(term => ({ [Op.iLike]: `%${term}%` }))
}
},
]
} : {},
},
];
if (filter) {
if (filter.id) {
where = {
...where,
['id']: Utils.uuid(filter.id),
};
}
if (filter.title) {
where = {
...where,
[Op.and]: Utils.ilike(
'knowledge_articles',
'title',
filter.title,
),
};
}
if (filter.slug) {
where = {
...where,
[Op.and]: Utils.ilike(
'knowledge_articles',
'slug',
filter.slug,
),
};
}
if (filter.content) {
where = {
...where,
[Op.and]: Utils.ilike(
'knowledge_articles',
'content',
filter.content,
),
};
}
if (filter.tags) {
where = {
...where,
[Op.and]: Utils.ilike(
'knowledge_articles',
'tags',
filter.tags,
),
};
}
if (filter.external_ref) {
where = {
...where,
[Op.and]: Utils.ilike(
'knowledge_articles',
'external_ref',
filter.external_ref,
),
};
}
if (filter.quality_scoreRange) {
const [start, end] = filter.quality_scoreRange;
if (start !== undefined && start !== null && start !== '') {
where = {
...where,
quality_score: {
...where.quality_score,
[Op.gte]: start,
},
};
}
if (end !== undefined && end !== null && end !== '') {
where = {
...where,
quality_score: {
...where.quality_score,
[Op.lte]: end,
},
};
}
}
if (filter.published_atRange) {
const [start, end] = filter.published_atRange;
if (start !== undefined && start !== null && start !== '') {
where = {
...where,
published_at: {
...where.published_at,
[Op.gte]: start,
},
};
}
if (end !== undefined && end !== null && end !== '') {
where = {
...where,
published_at: {
...where.published_at,
[Op.lte]: end,
},
};
}
}
if (filter.last_reviewed_atRange) {
const [start, end] = filter.last_reviewed_atRange;
if (start !== undefined && start !== null && start !== '') {
where = {
...where,
last_reviewed_at: {
...where.last_reviewed_at,
[Op.gte]: start,
},
};
}
if (end !== undefined && end !== null && end !== '') {
where = {
...where,
last_reviewed_at: {
...where.last_reviewed_at,
[Op.lte]: end,
},
};
}
}
if (filter.active !== undefined) {
where = {
...where,
active: filter.active === true || filter.active === 'true'
};
}
if (filter.visibility) {
where = {
...where,
visibility: filter.visibility,
};
}
if (filter.createdAtRange) {
const [start, end] = filter.createdAtRange;
if (start !== undefined && start !== null && start !== '') {
where = {
...where,
['createdAt']: {
...where.createdAt,
[Op.gte]: start,
},
};
}
if (end !== undefined && end !== null && end !== '') {
where = {
...where,
['createdAt']: {
...where.createdAt,
[Op.lte]: end,
},
};
}
}
}
const queryOptions = {
where,
include,
distinct: true,
order: filter.field && filter.sort
? [[filter.field, filter.sort]]
: [['createdAt', 'desc']],
transaction: options?.transaction,
logging: console.log
};
if (!options?.countOnly) {
queryOptions.limit = limit ? Number(limit) : undefined;
queryOptions.offset = offset ? Number(offset) : undefined;
}
try {
const { rows, count } = await db.knowledge_articles.findAndCountAll(queryOptions);
return {
rows: options?.countOnly ? [] : rows,
count: count
};
} catch (error) {
console.error('Error executing query:', error);
throw error;
}
}
static async findAllAutocomplete(query, limit, offset, ) {
let where = {};
if (query) {
where = {
[Op.or]: [
{ ['id']: Utils.uuid(query) },
Utils.ilike(
'knowledge_articles',
'title',
query,
),
],
};
}
const records = await db.knowledge_articles.findAll({
attributes: [ 'id', 'title' ],
where,
limit: limit ? Number(limit) : undefined,
offset: offset ? Number(offset) : undefined,
orderBy: [['title', 'ASC']],
});
return records.map((record) => ({
id: record.id,
label: record.title,
}));
}
};