kamp/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2024-11-15 02:50:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

Extend support for some chars in slugify of workspace (#1890)

Browse Source 
resolves #1877
This commit is contained in:
Timothy Carambat 2024-07-18 10:35:02 -07:00 committed by GitHub
parent e909b25b29
commit 75b6710052
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
server/models/workspace.js
View File

@ -1,5 +1,5 @@
const prisma = require("../utils/prisma"); const prisma = require("../utils/prisma");
const slugify = require("slugify"); const slugifyModule = require("slugify");
const { Document } = require("./documents"); const { Document } = require("./documents");
const { WorkspaceUser } = require("./workspaceUsers"); const { WorkspaceUser } = require("./workspaceUsers");
const { ROLES } = require("../utils/middleware/multiUserProtected"); const { ROLES } = require("../utils/middleware/multiUserProtected");
@ -28,16 +28,27 @@ const Workspace = {
"agentModel", "agentModel",
"queryRefusalResponse", "queryRefusalResponse",
], ],
/**
* The default Slugify module requires some additional mapping to prevent downstream issues
* with some vector db providers and instead of building a normalization method for every provider
* we can capture this on the table level to not have to worry about it.
* @param {...any} args - slugify args for npm package.
* @returns {string}
*/
slugify: function (...args) {
slugifyModule.extend({ "+": " plus ", "!": " bang " });
return slugifyModule(...args);
},
new: async function (name = null, creatorId = null) { new: async function (name = null, creatorId = null) {
if (!name) return { result: null, message: "name cannot be null" }; if (!name) return { result: null, message: "name cannot be null" };
var slug = slugify(name, { lower: true }); var slug = this.slugify(name, { lower: true });
slug = slug || uuidv4(); slug = slug || uuidv4();
const existingBySlug = await this.get({ slug }); const existingBySlug = await this.get({ slug });
if (existingBySlug !== null) { if (existingBySlug !== null) {
const slugSeed = Math.floor(10000000 + Math.random() * 90000000); const slugSeed = Math.floor(10000000 + Math.random() * 90000000);
slug = slugify(`${name}-${slugSeed}`, { lower: true }); slug = this.slugify(`${name}-${slugSeed}`, { lower: true });
} }
try { try {