From 75b67100529b8b3c29ce624eab2fd1cc854e8dfc Mon Sep 17 00:00:00 2001
From: Timothy Carambat <rambat1010@gmail.com>
Date: Thu, 18 Jul 2024 10:35:02 -0700
Subject: [PATCH] Extend support for some chars in slugify of workspace (#1890)

resolves #1877
---
 server/models/workspace.js | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/server/models/workspace.js b/server/models/workspace.js
index e59ac4d8..d80f00dd 100644
--- a/server/models/workspace.js
+++ b/server/models/workspace.js
@@ -1,5 +1,5 @@
 const prisma = require("../utils/prisma");
-const slugify = require("slugify");
+const slugifyModule = require("slugify");
 const { Document } = require("./documents");
 const { WorkspaceUser } = require("./workspaceUsers");
 const { ROLES } = require("../utils/middleware/multiUserProtected");
@@ -28,16 +28,27 @@ const Workspace = {
     "agentModel",
     "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) {
     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();
 
     const existingBySlug = await this.get({ slug });
     if (existingBySlug !== null) {
       const slugSeed = Math.floor(10000000 + Math.random() * 90000000);
-      slug = slugify(`${name}-${slugSeed}`, { lower: true });
+      slug = this.slugify(`${name}-${slugSeed}`, { lower: true });
     }
 
     try {