anything-llm/server/models/workspaceChats.js

117 lines
3.1 KiB
JavaScript
Raw Normal View History

const { checkForMigrations } = require("../utils/database");
2023-06-04 04:28:07 +02:00
const WorkspaceChats = {
2023-06-08 06:31:35 +02:00
tablename: "workspace_chats",
2023-06-04 04:28:07 +02:00
colsInit: `
id INTEGER PRIMARY KEY AUTOINCREMENT,
workspaceId INTEGER NOT NULL,
prompt TEXT NOT NULL,
response TEXT NOT NULL,
include BOOL DEFAULT true,
createdAt TEXT DEFAULT CURRENT_TIMESTAMP,
lastUpdatedAt TEXT DEFAULT CURRENT_TIMESTAMP
`,
migrateTable: async function () {
console.log(
`\x1b[34m[MIGRATING]\x1b[0m Checking for WorkspaceChats migrations`
);
const db = await this.db(false);
await checkForMigrations(this, db);
},
migrations: function () {
return [];
},
db: async function (tracing = true) {
2023-06-08 06:31:35 +02:00
const sqlite3 = require("sqlite3").verbose();
const { open } = require("sqlite");
2023-06-04 04:28:07 +02:00
const db = await open({
filename: `${
!!process.env.STORAGE_DIR ? `${process.env.STORAGE_DIR}/` : "storage/"
}anythingllm.db`,
2023-06-08 06:31:35 +02:00
driver: sqlite3.Database,
});
2023-06-04 04:28:07 +02:00
2023-06-08 06:31:35 +02:00
await db.exec(
`CREATE TABLE IF NOT EXISTS ${this.tablename} (${this.colsInit})`
);
if (tracing) db.on("trace", (sql) => console.log(sql));
2023-06-08 06:31:35 +02:00
return db;
2023-06-04 04:28:07 +02:00
},
new: async function ({ workspaceId, prompt, response = {} }) {
2023-06-08 06:31:35 +02:00
const db = await this.db();
const { id, success, message } = await db
.run(
`INSERT INTO ${this.tablename} (workspaceId, prompt, response) VALUES (?, ?, ?)`,
[workspaceId, prompt, JSON.stringify(response)]
)
2023-06-04 04:28:07 +02:00
.then((res) => {
2023-06-08 06:31:35 +02:00
return { id: res.lastID, success: true, message: null };
2023-06-04 04:28:07 +02:00
})
.catch((error) => {
2023-06-08 06:31:35 +02:00
return { id: null, success: false, message: error.message };
});
if (!success) {
db.close();
return { chat: null, message };
}
2023-06-04 04:28:07 +02:00
2023-06-08 06:31:35 +02:00
const chat = await db.get(
`SELECT * FROM ${this.tablename} WHERE id = ${id}`
);
db.close();
2023-06-08 06:31:35 +02:00
return { chat, message: null };
2023-06-04 04:28:07 +02:00
},
forWorkspace: async function (workspaceId = null) {
if (!workspaceId) return [];
2023-06-08 06:31:35 +02:00
return await this.where(
`workspaceId = ${workspaceId} AND include = true`,
null,
"ORDER BY id ASC"
);
2023-06-04 04:28:07 +02:00
},
markHistoryInvalid: async function (workspaceId = null) {
if (!workspaceId) return;
2023-06-08 06:31:35 +02:00
const db = await this.db();
await db.run(
`UPDATE ${this.tablename} SET include = false WHERE workspaceId = ?`,
[workspaceId]
);
db.close();
2023-06-04 04:28:07 +02:00
return;
},
2023-06-08 06:31:35 +02:00
get: async function (clause = "") {
const db = await this.db();
const result = await db
.get(`SELECT * FROM ${this.tablename} WHERE ${clause}`)
.then((res) => res || null);
db.close();
2023-06-04 04:28:07 +02:00
if (!result) return null;
2023-06-08 06:31:35 +02:00
return result;
2023-06-04 04:28:07 +02:00
},
2023-06-08 06:31:35 +02:00
delete: async function (clause = "") {
const db = await this.db();
await db.get(`DELETE FROM ${this.tablename} WHERE ${clause}`);
db.close();
2023-06-08 06:31:35 +02:00
return true;
2023-06-04 04:28:07 +02:00
},
2023-06-08 06:31:35 +02:00
where: async function (clause = "", limit = null, order = null) {
const db = await this.db();
const results = await db.all(
`SELECT * FROM ${this.tablename} ${clause ? `WHERE ${clause}` : ""} ${
!!limit ? `LIMIT ${limit}` : ""
} ${!!order ? order : ""}`
);
db.close();
2023-06-08 06:31:35 +02:00
return results;
2023-06-04 04:28:07 +02:00
},
2023-06-08 06:31:35 +02:00
};
2023-06-04 04:28:07 +02:00
2023-06-08 06:31:35 +02:00
module.exports = { WorkspaceChats };