anything-llm/server/models/workspaceThread.js
Imtiaz Mehmood ad778dd36d
Feat/quick delete chat (#1302)
* feat:quick delete chat thread

* update:pull request template

* refactor bulk-deletion implementation

* unset pull_request_changes

* add border none for desktop support

* unset marks when toggling bulk mode

---------

Co-authored-by: timothycarambat <rambat1010@gmail.com>
2024-05-08 11:10:00 -07:00

90 lines
2.2 KiB
JavaScript

const prisma = require("../utils/prisma");
const { v4: uuidv4 } = require("uuid");
const WorkspaceThread = {
writable: ["name"],
new: async function (workspace, userId = null) {
try {
const thread = await prisma.workspace_threads.create({
data: {
name: "New thread",
slug: uuidv4(),
user_id: userId ? Number(userId) : null,
workspace_id: workspace.id,
},
});
return { thread, message: null };
} catch (error) {
console.error(error.message);
return { thread: null, message: error.message };
}
},
update: async function (prevThread = null, data = {}) {
if (!prevThread) throw new Error("No thread id provided for update");
const validData = {};
Object.entries(data).forEach(([key, value]) => {
if (!this.writable.includes(key)) return;
validData[key] = value;
});
if (Object.keys(validData).length === 0)
return { thread: prevThread, message: "No valid fields to update!" };
try {
const thread = await prisma.workspace_threads.update({
where: { id: prevThread.id },
data: validData,
});
return { thread, message: null };
} catch (error) {
console.error(error.message);
return { thread: null, message: error.message };
}
},
get: async function (clause = {}) {
try {
const thread = await prisma.workspace_threads.findFirst({
where: clause,
});
return thread || null;
} catch (error) {
console.error(error.message);
return null;
}
},
delete: async function (clause = {}) {
try {
await prisma.workspace_threads.deleteMany({
where: clause,
});
return true;
} catch (error) {
console.error(error.message);
return false;
}
},
where: async function (clause = {}, limit = null, orderBy = null) {
try {
const results = await prisma.workspace_threads.findMany({
where: clause,
...(limit !== null ? { take: limit } : {}),
...(orderBy !== null ? { orderBy } : {}),
});
return results;
} catch (error) {
console.error(error.message);
return [];
}
},
};
module.exports = { WorkspaceThread };