Implement workspace threading that is backwards compatible (#699)

* Implement workspace thread that is compatible with legacy versions * last touches * comment on chat qty enforcement
2026-04-26 01:25:15 +02:00 · 2024-02-08 18:37:22 -08:00
parent b985524901
commit 406732830f
21 changed files with 1087 additions and 83 deletions
--- a/server/endpoints/workspaceThreads.js
+++ b/server/endpoints/workspaceThreads.js
@@ -0,0 +1,150 @@
+const { multiUserMode, userFromSession, reqBody } = require("../utils/http");
+const { validatedRequest } = require("../utils/middleware/validatedRequest");
+const { Telemetry } = require("../models/telemetry");
+const {
+  flexUserRoleValid,
+  ROLES,
+} = require("../utils/middleware/multiUserProtected");
+const { EventLogs } = require("../models/eventLogs");
+const { WorkspaceThread } = require("../models/workspaceThread");
+const {
+  validWorkspaceSlug,
+  validWorkspaceAndThreadSlug,
+} = require("../utils/middleware/validWorkspace");
+const { WorkspaceChats } = require("../models/workspaceChats");
+const { convertToChatHistory } = require("../utils/chats");
+
+function workspaceThreadEndpoints(app) {
+  if (!app) return;
+
+  app.post(
+    "/workspace/:slug/thread/new",
+    [validatedRequest, flexUserRoleValid([ROLES.all]), validWorkspaceSlug],
+    async (request, response) => {
+      try {
+        const user = await userFromSession(request, response);
+        const workspace = response.locals.workspace;
+        const { thread, message } = await WorkspaceThread.new(
+          workspace,
+          user?.id
+        );
+        await Telemetry.sendTelemetry(
+          "workspace_thread_created",
+          {
+            multiUserMode: multiUserMode(response),
+            LLMSelection: process.env.LLM_PROVIDER || "openai",
+            Embedder: process.env.EMBEDDING_ENGINE || "inherit",
+            VectorDbSelection: process.env.VECTOR_DB || "pinecone",
+          },
+          user?.id
+        );
+
+        await EventLogs.logEvent(
+          "workspace_thread_created",
+          {
+            workspaceName: workspace?.name || "Unknown Workspace",
+          },
+          user?.id
+        );
+        response.status(200).json({ thread, message });
+      } catch (e) {
+        console.log(e.message, e);
+        response.sendStatus(500).end();
+      }
+    }
+  );
+
+  app.get(
+    "/workspace/:slug/threads",
+    [validatedRequest, flexUserRoleValid([ROLES.all]), validWorkspaceSlug],
+    async (request, response) => {
+      try {
+        const user = await userFromSession(request, response);
+        const workspace = response.locals.workspace;
+        const threads = await WorkspaceThread.where({
+          workspace_id: workspace.id,
+          user_id: user?.id || null,
+        });
+        response.status(200).json({ threads });
+      } catch (e) {
+        console.log(e.message, e);
+        response.sendStatus(500).end();
+      }
+    }
+  );
+
+  app.delete(
+    "/workspace/:slug/thread/:threadSlug",
+    [
+      validatedRequest,
+      flexUserRoleValid([ROLES.all]),
+      validWorkspaceAndThreadSlug,
+    ],
+    async (_, response) => {
+      try {
+        const thread = response.locals.thread;
+        await WorkspaceThread.delete({ id: thread.id });
+        response.sendStatus(200).end();
+      } catch (e) {
+        console.log(e.message, e);
+        response.sendStatus(500).end();
+      }
+    }
+  );
+
+  app.get(
+    "/workspace/:slug/thread/:threadSlug/chats",
+    [
+      validatedRequest,
+      flexUserRoleValid([ROLES.all]),
+      validWorkspaceAndThreadSlug,
+    ],
+    async (request, response) => {
+      try {
+        const user = await userFromSession(request, response);
+        const workspace = response.locals.workspace;
+        const thread = response.locals.thread;
+        const history = await WorkspaceChats.where(
+          {
+            workspaceId: workspace.id,
+            user_id: user?.id || null,
+            thread_id: thread.id,
+            include: true,
+          },
+          null,
+          { id: "asc" }
+        );
+
+        response.status(200).json({ history: convertToChatHistory(history) });
+      } catch (e) {
+        console.log(e.message, e);
+        response.sendStatus(500).end();
+      }
+    }
+  );
+
+  app.post(
+    "/workspace/:slug/thread/:threadSlug/update",
+    [
+      validatedRequest,
+      flexUserRoleValid([ROLES.all]),
+      validWorkspaceAndThreadSlug,
+    ],
+    async (request, response) => {
+      try {
+        const data = reqBody(request);
+        const currentThread = response.locals.thread;
+        const { thread, message } = await WorkspaceThread.update(
+          currentThread,
+          data
+        );
+        response.status(200).json({ thread, message });
+      } catch (e) {
+        console.log(e.message, e);
+        response.sendStatus(500).end();
+      }
+    }
+  );
+}
+
+module.exports = { workspaceThreadEndpoints };